Golang数据库生成CRUD代码方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
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)
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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)
}