go语言 | 快速生成数据库表的 model 和 queryset

在Go语言中，没有类似Python的ORM框架，所以我们需要自己实现一个简单的数据库表的model和queryset。以下是一个简单的例子：

package main
 
import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "log"
)
 
type Model struct {
    db *sql.DB
}
 
func NewModel(db *sql.DB) *Model {
    return &Model{db: db}
}
 
type QuerySet struct {
    model *Model
}
 
func (m *Model) Query() *QuerySet {
    return &QuerySet{model: m}
}
 
func (q *QuerySet) Get(tableName string, id int) (map[string]interface{}, error) {
    query := fmt.Sprintf("SELECT * FROM %s WHERE id = ?", tableName)
    row := q.model.db.QueryRow(query, id)
    // 假设我们有一个名为User的表，它有id和name两个字段
    var id_, name sql.NullString
    err := row.Scan(&id_, &name)
    if err != nil {
        return nil, err
    }
    result := make(map[string]interface{})
    result["id"] = id_.String
    result["name"] = name.String
    return result, nil
}
 
func main() {
    db, err := sql.Open("mysql", "user:password@/dbname")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()
 
    model := NewModel(db)
    querySet := model.Query()
    result, err := querySet.Get("User", 1)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(result)
}

这个例子中，我们定义了一个Model结构体，它包含了一个*sql.DB类型的字段。我们还定义了一个QuerySet结构体，它包含了一个Model类型的字段。在QuerySet的Get方法中，我们使用QueryRow来执行SQL查询，并假设返回结果是一个map。

在main函数中，我们创建了一个Model实例，并使用它来创建一个QuerySet实例。然后我们调用QuerySet的Get方法来获取特定ID的记录。

请注意，这个例子是为了演示如何快速生成model和queryset，并不是一个完整的ORM实现。在实际应用中，你需要根据你的数据库模型来扩展这个例子，并添加更多的功能，比如支持更复杂的查询，事务处理，跨表和一对多等关系处理等。