Go语言中基于嵌入实现通用CRUD操作的策略与gorp反射机制解析

package models

import (
    "database/sql"
    "github.com/coopernurse/gorp"
    _ "github.com/go-sql-driver/mysql" // MySQL驱动
)

// GorpModel 包含通用的数据库模型属性
type GorpModel struct {
    New bool `db:"-"` // 用于标记是否为新记录
}

// dbm 是gorp的DbMap实例，通常作为全局或单例管理
var dbm *gorp.DbMap = nil

// DbInit 初始化数据库连接和gorp DbMap
func (gm *GorpModel) DbInit() {
    gm.New = true
    if dbm == nil {
        db, err := sql.Open("mysql", "username:password@tcp(127.0.0.1:3306)/my_db?charset=utf8")
        if err != nil {
            panic(err) // 实际应用中应进行更优雅的错误处理
        }
        dbm = &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}
        // 注意：这里需要为每个具体的模型添加表映射，例如 dbm.AddTable(User{}).SetKeys(true, "Id")
        // dbm.CreateTables() // 仅在开发环境或首次运行时调用
    }
}

// Create 方法试图将GorpModel实例插入数据库
func (gm *GorpModel) Create() {
    // 问题所在：gorp会反射gm的类型，即GorpModel，而非嵌入它的具体类型
    err := dbm.Insert(gm)
    if err != nil {
        panic(err)
    }
}

// Delete 方法试图删除GorpModel实例
func (gm *GorpModel) Delete() int64 {
    nrows, err := dbm.Delete(gm)
    if err != nil {
        panic(err)
    }
    return nrows
}

// Update 方法试图更新GorpModel实例
func (gm *GorpModel) Update() {
    _, err := dbm.Update(gm)
    if err != nil {
        panic(err)
    }
}

package models

import (
    "database/sql"
    "fmt"
    "github.com/coopernurse/gorp"
    _ "github.com/go-sql-driver/mysql" // MySQL驱动
)

// GorpModel 包含通用的数据库模型属性，不再包含CRUD方法
type GorpModel struct {
    New bool `db:"-"` // 用于标记是否为新记录
}

// 定义一个具体的业务模型，例如 User
type User struct {
    GorpModel `db:"-"` // 嵌入GorpModel
    Id        int64  `db:"id"`
    Name      string `db:"name"`
    Email     string `db:"email"`
}

// dbm 是gorp的DbMap实例
var dbm *gorp.DbMap = nil

// InitDbMap 初始化数据库连接和gorp DbMap
// 这是一个包级函数，负责初始化全局的dbm
func InitDbMap() {
    if dbm == nil {
        db, err := sql.Open("mysql", "username:password@tcp(127.0.0.1:3306)/my_db?charset=utf8")
        if err != nil {
            panic(fmt.Errorf("failed to open database connection: %w", err))
        }
        dbm = &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}

        // ！！！重要：为每个具体的业务模型添加表映射
        // gorp会根据这里注册的类型来推断表名
        dbm.AddTable(User{}).SetKeys(true, "Id")
        // dbm.AddTable(AnotherModel{}).SetKeys(true, "Id") // 如果有其他模型，也需要在这里添加

        // 仅在开发环境或首次运行时调用，用于创建表
        err = dbm.CreateTablesIfNotExists()
        if err != nil {
            panic(fmt.Errorf("failed to create tables: %w", err))
        }
    }
}

// CreateEntity 通用创建实体函数
// 接受一个interface{}参数，gorp将对传入的实际类型进行反射
func CreateEntity(entity interface{}) error {
    if dbm == nil {
        return fmt.Errorf("database map is not initialized")
    }
    err := dbm.Insert(entity)
    if err != nil {
        return fmt.Errorf("failed to create entity: %w", err)
    }
    return nil
}

// UpdateEntity 通用更新实体函数
func UpdateEntity(entity interface{}) (int64, error) {
    if dbm == nil {
        return 0, fmt.Errorf("database map is not initialized")
    }
    rowsAffected, err := dbm.Update(entity)
    if err != nil {
        return 0, fmt.Errorf("failed to update entity: %w", err)
    }
    return rowsAffected, nil
}

// DeleteEntity 通用删除实体函数
func DeleteEntity(entity interface{}) (int64, error) {
    if dbm == nil {
        return 0, fmt.Errorf("database map is not initialized")
    }
    rowsAffected, err := dbm.Delete(entity)
    if err != nil {
        return 0, fmt.Errorf("failed to delete entity: %w", err)
    }
    return rowsAffected, nil
}

// 示例：如何使用这些通用函数
func main() {
    InitDbMap() // 初始化数据库

    user := &User{
        Name:  "Alice",
        Email: "alice@example.com",
    }
    user.New = true // 标记为新记录

    // 使用通用函数创建用户
    err := CreateEntity(user)
    if err != nil {
        fmt.Printf("Error creating user: %v\n", err)
        return
    }
    fmt.Printf("User created with ID: %d\n", user.Id)

    // 更新用户
    user.Name = "Alice Smith"
    rows, err := UpdateEntity(user)
    if err != nil {
        fmt.Printf("Error updating user: %v\n", err)
        return
    }
    fmt.Printf("User updated, rows affected: %d\n", rows)

    // 删除用户
    // rows, err = DeleteEntity(user)
    // if err != nil {
    //     fmt.Printf("Error deleting user: %v\n", err)
    //     return
    // }
    // fmt.Printf("User deleted, rows affected: %d\n", rows)
}

dbm.AddTable(User{}).SetKeys(true, "Id")
dbm.AddTable(Product{}).SetKeys(true, "Id")

// SaveEntity 通用保存实体函数 (根据New字段判断是创建还是更新)
func SaveEntity(entity interface{}, isNew bool) error {
    if isNew {
        return CreateEntity(entity)
    }
    _, err := UpdateEntity(entity)
    return err
}