Cleanup

planetarium · Nov 15, 2023 · 2cb9598 · 2cb9598
1 parent 36edcf8
commit 2cb9598
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 17 deletions.
diff --git a/Libplanet.Extensions.Cocona/Commands/MptCommand.cs b/Libplanet.Extensions.Cocona/Commands/MptCommand.cs
@@ -69,14 +69,12 @@ public void Diff(
         kvStoreUri = ConvertKVStoreUri(kvStoreUri, toolConfiguration);
         otherKvStoreUri = ConvertKVStoreUri(otherKvStoreUri, toolConfiguration);
 
-        IKeyValueStore keyValueStore = LoadKVStoreFromURI(kvStoreUri);
-        IKeyValueStore otherKeyValueStore = LoadKVStoreFromURI(otherKvStoreUri);
-        var trie = new MerkleTrie(
-            keyValueStore,
-            HashDigest<SHA256>.FromString(stateRootHashHex));
-        var otherTrie = new MerkleTrie(
-            otherKeyValueStore,
-            HashDigest<SHA256>.FromString(otherStateRootHashHex));
+        IStateStore stateStore = new TrieStateStore(LoadKVStoreFromURI(kvStoreUri));
+        IStateStore otherStateStore = new TrieStateStore(LoadKVStoreFromURI(otherKvStoreUri));
+        var trie =
+            stateStore.GetStateRoot(HashDigest<SHA256>.FromString(stateRootHashHex));
+        var otherTrie =
+            otherStateStore.GetStateRoot(HashDigest<SHA256>.FromString(otherStateRootHashHex));
 
         var codec = new Codec();
         HashDigest<SHA256> originRootHash = trie.Hash;
@@ -113,10 +111,8 @@ public void Export(
         ToolConfiguration toolConfiguration = configurationService.Load();
         kvStoreUri = ConvertKVStoreUri(kvStoreUri, toolConfiguration);
 
-        IKeyValueStore keyValueStore = LoadKVStoreFromURI(kvStoreUri);
-        var trie = new MerkleTrie(
-            keyValueStore,
-            HashDigest<SHA256>.FromString(stateRootHashHex));
+        IStateStore stateStore = new TrieStateStore(LoadKVStoreFromURI(kvStoreUri));
+        var trie = stateStore.GetStateRoot(HashDigest<SHA256>.FromString(stateRootHashHex));
         var codec = new Codec();
 
         // This assumes the original key was encoded from a sensible string.

diff --git a/Libplanet.Store/HashNodeCache.cs b/Libplanet.Store/HashNodeCache.cs
@@ -20,7 +20,7 @@ public class HashNodeCache
         private int _hits;
         private object _reportLock;
 
-        public HashNodeCache()
+        internal HashNodeCache()
         {
             _cache = new LruCache<HashDigest<SHA256>, IValue>(_cahceSize);
             _stopwatch = new Stopwatch();

diff --git a/Libplanet.Store/Trie/MerkleTrie.cs b/Libplanet.Store/Trie/MerkleTrie.cs
@@ -38,12 +38,12 @@ static MerkleTrie()
         /// nodes.</param>
         /// <param name="rootHash">The root <see cref="ITrie.Hash"/> of
         /// <see cref="MerkleTrie"/>.</param>
-        /// <param name="cache">The <see cref="HashNodeCache"/> to use as a cache.</param>
+        /// <param name="cache">The <see cref="HashNodeCache"/> to use as cache.</param>
         public MerkleTrie(
             IKeyValueStore keyValueStore,
             HashDigest<SHA256> rootHash,
             HashNodeCache? cache = null)
-            : this(keyValueStore, new HashNode(rootHash))
+            : this(keyValueStore, new HashNode(rootHash), cache)
         {
         }
 
@@ -54,9 +54,11 @@ public MerkleTrie(
         /// nodes.</param>
         /// <param name="root">The root node of <see cref="MerkleTrie"/>.  If it is
         /// <see langword="null"/>, it will be treated like empty trie.</param>
-        /// <param name="cache">The <see cref="HashNodeCache"/> to use as a cache.</param>
+        /// <param name="cache">The <see cref="HashNodeCache"/> to use as cache.</param>
         public MerkleTrie(
-            IKeyValueStore keyValueStore, INode? root = null, HashNodeCache? cache = null)
+            IKeyValueStore keyValueStore,
+            INode? root = null,
+            HashNodeCache? cache = null)
         {
             // FIXME: It might be a good idea to have something like IReadOnlyKeyValueStore.
             KeyValueStore = keyValueStore;