fix

dydxprotocol · Nov 8, 2023 · 8467fdb · 8467fdb
1 parent 82f7a60
commit 8467fdb
Showing 1 changed file with 12 additions and 16 deletions.
diff --git a/...res/src/db/migrations/migration_files/20231101173444_update_fills_type_with_deleverage.ts b/...res/src/db/migrations/migration_files/20231101173444_update_fills_type_with_deleverage.ts
@@ -1,23 +1,19 @@
 import * as Knex from 'knex';
 
-export async function up(knex: Knex): Promise<void> {
-  return knex.raw(`
-    ALTER TABLE ONLY fills
-    DROP CONSTRAINT IF EXISTS fills_type_check;
+import { formatAlterTableEnumSql } from '../helpers';
 
-    ALTER TABLE ONLY fills
-    ADD CONSTRAINT fills_type_check
-    CHECK (type = ANY (ARRAY['MARKET'::text, 'LIMIT'::text, 'LIQUIDATED'::text, 'LIQUIDATION'::text, 'DELEVERAGED'::text, 'OFFSETTING'::text]));
-  `);
+export async function up(knex: Knex): Promise<void> {
+  return knex.raw(formatAlterTableEnumSql(
+    'fills',
+    'type',
+    ['MARKET', 'LIMIT', 'LIQUIDATED', 'LIQUIDATION', 'DELEVERAGED', 'OFFSETTING'],
+  ));
 }
 
 export async function down(knex: Knex): Promise<void> {
-  return knex.raw(`
-    ALTER TABLE ONLY fills
-    DROP CONSTRAINT IF EXISTS fills_type_check;
-
-    ALTER TABLE ONLY fills
-    ADD CONSTRAINT fills_type_check
-    CHECK (type = ANY (ARRAY['MARKET'::text, 'LIMIT'::text, 'LIQUIDATED'::text, 'LIQUIDATION'::text]));
-  `);
+  return knex.raw(formatAlterTableEnumSql(
+    'fills',
+    'type',
+    ['MARKET', 'LIMIT', 'LIQUIDATED', 'LIQUIDATION'],
+  ));
 }