forked from mirror/ledisdb
59 lines
1.0 KiB
Go
59 lines
1.0 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/ledisdb/ledisdb/config"
|
|
"github.com/ledisdb/ledisdb/store/driver"
|
|
|
|
_ "github.com/ledisdb/ledisdb/store/goleveldb" // register goleveldb
|
|
_ "github.com/ledisdb/ledisdb/store/leveldb" // register leveldb
|
|
_ "github.com/ledisdb/ledisdb/store/rocksdb" // register rocksdb
|
|
)
|
|
|
|
func getStorePath(cfg *config.Config) string {
|
|
if len(cfg.DBPath) > 0 {
|
|
return cfg.DBPath
|
|
}
|
|
return path.Join(cfg.DataDir, fmt.Sprintf("%s_data", cfg.DBName))
|
|
}
|
|
|
|
func Open(cfg *config.Config) (*DB, error) {
|
|
s, err := driver.GetStore(cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
path := getStorePath(cfg)
|
|
|
|
if err := os.MkdirAll(path, 0755); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
idb, err := s.Open(path, cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
db := new(DB)
|
|
db.db = idb
|
|
db.name = s.String()
|
|
db.st = &Stat{}
|
|
db.cfg = cfg
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func Repair(cfg *config.Config) error {
|
|
s, err := driver.GetStore(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
path := getStorePath(cfg)
|
|
|
|
return s.Repair(path, cfg)
|
|
}
|