add whitelist

planetarium · Oct 25, 2023 · 76454a6 · 76454a6
1 parent fdae923
commit 76454a6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/Lib9c.Policy/AccessControlService/IAccessControlService.cs b/Lib9c.Policy/AccessControlService/IAccessControlService.cs
@@ -5,5 +5,7 @@ namespace Nekoyume.Blockchain
     public interface IAccessControlService
     {
         public bool IsAccessDenied(Address address);
+
+        public int GetAccessLevel(Address address);
     }
 }
diff --git a/Lib9c.Policy/NCStagePolicy.cs b/Lib9c.Policy/NCStagePolicy.cs
@@ -61,7 +61,11 @@ public IEnumerable<Transaction> Iterate(BlockChain blockChain, bool filtered = t
                     s.Add(tx);
                     if (s.Count > _quotaPerSigner)
                     {
-                        s.Remove(s.Max);
+                        var accessLevel = _accessControlService?.GetAccessLevel(tx.Signer);
+                        if (accessLevel <= 0)
+                        {
+                            s.Remove(s.Max);
+                        }
                     }
                 }
 
@@ -77,7 +81,8 @@ public IEnumerable<Transaction> Iterate(BlockChain blockChain, bool filtered = t
 
         public bool Stage(BlockChain blockChain, Transaction transaction)
         {
-            if (_accessControlService != null && _accessControlService.IsAccessDenied(transaction.Signer))
+            var accessLevel = _accessControlService?.GetAccessLevel(transaction.Signer);
+            if (accessLevel == 0)
             {
                 return false;
             }