任务一完成

internal/data/gorm.go

@@ -0,0 +1,74 @@
+package data
+
+import (
+	"time"
+
+	"kra/internal/conf"
+
+	"github.com/go-kratos/kratos/v2/log"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+)
+
+// NewGormDB 创建GORM数据库连接
+func NewGormDB(c *conf.Data, l log.Logger) (*gorm.DB, func(), error) {
+	helper := log.NewHelper(l)
+
+	var dialector gorm.Dialector
+	switch c.Database.Driver {
+	case "mysql":
+		dialector = mysql.Open(c.Database.Source)
+	case "postgres":
+		dialector = postgres.Open(c.Database.Source)
+	case "sqlite":
+		dialector = sqlite.Open(c.Database.Source)
+	default:
+		dialector = mysql.Open(c.Database.Source)
+	}
+
+	gormConfig := &gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   c.Database.TablePrefix,
+			SingularTable: true,
+		},
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
+
+	// 设置日志级别
+	if c.Database.LogMode {
+		gormConfig.Logger = logger.Default.LogMode(logger.Info)
+	} else {
+		gormConfig.Logger = logger.Default.LogMode(logger.Silent)
+	}
+
+	db, err := gorm.Open(dialector, gormConfig)
+	if err != nil {
+		helper.Errorf("failed to connect database: %v", err)
+		return nil, nil, err
+	}
+
+	sqlDB, err := db.DB()
+	if err != nil {
+		helper.Errorf("failed to get sql.DB: %v", err)
+		return nil, nil, err
+	}
+
+	// 连接池配置
+	sqlDB.SetMaxIdleConns(int(c.Database.MaxIdleConns))
+	sqlDB.SetMaxOpenConns(int(c.Database.MaxOpenConns))
+	sqlDB.SetConnMaxLifetime(time.Duration(c.Database.MaxLifetime) * time.Second)
+
+	cleanup := func() {
+		helper.Info("closing database connection")
+		if err := sqlDB.Close(); err != nil {
+			helper.Errorf("failed to close database: %v", err)
+		}
+	}
+
+	helper.Info("database connected successfully")
+	return db, cleanup, nil
+}

internal/data/model/common.go

@@ -0,0 +1,30 @@
+package model
+
+import (
+	"time"
+
+	"gorm.io/gorm"
+)
+
+// Model 基础模型
+type Model struct {
+	ID        uint           `gorm:"primarykey" json:"id"`
+	CreatedAt time.Time      `json:"createdAt"`
+	UpdatedAt time.Time      `json:"updatedAt"`
+	DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
+}
+
+// PageInfo 分页请求
+type PageInfo struct {
+	Page     int    `json:"page" form:"page"`
+	PageSize int    `json:"pageSize" form:"pageSize"`
+	Keyword  string `json:"keyword" form:"keyword"`
+}
+
+// PageResult 分页响应
+type PageResult struct {
+	List     interface{} `json:"list"`
+	Total    int64       `json:"total"`
+	Page     int         `json:"page"`
+	PageSize int         `json:"pageSize"`
+}

internal/data/redis.go

@@ -0,0 +1,47 @@
+package data
+
+import (
+	"context"
+
+	"kra/internal/conf"
+
+	"github.com/go-kratos/kratos/v2/log"
+	"github.com/redis/go-redis/v9"
+)
+
+// NewRedisClient 创建Redis客户端
+func NewRedisClient(c *conf.Data, l log.Logger) (redis.UniversalClient, func(), error) {
+	helper := log.NewHelper(l)
+
+	var client redis.UniversalClient
+
+	if c.Redis.UseCluster {
+		client = redis.NewClusterClient(&redis.ClusterOptions{
+			Addrs:    c.Redis.ClusterAddrs,
+			Password: c.Redis.Password,
+		})
+	} else {
+		client = redis.NewClient(&redis.Options{
+			Addr:     c.Redis.Addr,
+			Password: c.Redis.Password,
+			DB:       int(c.Redis.Db),
+		})
+	}
+
+	// 测试连接
+	ctx := context.Background()
+	if _, err := client.Ping(ctx).Result(); err != nil {
+		helper.Errorf("failed to connect redis: %v", err)
+		return nil, nil, err
+	}
+
+	cleanup := func() {
+		helper.Info("closing redis connection")
+		if err := client.Close(); err != nil {
+			helper.Errorf("failed to close redis: %v", err)
+		}
+	}
+
+	helper.Info("redis connected successfully")
+	return client, cleanup, nil
+}