Merge pull request #2313 from planetarium/fix-stagepolicy

use concurrent dictionary in stagepolicy
planetarium · Dec 19, 2023 · b606e21 · b606e21
2 parents 0237a58 + b7f69a1
commit b606e21
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Lib9c.Policy/NCStagePolicy.cs b/Lib9c.Policy/NCStagePolicy.cs
@@ -15,7 +15,7 @@ public class NCStagePolicy : IStagePolicy
         private readonly VolatileStagePolicy _impl;
         private readonly ConcurrentDictionary<Address, SortedList<Transaction, TxId>> _txs;
         private readonly int _quotaPerSigner;
-        private readonly Dictionary<Address, int> _quotaPerSignerList;
+        private readonly ConcurrentDictionary<Address, int> _quotaPerSignerList;
         private IAccessControlService? _accessControlService;
 
         public NCStagePolicy(TimeSpan txLifeTime, int quotaPerSigner, IAccessControlService? accessControlService = null)
@@ -32,7 +32,7 @@ public NCStagePolicy(TimeSpan txLifeTime, int quotaPerSigner, IAccessControlServ
                 ? new VolatileStagePolicy()
                 : new VolatileStagePolicy(txLifeTime);
 
-            _quotaPerSignerList = new Dictionary<Address, int>();
+            _quotaPerSignerList = new ConcurrentDictionary<Address, int>();
             _accessControlService = accessControlService;
         }
 
@@ -52,7 +52,7 @@ public IEnumerable<Transaction> Iterate(BlockChain blockChain, bool filtered = t
         {
             if (filtered)
             {
-                var txsPerSigner = new Dictionary<Address, SortedSet<Transaction>>();
+                var txsPerSigner = new ConcurrentDictionary<Address, SortedSet<Transaction>>();
                 foreach (Transaction tx in _impl.Iterate(blockChain, filtered))
                 {
                     if (!txsPerSigner.TryGetValue(tx.Signer, out var s))