fix: broadcast address info updated

nervosnetwork · Jul 29, 2019 · 7aa4c83 · 7aa4c83
1 parent ce34ee6
commit 7aa4c83
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/neuron-wallet/src/database/address/entities/address.ts b/packages/neuron-wallet/src/database/address/entities/address.ts
@@ -1,3 +1,4 @@
+import { remote } from 'electron'
 import { Entity, BaseEntity, PrimaryColumn, Column, AfterInsert, AfterUpdate, AfterRemove } from 'typeorm'
 import { AddressType } from '../../../models/keys/address'
 import { Address as AddressInterface } from '../dao'
@@ -8,6 +9,11 @@ export enum AddressVersion {
   Mainnet = 'mainnet',
 }
 
+const isRenderer = process && process.type === 'renderer'
+const addressDbChangedSubject = isRenderer
+  ? remote.require('./models/subjects/address-db-changed-subject').default.getSubject()
+  : AddressDbChangedSubject.getSubject()
+
 @Entity()
 export default class Address extends BaseEntity {
   @PrimaryColumn({
@@ -105,6 +111,6 @@ export default class Address extends BaseEntity {
   }
 
   private changed = (event: string) => {
-    AddressDbChangedSubject.getSubject().next(event)
+    addressDbChangedSubject.next(event)
   }
 }