Shargon's Feedback

neo-project · Nov 3, 2020 · 70fb527 · 70fb527
1 parent 94965aa
commit 70fb527
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/src/neo/Ledger/ContractState.cs b/src/neo/Ledger/ContractState.cs
@@ -75,17 +75,20 @@ public JObject ToJson()
         {
             JObject json = new JObject();
             json["id"] = Id;
+
             if (Id < 0)
             {
-                foreach (var native in NativeContract.Contracts)
+                var native = NativeContract.GetContract(Id);
+                if (native != null) json["name"] = native.Name;
+            }
+            else
+            {
+                if (Manifest.Extra != null && Manifest.Extra["name"] != null)
                 {
-                    if (Id == native.Id)
-                    {
-                        json["name"] = native.Name;
-                        break;
-                    }
+                    json["name"] = Manifest.Extra["name"];
                 }
             }
+
             json["hash"] = ScriptHash.ToString();
             json["script"] = Convert.ToBase64String(Script);
             json["manifest"] = Manifest.ToJson();

diff --git a/src/neo/SmartContract/Native/NativeContract.cs b/src/neo/SmartContract/Native/NativeContract.cs
@@ -17,6 +17,7 @@ namespace Neo.SmartContract.Native
     public abstract class NativeContract
     {
         private static readonly List<NativeContract> contractsList = new List<NativeContract>();
+        private static readonly Dictionary<int, NativeContract> contractsIdDictionary = new Dictionary<int, NativeContract>();
         private static readonly Dictionary<string, NativeContract> contractsNameDictionary = new Dictionary<string, NativeContract>();
         private static readonly Dictionary<UInt160, NativeContract> contractsHashDictionary = new Dictionary<UInt160, NativeContract>();
         private readonly Dictionary<string, ContractMethodMetadata> methods = new Dictionary<string, ContractMethodMetadata>();
@@ -77,6 +78,7 @@ protected NativeContract()
                 Extra = null
             };
             contractsList.Add(this);
+            contractsIdDictionary.Add(Id, this);
             contractsNameDictionary.Add(Name, this);
             contractsHashDictionary.Add(Hash, this);
         }
@@ -104,6 +106,12 @@ public static NativeContract GetContract(string name)
             return contract;
         }
 
+        public static NativeContract GetContract(int id)
+        {
+            contractsIdDictionary.TryGetValue(id, out var contract);
+            return contract;
+        }
+
         internal void Invoke(ApplicationEngine engine)
         {
             if (!engine.CurrentScriptHash.Equals(Hash))