1 files changed, 31 insertions, 0 deletions
diff --git a/packages/pipeline/migrations/1549499426238-AddTxHashToERC20ApprovalEventPrimaryKey.ts b/packages/pipeline/migrations/1549499426238-AddTxHashToERC20ApprovalEventPrimaryKey.ts
new file mode 100644
index 000000000..874713e67
--- /dev/null
+++ b/packages/pipeline/migrations/1549499426238-AddTxHashToERC20ApprovalEventPrimaryKey.ts
@@ -0,0 +1,31 @@
+import { MigrationInterface, QueryRunner } from 'typeorm';
+
+const oldPrimaryColumns = ['token_address', 'log_index', 'block_number'];
+
+const newPrimaryColumns = ['transaction_hash'];
+
+async function updatePrimaryKeysAsync(queryRunner: QueryRunner, columnNames: string[]): Promise<void> {
+    const table = await queryRunner.getTable(`raw.erc20_approval_events`);
+    if (table === undefined) {
+        throw new Error(`Couldn't get table 'raw.erc20_approval_events'`);
+    }
+    const columns = [];
+    for (const columnName of columnNames) {
+        const column = table.findColumnByName(columnName);
+        if (column === undefined) {
+            throw new Error(`Couldn't get column '${columnName}' from table 'raw.erc20_approval_events'`);
+        }
+        columns.push(column);
+    }
+    await queryRunner.updatePrimaryKeys(table, columns);
+}
+
+export class AddTxHashToERC20ApprovalEventPrimaryKey1549499426238 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<any> {
+        await updatePrimaryKeysAsync(queryRunner, oldPrimaryColumns.concat(newPrimaryColumns));
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<any> {
+        await updatePrimaryKeysAsync(queryRunner, oldPrimaryColumns);
+    }
+}