mirror of
https://github.com/Kugelschieber/migo.git
synced 2026-01-18 14:50:29 +00:00
38 lines
558 B
Go
38 lines
558 B
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
badger "github.com/dgraph-io/badger/v3"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
db *badger.DB
|
|
)
|
|
|
|
// Init initializes the database.
|
|
func Init() error {
|
|
dataDir := os.Getenv("MIGO_DATA_DIR")
|
|
|
|
if dataDir == "" {
|
|
return errors.New("MIGO_DATA_DIR not configured")
|
|
}
|
|
|
|
var err error
|
|
db, err = badger.Open(badger.DefaultOptions(dataDir))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Close closes the database connection.
|
|
func Close() {
|
|
if err := db.Close(); err != nil {
|
|
log.Printf("Error closing database connection: %v", err)
|
|
}
|
|
}
|