59 lines
925 B
Go
59 lines
925 B
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
)
|
|
|
|
var (
|
|
once sync.Once
|
|
)
|
|
|
|
type LdbInstance struct {
|
|
db *leveldb.DB
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func GetLdbInstance() *LdbInstance {
|
|
in := LdbInstance{}
|
|
once.Do(func() {
|
|
var err error
|
|
in.db, err = leveldb.OpenFile("./data/honoka-chan.db", nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
})
|
|
return &in
|
|
}
|
|
|
|
func (in *LdbInstance) Close() error {
|
|
in.mu.Lock()
|
|
defer in.mu.Unlock()
|
|
if in.db != nil {
|
|
return in.db.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (in *LdbInstance) Get(key []byte) ([]byte, error) {
|
|
if len(key) == 0 {
|
|
return nil, errors.New("empty key")
|
|
}
|
|
res, err := in.db.Get(key, nil)
|
|
if err == leveldb.ErrNotFound {
|
|
return nil, nil
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (in *LdbInstance) Set(key, value []byte) error {
|
|
if len(key) == 0 {
|
|
return errors.New("empty key")
|
|
}
|
|
return in.db.Put(key, value, nil)
|
|
}
|