add config

hstreamdb · Jun 19, 2024 · 23dc65d · 23dc65d
1 parent 1de7b0b
commit 23dc65d
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/conf/hstream.yaml b/conf/hstream.yaml
@@ -141,6 +141,10 @@ hserver:
       las: hstreamdb/sink-las:latest
       elasticsearch: hstreamdb/sink-elasticsearch:latest
 
+  cache-store:
+    # the cache store work directory
+    cache-store-path: /data/cache-store
+
   # Internal grpc settings (typically, there's no need to modify these)
   #grpc:
   #  channel-args:

diff --git a/hstream/src/HStream/Server/Config.hs b/hstream/src/HStream/Server/Config.hs
@@ -126,6 +126,8 @@ data ServerOpts = ServerOpts
   , _querySnapshotPath            :: !FilePath
   , experimentalFeatures          :: ![ExperimentalFeature]
 
+  , _cacheStorePath               :: !FilePath
+
 #ifndef HStreamUseGrpcHaskell
   , grpcChannelArgs               :: ![HsGrpc.ChannelArg]
 #endif
@@ -263,6 +265,10 @@ parseJSONToOptions CliOptions{..} obj = do
 
   let experimentalFeatures = cliExperimentalFeatures
 
+  cacheStoreCfg <- nodeCfgObj .:? "cache-store" .!= mempty
+  cacheStorePath <- cacheStoreCfg .:? "cache-store-path" .!= "/data/cache_store"
+  let _cacheStorePath = fromMaybe cacheStorePath cliCacheStorePath
+
 #ifndef HStreamUseGrpcHaskell
   grpcCfg <- nodeCfgObj .:? "grpc" .!= mempty
   grpcChanArgsCfg <- grpcCfg .:? "channel-args" .!= mempty

diff --git a/hstream/src/HStream/Server/Configuration/Cli.hs b/hstream/src/HStream/Server/Configuration/Cli.hs
@@ -138,6 +138,7 @@ data CliOptions = CliOptions
 
     -- Experimental Features
   , cliExperimentalFeatures         :: ![ExperimentalFeature]
+  , cliCacheStorePath               :: !(Maybe FilePath)
 
     -- Internal options
   , cliStoreCompression             :: !(Maybe Compression)
@@ -182,6 +183,8 @@ cliOptionsParser = do
 
   cliExperimentalFeatures <- O.many experimentalFeatureParser
 
+  cliCacheStorePath     <- optional cacheStorePathParser
+
   cliStoreCompression <- optional storeCompressionParser
 
   return CliOptions{..}
@@ -455,6 +458,12 @@ querySnapshotPathParser = strOption
   <> metavar "PATH" <> value "/data/query_snapshots"
   <> help "hstream query snapshot store path"
 
+cacheStorePathParser :: O.Parser FilePath
+cacheStorePathParser = strOption
+  $  long "cache-store-path"
+  <> metavar "PATH" <> value "/data/cache_store"
+  <> help "hstream cache store path"
+
 data ExperimentalFeature
   = ExperimentalStreamV2
   deriving (Show, Eq)