feat: add totalBalance to address entity

nervosnetwork · Sep 3, 2019 · 5746438 · 5746438
1 parent a952152
commit 5746438
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 3 deletions.
diff --git a/packages/neuron-wallet/src/database/address/entities/address.ts b/packages/neuron-wallet/src/database/address/entities/address.ts
@@ -73,6 +73,9 @@ export default class Address extends BaseEntity {
   @Column()
   pendingBalance: string = '0'
 
+  @Column()
+  totalBalance: string = '0'
+
   public balance = (): string => {
     return (BigInt(this.liveBalance) + BigInt(this.sentBalance)).toString()
   }

diff --git a/packages/neuron-wallet/src/database/address/migrations/1562126909151-extendBalance.ts b/packages/neuron-wallet/src/database/address/migrations/1562126909151-extendBalance.ts
@@ -13,8 +13,8 @@ export class extendBalance1562126909151 implements MigrationInterface {
     }
 
     public async down(queryRunner: QueryRunner): Promise<any> {
-      await queryRunner.dropColumn('transaction', 'sentBalance')
-      await queryRunner.dropColumn('transaction', 'pendingBalance')
+      await queryRunner.dropColumn('address', 'sentBalance')
+      await queryRunner.dropColumn('address', 'pendingBalance')
       await queryRunner.changeColumn('address', 'liveBalance', new TableColumn({
         name: 'balance',
         type: 'varchar',

diff --git a/packages/neuron-wallet/src/database/address/migrations/1567485550388-AddTotalBalance.ts b/packages/neuron-wallet/src/database/address/migrations/1567485550388-AddTotalBalance.ts
@@ -0,0 +1,17 @@
+import {MigrationInterface, QueryRunner, TableColumn} from "typeorm";
+
+export class AddTotalBalance1567485550388 implements MigrationInterface {
+
+  public async up(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.addColumn('address', new TableColumn({
+      name: 'totalBalance',
+      type: 'varchar',
+      default: '0',
+    }))
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.dropColumn('address', 'totalBalance')
+  }
+
+}
diff --git a/packages/neuron-wallet/src/database/address/ormconfig.ts b/packages/neuron-wallet/src/database/address/ormconfig.ts
@@ -8,6 +8,7 @@ import Address from './entities/address'
 
 import { AddAddress1561461669542 } from './migrations/1561461669542-AddAddress'
 import { extendBalance1562126909151 } from './migrations/1562126909151-extendBalance'
+import { AddTotalBalance1567485550388 } from './migrations/1567485550388-AddTotalBalance'
 
 const dbPath = path.join(env.fileBasePath, 'address.sqlite')
 
@@ -20,7 +21,7 @@ const connectOptions = (): SqliteConnectionOptions => {
     type: 'sqlite',
     database,
     entities: [Address],
-    migrations: [AddAddress1561461669542, extendBalance1562126909151],
+    migrations: [AddAddress1561461669542, extendBalance1562126909151, AddTotalBalance1567485550388],
     synchronize: false,
     migrationsRun: true,
     logging: ['error'],