mirror of
https://github.com/astaxie/beego.git
synced 2024-12-25 14:00:49 +00:00
205 lines
3.6 KiB
Go
205 lines
3.6 KiB
Go
package goleveldb
|
|
|
|
import (
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
"github.com/syndtr/goleveldb/leveldb/cache"
|
|
"github.com/syndtr/goleveldb/leveldb/filter"
|
|
"github.com/syndtr/goleveldb/leveldb/opt"
|
|
"github.com/syndtr/goleveldb/leveldb/storage"
|
|
"github.com/syndtr/goleveldb/leveldb/util"
|
|
|
|
"github.com/siddontang/ledisdb/config"
|
|
"github.com/siddontang/ledisdb/store/driver"
|
|
|
|
"os"
|
|
)
|
|
|
|
const defaultFilterBits int = 10
|
|
|
|
type Store struct {
|
|
}
|
|
|
|
func (s Store) String() string {
|
|
return DBName
|
|
}
|
|
|
|
type MemStore struct {
|
|
}
|
|
|
|
func (s MemStore) String() string {
|
|
return MemDBName
|
|
}
|
|
|
|
type DB struct {
|
|
path string
|
|
|
|
cfg *config.LevelDBConfig
|
|
|
|
db *leveldb.DB
|
|
|
|
opts *opt.Options
|
|
|
|
iteratorOpts *opt.ReadOptions
|
|
|
|
syncOpts *opt.WriteOptions
|
|
|
|
cache cache.Cache
|
|
|
|
filter filter.Filter
|
|
}
|
|
|
|
func (s Store) Open(path string, cfg *config.Config) (driver.IDB, error) {
|
|
if err := os.MkdirAll(path, 0755); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
db := new(DB)
|
|
db.path = path
|
|
db.cfg = &cfg.LevelDB
|
|
|
|
db.initOpts()
|
|
|
|
var err error
|
|
db.db, err = leveldb.OpenFile(db.path, db.opts)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func (s Store) Repair(path string, cfg *config.Config) error {
|
|
db, err := leveldb.RecoverFile(path, newOptions(&cfg.LevelDB))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
db.Close()
|
|
return nil
|
|
}
|
|
|
|
func (s MemStore) Open(path string, cfg *config.Config) (driver.IDB, error) {
|
|
db := new(DB)
|
|
db.path = path
|
|
db.cfg = &cfg.LevelDB
|
|
|
|
db.initOpts()
|
|
|
|
var err error
|
|
db.db, err = leveldb.Open(storage.NewMemStorage(), db.opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func (s MemStore) Repair(path string, cfg *config.Config) error {
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) initOpts() {
|
|
db.opts = newOptions(db.cfg)
|
|
|
|
db.iteratorOpts = &opt.ReadOptions{}
|
|
db.iteratorOpts.DontFillCache = true
|
|
|
|
db.syncOpts = &opt.WriteOptions{}
|
|
db.syncOpts.Sync = true
|
|
}
|
|
|
|
func newOptions(cfg *config.LevelDBConfig) *opt.Options {
|
|
opts := &opt.Options{}
|
|
opts.ErrorIfMissing = false
|
|
|
|
opts.BlockCacheCapacity = cfg.CacheSize
|
|
|
|
//we must use bloomfilter
|
|
opts.Filter = filter.NewBloomFilter(defaultFilterBits)
|
|
|
|
if !cfg.Compression {
|
|
opts.Compression = opt.NoCompression
|
|
} else {
|
|
opts.Compression = opt.SnappyCompression
|
|
}
|
|
|
|
opts.BlockSize = cfg.BlockSize
|
|
opts.WriteBuffer = cfg.WriteBufferSize
|
|
opts.OpenFilesCacheCapacity = cfg.MaxOpenFiles
|
|
|
|
//here we use default value, later add config support
|
|
opts.CompactionTableSize = 32 * 1024 * 1024
|
|
opts.WriteL0SlowdownTrigger = 16
|
|
opts.WriteL0PauseTrigger = 64
|
|
|
|
return opts
|
|
}
|
|
|
|
func (db *DB) Close() error {
|
|
return db.db.Close()
|
|
}
|
|
|
|
func (db *DB) Put(key, value []byte) error {
|
|
return db.db.Put(key, value, nil)
|
|
}
|
|
|
|
func (db *DB) Get(key []byte) ([]byte, error) {
|
|
v, err := db.db.Get(key, nil)
|
|
if err == leveldb.ErrNotFound {
|
|
return nil, nil
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (db *DB) Delete(key []byte) error {
|
|
return db.db.Delete(key, nil)
|
|
}
|
|
|
|
func (db *DB) SyncPut(key []byte, value []byte) error {
|
|
return db.db.Put(key, value, db.syncOpts)
|
|
}
|
|
|
|
func (db *DB) SyncDelete(key []byte) error {
|
|
return db.db.Delete(key, db.syncOpts)
|
|
}
|
|
|
|
func (db *DB) NewWriteBatch() driver.IWriteBatch {
|
|
wb := &WriteBatch{
|
|
db: db,
|
|
wbatch: new(leveldb.Batch),
|
|
}
|
|
return wb
|
|
}
|
|
|
|
func (db *DB) NewIterator() driver.IIterator {
|
|
it := &Iterator{
|
|
db.db.NewIterator(nil, db.iteratorOpts),
|
|
}
|
|
|
|
return it
|
|
}
|
|
|
|
func (db *DB) NewSnapshot() (driver.ISnapshot, error) {
|
|
snapshot, err := db.db.GetSnapshot()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := &Snapshot{
|
|
db: db,
|
|
snp: snapshot,
|
|
}
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func (db *DB) Compact() error {
|
|
return db.db.CompactRange(util.Range{nil, nil})
|
|
}
|
|
|
|
func init() {
|
|
driver.Register(Store{})
|
|
driver.Register(MemStore{})
|
|
}
|