write-test-message.sh in message-db-3.0.0

- old
+ new

@@ -1,9 +1,13 @@
 #!/usr/bin/env bash
 
-set -u
+set -ue
 
+function run_psql {
+  psql -v ON_ERROR_STOP=1 "$@"
+}
+
 instances=1
 if [ ! -z ${INSTANCES+x} ]; then
   instances=$INSTANCES
 fi
 
@@ -11,10 +15,15 @@
 stream_name="testStream-$uuid"
 if [ ! -z ${STREAM_NAME+x} ]; then
   stream_name=$STREAM_NAME
 fi
 
+type="SomeType"
+if [ ! -z ${TYPE+x} ]; then
+  type=$TYPE
+fi
+
 title="Writing $instances Messages to Stream $stream_name"
 if [ -z ${METADATA+x} ]; then
   metadata="'{\"metaAttribute\": \"some meta value\"}'"
 else
   metadata="$METADATA"
@@ -51,13 +60,13 @@
 for (( i=1; i<=instances; i++ )); do
   uuid=$(echo $(uuidgen) | tr '[:upper:]' '[:lower:]')
 
   echo "Instance: $i, Message ID: $uuid"
 
-  psql $database -U $user -c "SELECT write_message('$uuid'::varchar, '$stream_name'::varchar, 'SomeType'::varchar, '{\"attribute\": \"some value\"}'::jsonb, $metadata);" > /dev/null
+  run_psql $database -U $user -c "SELECT write_message('$uuid'::varchar, '$stream_name'::varchar, '$type'::varchar, '{\"attribute\": \"some value\"}'::jsonb, $metadata);" > /dev/null
 done
 
 
 echo
-psql $database -U $user -P pager=off -x -c "SELECT * FROM messages WHERE stream_name = '$stream_name';"
+run_psql $database -U $user -P pager=off -x -c "SELECT * FROM messages WHERE stream_name = '$stream_name';"
 
 echo