Golang SQL函数实现

最近在学习Golang操作数据库，想请教下如何优雅地实现SQL相关函数？比如连接数据库、执行查询、事务处理这些常用操作。有没有什么最佳实践或推荐的第三方库？自己写的时候总感觉代码不够简洁，特别是错误处理和资源释放这块容易出问题。希望能分享些实际项目中的经验，谢谢！

2 回复

在Golang中，使用database/sql包实现SQL函数操作。常用方法包括：

  1. db.Query() 执行查询
  2. db.Exec() 执行插入/更新
  3. 预处理语句db.Prepare()
  4. 事务db.Begin()

示例：

rows, err := db.Query("SELECT * FROM users")

需导入数据库驱动，如_ "github.com/go-sql-driver/mysql"。

更多关于Golang SQL函数实现的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在Golang中，SQL函数实现主要涉及数据库操作，通常使用database/sql包配合相应数据库驱动。以下是常见SQL函数实现示例：

1. 数据库连接

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

func connectDB() (*sql.DB, error) {
    db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/dbname")
    if err != nil {
        return nil, err
    }
    return db, nil
}

2. 查询函数

type User struct {
    ID   int
    Name string
    Age  int
}

func GetUserByID(db *sql.DB, id int) (*User, error) {
    var user User
    err := db.QueryRow("SELECT id, name, age FROM users WHERE id = ?", id).
        Scan(&user.ID, &user.Name, &user.Age)
    if err != nil {
        return nil, err
    }
    return &user, nil
}

3. 插入函数

func CreateUser(db *sql.DB, name string, age int) (int64, error) {
    result, err := db.Exec("INSERT INTO users (name, age) VALUES (?, ?)", name, age)
    if err != nil {
        return 0, err
    }
    return result.LastInsertId()
}

4. 更新函数

func UpdateUserAge(db *sql.DB, id int, age int) error {
    _, err := db.Exec("UPDATE users SET age = ? WHERE id = ?", age, id)
    return err
}

5. 事务处理

func TransferMoney(db *sql.DB, from, to int, amount float64) error {
    tx, err := db.Begin()
    if err != nil {
        return err
    }
    
    defer func() {
        if p := recover(); p != nil {
            tx.Rollback()
            panic(p)
        }
    }()
    
    // 执行转账操作
    _, err = tx.Exec("UPDATE accounts SET balance = balance - ? WHERE id = ?", amount, from)
    if err != nil {
        tx.Rollback()
        return err
    }
    
    _, err = tx.Exec("UPDATE accounts SET balance = balance + ? WHERE id = ?", amount, to)
    if err != nil {
        tx.Rollback()
        return err
    }
    
    return tx.Commit()
}

6. 批量查询

func GetAllUsers(db *sql.DB) ([]User, error) {
    rows, err := db.Query("SELECT id, name, age FROM users")
    if err != nil {
        return nil, err
    }
    defer rows.Close()
    
    var users []User
    for rows.Next() {
        var user User
        if err := rows.Scan(&user.ID, &user.Name, &user.Age); err != nil {
            return nil, err
        }
        users = append(users, user)
    }
    return users, nil
}

关键要点：

  • 使用sql.Open()建立连接
  • QueryRow()用于单行查询
  • Query()用于多行查询
  • Exec()用于执行INSERT/UPDATE/DELETE
  • 使用事务保证数据一致性
  • 及时关闭rows和数据库连接

这些函数涵盖了基本的SQL操作，可以根据具体需求进行扩展和优化。

回到顶部