Get staged txs from StagePolicy

planetarium · Sep 25, 2024 · f075b75 · f075b75
1 parent 2b3dfb4
commit f075b75
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/NineChronicles.Headless/GraphTypes/NodeStatus.cs b/NineChronicles.Headless/GraphTypes/NodeStatus.cs
@@ -99,15 +99,15 @@ public NodeStatusType(StandaloneContext context, IBlockChainRepository blockChai
 
                     if (!fieldContext.HasArgument("address"))
                     {
-                        return context.BlockChain.GetStagedTransactionIds();
+                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain).Select(tx => tx.Id)
+                            .ToImmutableHashSet();
                     }
                     else
                     {
                         Address address = fieldContext.GetArgument<Address>("address");
-                        IImmutableSet<TxId> stagedTransactionIds = context.BlockChain.GetStagedTransactionIds();
 
-                        return stagedTransactionIds.Where(txId =>
-                        context.BlockChain.GetTransaction(txId).Signer.Equals(address));
+                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain)
+                            .Where(tx => tx.Signer.Equals(address)).Select(tx => tx.Id).ToImmutableHashSet();
                     }
                 }
             );
@@ -121,7 +121,7 @@ public NodeStatusType(StandaloneContext context, IBlockChainRepository blockChai
                         throw new InvalidOperationException($"{nameof(context.BlockChain)} is null.");
                     }
 
-                    return context.BlockChain.GetStagedTransactionIds().Count;
+                    return context.BlockChain.StagePolicy.Iterate(context.BlockChain).ToImmutableHashSet().Count;
                 }
             );
             Field<NonNullGraphType<BlockHeaderType>>(