Adding Neo.Transaction.GetUnspentCoins for issue #114 (#123)

* Adding Neo.Transaction.GetUnspentCoins for issue #114 * Add TestBlockchain.GetUnspent(UInt256 hash)
neo-project · Dec 11, 2017 · 46b4ce0 · 46b4ce0
1 parent b3b195e
commit 46b4ce0
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 0 deletions.
diff --git a/neo.UnitTests/TestBlockchain.cs b/neo.UnitTests/TestBlockchain.cs
@@ -133,6 +133,11 @@ public override TransactionOutput GetUnspent(UInt256 hash, ushort index)
             throw new NotImplementedException();
         }
 
+        public override IEnumerable<TransactionOutput> GetUnspent(UInt256 hash)
+        {
+            throw new NotImplementedException();
+        }
+
         public override IEnumerable<VoteState> GetVotes(IEnumerable<Transaction> others)
         {
             VoteState vs = new VoteState() { Count = Fixed8.FromDecimal(1), PublicKeys = TestUtils.StandbyValidators};            

diff --git a/neo/Core/Blockchain.cs b/neo/Core/Blockchain.cs
@@ -425,6 +425,9 @@ public Transaction GetTransaction(UInt256 hash)
         /// <returns>返回一个交易输出，表示一个未花费的资产</returns>
         public abstract TransactionOutput GetUnspent(UInt256 hash, ushort index);
 
+        public abstract IEnumerable<TransactionOutput> GetUnspent(UInt256 hash);
+
+
         /// <summary>
         /// 获取选票信息
         /// </summary>

diff --git a/neo/Implementations/Blockchains/LevelDB/LevelDBBlockchain.cs b/neo/Implementations/Blockchains/LevelDB/LevelDBBlockchain.cs
@@ -357,6 +357,30 @@ public override TransactionOutput GetUnspent(UInt256 hash, ushort index)
             }
         }
 
+        public override IEnumerable<TransactionOutput> GetUnspent(UInt256 hash)
+        {
+            ReadOptions options = new ReadOptions();
+            using (options.Snapshot = db.GetSnapshot())
+            {
+                List<TransactionOutput> outputs = new List<TransactionOutput>();
+                UnspentCoinState state = db.TryGet<UnspentCoinState>(options, DataEntryPrefix.ST_Coin, hash);
+                if (state != null)
+                {
+                    int height;
+                    Transaction tx = GetTransaction(options, hash, out height);
+                    for (int i = 0; i < state.Items.Length; i++)
+                    {
+                        if (!state.Items[i].HasFlag(CoinState.Spent))
+                        {
+                            outputs.Add(tx.Outputs[i]);
+                        }
+
+                    }
+                }
+                return outputs;
+            }
+        }
+
         public override IEnumerable<VoteState> GetVotes(IEnumerable<Transaction> others)
         {
             ReadOptions options = new ReadOptions();

diff --git a/neo/SmartContract/ApplicationEngine.cs b/neo/SmartContract/ApplicationEngine.cs
@@ -350,6 +350,7 @@ protected virtual long GetPriceForSysCall()
                     return 100;
                 case "Neo.Transaction.GetReferences":
                 case "AntShares.Transaction.GetReferences":
+                case "Neo.Transaction.GetUnspentCoins":    
                     return 200;
                 case "Neo.Account.SetVotes":
                 case "AntShares.Account.SetVotes":

diff --git a/neo/SmartContract/StateReader.cs b/neo/SmartContract/StateReader.cs
@@ -46,6 +46,7 @@ public StateReader()
             Register("Neo.Transaction.GetInputs", Transaction_GetInputs);
             Register("Neo.Transaction.GetOutputs", Transaction_GetOutputs);
             Register("Neo.Transaction.GetReferences", Transaction_GetReferences);
+            Register("Neo.Transaction.GetUnspentCoins", Transaction_GetUnspentCoins);
             Register("Neo.Attribute.GetUsage", Attribute_GetUsage);
             Register("Neo.Attribute.GetData", Attribute_GetData);
             Register("Neo.Input.GetHash", Input_GetHash);
@@ -418,6 +419,14 @@ protected virtual bool Transaction_GetReferences(ExecutionEngine engine)
             return true;
         }
 
+        protected virtual bool Transaction_GetUnspentCoins(ExecutionEngine engine)
+        {
+            Transaction tx = engine.EvaluationStack.Pop().GetInterface<Transaction>();
+            if (tx == null) return false;
+            engine.EvaluationStack.Push(Blockchain.Default.GetUnspent(tx.Hash).Select(p => StackItem.FromInterface(p)).ToArray());
+            return true;
+        }
+
         protected virtual bool Attribute_GetUsage(ExecutionEngine engine)
         {
             TransactionAttribute attr = engine.EvaluationStack.Pop().GetInterface<TransactionAttribute>();