Goqu: Golang 惯用 SQL 构建与查询库

Goqu 是一个强大的 Golang SQL 构建器和查询库，它提供了流畅的 API 来构建类型安全的 SQL 查询。下面我将介绍 Goqu 的主要功能和使用方法。

安装

go get github.com/doug-martin/goqu/v9

基本使用

初始化连接

import (
    "database/sql"
    "github.com/doug-martin/goqu/v9"
    _ "github.com/doug-martin/goqu/v9/dialect/mysql" // 根据数据库类型选择
)

func main() {
    // 创建数据库连接
    db, err := sql.Open("mysql", "user:password@/dbname")
    if err != nil {
        panic(err.Error())
    }
    defer db.Close()
    
    // 创建 goqu 数据库实例
    dialect := goqu.Dialect("mysql")
    gq := dialect.DB(db)
}

查询构建

简单查询

sql, args, _ := goqu.From("users").
    Select("id", "name", "email").
    Where(goqu.Ex{
        "status": "active",
        "age":    goqu.Op{"gt": 18},
    }).
    Order(goqu.I("created_at").Desc()).
    Limit(10).
    Offset(0).
    ToSQL()

// 执行查询
rows, err := gq.Query(sql, args...)

复杂条件

sql, _, _ := goqu.From("users").
    Select("*").
    Where(goqu.ExOr{
        "first_name": "John",
        "last_name":  "Doe",
    }).
    Where(goqu.C("age").Gt(18)).
    Where(goqu.C("created_at").Between(
        goqu.Range(goqu.L("2020-01-01"), goqu.L("2020-12-31")),
    )).
    ToSQL()

插入数据

// 单条插入
sql, args, _ := goqu.Insert("users").
    Rows(goqu.Record{
        "name":  "John Doe",
        "email": "john@example.com",
        "age":   30,
    }).
    ToSQL()

// 批量插入
sql, args, _ := goqu.Insert("users").
    Cols("name", "email", "age").
    Vals(
        goqu.Vals{"Jane Doe", "jane@example.com", 28},
        goqu.Vals{"Bob Smith", "bob@example.com", 35},
    ).
    ToSQL()

更新数据

sql, args, _ := goqu.Update("users").
    Set(goqu.Record{"status": "inactive"}).
    Where(goqu.Ex{"last_login": goqu.Op{"lt": "2020-01-01"}}).
    ToSQL()

删除数据

sql, args, _ := goqu.Delete("users").
    Where(goqu.Ex{"status": "banned"}).
    ToSQL()

高级查询

连接查询

sql, _, _ := goqu.From("users").
    Select("users.id", "users.name", "orders.amount").
    Join(goqu.T("orders"), goqu.On(goqu.Ex{"users.id": goqu.I("orders.user_id")})).
    Where(goqu.Ex{"users.status": "active"}).
    ToSQL()

子查询

subQuery := goqu.From("orders").
    Select("user_id", goqu.MAX("amount").As("max_amount")).
    GroupBy("user_id")

sql, _, _ := goqu.From("users").
    Select("users.*", "sub_query.max_amount").
    Join(subQuery.As("sub_query"), 
        goqu.On(goqu.I("users.id").Eq(goqu.I("sub_query.user_id")))).
    ToSQL()

函数和表达式

sql, _, _ := goqu.From("users").
    Select(
        goqu.COUNT("*").As("total"),
        goqu.AVG("age").As("avg_age"),
        goqu.MAX("created_at").As("last_created"),
    ).
    ToSQL()

执行查询

// 查询单条记录
var user struct {
    ID    int64  `db:"id"`
    Name  string `db:"name"`
    Email string `db:"email"`
}

found, err := gq.From("users").
    Select("id", "name", "email").
    Where(goqu.Ex{"id": 1}).
    ScanStruct(&user)

// 查询多条记录
var users []struct {
    ID    int64  `db:"id"`
    Name  string `db:"name"`
    Email string `db:"email"`
}

err := gq.From("users").
    Select("id", "name", "email").
    Where(goqu.Ex{"status": "active"}).
    ScanStructs(&users)

事务支持

err := gq.WithTx(func(tx *goqu.TxDatabase) error {
    // 在事务中执行操作
    if _, err := tx.Insert("users").Rows(...).Executor().Exec(); err != nil {
        return err
    }
    
    if _, err := tx.Update("accounts").Set(...).Executor().Exec(); err != nil {
        return err
    }
    
    return nil
})

日志记录

import "github.com/doug-martin/goqu/v9/exp"

// 自定义日志记录器
gq.Logger(func(sql string, args []interface{}, err error) {
    if err != nil {
        log.Printf("Error: %v\nSQL: %s\nArgs: %v", err, sql, args)
    } else {
        log.Printf("SQL: %s\nArgs: %v", sql, args)
    }
})

Goqu 提供了强大而灵活的 SQL 构建能力，同时保持了 Golang 的惯用风格。它支持多种数据库方言，包括 MySQL、PostgreSQL、SQLite 等，是替代原生 SQL 字符串拼接的优秀选择。