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.
60 lines
962 B
60 lines
962 B
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
"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) Set(key []byte, value []byte) {
|
|
err := db.db.Put(key, value, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (db *LevelDB) Get(key []byte) []byte {
|
|
res, err := db.db.Get(key, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return res
|
|
}
|
|
|
|
func (db *LevelDB) Delete(key []byte) {
|
|
err := db.db.Delete(key, nil)
|
|
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)
|
|
}
|
|
}
|