[lmdb facebookincubator#5] Enable RockDB cache in benchmarks

Author: simonmar

glean/bench/BenchDB.hs

@@ -39,7 +39,7 @@ withBenchDB conf num act = do
       | useLMDB conf = [setLMDBStorage]
       | otherwise = []
 
-  withEmptyTestDB settings $ \env repo -> do
+  withEmptyTestDB (enableRocksDBCache : settings) $ \env repo -> do
   withOpenDatabase env repo $ \odb ->
     void $ return odb
 

glean/test/lib/Glean/Database/Test.hs

@@ -22,6 +22,7 @@ module Glean.Database.Test
   , setMaxSetSize
   , enableTcDebug
   , enableQueryDebug
+  , enableRocksDBCache
   , withTestEnv
   , kickOffTestDB
   , waitUntilComplete
@@ -120,6 +121,13 @@ enableQueryDebug :: Setting
 enableQueryDebug cfg = cfg
   { cfgDebug = (cfgDebug cfg) { queryDebug = True } }
 
+enableRocksDBCache :: Setting
+enableRocksDBCache cfg = cfg
+  { cfgServerConfig = (cfgServerConfig cfg) <&> \scfg -> scfg
+      { ServerConfig.config_db_rocksdb_cache_mb =
+        ServerConfig.config_db_rocksdb_cache_mb def }
+  }
+
 withTestEnv
   :: [Setting]
   -> (Env -> IO a)