Merge pull request #33 from blugelabs/config-with-custom-directory

method to create config with custom diredtory
blugelabs · Oct 6, 2020 · 43335cc · 43335cc
2 parents e32d4b0 + 3b3051c
commit 43335cc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/config.go b/config.go
@@ -89,6 +89,10 @@ func InMemoryOnlyConfig() Config {
 	indexConfig := index.InMemoryOnlyConfig()
 	return defaultConfig(indexConfig)
 }
+func DefaultConfigWithDirectory(df func() index.Directory) Config {
+	indexConfig := index.DefaultConfigWithDirectory(df)
+	return defaultConfig(indexConfig)
+}
 
 func defaultConfig(indexConfig index.Config) Config {
 	rv := Config{

diff --git a/index/config.go b/index/config.go
@@ -145,6 +145,12 @@ func InMemoryOnlyConfig() Config {
 	return rv
 }
 
+func DefaultConfigWithDirectory(df func() Directory) Config {
+	rv := defaultConfig()
+	rv.DirectoryFunc = df
+	return rv
+}
+
 func defaultConfig() Config {
 	rv := Config{
 		SegmentType:      ice.Type,