Golang PostgreSQL数据库查询构建器插件库buildsqlx的使用

buildsqlx是一个Go语言的数据库查询构建器库，专门为PostgreSQL设计，提供了简洁高效的API来构建和执行SQL查询。

安装

go get -u github.com/arthurkushman/buildsqlx

基本使用示例

package main

import (
    "database/sql"
    "github.com/arthurkushman/buildsqlx"
    _ "github.com/lib/pq"
)

// 定义数据库连接
var db = buildsqlx.NewDb(buildsqlx.NewConnection("postgres", "user=postgres dbname=postgres password=postgres sslmode=disable"))

// 定义数据结构
type Post struct {
    ID     int64
    Title  string
    Body   string
    Points *int64
}

func main() {
    // 查询示例
    var posts []Post
    err := db.Table("posts").
        Select("id", "title", "body", "points").
        Where("points", ">", 100).
        OrderBy("created_at", "DESC").
        Limit(10).
        EachToStruct(func(rows *sql.Rows) error {
            var p Post
            err := db.Next(rows, &p)
            if err != nil {
                return err
            }
            posts = append(posts, p)
            return nil
        })
    
    if err != nil {
        panic(err)
    }
}

查询构建功能

基本查询

// 选择特定列
q := db.Table("users").Select("name", "email")

// 添加条件
q.Where("age", ">", 18)
  .AndWhere("status", "=", "active")
  .OrWhere("is_admin", "=", true)

// 排序和分页
q.OrderBy("created_at", "DESC")
  .Limit(20)
  .Offset(10)

// 执行查询
var users []User
err := q.ScanStruct(&users)

连接查询

err := db.Table("users").
    Select("users.name", "posts.title").
    LeftJoin("posts", "users.id", "=", "posts.user_id").
    Where("users.points", ">", 1000).
    ScanStruct(&result)

聚合函数

count, err := db.Table("users").Where("age", ">", 18).Count()
avgPoints, err := db.Table("users").Avg("points")
maxPoints, err := db.Table("users").Max("points")

数据操作

插入数据

// 单条插入
newPost := Post{
    Title: "New Post",
    Body: "This is the content",
}

id, err := db.Table("posts").InsertGetId(newPost)

// 批量插入
posts := []Post{
    {Title: "Post 1", Body: "Content 1"},
    {Title: "Post 2", Body: "Content 2"},
}
err := db.Table("posts").InsertBatch(posts)

更新数据

rowsAffected, err := db.Table("posts").
    Where("id", "=", 123).
    Update(Post{
        Title: "Updated Title",
        Body: "Updated content",
    })

删除数据

rowsAffected, err := db.Table("posts").
    Where("id", "=", 123).
    Delete()

高级功能

事务处理

err := db.InTransaction(func() (interface{}, error) {
    // 在事务中执行多个操作
    _, err := db.Table("users").Where("id", "=", 1).Update(map[string]interface{}{
        "points": 100,
    })
    if err != nil {
        return nil, err
    }
    
    _, err = db.Table("logs").Insert(map[string]interface{}{
        "user_id": 1,
        "action": "update_points",
    })
    
    return nil, err
})

表结构操作

// 创建表
res, err := db.Schema("users", func(table *buildsqlx.Table) error {
    table.Increments("id")
    table.String("name", 128).NotNull()
    table.String("email", 255).Unique()
    table.Integer("age").Default(0)
    table.DateTime("created_at", true)
    return nil
})

// 修改表
res, err := db.Schema("users", func(table *buildsqlx.Table) error {
    table.String("bio", 512).Change()
    table.DropColumn("old_column")
    return nil
})

调试SQL

// 打印SQL但不执行
db.Table("users").Select("name").Where("id", "=", 1).Dump()

// 打印SQL并退出
db.Table("users").Select("name").Where("id", "=", 1).Dd()

实用功能

检查表是否存在

exists, err := db.HasTable("public", "users")

分块处理大数据

var totalPoints int64
err := db.Table("users").
    Select("points").
    Chunk(&User{}, 100, func(users []any) bool {
        for _, u := range users {
            user := u.(User)
            totalPoints += user.Points
        }
        return true // 继续处理下一块
    })

获取单列值

var names []string
res, err := db.Table("users").Pluck(&User{})
for _, v := range res {
    user := v.(User)
    names = append(names, user.Name)
}

buildsqlx提供了简洁直观的API来构建复杂的SQL查询，同时保持了高性能和灵活性。它特别适合需要与PostgreSQL数据库交互的Go应用程序。