Skip to content

Commit 5746438

Browse files
committed
feat: add totalBalance to address entity
1 parent a952152 commit 5746438

4 files changed

Lines changed: 24 additions & 3 deletions

File tree

packages/neuron-wallet/src/database/address/entities/address.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,9 @@ export default class Address extends BaseEntity {
7373
@Column()
7474
pendingBalance: string = '0'
7575

76+
@Column()
77+
totalBalance: string = '0'
78+
7679
public balance = (): string => {
7780
return (BigInt(this.liveBalance) + BigInt(this.sentBalance)).toString()
7881
}

packages/neuron-wallet/src/database/address/migrations/1562126909151-extendBalance.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@ export class extendBalance1562126909151 implements MigrationInterface {
1313
}
1414

1515
public async down(queryRunner: QueryRunner): Promise<any> {
16-
await queryRunner.dropColumn('transaction', 'sentBalance')
17-
await queryRunner.dropColumn('transaction', 'pendingBalance')
16+
await queryRunner.dropColumn('address', 'sentBalance')
17+
await queryRunner.dropColumn('address', 'pendingBalance')
1818
await queryRunner.changeColumn('address', 'liveBalance', new TableColumn({
1919
name: 'balance',
2020
type: 'varchar',
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
import {MigrationInterface, QueryRunner, TableColumn} from "typeorm";
2+
3+
export class AddTotalBalance1567485550388 implements MigrationInterface {
4+
5+
public async up(queryRunner: QueryRunner): Promise<any> {
6+
await queryRunner.addColumn('address', new TableColumn({
7+
name: 'totalBalance',
8+
type: 'varchar',
9+
default: '0',
10+
}))
11+
}
12+
13+
public async down(queryRunner: QueryRunner): Promise<any> {
14+
await queryRunner.dropColumn('address', 'totalBalance')
15+
}
16+
17+
}

packages/neuron-wallet/src/database/address/ormconfig.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import Address from './entities/address'
88

99
import { AddAddress1561461669542 } from './migrations/1561461669542-AddAddress'
1010
import { extendBalance1562126909151 } from './migrations/1562126909151-extendBalance'
11+
import { AddTotalBalance1567485550388 } from './migrations/1567485550388-AddTotalBalance'
1112

1213
const dbPath = path.join(env.fileBasePath, 'address.sqlite')
1314

@@ -20,7 +21,7 @@ const connectOptions = (): SqliteConnectionOptions => {
2021
type: 'sqlite',
2122
database,
2223
entities: [Address],
23-
migrations: [AddAddress1561461669542, extendBalance1562126909151],
24+
migrations: [AddAddress1561461669542, extendBalance1562126909151, AddTotalBalance1567485550388],
2425
synchronize: false,
2526
migrationsRun: true,
2627
logging: ['error'],

0 commit comments

Comments
 (0)