go语言的database/sql结合squirrel工具sql生成器完成数据库操作

news2025/2/11 8:57:31

database/sql

database/sql是go语言内置数据库引擎，使用sql查询数据库，配置datasource后使用其数据库操作方法对数据库操作，如下：

package main

import (
	"database/sql"
	"fmt"

	_ "github.com/Masterminds/squirrel"
	_ "github.com/go-sql-driver/mysql"
)

func main() {
	datasource := "root:root@tcp(192.168.42.128:3306)/test"
	sqlDB, err := sql.Open("mysql", datasource)
	if err != nil {
		panic(err)
	}
	defer sqlDB.Close()

	r := sqlDB.QueryRow("select * from user")
	type User struct {
		Id       int
		Account  string
		Password string
	}

	var user User
	err = r.Scan(&user.Id, &user.Account, &user.Password)
	if err != nil {
		panic(err)
	}
	fmt.Println(user)
	//squirrel.Select("")
}

在这里插入图片描述

除了原生的方法外许多orm框架也是对sql进行了封装，如gorm，xorm等。使用原生database/sql就需要编写sql语句。

database/sql方法传入的是sql的字符串，再使用squirrel生成sql。

squirrel

squirrel是一个sql生成工具，能够帮助开发者快速构建sql，避免拼接原生sql，原生sql语句一般是很长的，特别是多表联查的时候。squirrel能够避免直接拼接太长SQL的过程，简化书写。

github地址

首先介绍squirrel的四大结构体

SelectBuilder
UpdateBuilder
InsertBuilder
DeleteBuilder

这些结构体都没有成员变量，都是用来实实现具体方法构造SQL的，其归属其类型都是builder.Builder。

但是由于不用DDL,DML的区别，这些结构体也会实现不同的方法，例如

select * from `user` where `id` = 1 inner join order on `user.orderid` = `order.id`

update `user` set `password` = 123456

delete from `user` where `id` = 1

insert into `user` (....) values (....)

在上述CURD中，不同的语句具有不同的关键字，这是四大结构体就会实现具体的方法，返回特定sql构造结构体。

在这里插入图片描述

那么就可以使用squirrel构造sql了，显然方法参数是sql的变量。

s, i, err := squirrel.Select("*").From("user").ToSql()
if err != nil {
	panic(err)
}
fmt.Println(s)
fmt.Print(i)

在这里插入图片描述

可以看到ToSql方法返回三个参数，分别是sql语句和参数为切片类型，代码所示没有传入参数因此未一个空切片。

ub := squirrel.Update("user").Set("password", 12345678).Where("id", 1)
s, i, _ := ub.ToSql()
fmt.Printf("sql = %s,elem = %v\n", s, i)

ub1 := squirrel.Update("user").Set("password = ?", 12345678).Where("id = ?", 1)
s1, i1, _ := ub1.ToSql()
fmt.Printf("sql = %s,elem = %v\n", s1, i1)

s2, i2, _ := squirrel.Delete("user").Where("id = ?", 1).ToSql()
fmt.Printf("sql = %s,elem = %v\n", s2, i2)

在这里插入图片描述

通过上面的测试可以发现set关键字后面只能是=因此会自动拼接，where后面不一定是=所以不会自动拼接。

数据库操作的关键是条件查询，对于where关键字来说是传入变量最多的，通过上述案例可以知道如下的传参方式：

s, i, err := squirrel.Select("*").From("user").Where("id = ? and name = ? and password = ?", 1, "xiaoux", "12345678").ToSql()
fmt.Printf("%s,%v,%v\n", s, i, err)

在这里插入图片描述

这样依旧可以传参成功但是并没有简便多少，另外squirrel还提供squirrel.Eq来传参

s2, i2, err2 := squirrel.Select("*").From("user").Where(squirrel.Eq{"id": 1, "name": "xiaoxu", "password": "123456"}).ToSql()
fmt.Printf("%s,%v,%v\n", s2, i2, err2)

在这里插入图片描述
和前一个生成sql一样，但是却更具有可观性。squirrel不支持结构体作为参数，只支持string和map两种类型。

s, i, err := squirrel.Select("user.id", "user.name", "order.id", "order.name").From("user").Where("user.orderid = order.id").Join("order").ToSql()
fmt.Println(s, i, err)