diff --git a/db_badger.go b/db_badger.go new file mode 100644 index 0000000..fdf1a3f --- /dev/null +++ b/db_badger.go @@ -0,0 +1,14 @@ +// +build badger + +package main + +import ( + "github.com/fiatjaf/eventstore/badger" +) + +func getDB() badger.BadgerBackend { + return badger.BadgerBackend{ + Path: getEnv("DB_PATH"), + } +} + diff --git a/db_default.go b/db_default.go new file mode 100644 index 0000000..1aa5937 --- /dev/null +++ b/db_default.go @@ -0,0 +1,14 @@ +// +build !badger + +package main + +import ( + "github.com/fiatjaf/eventstore/lmdb" +) + +func getDB() lmdb.LMDBBackend { + return lmdb.LMDBBackend{ + Path: getEnv("DB_PATH"), + } +} + diff --git a/main.go b/main.go index 83b376a..716f806 100644 --- a/main.go +++ b/main.go @@ -11,7 +11,6 @@ import ( "time" "github.com/cespare/xxhash" - "github.com/fiatjaf/eventstore/lmdb" "github.com/fiatjaf/khatru" "github.com/greatroar/blobloom" "github.com/joho/godotenv" @@ -62,9 +61,7 @@ func main() { relay.Info.Description = config.RelayDescription appendPubkey(config.RelayPubkey) - db := lmdb.LMDBBackend{ - Path: config.DBPath, - } + db := getDB() if err := db.Init(); err != nil { panic(err) }