feat(rocksdb): configure rocksdb options

planetarium · Nov 3, 2020 · b573215 · b573215
1 parent 8d21c1c
commit b573215
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/Libplanet.RocksDBStore/RocksDBStore.cs b/Libplanet.RocksDBStore/RocksDBStore.cs
@@ -69,11 +69,17 @@ private readonly Dictionary<Guid, LruCache<string, Tuple<HashDigest<SHA256>, lon
         /// <param name="blockCacheSize">The capacity of the block cache.</param>
         /// <param name="txCacheSize">The capacity of the transaction cache.</param>
         /// <param name="statesCacheSize">The capacity of the states cache.</param>
+        /// <param name="maxTotalWalSize">The number to configure max_total_wal_size RocksDB option.
+        /// </param>
+        /// <param name="keepLogFileNum">The number to configure keep_log_file_num RocksDB option.
+        /// </param>
         public RocksDBStore(
             string path,
             int blockCacheSize = 512,
             int txCacheSize = 1024,
-            int statesCacheSize = 10000
+            int statesCacheSize = 10000,
+            ulong? maxTotalWalSize = null,
+            ulong? keepLogFileNum = null
         )
         {
             _logger = Log.ForContext<RocksDBStore>();
@@ -102,6 +108,16 @@ public RocksDBStore(
             _options = new DbOptions()
                 .SetCreateIfMissing();
 
+            if (maxTotalWalSize is ulong maxTotalWalSizeValue)
+            {
+                _options = _options.SetMaxTotalWalSize(maxTotalWalSizeValue);
+            }
+
+            if (keepLogFileNum is ulong keepLogFileNumValue)
+            {
+                _options = _options.SetKeepLogFileNum(keepLogFileNumValue);
+            }
+
             _blockDb = RocksDBUtils.OpenRocksDb(_options, RocksDbPath(BlockDbName));
             _txDb = RocksDBUtils.OpenRocksDb(_options, RocksDbPath(TxDbName));
             _stateDb = RocksDBUtils.OpenRocksDb(_options, RocksDbPath(StateDbName));