Introduce filtered argument at .nodeStatus.stagedTxIds[Count]

moreal · Aug 30, 2024 · e97ef84 · e97ef84
1 parent 11b8caa
commit e97ef84
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/NineChronicles.Headless/GraphTypes/NodeStatus.cs b/NineChronicles.Headless/GraphTypes/NodeStatus.cs
@@ -95,6 +95,12 @@ public NodeStatusType(StandaloneContext context)
                     {
                         Name = "address",
                         Description = "Target address to query"
+                    },
+                    new QueryArgument<BooleanGraphType>
+                    {
+                        Name = "filtered",
+                        Description = "Whether to filter masked staged Transactions or not.",
+                        DefaultValue = true,
                     }
                 ),
                 description: "Ids of staged transactions from the current node.",
@@ -104,32 +110,42 @@ public NodeStatusType(StandaloneContext context)
                     {
                         throw new InvalidOperationException($"{nameof(context.BlockChain)} is null.");
                     }
-
+
+                    var filtered = fieldContext.GetArgument<bool>("filtered");
                     if (!fieldContext.HasArgument("address"))
                     {
-                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain).Select(tx => tx.Id)
+                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain, filtered).Select(tx => tx.Id)
                             .ToImmutableHashSet();
                     }
                     else
                     {
                         Address address = fieldContext.GetArgument<Address>("address");
 
-                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain)
+                        return context.BlockChain.StagePolicy.Iterate(context.BlockChain, filtered)
                             .Where(tx => tx.Signer.Equals(address)).Select(tx => tx.Id).ToImmutableHashSet();
                     }
                 }
             );
             Field<IntGraphType>(
                 name: "stagedTxIdsCount",
                 description: "The number of ids of staged transactions from the current node.",
+                arguments: new QueryArguments(
+                    new QueryArgument<BooleanGraphType>
+                    {
+                        Name = "filtered",
+                        Description = "Whether to filter masked staged Transactions or not.",
+                        DefaultValue = true,
+                    }
+                ),
                 resolve: fieldContext =>
                 {
                     if (context.BlockChain is null)
                     {
                         throw new InvalidOperationException($"{nameof(context.BlockChain)} is null.");
                     }
 
-                    return context.BlockChain.StagePolicy.Iterate(context.BlockChain).ToImmutableHashSet().Count;
+                    var filtered = fieldContext.GetArgument<bool>("filtered");
+                    return context.BlockChain.StagePolicy.Iterate(context.BlockChain, filtered).ToImmutableHashSet().Count;
                 }
             );
             Field<NonNullGraphType<BlockHeaderType>>(