message-db 0.0.0 → 1.1.3

data/database/functions/hash-64.sql

@@ -0,0 +1,13 @@
+CREATE OR REPLACE FUNCTION message_store.hash_64(
+  value varchar
+)
+RETURNS bigint
+AS $$
+DECLARE
+  _hash bigint;
+BEGIN
+  SELECT left('x' || md5(hash_64.value), 17)::bit(64)::bigint INTO _hash;
+  return _hash;
+END;
+$$ LANGUAGE plpgsql
+IMMUTABLE;

data/database/functions/id.sql

@@ -0,0 +1,18 @@
+CREATE OR REPLACE FUNCTION message_store.id(
+  stream_name varchar
+)
+RETURNS varchar
+AS $$
+DECLARE
+  _id_separator_position integer;
+BEGIN
+  _id_separator_position := STRPOS(id.stream_name, '-');
+
+  IF _id_separator_position = 0 THEN
+    RETURN NULL;
+  END IF;
+
+  RETURN SUBSTRING(id.stream_name, _id_separator_position + 1);
+END;
+$$ LANGUAGE plpgsql
+IMMUTABLE;

data/database/functions/is-category.sql

@@ -0,0 +1,14 @@
+CREATE OR REPLACE FUNCTION message_store.is_category(
+  stream_name varchar
+)
+RETURNS boolean
+AS $$
+BEGIN
+  IF NOT STRPOS(is_category.stream_name, '-') = 0 THEN
+    RETURN FALSE;
+  END IF;
+
+  RETURN TRUE;
+END;
+$$ LANGUAGE plpgsql
+IMMUTABLE;

data/database/functions/message-store-version.sql

@@ -0,0 +1,8 @@
+CREATE OR REPLACE FUNCTION message_store.message_store_version()
+RETURNS varchar
+AS $$
+BEGIN
+  RETURN '1.1.3';
+END;
+$$ LANGUAGE plpgsql
+VOLATILE;

data/database/functions/stream-version.sql

@@ -0,0 +1,19 @@
+CREATE OR REPLACE FUNCTION message_store.stream_version(
+  stream_name varchar
+)
+RETURNS bigint
+AS $$
+DECLARE
+  _stream_version bigint;
+BEGIN
+  SELECT
+    max(position) into _stream_version
+  FROM
+    messages
+  WHERE
+    messages.stream_name = stream_version.stream_name;
+
+  RETURN _stream_version;
+END;
+$$ LANGUAGE plpgsql
+VOLATILE;

data/database/functions/write-message.sql

@@ -0,0 +1,73 @@
+CREATE OR REPLACE FUNCTION message_store.write_message(
+  id varchar,
+  stream_name varchar,
+  "type" varchar,
+  data jsonb,
+  metadata jsonb DEFAULT NULL,
+  expected_version bigint DEFAULT NULL
+)
+RETURNS bigint
+AS $$
+DECLARE
+  _message_id uuid;
+  _stream_version bigint;
+  _next_position bigint;
+BEGIN
+  PERFORM acquire_lock(write_message.stream_name);
+
+  _stream_version := stream_version(write_message.stream_name);
+
+  IF _stream_version IS NULL THEN
+    _stream_version := -1;
+  END IF;
+
+  IF write_message.expected_version IS NOT NULL THEN
+    IF write_message.expected_version != _stream_version THEN
+      RAISE EXCEPTION
+        'Wrong expected version: % (Stream: %, Stream Version: %)',
+        write_message.expected_version,
+        write_message.stream_name,
+        _stream_version;
+    END IF;
+  END IF;
+
+  _next_position := _stream_version + 1;
+
+  _message_id = uuid(write_message.id);
+
+  INSERT INTO messages
+    (
+      id,
+      stream_name,
+      position,
+      type,
+      data,
+      metadata
+    )
+  VALUES
+    (
+      _message_id,
+      write_message.stream_name,
+      _next_position,
+      write_message.type,
+      write_message.data,
+      write_message.metadata
+    )
+  ;
+
+  IF current_setting('message_store.debug_write', true) = 'on' OR current_setting('message_store.debug', true) = 'on' THEN
+    RAISE NOTICE '» write_message';
+    RAISE NOTICE 'id ($1): %', write_message.id;
+    RAISE NOTICE 'stream_name ($2): %', write_message.stream_name;
+    RAISE NOTICE 'type ($3): %', write_message.type;
+    RAISE NOTICE 'data ($4): %', write_message.data;
+    RAISE NOTICE 'metadata ($5): %', write_message.metadata;
+    RAISE NOTICE 'expected_version ($6): %', write_message.expected_version;
+    RAISE NOTICE '_stream_version: %', _stream_version;
+    RAISE NOTICE '_next_position: %', _next_position;
+  END IF;
+
+  RETURN _next_position;
+END;
+$$ LANGUAGE plpgsql
+VOLATILE;

