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

背景介绍

dotsql不是ORM,也不是SQL查询语句的构建器,而是可以在一个SQL文件中拎出某几行来执行的工具,非常类似于ini配置文件的读取。如果还不理解,我们来看如下内容。

-- name: create-users-table
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("htmlcode">
-- 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("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.Merge(dot1, dot2)

总结

广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!