Go語言庫系列之dotsql

Go語言庫系列之dotsql

導讀：能單獨拎出SQL檔案的某一行或幾行執行，是不是非常有趣？今天我們來介紹一下這個有意思的庫--dotsql。

背景介紹#

dotsql不是ORM，也不是SQL查詢語句的建構器，而是可以在一個SQL檔案中拎出某幾行來執行的工具，非常類似于ini配置檔案的讀取。如果還不了解，我們來看如下内容。

Copy

-- 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檔案中定義的語句，我們可以很清晰地看出，每條語句上方都以-- name的方式打上了“注解”，而作為開發人員，可以根據打了标記的名稱挑選語句執行。

快速上手#

準備工作#

目錄結構概覽

.

├── data.sql

├── go.mod

├── go.sum

└── main.go

初始化項目

go mod init dotsql

建立data.sql檔案，鍵入如下SQL，隻是示例，内容可以自定義。

DROP TABLE IF EXISTS users;

id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
name VARCHAR(255),
email VARCHAR(255)           

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

從注釋可以找到對應的語句，是一個建立表的操作

id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
name VARCHAR(255),
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, 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)

感謝大家的觀看，如果覺得文章對你有所幫助，歡迎關注公衆号「平也」，聚焦Go語言與技術原理。

作者： 平也

出處：

https://www.cnblogs.com/enochzzg/p/12620058.html