golang gorm 使用泛型实现公共方法

package repository

import (
    "gorm.io/gorm"
)

// 定义模型必须实现的接口（例如包含ID字段）
type Model interface {
    TableName() string
    // 可以添加其他需要实现的约束方法，例如:
    // GetID() uint
}

// 通用 Repository 结构体
type GenericRepository[T Model] struct {
    DB *gorm.DB
}

func NewRepository[T Model](db *gorm.DB) *GenericRepository[T] {
    return &GenericRepository[T]{DB: db}
}

// FindByID 根据ID查询记录
func (r *GenericRepository[T]) FindByID(id uint) (*T, error) {
    var entity T
    if err := r.DB.First(&entity, id).Error; err != nil {
        return nil, err
    }
    return &entity, nil
}

// Create 创建新记录
func (r *GenericRepository[T]) Create(entity *T) error {
    return r.DB.Create(entity).Error
}

// Update 更新记录
func (r *GenericRepository[T]) Update(entity *T) error {
    return r.DB.Save(entity).Error
}

// Delete 删除记录
func (r *GenericRepository[T]) Delete(id uint) error {
    var entity T
    return r.DB.Delete(&entity, id).Error
}

// FindAll 查询所有记录（带分页）
func (r *GenericRepository[T]) FindAll(page, pageSize int) ([]T, int64, error) {
    var (
        entities []T
        count    int64
    )

    db := r.DB.Model(&entities)
    db.Count(&count)

    if page > 0 && pageSize > 0 {
        offset := (page - 1) * pageSize
        db = db.Offset(offset).Limit(pageSize)
    }

    if err := db.Find(&entities).Error; err != nil {
        return nil, 0, err
    }

    return entities, count, nil
}

// Where 条件查询
func (r *GenericRepository[T]) Where(query interface{}, args ...interface{}) *GenericRepository[T] {
    r.DB = r.DB.Where(query, args...)
    return r
}

// WithPreloads 预加载关联
func (r *GenericRepository[T]) WithPreloads(preloads ...string) *GenericRepository[T] {
    for _, preload := range preloads {
        r.DB = r.DB.Preload(preload)
    }
    return r
}

// 定义模型
type User struct {
    gorm.Model
    Name  string
    Email string `gorm:"uniqueIndex"`
}

// 实现 Model 接口
func (User) TableName() string {
    return "users"
}

func main() {
    // 初始化数据库连接
    db, _ := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})

    // 自动迁移
    db.AutoMigrate(&User{})

    // 创建 Repository
    userRepo := repository.NewRepository[User](db)

    // 使用示例
    newUser := &User{Name: "John", Email: "[email protected]"}
    userRepo.Create(newUser)

    // 查询用户
    user, _ := userRepo.FindByID(1)
    fmt.Println(user)

    // 条件查询
    users, _, _ := userRepo.Where("name LIKE ?", "%John%").FindAll(1, 10)
}