Go語(yǔ)言庫(kù)系列之dotsql詳解

導(dǎo)讀：能單獨(dú)拎出SQL文件的某一行或幾行執(zhí)行，是不是非常有趣？今天我們來(lái)介紹一下這個(gè)有意思的庫(kù)--dotsql。

背景介紹

dotsql不是ORM，也不是SQL查詢語(yǔ)句的構(gòu)建器，而是可以在一個(gè)SQL文件中拎出某幾行來(lái)執(zhí)行的工具，非常類似于ini配置文件的讀取。如果還不理解，我們來(lái)看如下內(nèi)容。

-- 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(?, ?)

-- 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

上面是SQL文件中定義的語(yǔ)句，我們可以很清晰地看出，每條語(yǔ)句上方都以 -- name 的方式打上了“注解”，而作為開(kāi)發(fā)人員，可以根據(jù)打了標(biāo)記的名稱挑選語(yǔ)句執(zhí)行。

快速上手
準(zhǔn)備工作

目錄結(jié)構(gòu)概覽

├── data.sql
├── go.mod
├── go.sum
└── main.go

初始化項(xiàng)目

go mod init dotsql

創(chuàng)建data.sql文件，鍵入如下SQL，只是示例，內(nèi)容可以自定義。

-- 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來(lái)演示，并存儲(chǔ)在內(nèi)存當(dāng)中，所以要先安裝sqlite驅(qū)動(dòng)

go get github.com/mattn/go-sqlite3

代碼演示

現(xiàn)在來(lái)寫(xiě)代碼，導(dǎo)入go-sqlite3庫(kù)

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

獲取sqlite3的數(shù)據(jù)庫(kù)句柄

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

加載data.sql文件

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

挑選文件中的一個(gè)標(biāo)簽來(lái)執(zhí)行，Exec方法的第一個(gè)參數(shù)需要傳入句柄

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

從注釋可以找到對(duì)應(yīng)的語(yǔ)句，是一個(gè)創(chuàng)建表的操作

-- 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)
);

再來(lái)執(zhí)行第二條語(yǔ)句，插入表數(shù)據(jù)

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

我們?cè)賮?lái)嘗試查詢表數(shù)據(jù)，這里要注意，目前所有的操作都是基于定義的標(biāo)簽來(lái)選擇執(zhí)行的

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類型，同學(xué)們可以自行遍歷取值測(cè)試，大功告成！

其他玩法

我們可以先預(yù)準(zhǔn)備SQL語(yǔ)句，再在合適的時(shí)機(jī)執(zhí)行

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

同樣，我們也可以將多個(gè)SQL文件合并再進(jìn)行取值操作

dot1, err := dotsql.LoadFromFile("queries1.sql")
dot2, err := dotsql.LoadFromFile("queries2.sql")
dot := dotsql.Merge(dot1, dot2)

總結(jié)

到此這篇關(guān)于Go語(yǔ)言庫(kù)系列之dotsql的文章就介紹到這了,更多相關(guān)go語(yǔ)言庫(kù) dotsql內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！