使用Pogreb进行Golang嵌入式键值存储

Pogreb是一个用纯Go编写的嵌入式键值存储库，它简单、高效且适合嵌入式应用场景。下面我将详细介绍Pogreb的使用方法，并提供示例代码。

Pogreb特点

• 纯Go实现，无外部依赖
• 支持持久化存储
• 简单的API接口
• 支持事务
• 线程安全
• 内存占用低

安装Pogreb

go get github.com/akrylysov/pogreb

基本使用示例

1. 打开数据库

package main

import (
	"log"
	"github.com/akrylysov/pogreb"
)

func main() {
	// 打开数据库(如果不存在则创建)
	db, err := pogreb.Open("example.db", nil)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
	
	// 数据库操作...
}

2. 基本的CRUD操作

// 写入数据
err = db.Put([]byte("key1"), []byte("value1"))
if err != nil {
	log.Fatal(err)
}

// 读取数据
val, err := db.Get([]byte("key1"))
if err != nil {
	log.Fatal(err)
}
fmt.Printf("key1: %s\n", val) // 输出: key1: value1

// 删除数据
err = db.Delete([]byte("key1"))
if err != nil {
	log.Fatal(err)
}

// 检查键是否存在
has, err := db.Has([]byte("key1"))
if err != nil {
	log.Fatal(err)
}
fmt.Println("key1 exists:", has) // 输出: key1 exists: false

3. 批量操作

// 批量写入
err = db.Put([]byte("name"), []byte("Alice"))
err = db.Put([]byte("age"), []byte("30"))
err = db.Put([]byte("city"), []byte("New York"))

// 遍历所有键值对
it := db.Items()
for {
	key, val, err := it.Next()
	if err == pogreb.ErrIterationDone {
		break
	}
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("%s: %s\n", key, val)
}

4. 使用事务

// 开始事务
tx, err := db.Begin()
if err != nil {
	log.Fatal(err)
}

// 在事务中执行操作
err = tx.Put([]byte("tx_key"), []byte("tx_value"))
if err != nil {
	tx.Rollback()
	log.Fatal(err)
}

// 提交事务
err = tx.Commit()
if err != nil {
	log.Fatal(err)
}

5. 性能优化选项

opts := &pogreb.Options{
	BackgroundCompactionInterval: time.Minute * 5, // 后台压缩间隔
	BackgroundSyncInterval:       time.Second * 30, // 后台同步间隔
}

db, err := pogreb.Open("optimized.db", opts)
if err != nil {
	log.Fatal(err)
}
defer db.Close()

实际应用示例

下面是一个更完整的示例，展示如何使用Pogreb构建一个简单的缓存系统：

package main

import (
	"fmt"
	"log"
	"time"
	
	"github.com/akrylysov/pogreb"
)

type Cache struct {
	db *pogreb.DB
}

func NewCache(dbPath string) (*Cache, error) {
	db, err := pogreb.Open(dbPath, nil)
	if err != nil {
		return nil, err
	}
	return &Cache{db: db}, nil
}

func (c *Cache) Set(key, value string) error {
	return c.db.Put([]byte(key), []byte(value))
}

func (c *Cache) Get(key string) (string, error) {
	val, err := c.db.Get([]byte(key))
	if err != nil {
		return "", err
	}
	return string(val), nil
}

func (c *Cache) Delete(key string) error {
	return c.db.Delete([]byte(key))
}

func (c *Cache) Close() error {
	return c.db.Close()
}

func main() {
	cache, err := NewCache("cache.db")
	if err != nil {
		log.Fatal(err)
	}
	defer cache.Close()

	// 设置缓存
	err = cache.Set("session_token", "abc123xyz")
	if err != nil {
		log.Fatal(err)
	}

	// 获取缓存
	token, err := cache.Get("session_token")
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Session token:", token)

	// 删除缓存
	err = cache.Delete("session_token")
	if err != nil {
		log.Fatal(err)
	}
}

注意事项

1. Pogreb适合存储中小规模的数据(GB级别)
2. 所有键和值都是[]byte类型，需要自行处理序列化
3. 频繁写入时性能可能不如内存数据库
4. 数据库文件会随着写入操作增长，删除操作不会立即释放空间
5. 定期调用db.Compact()可以优化存储空间

Pogreb是一个简单高效的嵌入式键值存储解决方案，特别适合需要持久化存储但又不想引入复杂数据库系统的Go应用程序。