From ba927b9a380a4b608f1d08c8deb9f692296204f4 Mon Sep 17 00:00:00 2001
From: Nick Zelei <2420177+nickzelei@users.noreply.github.com>
Date: Tue, 21 May 2024 21:04:20 -0700
Subject: [PATCH] seeds temporal on compose boot (#2001)

---
 compose.yml              | 14 ++++++++++++++
 hack/create_schedules.sh | 24 ++++++++++++++++++++++++
 hack/seed-neosync.sql    | 40 ++++++++++++++++++++--------------------
 3 files changed, 58 insertions(+), 20 deletions(-)
 create mode 100755 hack/create_schedules.sh

diff --git a/compose.yml b/compose.yml
index 3a6823e45..39efc0532 100644
--- a/compose.yml
+++ b/compose.yml
@@ -94,6 +94,20 @@ services:
     networks:
       - neosync-network
 
+  temporal-seed:
+    container_name: neosync-temporal-seed
+    image: temporalio/admin-tools:latest
+    environment:
+      - TEMPORAL_ADDRESS=temporal:7233
+    entrypoint: /bin/sh -c "/create_schedules.sh"
+    volumes:
+      - ./hack/create_schedules.sh:/create_schedules.sh
+    depends_on:
+      - api
+    networks:
+      - neosync-network
+      - temporal-network
+
   worker:
     container_name: neosync-worker
     image: ghcr.io/nucleuscloud/neosync/worker:latest
diff --git a/hack/create_schedules.sh b/hack/create_schedules.sh
new file mode 100755
index 000000000..2b5d46a68
--- /dev/null
+++ b/hack/create_schedules.sh
@@ -0,0 +1,24 @@
+#!/bin/sh
+
+# note, this command is slightly different when run on a mac.
+# this is expected to be run using the docker admin-tools image
+# for whatever reason (probably version difference) the CLI flags are different.
+# this may change in the future as they continue to merge these things...
+
+temporal schedule create \
+      --schedule-id "6638b7bf-7905-468f-8287-5c2df4732bf0" \
+      --task-queue "sync-job" \
+      --workflow-type "Workflow" \
+      --cron "0 0 1 1 *" \
+      --pause \
+      --workflow-id "6638b7bf-7905-468f-8287-5c2df4732bf0" \
+      --input '{"JobId": "6638b7bf-7905-468f-8287-5c2df4732bf0"}'
+
+temporal schedule create \
+      --schedule-id "2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600" \
+      --task-queue "sync-job" \
+      --workflow-type "Workflow" \
+      --cron "0 0 1 1 *" \
+      --pause \
+      --workflow-id "2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600" \
+      --input '{"JobId": "2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600"}'
diff --git a/hack/seed-neosync.sql b/hack/seed-neosync.sql
index 091574e61..21d9c0f48 100644
--- a/hack/seed-neosync.sql
+++ b/hack/seed-neosync.sql
@@ -21,25 +21,25 @@ ON CONFLICT DO NOTHING;
 --
 
 -- Generate Job
--- INSERT INTO neosync_api.jobs
--- (id, "name", account_id, status, connection_options, mappings, cron_schedule, created_by_id, updated_by_id, workflow_options, sync_options)
--- VALUES('6638b7bf-7905-468f-8287-5c2df4732bf0', 'seed-test-prod-db', '43c71652-b3f7-4dbb-87c2-508075496054', 1, '{"generateOptions": {"schemas": [{"schema": "public", "tables": [{"table": "users", "rowCount": 1000}]}], "fkSourceConnectionId": "e53ff29d-f079-4a4d-8db5-359fe480e740"}}'::jsonb, '[{"table": "users", "column": "id", "schema": "public", "jobMappingTransformerModel": {"config": {"generateUuid": {"includeHyphens": true}}, "source": 29}}, {"table": "users", "column": "age", "schema": "public", "jobMappingTransformerModel": {"config": {"GenerateInt64": {"max": 90, "min": 21, "randomizeSign": false}}, "source": 16}}, {"table": "users", "column": "current_salary", "schema": "public", "jobMappingTransformerModel": {"config": {"generateFloat64": {"max": 150000, "min": 50000, "precision": 6, "randomizeSign": false}}, "source": 11}}, {"table": "users", "column": "first_name", "schema": "public", "jobMappingTransformerModel": {"config": {"generateFirstName": {}}, "source": 10}}, {"table": "users", "column": "last_name", "schema": "public", "jobMappingTransformerModel": {"config": {"generateLastName": {}}, "source": 18}}]'::jsonb, '0 0 1 1 *', '00000000-0000-0000-0000-000000000000', '00000000-0000-0000-0000-000000000000', '{}'::jsonb, '{}'::jsonb)
--- ON CONFLICT DO NOTHING;
--- --
--- INSERT INTO neosync_api.job_destination_connection_associations
--- (id, job_id, connection_id, "options")
--- VALUES('30c56207-2c9e-4cef-a903-1cede68fd0f0', '6638b7bf-7905-468f-8287-5c2df4732bf0', 'e53ff29d-f079-4a4d-8db5-359fe480e740', '{"postgresOptions": {"initTableSchema": false, "onConflictConfig": {"doNothing": true}, "truncateTableconfig": {"truncateCascade": true, "truncateBeforeInsert": true}}}'::jsonb)
--- ON CONFLICT DO NOTHING;
--- -- End Generate Job
+INSERT INTO neosync_api.jobs
+(id, "name", account_id, status, connection_options, mappings, cron_schedule, created_by_id, updated_by_id, workflow_options, sync_options)
+VALUES('6638b7bf-7905-468f-8287-5c2df4732bf0', 'seed-test-prod-db', '43c71652-b3f7-4dbb-87c2-508075496054', 1, '{"generateOptions": {"schemas": [{"schema": "public", "tables": [{"table": "users", "rowCount": 1000}]}], "fkSourceConnectionId": "e53ff29d-f079-4a4d-8db5-359fe480e740"}}'::jsonb, '[{"table": "users", "column": "id", "schema": "public", "jobMappingTransformerModel": {"config": {"generateUuid": {"includeHyphens": true}}, "source": 29}}, {"table": "users", "column": "age", "schema": "public", "jobMappingTransformerModel": {"config": {"GenerateInt64": {"max": 90, "min": 21, "randomizeSign": false}}, "source": 16}}, {"table": "users", "column": "current_salary", "schema": "public", "jobMappingTransformerModel": {"config": {"generateFloat64": {"max": 150000, "min": 50000, "precision": 6, "randomizeSign": false}}, "source": 11}}, {"table": "users", "column": "first_name", "schema": "public", "jobMappingTransformerModel": {"config": {"generateFirstName": {}}, "source": 10}}, {"table": "users", "column": "last_name", "schema": "public", "jobMappingTransformerModel": {"config": {"generateLastName": {}}, "source": 18}}]'::jsonb, '0 0 1 1 *', '00000000-0000-0000-0000-000000000000', '00000000-0000-0000-0000-000000000000', '{}'::jsonb, '{}'::jsonb)
+ON CONFLICT DO NOTHING;
+--
+INSERT INTO neosync_api.job_destination_connection_associations
+(id, job_id, connection_id, "options")
+VALUES('30c56207-2c9e-4cef-a903-1cede68fd0f0', '6638b7bf-7905-468f-8287-5c2df4732bf0', 'e53ff29d-f079-4a4d-8db5-359fe480e740', '{"postgresOptions": {"initTableSchema": false, "onConflictConfig": {"doNothing": true}, "truncateTableconfig": {"truncateCascade": true, "truncateBeforeInsert": true}}}'::jsonb)
+ON CONFLICT DO NOTHING;
+-- End Generate Job
 
--- -- Sync Job
--- INSERT INTO neosync_api.jobs
--- (id, "name", account_id, status, connection_options, mappings, cron_schedule, created_by_id, updated_by_id, workflow_options, sync_options)
--- VALUES('2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600', 'prod-to-stage-sync', '43c71652-b3f7-4dbb-87c2-508075496054', 1, '{"postgresOptions": {"schemas": [], "connectionId": "e53ff29d-f079-4a4d-8db5-359fe480e740", "haltOnNewColumnAddition": false, "subsetByForeignKeyConstraints": true}}'::jsonb, '[{"table": "users", "column": "id", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "age", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "current_salary", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "first_name", "schema": "public", "jobMappingTransformerModel": {"config": {"transformFirstName": {"preserveLength": false}}, "source": 32}}, {"table": "users", "column": "last_name", "schema": "public", "jobMappingTransformerModel": {"config": {"transformLastName": {"preserveLength": false}}, "source": 37}}]'::jsonb, '0 0 1 1 *', '00000000-0000-0000-0000-000000000000', '00000000-0000-0000-0000-000000000000', '{}'::jsonb, '{}'::jsonb)
--- ON CONFLICT DO NOTHING;
+-- Sync Job
+INSERT INTO neosync_api.jobs
+(id, "name", account_id, status, connection_options, mappings, cron_schedule, created_by_id, updated_by_id, workflow_options, sync_options)
+VALUES('2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600', 'prod-to-stage-sync', '43c71652-b3f7-4dbb-87c2-508075496054', 1, '{"postgresOptions": {"schemas": [], "connectionId": "e53ff29d-f079-4a4d-8db5-359fe480e740", "haltOnNewColumnAddition": false, "subsetByForeignKeyConstraints": true}}'::jsonb, '[{"table": "users", "column": "id", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "age", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "current_salary", "schema": "public", "jobMappingTransformerModel": {"config": {"passthrough": {}}, "source": 1}}, {"table": "users", "column": "first_name", "schema": "public", "jobMappingTransformerModel": {"config": {"transformFirstName": {"preserveLength": false}}, "source": 32}}, {"table": "users", "column": "last_name", "schema": "public", "jobMappingTransformerModel": {"config": {"transformLastName": {"preserveLength": false}}, "source": 37}}]'::jsonb, '0 0 1 1 *', '00000000-0000-0000-0000-000000000000', '00000000-0000-0000-0000-000000000000', '{}'::jsonb, '{}'::jsonb)
+ON CONFLICT DO NOTHING;
 
--- INSERT INTO neosync_api.job_destination_connection_associations
--- (id, job_id, connection_id, "options")
--- VALUES('15d0f1c0-55c3-4340-9bbb-5cd55c1baabe', '2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600', '3c592989-c672-46ea-b709-2e338cd1daff', '{"postgresOptions": {"initTableSchema": true, "onConflictConfig": {"doNothing": true}, "truncateTableconfig": {"truncateCascade": true, "truncateBeforeInsert": true}}}'::jsonb)
--- ON CONFLICT DO NOTHING;
--- -- End Sync Job
+INSERT INTO neosync_api.job_destination_connection_associations
+(id, job_id, connection_id, "options")
+VALUES('15d0f1c0-55c3-4340-9bbb-5cd55c1baabe', '2a5d5caa-7f09-4fdf-a4a7-6a2e341aa600', '3c592989-c672-46ea-b709-2e338cd1daff', '{"postgresOptions": {"initTableSchema": true, "onConflictConfig": {"doNothing": true}, "truncateTableconfig": {"truncateCascade": true, "truncateBeforeInsert": true}}}'::jsonb)
+ON CONFLICT DO NOTHING;
+-- End Sync Job