DiceDB - Golang高性能内存数据库插件库使用指南

DiceDB是一个基于Go语言开发的高性能内存数据库插件库，它提供了类似Redis的键值存储功能，但完全嵌入在Go应用中运行，无需额外服务。

主要特性

• 纯Go实现，无外部依赖
• 支持字符串、哈希、列表、集合、有序集合等数据结构
• 高性能，基于内存操作
• 支持数据持久化
• 简单易用的API接口

安装

go get github.com/douyu/jupiter/pkg/store/dicedb

基本使用示例

1. 初始化DiceDB

package main

import (
	"fmt"
	"github.com/douyu/jupiter/pkg/store/dicedb"
)

func main() {
	// 初始化DiceDB
	db := dicedb.New()
	defer db.Close()

	// 设置键值
	err := db.Set("name", "DiceDB User")
	if err != nil {
		fmt.Println("Set error:", err)
		return
	}

	// 获取值
	val, err := db.Get("name")
	if err != nil {
		fmt.Println("Get error:", err)
		return
	}
	fmt.Println("name:", val)
}

2. 使用哈希表

// 哈希表操作示例
func hashExample(db *dicedb.DiceDB) {
	// 设置哈希字段
	db.HSet("user:1", "name", "Alice")
	db.HSet("user:1", "age", "25")
	db.HSet("user:1", "email", "alice@example.com")

	// 获取单个字段
	name, err := db.HGet("user:1", "name")
	if err != nil {
		fmt.Println("HGet error:", err)
		return
	}
	fmt.Println("User name:", name)

	// 获取所有字段
	allFields, err := db.HGetAll("user:1")
	if err != nil {
		fmt.Println("HGetAll error:", err)
		return
	}
	fmt.Println("All fields:", allFields)
}

3. 列表操作

// 列表操作示例
func listExample(db *dicedb.DiceDB) {
	// 从左侧推入元素
	db.LPush("tasks", "task1", "task2", "task3")

	// 从右侧推入元素
	db.RPush("tasks", "task4")

	// 获取列表长度
	length, err := db.LLen("tasks")
	if err != nil {
		fmt.Println("LLen error:", err)
		return
	}
	fmt.Println("Tasks length:", length)

	// 获取列表元素
	tasks, err := db.LRange("tasks", 0, -1)
	if err != nil {
		fmt.Println("LRange error:", err)
		return
	}
	fmt.Println("All tasks:", tasks)
}

4. 集合操作

// 集合操作示例
func setExample(db *dicedb.DiceDB) {
	// 添加元素
	db.SAdd("tags", "golang", "database", "memory")

	// 检查元素是否存在
	exists, err := db.SIsMember("tags", "golang")
	if err != nil {
		fmt.Println("SIsMember error:", err)
		return
	}
	fmt.Println("Is 'golang' in tags:", exists)

	// 获取所有成员
	members, err := db.SMembers("tags")
	if err != nil {
		fmt.Println("SMembers error:", err)
		return
	}
	fmt.Println("All tags:", members)
}

5. 有序集合操作

// 有序集合示例
func zsetExample(db *dicedb.DiceDB) {
	// 添加带分数的成员
	db.ZAdd("leaderboard", 100, "player1")
	db.ZAdd("leaderboard", 200, "player2")
	db.ZAdd("leaderboard", 150, "player3")

	// 获取成员排名(从高到低)
	rank, err := db.ZRevRank("leaderboard", "player1")
	if err != nil {
		fmt.Println("ZRevRank error:", err)
		return
	}
	fmt.Println("Player1 rank:", rank)

	// 获取分数范围内的成员
	players, err := db.ZRangeByScore("leaderboard", 100, 200)
	if err != nil {
		fmt.Println("ZRangeByScore error:", err)
		return
	}
	fmt.Println("Players with score 100-200:", players)
}

高级功能

数据持久化

func persistenceExample() {
	// 初始化带持久化的DiceDB
	db := dicedb.New(
		dicedb.WithPersistence(true),
		dicedb.WithPersistenceFile("mydb.dump"),
		dicedb.WithPersistenceInterval(60), // 60秒保存一次
	)
	defer db.Close()

	// 正常操作...
	db.Set("persistent_key", "this will be saved to disk")
}

性能优化配置

func performanceExample() {
	// 初始化带性能优化的DiceDB
	db := dicedb.New(
		dicedb.WithShardCount(32), // 分片数，默认为16
		dicedb.WithLRUCache(10000), // LRU缓存大小
	)
	defer db.Close()

	// 高性能操作...
}

最佳实践

1. 合理使用数据结构：根据场景选择最适合的数据结构，如频繁查询用哈希，排序用有序集合等。

2. 批量操作：尽可能使用批量操作命令减少网络开销。

3. 内存管理：监控内存使用情况，大数据量时考虑分片或LRU策略。

4. 错误处理：始终检查返回的错误，特别是在生产环境中。

5. 并发安全：DiceDB是并发安全的，但要注意业务逻辑中的竞态条件。

DiceDB非常适合需要高性能内存存储的Go应用场景，如缓存、会话存储、实时排行榜等。它的API设计借鉴了Redis，对于熟悉Redis的开发者来说学习成本很低。