go-redis/database/standalone_database.go

// Author: simon (ynwdlxm@163.com)
// Date: 2025/9/25 11:27
// Desc: 处理 RESP-SPEC 命令

package database

import (
	"strconv"
	"strings"

	"github.com/runningwater/go-redis/aof"
	"github.com/runningwater/go-redis/config"
	"github.com/runningwater/go-redis/interface/resp"
	"github.com/runningwater/go-redis/lib/logger"
	"github.com/runningwater/go-redis/resp/reply"
)

// StandaloneDatabase 表示独立运行的Redis数据库实例
//
//	包含多个数据库以及AOF持久化处理器
type StandaloneDatabase struct {
	dbSet      []*DB        // 数据库集合，每个元素代表一个独立的数据库
	aofHandler *aof.Handler // AOF持久化处理器，负责处理AOF相关操作
}

// NewStandaloneDatabase 创建并初始化一个新的独立数据库实例
//
//	根据配置决定数据库数量和是否启用AOF持久化
func NewStandaloneDatabase() *StandaloneDatabase {
	// 如果未设置数据库数量，默认创建16个数据库
	if config.Properties.Databases == 0 {
		config.Properties.Databases = 16
	}

	// 创建数据库实例
	database := &StandaloneDatabase{}

	// 初始化数据库集合
	database.dbSet = make([]*DB, config.Properties.Databases)

	// 为每个数据库索引创建对应的DB实例
	for i := range database.dbSet {
		db := NewDB()
		db.index = i
		database.dbSet[i] = db
	}

	// 如果启用了AOF持久化，则初始化AOF处理器
	if config.Properties.AppendOnly {
		aofHandler, err := aof.NewAofHandler(database)
		if err != nil {
			panic(err)
		}
		database.aofHandler = aofHandler
	}

	// 为每个数据库设置AOF记录函数
	for _, db := range database.dbSet {
		d := db
		// addAof 函数用于向AOF日志中添加命令记录
		d.addAof = func(line CmdLine) {
			database.aofHandler.AddAof(d.index, line)
		}
	}

	return database
}

// Exec 执行客户端发送的命令
//
//	client: 客户端连接对象
//	args: 命令参数列表，第一个参数为命令名称
//	返回命令执行结果
func (d *StandaloneDatabase) Exec(client resp.Connection, args [][]byte) resp.Reply {
	// 使用defer捕获可能发生的panic，防止程序崩溃
	defer func() {
		if err := recover(); err != nil {
			logger.Error(err)
		}
	}()

	// 获取命令名称并转换为小写
	cmdName := strings.ToLower(string(args[0]))

	// 处理SELECT命令 - 切换数据库
	if cmdName == "select" {
		// 检查参数数量是否正确
		if len(args) != 2 {
			return reply.NewErrReply("ERR wrong number of arguments for 'select' command")
		}
		// 执行数据库选择操作
		return execSelect(client, d, args[1:])
	} else if cmdName == "command" {
		// COMMAND命令返回OK
		return reply.NewOkReply()
	}

	// 获取客户端当前选择的数据库索引
	index := client.GetDBIndex()

	// 在对应的数据库中执行命令
	return d.dbSet[index].Exec(client, args)
}

// Close 关闭数据库连接
// TODO 实现数据库关闭逻辑
func (d *StandaloneDatabase) Close() {
	// TODO implement me
}

// AfterClientClose 处理客户端连接关闭后的清理工作
// TODO 实现客户端关闭后的处理逻辑
func (d *StandaloneDatabase) AfterClientClose(client resp.Connection) {
	// TODO implement me
}

// execSelect 处理数据库选择命令
//
// 参数:
//
//	c: 客户端连接对象，用于执行数据库切换操作
//	database: 数据库实例，包含所有可用的数据库集合
//	args: 命令参数，第一个参数为要选择的数据库索引
//
// 返回值:
//
//	resp.Reply: 执行结果回复，成功返回OK，失败返回错误信息
func execSelect(c resp.Connection, database *StandaloneDatabase, args [][]byte) resp.Reply {
	// 解析数据库索引参数
	dbIndex, err := strconv.Atoi(string(args[0]))
	if err != nil {
		return reply.NewErrReply("ERR invalid DB index")
	}

	// 验证数据库索引范围是否有效
	if dbIndex >= len(database.dbSet) {
		return reply.NewErrReply("ERR DB index is out of range")
	}

	// 执行数据库切换操作
	c.SelectDB(dbIndex)
	return reply.NewOkReply()
}