diff --git a/cmd/db/main.go b/cmd/db/main.go index fd7d09a..6dbbca4 100644 --- a/cmd/db/main.go +++ b/cmd/db/main.go @@ -124,7 +124,11 @@ func printDB() { } func main() { - DB = kv.InitDB() + var conf kv.Config = kv.Config{ + Path: "./data/", + } + + DB = kv.InitDB(conf) defer DB.Close() printDB() diff --git a/pkg/db/kv/init.go b/pkg/db/kv/init.go index ebc2f1f..18148df 100644 --- a/pkg/db/kv/init.go +++ b/pkg/db/kv/init.go @@ -4,8 +4,8 @@ import ( "github.com/dgraph-io/badger/v4" ) -func InitDB() *badger.DB { - opts := badger.DefaultOptions("./data") +func InitDB(conf Config) *badger.DB { + opts := badger.DefaultOptions(conf.Path) opts.Logger = nil opts.ValueLogFileSize = 10 << 20 // 10 Mo par fichier vlog opts.NumMemtables = 1 diff --git a/pkg/db/kv/struct.go b/pkg/db/kv/struct.go new file mode 100644 index 0000000..71b195f --- /dev/null +++ b/pkg/db/kv/struct.go @@ -0,0 +1,5 @@ +package kv + +type Config struct { + Path string +}