Optimize RocksDb startup (neo-project#333)

chenzhitong · Oct 13, 2020 · e522943 · e522943
1 parent 9a5c289
commit e522943
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/RocksDBStore/Plugins/Storage/Store.cs b/src/RocksDBStore/Plugins/Storage/Store.cs
@@ -20,8 +20,16 @@ internal class Store : IStore
         public Store(string path)
         {
             var families = new ColumnFamilies();
-            for (int x = 0; x <= byte.MaxValue; x++)
-                families.Add(new ColumnFamilies.Descriptor(x.ToString(), new ColumnFamilyOptions()));
+
+            try
+            {
+                foreach (var family in RocksDb.ListColumnFamilies(Options.Default, Path.GetFullPath(path)))
+                {
+                    families.Add(new ColumnFamilies.Descriptor(family, new ColumnFamilyOptions()));
+                }
+            }
+            catch { }
+
             db = RocksDb.Open(Options.Default, Path.GetFullPath(path), families);
 
             ColumnFamilyHandle defaultFamily = db.GetDefaultColumnFamily();