handle deletions

Author: antoinedc

run/models/tokenbalancechange.js

@@ -17,6 +17,7 @@ module.exports = (sequelize, DataTypes) => {
       TokenBalanceChange.belongsTo(models.Transaction, { foreignKey: 'transactionId', as: 'transaction' });
       TokenBalanceChange.belongsTo(models.Workspace, { foreignKey: 'workspaceId', as: 'workspace' });
       TokenBalanceChange.belongsTo(models.TokenTransfer, { foreignKey: 'tokenTransferId', as: 'tokenTransfer' });
+      TokenBalanceChange.hasOne(models.TokenBalanceChangeEvent, { foreignKey: 'tokenBalanceChangeId', as: 'event' });
       TokenBalanceChange.hasOne(models.Contract, {
           sourceKey: 'token',
           foreignKey: 'address',
@@ -40,6 +41,14 @@ module.exports = (sequelize, DataTypes) => {
       });
     }
 
+    async safeDestroy(transaction) {
+      const event = await this.getEvent();
+      if (event)
+        await event.destroy({ transaction });
+
+      return this.destroy({ transaction });
+    }
+
     async insertAnalyticEvent(sequelizeTransaction) {
       const transaction = await this.getTransaction();
       const contract = await this.getContract();

run/models/tokentransfer.js

@@ -32,13 +32,19 @@ module.exports = (sequelize, DataTypes) => {
           },
           constraints: false
       });
+      TokenTransfer.hasOne(models.TokenTransferEvent, { foreignKey: 'tokenTransferId', as: 'event' });
       TokenTransfer.hasMany(models.TokenBalanceChange, { foreignKey: 'tokenTransferId', as: 'tokenBalanceChanges' });
     }
 
     async safeDestroy(transaction) {
         const tokenBalanceChanges = await this.getTokenBalanceChanges();
         for (let i = 0; i < tokenBalanceChanges.length; i++)
-            await tokenBalanceChanges[i].destroy({ transaction});
+            await tokenBalanceChanges[i].safeDestroy(transaction);
+
+        const event = await this.getEvent();
+        if (event)
+            await event.destroy({ transaction });
+
         return this.destroy({ transaction });
     }
 

run/models/tokentransferevent.js

@@ -10,7 +10,6 @@ module.exports = (sequelize, DataTypes) => {
      * The `models/index` file will call this method automatically.
      */
     static associate(models) {
-      // define association here
     }
   }
   TokenTransferEvent.init({

run/models/transaction.js

@@ -36,6 +36,7 @@ module.exports = (sequelize, DataTypes) => {
           constraints: false
       });
       Transaction.hasOne(models.TransactionReceipt, { foreignKey: 'transactionId', as: 'receipt' });
+      Transaction.hasOne(models.TransactionEvent, { foreignKey: 'transactionId', as: 'event' });
       Transaction.hasMany(models.TokenTransfer, { foreignKey: 'transactionId', as: 'tokenTransfers' });
       Transaction.hasMany(models.TokenBalanceChange, { foreignKey: 'transactionId', as: 'tokenBalanceChanges' });
       Transaction.hasMany(models.TransactionTraceStep, { foreignKey: 'transactionId', as: 'traceSteps' });
@@ -45,9 +46,15 @@ module.exports = (sequelize, DataTypes) => {
         const receipt = await this.getReceipt();
         if (receipt)
             await receipt.safeDestroy(transaction);
+
         const traceSteps = await this.getTraceSteps();
         for (let i = 0; i < traceSteps.length; i++)
             await traceSteps[i].destroy(transaction);
+
+        const event = await this.getEvent();
+        if (event)
+            await event.destroy({ transaction });
+        
         return this.destroy({ transaction });
     }
 

run/models/transactionreceipt.js

@@ -16,8 +16,8 @@ module.exports = (sequelize, DataTypes) => {
      * The `models/index` file will call this method automatically.
      */
     static associate(models) {
-      TransactionReceipt.belongsTo(models.Transaction, { foreignKey: 'transactionId', as: 'transaction' });
-      TransactionReceipt.hasMany(models.TransactionLog, { foreignKey: 'transactionReceiptId', as: 'logs' });
+        TransactionReceipt.hasMany(models.TransactionLog, { foreignKey: 'transactionReceiptId', as: 'logs' });
+        TransactionReceipt.belongsTo(models.Transaction, { foreignKey: 'transactionId', as: 'transaction' });
     }
 
     async insertAnalyticEvent(sequelizeTransaction) {