data/database/indexes/messages-category.sql

@@ -0,0 +1,7 @@
+DROP INDEX IF EXISTS message_store.messages_category;
+
+CREATE INDEX messages_category ON message_store.messages (
+  message_store.category(stream_name),
+  global_position,
+  message_store.category(metadata->>'correlationStreamName')
+);

data/database/indexes/messages-id.sql

@@ -0,0 +1,5 @@
+DROP INDEX IF EXISTS message_store.messages_id;
+
+CREATE UNIQUE INDEX messages_id ON message_store.messages (
+  id
+);

data/database/indexes/messages-stream.sql

@@ -0,0 +1,6 @@
+DROP INDEX IF EXISTS messages_stream;
+
+CREATE UNIQUE INDEX messages_stream ON message_store.messages (
+  stream_name,
+  position
+);

data/database/install-functions.sh

@@ -0,0 +1,79 @@
+#!/usr/bin/env bash
+
+set -e
+
+function script_dir {
+  val="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+  echo "$val"
+}
+
+function create-types {
+  if [ -z ${DATABASE_NAME+x} ]; then
+    database=message_store
+    echo "(DATABASE_NAME is not set. Using: $database.)"
+  else
+    database=$DATABASE_NAME
+  fi
+
+  base=$(script_dir)
+
+  echo "» message type"
+  psql $database -q -f $base/types/message.sql
+}
+
+function create-functions {
+  if [ -z ${DATABASE_NAME+x} ]; then
+    database=message_store
+    echo "(DATABASE_NAME is not set. Using: $database.)"
+  else
+    database=$DATABASE_NAME
+  fi
+
+  base=$(script_dir)
+
+  echo "» message_store_version function"
+  psql $database -q -f $base/functions/message-store-version.sql
+
+  echo "» hash_64 function"
+  psql $database -q -f $base/functions/hash-64.sql
+
+  echo "» acquire_lock function"
+  psql $database -q -f $base/functions/acquire-lock.sql
+
+  echo "» category function"
+  psql $database -q -f $base/functions/category.sql
+
+  echo "» is_category function"
+  psql $database -q -f $base/functions/is-category.sql
+
+  echo "» id function"
+  psql $database -q -f $base/functions/id.sql
+
+  echo "» cardinal_id function"
+  psql $database -q -f $base/functions/cardinal-id.sql
+
+  echo "» stream_version function"
+  psql $database -q -f $base/functions/stream-version.sql
+
+  echo "» write_message function"
+  psql $database -q -f $base/functions/write-message.sql
+
+  echo "» get_stream_messages function"
+  psql $database -q -f $base/functions/get-stream-messages.sql
+
+  echo "» get_category_messages function"
+  psql $database -q -f $base/functions/get-category-messages.sql
+
+  echo "» get_last_stream_message function"
+  psql $database -q -f $base/functions/get-last-stream-message.sql
+}
+
+echo "Creating Types"
+echo "- - -"
+create-types
+echo
+
+echo "Creating Functions"
+echo "- - -"
+create-functions
+echo

data/database/install-indexes.sh

@@ -0,0 +1,33 @@
+#!/usr/bin/env bash
+
+set -e
+
+function script_dir {
+  val="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+  echo "$val"
+}
+
+function create-indexes {
+  if [ -z ${DATABASE_NAME+x} ]; then
+    database=message_store
+    echo "(DATABASE_NAME is not set. Using: $database.)"
+  else
+    database=$DATABASE_NAME
+  fi
+
+  base=$(script_dir)
+
+  echo "» messages_id index"
+  psql $database -q -f $base/indexes/messages-id.sql
+
+  echo "» messages_stream index"
+  psql $database -q -f $base/indexes/messages-stream.sql
+
+  echo "» messages_category index"
+  psql $database -q -f $base/indexes/messages-category.sql
+}
+
+echo "Creating Indexes"
+echo "- - -"
+create-indexes
+echo

