Go语言库系列之dotsql

导读：能单独拎出SQL文件的某些行或几行执行，是不是非常有趣？今天我们来介绍一下这个有意思的库–dotsql。

背景介绍

dotsql不是ORM，也不是SQL查询语句的生成器，或者可以在一个SQL文件中拎出某几行来执行的工具，非常地ini配置文件的读取。如果还不理解，我们来看如下内容。

SQL-名称：创建用户表CREATE TABLE用户（ID整数主键自动递增非空，名称VARCHAR（255），电子邮件VARCHAR（255））；-名称：创建用户INSERT INTO用户（名称，电子邮件）VALUES （？，？）-名称：通过电子邮件查找用户SELECT ID，名称

，，，，，，，，，，，，，，，，，，，，，从用户那里发送电子邮件，其中电子邮件=？

-名称：通过电子邮件查找一个用户SELECT ID，名称，来​​自用户的电子邮件，其中电子邮件=？限制1

-名称：拖放式的用户表DROP TABLE用户 是SQL文件中定义的语句，我们可以很清晰地研磨，每条语句上方都以 -name ‘的方式打上了’注解’，而作为开发人员，可以根据打了标记的名称选择语句执行。###快速上手###准备工作目录结构概述”shell。├──data.sql├──go.mod├──go的.sum └──main.go

初始化项目

go mod init dotsql

创建data.sql文件，键入如下SQL，只是示例，内容可以自定义。

-- name: create-users-table
DROP TABLE IF EXISTS users;
CREATE TABLE users (
    id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
    name VARCHAR(255),
    email VARCHAR(255)
);

-- name: create-user
INSERT INTO users (name, email) VALUES(?, ?)

-- name: find-users-by-email
SELECT id,name,email FROM users WHERE email = ?

-- name: find-one-user-by-email
SELECT id,name,email FROM users WHERE email = ? LIMIT 1

--name: drop-users-table
DROP TABLE users

为了方便，我们用sqlite来演示，并存储在内存当中，所以要先安装sqlite驱动

go get github.com/mattn/go-sqlite3

代码演示

现在来写代码，导入go-sqlite3库

import _ "github.com/mattn/go-sqlite3"

获取sqlite3的数据库句柄

db, _ := sql.Open("sqlite3", ":memory:")

加载data.sql文件

dot, _ := dotsql.LoadFromFile("data.sql")

挑选文件中的一个标签来执行，Exec方法的第一个参数需要传入句柄

dot.Exec(db, "create-users-table")

从注释可以找到对应的语句，是一个创建表的操作

-- name: create-users-table
DROP TABLE IF EXISTS users;
CREATE TABLE users (
    id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
    name VARCHAR(255),
    email VARCHAR(255)
);

再来执行第二条语句，插入表数据

dot.Exec(db, "create-user", "User Name", "main@example.com")

我们再来尝试查询表数据，这里要注意，目前所有的操作都是基于定义的标签来选择执行的

	rows, _ := dot.Query(db, "find-users-by-email", "main@example.com")
	var (
		id    int
		name  string
		email string
	)
	for rows.Next() {
		rows.Scan(&id, &name, &email)
		fmt.Println(id, name, email)
	}

Query方法返回的是*sql.Rows类型，同学们可以自行遍历取值测试，大功告成！

其他玩法

我们可以先预准备SQL语句，再在合适的时机执行

stmt, err := dot.Prepare(db, "drop-users-table")
result, err := stmt.Exec()

同样，我们也可以将多个SQL文件合并再进行取值操作

去
dot1 ， err ：= dotsql 。LoadFromFile （“ queries1.sql” ）
dot2 ， err ：= dotsql 。LoadFromFile （“ queries2.sql” ）
dot ：= dotsql 。合并（ dot1 ， dot2 ）```