tendermint/c_level_db.go

133 lines
2.4 KiB
Go
Raw Normal View History

// +build gcc
2016-11-29 16:06:36 -08:00
package db
import (
"fmt"
"path"
"github.com/jmhodges/levigo"
. "github.com/tendermint/go-common"
)
func init() {
registerDBCreator(CLevelDBBackendStr, func(name string, dir string) (DB, error) {
return NewCLevelDB(name, dir)
}, false)
}
type CLevelDB struct {
2016-11-29 16:06:36 -08:00
db *levigo.DB
ro *levigo.ReadOptions
wo *levigo.WriteOptions
woSync *levigo.WriteOptions
}
func NewCLevelDB(name string, dir string) (*CLevelDB, error) {
dbPath := path.Join(dir, name+".db")
2016-11-29 16:06:36 -08:00
opts := levigo.NewOptions()
opts.SetCache(levigo.NewLRUCache(1 << 30))
opts.SetCreateIfMissing(true)
db, err := levigo.Open(dbPath, opts)
if err != nil {
return nil, err
}
ro := levigo.NewReadOptions()
wo := levigo.NewWriteOptions()
woSync := levigo.NewWriteOptions()
woSync.SetSync(true)
database := &CLevelDB{
2016-11-29 16:06:36 -08:00
db: db,
ro: ro,
wo: wo,
woSync: woSync,
}
return database, nil
}
func (db *CLevelDB) Get(key []byte) []byte {
2016-11-29 16:06:36 -08:00
res, err := db.db.Get(db.ro, key)
if err != nil {
PanicCrisis(err)
}
return res
}
func (db *CLevelDB) Set(key []byte, value []byte) {
2016-11-29 16:06:36 -08:00
err := db.db.Put(db.wo, key, value)
if err != nil {
PanicCrisis(err)
}
}
func (db *CLevelDB) SetSync(key []byte, value []byte) {
2016-11-29 16:06:36 -08:00
err := db.db.Put(db.woSync, key, value)
if err != nil {
PanicCrisis(err)
}
}
func (db *CLevelDB) Delete(key []byte) {
2016-11-29 16:06:36 -08:00
err := db.db.Delete(db.wo, key)
if err != nil {
PanicCrisis(err)
}
}
func (db *CLevelDB) DeleteSync(key []byte) {
2016-11-29 16:06:36 -08:00
err := db.db.Delete(db.woSync, key)
if err != nil {
PanicCrisis(err)
}
}
func (db *CLevelDB) DB() *levigo.DB {
2016-11-29 16:06:36 -08:00
return db.db
}
func (db *CLevelDB) Close() {
2016-11-29 16:06:36 -08:00
db.db.Close()
db.ro.Close()
db.wo.Close()
db.woSync.Close()
}
func (db *CLevelDB) Print() {
2016-11-29 16:06:36 -08:00
iter := db.db.NewIterator(db.ro)
defer iter.Close()
for iter.Seek(nil); iter.Valid(); iter.Next() {
key := iter.Key()
value := iter.Value()
fmt.Printf("[%X]:\t[%X]\n", key, value)
}
}
func (db *CLevelDB) NewBatch() Batch {
2016-11-29 16:06:36 -08:00
batch := levigo.NewWriteBatch()
return &cLevelDBBatch{db, batch}
2016-11-29 16:06:36 -08:00
}
//--------------------------------------------------------------------------------
type cLevelDBBatch struct {
db *CLevelDB
2016-11-29 16:06:36 -08:00
batch *levigo.WriteBatch
}
func (mBatch *cLevelDBBatch) Set(key, value []byte) {
2016-11-29 16:06:36 -08:00
mBatch.batch.Put(key, value)
}
func (mBatch *cLevelDBBatch) Delete(key []byte) {
2016-11-29 16:06:36 -08:00
mBatch.batch.Delete(key)
}
func (mBatch *cLevelDBBatch) Write() {
2016-11-29 16:06:36 -08:00
err := mBatch.db.db.Write(mBatch.db.wo, mBatch.batch)
if err != nil {
PanicCrisis(err)
}
}