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) } }