feat: add inputIndex to inputs

nervosnetwork · Nov 11, 2019 · c760db5 · c760db5
1 parent 5a3f8a5
commit c760db5
Show file tree

Hide file tree

Showing 7 changed files with 34 additions and 3 deletions.
diff --git a/packages/neuron-wallet/src/database/chain/entities/input.ts b/packages/neuron-wallet/src/database/chain/entities/input.ts
@@ -48,6 +48,12 @@ export default class Input extends BaseEntity {
   })
   capacity: string | null = null
 
+  @Column({
+    type: 'varchar',
+    nullable: true,
+  })
+  inputIndex: string | null = null
+
   public previousOutput(): OutPoint | null {
     if (!this.outPointTxHash || !this.outPointIndex) {
       return null

diff --git a/packages/neuron-wallet/src/database/chain/migrations/1573461100330-AddInputIndexToInput.ts b/packages/neuron-wallet/src/database/chain/migrations/1573461100330-AddInputIndexToInput.ts
@@ -0,0 +1,17 @@
+import {MigrationInterface, QueryRunner, TableColumn} from "typeorm";
+
+export class AddInputIndexToInput1573461100330 implements MigrationInterface {
+
+  public async up(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.addColumn('input', new TableColumn({
+      name: 'inputIndex',
+      type: 'varchar',
+      isNullable: true,
+    }))
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.dropColumn('input', 'inputIndex')
+  }
+
+}
diff --git a/packages/neuron-wallet/src/database/chain/ormconfig.ts b/packages/neuron-wallet/src/database/chain/ormconfig.ts
@@ -16,6 +16,7 @@ import { AddLockToInput1570522869590 } from './migrations/1570522869590-AddLockT
 import { AddIndices1572006450765 } from './migrations/1572006450765-AddIndices'
 import { AddIndexToTxTimestamp1572137226866 } from './migrations/1572137226866-AddIndexToTxTimestamp'
 import { AddOutputIndex1572226722928 } from './migrations/1572226722928-AddOutputIndex'
+import { AddInputIndexToInput1573461100330 } from './migrations/1573461100330-AddInputIndexToInput'
 
 export const CONNECTION_NOT_FOUND_NAME = 'ConnectionNotFoundError'
 
@@ -43,7 +44,8 @@ const connectOptions = async (genesisBlockHash: string): Promise<SqliteConnectio
       AddLockToInput1570522869590,
       AddIndices1572006450765,
       AddIndexToTxTimestamp1572137226866,
-      AddOutputIndex1572226722928
+      AddOutputIndex1572226722928,
+      AddInputIndexToInput1573461100330,
     ],
     logging,
     maxQueryExecutionTime: 30

diff --git a/packages/neuron-wallet/src/services/indexer/queue.ts b/packages/neuron-wallet/src/services/indexer/queue.ts
@@ -214,7 +214,7 @@ export default class IndexerQueue {
           let txEntity: TransactionEntity | undefined = await TransactionPersistor.get(transaction.hash)
           if (!txEntity || !txEntity.blockHash) {
             if (!txEntity) {
-              for (const input of transaction.inputs!) {
+              for (const [inputIndex, input] of transaction.inputs!.entries()) {
                 if (input.previousOutput!.txHash === this.emptyTxHash) {
                   continue
                 }
@@ -224,6 +224,7 @@ export default class IndexerQueue {
                 input.lock = previousOutput.lock
                 input.lockHash = LockUtils.lockScriptToHash(input.lock)
                 input.capacity = previousOutput.capacity
+                input.inputIndex = inputIndex.toString()
               }
             }
             const { blockHash } = transactionWithStatus.txStatus

diff --git a/packages/neuron-wallet/src/services/sync/get-blocks.ts b/packages/neuron-wallet/src/services/sync/get-blocks.ts
@@ -48,7 +48,7 @@ export default class GetBlocks {
         const addresses = await checkTx.check(lockHashes)
         if (addresses.length > 0) {
           if (i > 0) {
-            for (const input of tx.inputs!) {
+            for (const [inputIndex, input] of tx.inputs!.entries()) {
               const previousTxHash = input.previousOutput!.txHash
               let previousTxWithStatus = cachedPreviousTxs.get(previousTxHash)
               if (!previousTxWithStatus) {
@@ -60,6 +60,7 @@ export default class GetBlocks {
               input.lock = previousOutput.lock
               input.lockHash = LockUtils.lockScriptToHash(input.lock)
               input.capacity = previousOutput.capacity
+              input.inputIndex = inputIndex.toString()
             }
           }
           await TransactionPersistor.saveFetchTx(tx)

diff --git a/packages/neuron-wallet/src/services/tx/transaction-persistor.ts b/packages/neuron-wallet/src/services/tx/transaction-persistor.ts
@@ -134,6 +134,9 @@ export class TransactionPersistor {
       input.lockHash = i.lockHash || null
       input.lock = i.lock || null
       input.since = i.since!
+      if (i.inputIndex) {
+        input.inputIndex = i.inputIndex
+      }
       inputs.push(input)
 
       if (outPoint) {

diff --git a/packages/neuron-wallet/src/types/cell-types.ts b/packages/neuron-wallet/src/types/cell-types.ts
@@ -79,6 +79,7 @@ export interface Input {
   capacity?: string | null
   lockHash?: string | null
   lock?: Script | null
+  inputIndex?: string
 }
 
 export interface Cell {