Golang数据库生成CRUD代码方法

在Golang中，我们可以根据数据库结构自动生成基本的CRUD(创建、读取、更新、删除)操作代码。以下是几种常见的方法：

1. 使用SQLx + 手动编写

package main

import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/jmoiron/sqlx"
)

type User struct {
    ID       int    `db:"id"`
    Username string `db:"username"`
    Email    string `db:"email"`
    Password string `db:"password"`
}

// 初始化数据库连接
func InitDB() (*sqlx.DB, error) {
    db, err := sqlx.Connect("mysql", "user:password@tcp(127.0.0.1:3306)/dbname?parseTime=true")
    if err != nil {
        return nil, err
    }
    return db, nil
}

// 创建用户
func CreateUser(db *sqlx.DB, user *User) error {
    query := `INSERT INTO users (username, email, password) VALUES (:username, :email, :password)`
    _, err := db.NamedExec(query, user)
    return err
}

// 获取用户
func GetUser(db *sqlx.DB, id int) (*User, error) {
    user := &User{}
    err := db.Get(user, "SELECT * FROM users WHERE id=?", id)
    return user, err
}

// 更新用户
func UpdateUser(db *sqlx.DB, user *User) error {
    query := `UPDATE users SET username=:username, email=:email, password=:password WHERE id=:id`
    _, err := db.NamedExec(query, user)
    return err
}

// 删除用户
func DeleteUser(db *sqlx.DB, id int) error {
    _, err := db.Exec("DELETE FROM users WHERE id=?", id)
    return err
}

func main() {
    db, err := InitDB()
    if err != nil {
        panic(err)
    }
    defer db.Close()
    
    // 使用示例
    user := &User{Username: "test", Email: "test@example.com", Password: "hashed_password"}
    err = CreateUser(db, user)
    if err != nil {
        fmt.Println("创建失败:", err)
    }
}

2. 使用GORM (ORM库)

package main

import (
    "fmt"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

type User struct {
    gorm.Model
    Username string `gorm:"unique"`
    Email    string
    Password string
}

func main() {
    dsn := "user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }

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

    // 创建
    db.Create(&User{Username: "user1", Email: "user1@example.com", Password: "hashed_password"})

    // 读取
    var user User
    db.First(&user, 1) // 根据主键查找
    db.First(&user, "username = ?", "user1") // 根据username查找

    // 更新
    db.Model(&user).Update("Email", "new_email@example.com")

    // 删除
    db.Delete(&user, 1)
}

3. 使用代码生成工具

有几个工具可以根据数据库表结构自动生成CRUD代码：

1. SQLBoiler: 强大的ORM代码生成器bash复制go install github.com/volatiletech/sqlboiler/v4@latest sqlboiler mysql
2. GORM Automigrate: GORM的自动迁移功能
3. XORM Reverse: 从数据库生成结构体bash复制xorm reverse mysql "user:password@/dbname?charset=utf8" templates/goxorm

4. 使用SQLBoiler生成CRUD代码示例

1. 首先安装SQLBoiler:bash复制go install github.com/volatiletech/sqlboiler/v4@latest go install github.com/volatiletech/sqlboiler/v4/drivers/sqlboiler-mysql@latest
2. 创建配置文件sqlboiler.toml:toml复制[mysql] dbname = "yourdb" user = "youruser" pass = "yourpass" host = "localhost" port = 3306 sslmode = "false"
3. 运行生成命令:bash复制sqlboiler mysql

这将为数据库中的每个表生成完整的模型和CRUD操作方法。

选择建议

• 对于简单项目: 使用GORM或SQLx手动编写
• 对于大型项目: 使用SQLBoiler等代码生成工具
• 需要快速原型开发: 使用GORM Automigrate

每种方法都有其优缺点，选择最适合你项目需求和团队熟悉度的方法。

最后

以上就是名字长了才好记最近收集整理的关于Golang数据库生成CRUD代码方法的全部内容，更多相关Golang数据库生成CRUD代码方法内容请搜索靠谱客的其他文章。