導讀：能單獨拎出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(?,?)

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

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

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

上面是SQL檔案中定義的語句，我們可以很清晰地看出，每條語句上方都以 -- name 的方式打上了“註解”，而作為開發人員，可以根據打了標記的名稱挑選語句執行。

快速上手
準備工作

目錄結構概覽

├── 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,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,email VARCHAR(255)
);

再來執行第二條語句，插入表資料

dot.Exec(db,"create-user","User Name","[email protected]")

我們再來嘗試查詢表資料，這裡要注意，目前所有的操作都是基於定義的標籤來選擇執行的

rows,_ := dot.Query(db,"find-users-by-email","[email protected]")
 var (
  id int
  name string
  email string
 )
 for rows.Next() {
  rows.Scan(&id,&name,&email)
  fmt.Println(id,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)

總結

到此這篇關於Go語言庫系列之dotsql的文章就介紹到這了,更多相關go語言庫 dotsql內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們！