You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.4 KiB
80 lines
1.4 KiB
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
"github.com/syndtr/goleveldb/leveldb/errors"
|
|
"github.com/syndtr/goleveldb/leveldb/opt"
|
|
"path"
|
|
)
|
|
|
|
type LevelDB struct {
|
|
db *leveldb.DB
|
|
}
|
|
|
|
func NewLevelDB(name string) (*LevelDB, error) {
|
|
dbPath := path.Join(name)
|
|
db, err := leveldb.OpenFile(dbPath, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
database := &LevelDB{db: db}
|
|
return database, nil
|
|
}
|
|
|
|
func (db *LevelDB) Get(key []byte) []byte {
|
|
res, err := db.db.Get(key, nil)
|
|
if err != nil {
|
|
if err == errors.ErrNotFound {
|
|
return nil
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
func (db *LevelDB) Set(key []byte, value []byte) {
|
|
err := db.db.Put(key, value, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (db *LevelDB) SetSync(key []byte, value []byte) {
|
|
err := db.db.Put(key, value, &opt.WriteOptions{Sync: true})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (db *LevelDB) Delete(key []byte) {
|
|
err := db.db.Delete(key, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (db *LevelDB) DeleteSync(key []byte) {
|
|
err := db.db.Delete(key, &opt.WriteOptions{Sync: true})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (db *LevelDB) DB() *leveldb.DB {
|
|
return db.db
|
|
}
|
|
|
|
func (db *LevelDB) Close() {
|
|
db.db.Close()
|
|
}
|
|
|
|
func (db *LevelDB) Print() {
|
|
iter := db.db.NewIterator(nil, nil)
|
|
for iter.Next() {
|
|
key := iter.Key()
|
|
value := iter.Value()
|
|
fmt.Printf("[%X]:\t[%X]\n", key, value)
|
|
}
|
|
}
|