forked from mirror/ledisdb
71 lines
1.1 KiB
Go
71 lines
1.1 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/siddontang/ledisdb/store/driver"
|
|
"os"
|
|
)
|
|
|
|
const DefaultStoreName = "goleveldb"
|
|
|
|
type Store interface {
|
|
Open(cfg *Config) (driver.IDB, error)
|
|
Repair(cfg *Config) error
|
|
}
|
|
|
|
var dbs = map[string]Store{}
|
|
|
|
func Register(name string, store Store) {
|
|
if _, ok := dbs[name]; ok {
|
|
panic(fmt.Errorf("store %s is registered", name))
|
|
}
|
|
|
|
dbs[name] = store
|
|
}
|
|
|
|
func ListStores() []string {
|
|
s := []string{}
|
|
for k, _ := range dbs {
|
|
s = append(s, k)
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func Open(cfg *Config) (*DB, error) {
|
|
if len(cfg.Name) == 0 {
|
|
cfg.Name = DefaultStoreName
|
|
}
|
|
|
|
s, ok := dbs[cfg.Name]
|
|
if !ok {
|
|
return nil, fmt.Errorf("store %s is not registered", cfg.Name)
|
|
}
|
|
|
|
if err := os.MkdirAll(cfg.Path, os.ModePerm); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
idb, err := s.Open(cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
db := &DB{idb}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func Repair(cfg *Config) error {
|
|
if len(cfg.Name) == 0 {
|
|
cfg.Name = DefaultStoreName
|
|
}
|
|
|
|
s, ok := dbs[cfg.Name]
|
|
if !ok {
|
|
return fmt.Errorf("db %s is not registered", cfg.Name)
|
|
}
|
|
|
|
return s.Repair(cfg)
|
|
}
|