mirror of
https://github.com/pikami/cosmium.git
synced 2025-03-13 13:26:25 +00:00
38 lines
650 B
Go
38 lines
650 B
Go
package badgerdatastore
|
|
|
|
import (
|
|
"encoding/gob"
|
|
|
|
"github.com/dgraph-io/badger/v4"
|
|
"github.com/pikami/cosmium/internal/logger"
|
|
)
|
|
|
|
type BadgerDataStore struct {
|
|
db *badger.DB
|
|
}
|
|
|
|
func NewBadgerDataStore() *BadgerDataStore {
|
|
gob.Register([]interface{}{})
|
|
|
|
badgerOpts := badger.DefaultOptions("").WithInMemory(true)
|
|
|
|
db, err := badger.Open(badgerOpts)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &BadgerDataStore{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *BadgerDataStore) Close() {
|
|
r.db.Close()
|
|
r.db = nil
|
|
}
|
|
|
|
func (r *BadgerDataStore) DumpToJson() (string, error) {
|
|
logger.ErrorLn("Badger datastore does not support state export currently.")
|
|
return "{}", nil
|
|
}
|