data/database/install-privileges.sh

@@ -0,0 +1,39 @@
+#!/usr/bin/env bash
+
+set -e
+
+function script_dir {
+  val="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+  echo "$val"
+}
+
+function grant-privileges {
+  if [ -z ${DATABASE_NAME+x} ]; then
+    database=message_store
+    echo "(DATABASE_NAME is not set. Using: $database.)"
+  else
+    database=$DATABASE_NAME
+  fi
+
+  base=$(script_dir)
+
+  echo "» schema privileges"
+  psql $database -q -f $base/privileges/schema.sql
+
+  echo "» messages table privileges"
+  psql $database -q -f $base/privileges/table.sql
+
+  echo "» sequence privileges"
+  psql $database -q -f $base/privileges/sequence.sql
+
+  echo "» functions privileges"
+  psql $database -q -f $base/privileges/functions.sql
+
+  echo "» views privileges"
+  psql $database -q -f $base/privileges/views.sql
+}
+
+echo "Granting Privileges"
+echo "- - -"
+grant-privileges
+echo

data/database/install-views.sh

@@ -0,0 +1,42 @@
+#!/usr/bin/env bash
+
+set -e
+
+function script_dir {
+  val="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+  echo "$val"
+}
+
+function create-views {
+  if [ -z ${DATABASE_NAME+x} ]; then
+    database=message_store
+    echo "(DATABASE_NAME is not set. Using: $database.)"
+  else
+    database=$DATABASE_NAME
+  fi
+
+  base=$(script_dir)
+
+  echo "» stream_summary view"
+  psql $database -q -f $base/views/stream-summary.sql
+
+  echo "» type_summary view"
+  psql $database -q -f $base/views/type-summary.sql
+
+  echo "» stream_type_summary view"
+  psql $database -q -f $base/views/stream-type-summary.sql
+
+  echo "» type_stream_summary view"
+  psql $database -q -f $base/views/type-stream-summary.sql
+
+  echo "» category_type_summary view"
+  psql $database -q -f $base/views/category-type-summary.sql
+
+  echo "» type_category_summary view"
+  psql $database -q -f $base/views/type-category-summary.sql
+}
+
+echo "Creating Views"
+echo "- - -"
+create-views
+echo

data/database/install.sh

@@ -0,0 +1,102 @@
+#!/usr/bin/env bash
+
+set -e
+
+function script_dir {
+  val="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+  echo "$val"
+}
+
+base=$(script_dir)
+
+echo
+echo "Installing Database"
+echo "Version: $(cat $base/VERSION.txt)"
+echo "= = ="
+
+if [ -z ${DATABASE_NAME+x} ]; then
+  database=message_store
+  echo "DATABASE_NAME is not set. Using: $database."
+  export DATABASE_NAME=$database
+else
+  database=$DATABASE_NAME
+fi
+
+if [ -z ${PGOPTIONS+x} ]; then
+  export PGOPTIONS='-c client_min_messages=warning'
+fi
+
+function create-user {
+  base=$(script_dir)
+
+  echo "» message_store role"
+  psql -q -f $base/roles/message-store.sql
+}
+
+function create-database {
+  echo "» $database database"
+  createdb $database
+}
+
+function create-schema {
+  echo "» message_store schema"
+  psql $database -q -f $base/schema/message-store.sql
+}
+
+function create-extensions {
+  base=$(script_dir)
+
+  echo "» pgcrypto extension"
+  psql $database -q -f $base/extensions/pgcrypto.sql
+}
+
+function create-table {
+  base=$(script_dir)
+
+  echo "» messages table"
+  psql $database -q -f $base/tables/messages.sql
+}
+
+echo
+
+echo "Creating User"
+echo "- - -"
+create-user
+echo
+
+echo "Creating Database"
+echo "- - -"
+create-database
+echo
+
+echo "Creating Schema"
+echo "- - -"
+create-schema
+echo
+
+echo "Creating Extensions"
+echo "- - -"
+create-extensions
+echo
+
+echo "Creating Table"
+echo "- - -"
+create-table
+echo
+
+# Install functions
+source $base/install-functions.sh
+
+# Install indexes
+source $base/install-indexes.sh
+
+# Install views
+source $base/install-views.sh
+
+# Install privileges
+source $base/install-privileges.sh
+
+echo "= = ="
+echo "Done Installing Database"
+echo "Version: $(cat $base/VERSION.txt)"
+echo