feat(cache): 优化缓存系统并重构数据库连接管理

- 将Redis连接方式改为连接池模式，提升连接复用效率 - 修复缓存注释错误，统一标识数据库缓存逻辑 - 添加数据检索结果非空验证，避免空指针异常 - 在数据库操作中添加读写锁保护，确保并发安全性 - 实现数据库查询和执行操作的重试机制，增强稳定性 - 更新配置文件中的缓存和数据库设置，优化缓存策略 - 重构README文档，补充框架特性和性能测试数据 - 添加示例路由配置，完善快速入门指南
2026-01-22 04:36:52 +08:00
parent 3455fb0a1c
commit b755519fc6
17 changed files with 2991 additions and 511 deletions
@@ -0,0 +1,100 @@
+package db
+
+import (
+	"code.hoteas.com/golang/hotime/cache"
+	. "code.hoteas.com/golang/hotime/common"
+	"database/sql"
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/sirupsen/logrus"
+	"sync"
+)
+
+// HoTimeDB 数据库操作核心结构体
+type HoTimeDB struct {
+	*sql.DB
+	ContextBase
+	DBName string
+	*cache.HoTimeCache
+	Log         *logrus.Logger
+	Type        string // 数据库类型: mysql, sqlite3, postgres
+	Prefix      string
+	LastQuery   string
+	LastData    []interface{}
+	ConnectFunc func(err ...*Error) (*sql.DB, *sql.DB)
+	LastErr     *Error
+	limit       Slice
+	*sql.Tx             //事务对象
+	SlaveDB     *sql.DB // 从数据库
+	Mode        int     // mode为0生产模式,1为测试模式,2为开发模式
+	mu          sync.RWMutex
+	limitMu     sync.Mutex
+	Dialect     Dialect // 数据库方言适配器
+}
+
+// SetConnect 设置数据库配置连接
+func (that *HoTimeDB) SetConnect(connect func(err ...*Error) (master, slave *sql.DB), err ...*Error) {
+	that.ConnectFunc = connect
+	_ = that.InitDb(err...)
+}
+
+// InitDb 初始化数据库连接
+func (that *HoTimeDB) InitDb(err ...*Error) *Error {
+	if len(err) != 0 {
+		that.LastErr = err[0]
+	}
+	that.DB, that.SlaveDB = that.ConnectFunc(that.LastErr)
+	if that.DB == nil {
+		return that.LastErr
+	}
+	e := that.DB.Ping()
+
+	that.LastErr.SetError(e)
+
+	if that.SlaveDB != nil {
+		e := that.SlaveDB.Ping()
+		that.LastErr.SetError(e)
+	}
+
+	// 根据数据库类型初始化方言适配器
+	if that.Dialect == nil {
+		that.initDialect()
+	}
+
+	return that.LastErr
+}
+
+// initDialect 根据数据库类型初始化方言
+func (that *HoTimeDB) initDialect() {
+	switch that.Type {
+	case "postgres", "postgresql":
+		that.Dialect = &PostgreSQLDialect{}
+	case "sqlite3", "sqlite":
+		that.Dialect = &SQLiteDialect{}
+	default:
+		that.Dialect = &MySQLDialect{}
+	}
+}
+
+// GetDialect 获取当前方言适配器
+func (that *HoTimeDB) GetDialect() Dialect {
+	if that.Dialect == nil {
+		that.initDialect()
+	}
+	return that.Dialect
+}
+
+// SetDialect 设置方言适配器
+func (that *HoTimeDB) SetDialect(dialect Dialect) {
+	that.Dialect = dialect
+}
+
+// GetType 获取数据库类型
+func (that *HoTimeDB) GetType() string {
+	return that.Type
+}
+
+// GetPrefix 获取表前缀
+func (that *HoTimeDB) GetPrefix() string {
+	return that.Prefix
+}