2015-10-22 12:31:02 -07:00
|
|
|
package db
|
|
|
|
|
2017-12-12 14:00:03 -08:00
|
|
|
import "fmt"
|
|
|
|
|
2017-12-28 18:30:56 -08:00
|
|
|
//----------------------------------------
|
2017-11-09 17:42:32 -05:00
|
|
|
// Main entry
|
2015-10-22 12:31:02 -07:00
|
|
|
|
2018-02-02 18:50:24 +01:00
|
|
|
type DbBackendType string
|
2018-01-24 15:55:31 -08:00
|
|
|
|
2016-11-30 20:15:29 -08:00
|
|
|
const (
|
2018-02-02 18:50:24 +01:00
|
|
|
LevelDBBackend DbBackendType = "leveldb" // legacy, defaults to goleveldb unless +gcc
|
|
|
|
CLevelDBBackend DbBackendType = "cleveldb"
|
|
|
|
GoLevelDBBackend DbBackendType = "goleveldb"
|
|
|
|
MemDBBackend DbBackendType = "memdb"
|
|
|
|
FSDBBackend DbBackendType = "fsdb" // using the filesystem naively
|
2016-11-30 20:15:29 -08:00
|
|
|
)
|
|
|
|
|
|
|
|
type dbCreator func(name string, dir string) (DB, error)
|
|
|
|
|
2018-02-02 18:50:24 +01:00
|
|
|
var backends = map[DbBackendType]dbCreator{}
|
2016-11-30 20:15:29 -08:00
|
|
|
|
2018-02-02 18:50:24 +01:00
|
|
|
func registerDBCreator(backend DbBackendType, creator dbCreator, force bool) {
|
2016-11-30 20:15:29 -08:00
|
|
|
_, ok := backends[backend]
|
|
|
|
if !force && ok {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
backends[backend] = creator
|
|
|
|
}
|
2015-10-22 12:31:02 -07:00
|
|
|
|
2018-02-02 18:50:24 +01:00
|
|
|
func NewDB(name string, backend DbBackendType, dir string) DB {
|
2016-11-30 20:15:29 -08:00
|
|
|
db, err := backends[backend](name, dir)
|
|
|
|
if err != nil {
|
2017-12-12 14:00:03 -08:00
|
|
|
panic(fmt.Sprintf("Error initializing DB: %v", err))
|
2015-10-22 12:31:02 -07:00
|
|
|
}
|
2016-11-30 20:15:29 -08:00
|
|
|
return db
|
2015-10-22 12:31:02 -07:00
|
|
|
}
|