ledisdb/store/store.go

package store

import (
	"fmt"
	"os"
	"path"

	"git.internal/re/ledisdb/config"
	"git.internal/re/ledisdb/store/driver"

	_ "git.internal/re/ledisdb/store/goleveldb" // register goleveldb
	_ "git.internal/re/ledisdb/store/leveldb"   // register leveldb
	_ "git.internal/re/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)
}