Add code to update the foreign keys

brave-intl · Aug 23, 2024 · 09e04dd · 09e04dd
1 parent d606a0a
commit 09e04dd
Show file tree

Hide file tree

Showing 2 changed files with 95 additions and 18 deletions.
diff --git a/eyeshade/migrations/0021_reset_transactions_no_vbat/up.sql b/eyeshade/migrations/0021_reset_transactions_no_vbat/up.sql
@@ -1,4 +1,4 @@
-select execute($$
+select execute($outer$
 
 insert into migrations (id, description) values ('0021', 'reset_transactions_no_vbat');
 
@@ -102,12 +102,89 @@ ORDER BY difference DESC;
 -- SELECT * FROM temp_balances_past_last_payout_or_just_balance;
 
 
--- Set up the new TRANACTIONS table
--- Rename the existing table
-ALTER TABLE transactions RENAME TO old_transactions;
 
--- Create a new table with the same structure
-CREATE TABLE transactions (LIKE old_transactions INCLUDING ALL);
+
+
+
+
+
+DO $inner$
+DECLARE
+    fk_record RECORD;
+    old_table_name TEXT := 'transactions';
+    new_table_name TEXT := 'transactions_old';
+BEGIN
+    -- Step 1: Drop foreign keys referencing the transactions table
+    FOR fk_record IN
+        SELECT
+            tc.table_name AS referencing_table,
+            kcu.column_name AS referencing_column,
+            ccu.table_name AS foreign_table,
+            ccu.column_name AS foreign_column,
+            tc.constraint_name AS constraint_name
+        FROM
+            information_schema.table_constraints AS tc
+            JOIN information_schema.key_column_usage AS kcu
+                ON tc.constraint_name = kcu.constraint_name
+            JOIN information_schema.constraint_column_usage AS ccu
+                ON ccu.constraint_name = tc.constraint_name
+        WHERE
+            tc.constraint_type = 'FOREIGN KEY'
+            AND ccu.table_name = old_table_name
+    LOOP
+        -- Drop the foreign key constraint
+        EXECUTE format('ALTER TABLE %I DROP CONSTRAINT %I;', fk_record.referencing_table, fk_record.constraint_name);
+    END LOOP;
+    -- Step 2: Rename the original transactions table
+    EXECUTE format('ALTER TABLE %I RENAME TO %I;', old_table_name, new_table_name);
+    -- Step 3: Recreate the transactions table with the original name
+    EXECUTE format('CREATE TABLE %I (LIKE %I INCLUDING ALL);', old_table_name, new_table_name);
+    -- Step 4: Recreate the foreign keys
+    FOR fk_record IN
+        SELECT
+            tc.table_name AS referencing_table,
+            kcu.column_name AS referencing_column,
+            ccu.table_name AS foreign_table,
+            ccu.column_name AS foreign_column,
+            tc.constraint_name AS constraint_name
+        FROM
+            information_schema.table_constraints AS tc
+            JOIN information_schema.key_column_usage AS kcu
+                ON tc.constraint_name = kcu.constraint_name
+            JOIN information_schema.constraint_column_usage AS ccu
+                ON ccu.constraint_name = tc.constraint_name
+        WHERE
+            tc.constraint_type = 'FOREIGN KEY'
+            AND ccu.table_name = new_table_name
+    LOOP
+        -- Recreate the foreign key constraint to point to the new table
+        EXECUTE format('
+            ALTER TABLE %I
+            ADD CONSTRAINT %I FOREIGN KEY (%I) REFERENCES %I(%I);',
+            fk_record.referencing_table,
+            fk_record.constraint_name,
+            fk_record.referencing_column,
+            old_table_name,
+            fk_record.foreign_column
+        );
+    END LOOP;
+END $inner$;
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
 
 -- POPULATE THE NEW TRANSACTIONS TABLE
@@ -123,4 +200,4 @@ SELECT
 FROM temp_balances_past_last_payout_or_just_balance;
 
 
-$$) where not exists (select * from migrations where id = '0021');
+$outer$) where not exists (select * from migrations where id = '0021');
diff --git a/package.json b/package.json
@@ -19,26 +19,26 @@
     "coverage": "nyc report --reporter=text-lcov | coveralls",
     "test-all": "npm run test-isup && npm run test-unit && npm run test-integration",
     "test-isup": "ava -v -s test/isup.integration.test.js",
-    "docker-test": "docker-compose run --rm -v $(pwd)/test:/usr/src/app/test -v $(pwd)/eyeshade:/usr/src/app/eyeshade -v $(pwd)/bat-utils:/usr/src/app/bat-utils eyeshade-web npm run test",
+    "docker-test": "docker compose run --rm -v $(pwd)/test:/usr/src/app/test -v $(pwd)/eyeshade:/usr/src/app/eyeshade -v $(pwd)/bat-utils:/usr/src/app/bat-utils eyeshade-web npm run test",
     "lint": "npx standard",
     "lint-fix": "npm run lint -- --fix",
     "security": "npx standard --version && npm run lint && npm run moderate-audit",
     "moderate-audit": "npm audit --audit-level=moderate; echo $?",
     "test-integration": "ava -v -s test/*/*.integration.test.js",
     "test-settlements": "ava -v -s eyeshade/**/settlements.test.js",
     "test-unit": "ava -v -s bat-utils/**/*.test.js eyeshade/**/*.test.js",
-    "docker-reset": "docker-compose down && docker rmi $(docker images -a -q) --force",
-    "docker-build": "docker-compose build",
-    "docker-seed-eyeshade": "docker-compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web bash ./bin/seed.sh",
-    "docker-up": "docker-compose -f docker-compose.yml up $npm_package_config_dockerservices",
-    "docker-up-dev": "docker-compose -f docker-compose.yml up $npm_package_config_dockerservices $npm_package_config_dockermonitors",
-    "docker-up-detached": "docker-compose up -d $npm_package_config_dockerservices",
-    "docker-migrate-eyeshade-up": "docker-compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web npm run migrate-up",
+    "docker-reset": "docker compose down && docker rmi $(docker images -a -q) --force",
+    "docker-build": "docker compose build",
+    "docker-seed-eyeshade": "docker compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web bash ./bin/seed.sh",
+    "docker-up": "docker compose -f docker-compose.yml up $npm_package_config_dockerservices",
+    "docker-up-dev": "docker compose -f docker-compose.yml up $npm_package_config_dockerservices $npm_package_config_dockermonitors",
+    "docker-up-detached": "docker compose up -d $npm_package_config_dockerservices",
+    "docker-migrate-eyeshade-up": "docker compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web npm run migrate-up",
     "migrate-up": "./bin/migrate-up.sh",
     "docker-migrate-up": "npm run docker-migrate-eyeshade-up && npm run docker-seed-eyeshade",
-    "docker-migrate-down": "docker-compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web ./bin/migrate-down.sh",
-    "docker-migrate-down-all": "docker-compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web bash ./bin/migrate-down-all.sh",
-    "docker-log-detached": "docker-compose logs",
+    "docker-migrate-down": "docker compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web ./bin/migrate-down.sh",
+    "docker-migrate-down-all": "docker compose -f docker-compose.yml run --rm -w /usr/src/app/eyeshade eyeshade-web bash ./bin/migrate-down-all.sh",
+    "docker-log-detached": "docker compose logs",
     "postinstall": "test -e .git && { for file in `ls .git-hooks`; do ln -sf ../../.git-hooks/${file} .git/hooks/${file}; done } || true"
   },
   "config": {