Go入门笔记36-一个LevelDB的Go封装

Go LevelDB默认的Get,Put用着很不习惯,稍微封装了下,我的项目结构体比较多,所以默认按value结构体设计,项目中缓存用啥持续纠结,从map->SyncMap->leveldb->ssdb->ledisdb->redis->leveldb研究了一遍,最后还是暂时使用leveldb。

1、代码

package uleveldb

import (
	"encoding/json"
	"fmt"

	"github.com/syndtr/goleveldb/leveldb"
	"github.com/syndtr/goleveldb/leveldb/errors"
	"github.com/syndtr/goleveldb/leveldb/iterator"
	"github.com/syndtr/goleveldb/leveldb/opt"
	"github.com/syndtr/goleveldb/leveldb/util"
)

type Uleveldb struct {
	DB *leveldb.DB
}

//打开Db
func OpenDb(path string) *Uleveldb {
	newdb, err := leveldb.OpenFile(path, nil)
	if _, corrupted := err.(*errors.ErrCorrupted); corrupted {
		newdb, err = leveldb.RecoverFile(path, nil)
	}

	if err != nil {
		panic(err)
	}
	uleveldb := Uleveldb{}
	uleveldb.DB = newdb
	return &uleveldb
}

//释放资源
func (uleveldb *Uleveldb) Close() {
	uleveldb.DB.Close()
}

/**
 * @description: 存储数据
 * @param  {*}
 * @return {*}
 * @param {interface{}} key
 * @param {interface{}} value
 */
func (uleveldb *Uleveldb) Put(key string, value interface{}) error {

	result, err := json.Marshal(value)
	if err != nil {
		fmt.Println("error")
		return err
	}
	err = uleveldb.DB.Put([]byte(key), []byte(result), nil)
	return err
}

/**
 * @description: 读取数据
 * @param  {*}
 * @return {*}
 * @param {interface{}} key
 * @param {interface{}} value
 */
func (uleveldb *Uleveldb) Get(key interface{}) ([]byte, error) {

	return uleveldb.DB.Get([]byte(key.(string)), nil)
}

/**
 * @description: 删除数据
 * @param  {*}
 * @return {*}
 * @param {string} key
 * @param {interface{}} value
 */
func (uleveldb *Uleveldb) Delete(key string) error {

	return uleveldb.DB.Delete([]byte(key), nil)
}

/**
 * @description:  返回指定当前DB迭代器
 * @param  {*}
 * @return {*}
 * @param {string} key
 * @param {interface{}} value
 */
func (uleveldb *Uleveldb) NewIterator(slice *util.Range, ro *opt.ReadOptions) iterator.Iterator {

	return uleveldb.DB.NewIterator(slice, ro)
}

posted @ 2021-09-04 17:06  zhaogaojian  阅读(425)  评论(0编辑  收藏  举报