本文详解如何使用 go 从零构建一个轻量级 redis 兼容服务，涵盖协议解析、内存数据结构设计（map + 跳表）、多协程并发模型，并提供可运行的 get/set/zadd/zcard/zrange 等命令实现示例。
本文详解如何使用 go 从零构建一个轻量级 redis 兼容服务，涵盖协议解析、内存数据结构设计（map + 跳表）、多协程并发模型，并提供可运行的 get/set/zadd/zcard/zrange 等命令实现示例。
在 Go 中实现 Redis 的核心功能，本质是构建一个内存键值存储服务，支持 RESP（Redis Serialization Protocol）协议，并提供原子性、线程安全的常用命令。不同于直接使用 github.com/go-redis/redis 客户端连接外部 Redis，本文聚焦于自研简易 Redis 服务端——适合学习原理、定制协议或嵌入式场景。
✅ 基础架构设计
推荐采用「1 引擎 + N 连接」协程模型：
- 1 个主 goroutine：管理全局状态（如内存数据库 map[string]interface{}）、定时任务（如 SAVE）、跳表实例（用于有序集合）；
- 每客户端 2 个 goroutine：一个负责读取并解析 RESP 请求（conn.Read()），另一个负责写回响应（conn.Write()）。该分离模型天然支持流水线（pipelining），且避免读写竞争，无需显式锁即可保障单命令原子性。
⚠️ 注意：Go 不支持 fork()，因此无法完美复现 Redis 的 RDB 后台 save（依赖 copy-on-write）。实际中建议采用前台同步 SAVE（如序列化为 JSON 或自定义二进制格式）或AOF 日志追加写（更易实现且持久性强）。
✅ 核心数据结构实现（精简版）
// 内存数据库（简化版）
var db = sync.Map{} // key: string → value: *DataValue
type DataValue struct {
Type string // "string", "zset"
Value interface{} // string or *ZSet
}
// 有序集合：基于跳表（此处用 map+sort 替代，生产环境请用 github.com/huandu/skiplist）
type ZSet struct {
Members map[string]float64 // member → score
sorted []string // 缓存排序后的成员（按 score 升序）
}
func (z *ZSet) Add(member string, score float64) {
z.Members[member] = score
z.rebuildSorted()
}
func (z *ZSet) rebuildSorted() {
z.sorted = make([]string, 0, len(z.Members))
for m := range z.Members {
z.sorted = append(z.sorted, m)
}
sort.Slice(z.sorted, func(i, j int) bool {
return z.Members[z.sorted[i]] < z.Members[z.sorted[j]]
})
}✅ RESP 协程处理示例（关键逻辑）
func handleConnection(conn net.Conn) {
defer conn.Close()
reader := bufio.NewReader(conn)
writer := bufio.NewWriter(conn)
defer writer.Flush()
for {
line, err := reader.ReadString('\n')
if err != nil {
return
}
cmd := parseRESP(line) // 解析 *array, 支持 $, *, : 等类型
if len(cmd) == 0 {
continue
}
reply := executeCommand(cmd) // 执行命令，返回 RESP 格式字节切片
writer.Write(reply)
writer.Flush()
}
}
func executeCommand(cmd []interface{}) []byte {
if len(cmd) == 0 {
return formatError("ERR empty command")
}
switch strings.ToUpper(cmd[0].(string)) {
case "GET":
if len(cmd) != 2 {
return formatError("ERR wrong number of arguments for 'get'")
}
key := cmd[1].(string)
if val, ok := db.Load(key); ok {
if dv, ok := val.(*DataValue); ok && dv.Type == "string" {
return formatBulkString(dv.Value.(string))
}
}
return formatNull()
case "SET":
if len(cmd) < 3 {
return formatError("ERR wrong number of arguments for 'set'")
}
key, value := cmd[1].(string), cmd[2].(string)
db.Store(key, &DataValue{Type: "string", Value: value})
return formatStatus("OK")
case "ZADD":
if len(cmd) < 4 || (len(cmd)-2)%2 != 0 {
return formatError("ERR syntax error")
}
key := cmd[1].(string)
z, _ := getOrCreateZSet(key)
for i := 2; i < len(cmd); i += 2 {
score := parseFloat(cmd[i].(string))
member := cmd[i+1].(string)
z.Add(member, score)
}
return formatInteger(len(cmd)/2 - 1)
case "ZRANGE":
if len(cmd) != 4 {
return formatError("ERR wrong number of arguments for 'zrange'")
}
key := cmd[1].(string)
start := parseInt(cmd[2].(string))
stop := parseInt(cmd[3].(string))
if z, ok := getZSet(key); ok {
end := min(stop+1, len(z.sorted))
if start < 0 { start = 0 }
if end > len(z.sorted) { end = len(z.sorted) }
result := make([][]byte, 0, end-start)
for _, m := range z.sorted[start:end] {
result = append(result, formatBulkString(m))
}
return formatArray(result)
}
return formatArray(nil)
default:
return formatError("ERR unknown command '" + cmd[0].(string) + "'")
}
}✅ 关键注意事项与优化建议
- 原子性保障：所有命令在单 goroutine 中顺序执行（无并发修改同一 key），天然满足 Redis 的单命令原子性；跨命令事务需额外引入 WATCH/MULTI/EXEC 机制。
- 内存安全：使用 sync.Map 替代普通 map，避免并发读写 panic；对 ZSet 等复合结构，建议封装读写互斥锁（sync.RWMutex）。
- 协议兼容性：严格遵循 RESP v2 规范（如错误以 -ERR 开头，整数以 : 开头，空行为 $-1）；可借助 github.com/gomodule/redigo/redis 的 redis.Parse 工具辅助调试。
- 持久化扩展：SAVE 命令可实现为 json.Marshal(db) 写入文件；AOF 更推荐——每次命令追加到日志文件，重启时重放（注意 fsync 策略权衡性能与安全性）。
- 生产就绪参考：可借鉴 ledisdb（Go 实现的类 Redis DB）或 go-redis-server（Docker 官方轻量版），在其基础上裁剪或增强。
通过以上结构，你可在 500 行内完成一个支持字符串、有序集合及基础命令的 Redis 兼容服务——既是深入理解 Redis 架构的绝佳实践，也为定制化缓存中间件打下坚实基础。
