Export DbBackendType in order to fix IAVL tests
This commit is contained in:
parent
2e76546223
commit
cbc63518e5
|
@ -15,7 +15,7 @@ func cleanupDBDir(dir, name string) {
|
||||||
os.RemoveAll(filepath.Join(dir, name) + ".db")
|
os.RemoveAll(filepath.Join(dir, name) + ".db")
|
||||||
}
|
}
|
||||||
|
|
||||||
func testBackendGetSetDelete(t *testing.T, backend dbBackendType) {
|
func testBackendGetSetDelete(t *testing.T, backend DbBackendType) {
|
||||||
// Default
|
// Default
|
||||||
dir, dirname := cmn.Tempdir(fmt.Sprintf("test_backend_%s_", backend))
|
dir, dirname := cmn.Tempdir(fmt.Sprintf("test_backend_%s_", backend))
|
||||||
defer dir.Close()
|
defer dir.Close()
|
||||||
|
|
|
@ -45,7 +45,7 @@ func checkValuePanics(t *testing.T, itr Iterator) {
|
||||||
assert.Panics(t, func() { itr.Key() }, "checkValuePanics expected panic but didn't")
|
assert.Panics(t, func() { itr.Key() }, "checkValuePanics expected panic but didn't")
|
||||||
}
|
}
|
||||||
|
|
||||||
func newTempDB(t *testing.T, backend dbBackendType) (db DB) {
|
func newTempDB(t *testing.T, backend DbBackendType) (db DB) {
|
||||||
dir, dirname := cmn.Tempdir("test_go_iterator")
|
dir, dirname := cmn.Tempdir("test_go_iterator")
|
||||||
db = NewDB("testdb", backend, dirname)
|
db = NewDB("testdb", backend, dirname)
|
||||||
dir.Close()
|
dir.Close()
|
||||||
|
|
18
db/db.go
18
db/db.go
|
@ -5,21 +5,21 @@ import "fmt"
|
||||||
//----------------------------------------
|
//----------------------------------------
|
||||||
// Main entry
|
// Main entry
|
||||||
|
|
||||||
type dbBackendType string
|
type DbBackendType string
|
||||||
|
|
||||||
const (
|
const (
|
||||||
LevelDBBackend dbBackendType = "leveldb" // legacy, defaults to goleveldb unless +gcc
|
LevelDBBackend DbBackendType = "leveldb" // legacy, defaults to goleveldb unless +gcc
|
||||||
CLevelDBBackend dbBackendType = "cleveldb"
|
CLevelDBBackend DbBackendType = "cleveldb"
|
||||||
GoLevelDBBackend dbBackendType = "goleveldb"
|
GoLevelDBBackend DbBackendType = "goleveldb"
|
||||||
MemDBBackend dbBackendType = "memdb"
|
MemDBBackend DbBackendType = "memdb"
|
||||||
FSDBBackend dbBackendType = "fsdb" // using the filesystem naively
|
FSDBBackend DbBackendType = "fsdb" // using the filesystem naively
|
||||||
)
|
)
|
||||||
|
|
||||||
type dbCreator func(name string, dir string) (DB, error)
|
type dbCreator func(name string, dir string) (DB, error)
|
||||||
|
|
||||||
var backends = map[dbBackendType]dbCreator{}
|
var backends = map[DbBackendType]dbCreator{}
|
||||||
|
|
||||||
func registerDBCreator(backend dbBackendType, creator dbCreator, force bool) {
|
func registerDBCreator(backend DbBackendType, creator dbCreator, force bool) {
|
||||||
_, ok := backends[backend]
|
_, ok := backends[backend]
|
||||||
if !force && ok {
|
if !force && ok {
|
||||||
return
|
return
|
||||||
|
@ -27,7 +27,7 @@ func registerDBCreator(backend dbBackendType, creator dbCreator, force bool) {
|
||||||
backends[backend] = creator
|
backends[backend] = creator
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewDB(name string, backend dbBackendType, dir string) DB {
|
func NewDB(name string, backend DbBackendType, dir string) DB {
|
||||||
db, err := backends[backend](name, dir)
|
db, err := backends[backend](name, dir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(fmt.Sprintf("Error initializing DB: %v", err))
|
panic(fmt.Sprintf("Error initializing DB: %v", err))
|
||||||
|
|
Loading…
Reference in New Issue