Go沒(méi)有內(nèi)置的驅(qū)動(dòng)支持任何的數(shù)據(jù)庫(kù)，但是Go定義了database/sql接口，用戶(hù)可以基于驅(qū)動(dòng)接口開(kāi)發(fā)相應(yīng)數(shù)據(jù)庫(kù)的驅(qū)動(dòng)。

1季惩、sql.Register

注冊(cè)數(shù)據(jù)庫(kù)驅(qū)動(dòng)，當(dāng)?shù)谌介_(kāi)發(fā)者開(kāi)發(fā)數(shù)據(jù)庫(kù)驅(qū)動(dòng)時(shí)腻格，都會(huì)實(shí)現(xiàn)init函數(shù)画拾，在init里面會(huì)調(diào)用這個(gè)Register(name string, driver driver.Driver)完成本驅(qū)動(dòng)的注冊(cè)。

我們來(lái)看一下mymysql菜职、sqlite3的驅(qū)動(dòng)里面都是怎么調(diào)用的：

//https://github.com/mattn/go-sqlite3驅(qū)動(dòng)

func init() {

????sql.Register("sqlite3", &SQLiteDriver{})

}

//https://github.com/mikespook/mymysql驅(qū)動(dòng)

// Driver automatically registered in database/sql

var d = Driver{proto: "tcp", raddr: "127.0.0.1:3306"}

func init() {

????Register("SET NAMES utf8")

????sql.Register("mymysql", &d)

}

我們看到第三方數(shù)據(jù)庫(kù)驅(qū)動(dòng)都是通過(guò)調(diào)用這個(gè)函數(shù)來(lái)注冊(cè)自己的數(shù)據(jù)庫(kù)驅(qū)動(dòng)名稱(chēng)以及相應(yīng)的driver實(shí)現(xiàn)青抛。在database/sql內(nèi)部通過(guò)一個(gè)map來(lái)存儲(chǔ)用戶(hù)定義的相應(yīng)驅(qū)動(dòng)。

var drivers = make(map[string]driver.Driver)

drivers[name] = driver

因此通過(guò)database/sql的注冊(cè)函數(shù)可以同時(shí)注冊(cè)多個(gè)數(shù)據(jù)庫(kù)驅(qū)動(dòng)酬核，只要不重復(fù)蜜另。

2、driver.Driver

Driver是一個(gè)數(shù)據(jù)庫(kù)驅(qū)動(dòng)的接口嫡意，他定義了一個(gè)method： Open(name string)举瑰，這個(gè)方法返回一個(gè)數(shù)據(jù)庫(kù)的Conn接口。

type Driver interface {

????Open(name string) (Conn, error)

}

返回的Conn只能用來(lái)進(jìn)行一次goroutine的操作蔬螟，也就是說(shuō)不能把這個(gè)Conn應(yīng)用于Go的多個(gè)goroutine里面此迅。如下代碼會(huì)出現(xiàn)錯(cuò)誤

...

go goroutineA (Conn)? //執(zhí)行查詢(xún)操作

go goroutineB (Conn)? //執(zhí)行插入操作

...

上面這樣的代碼可能會(huì)使Go不知道某個(gè)操作究竟是由哪個(gè)goroutine發(fā)起的旧巾，從而導(dǎo)致數(shù)據(jù)混亂耸序，比如可能會(huì)把goroutineA里面執(zhí)行的查詢(xún)操作的結(jié)果返回給goroutineB從而使B錯(cuò)誤地把此結(jié)果當(dāng)成自己執(zhí)行的插入數(shù)據(jù)。

第三方驅(qū)動(dòng)都會(huì)定義這個(gè)函數(shù)鲁猩，它會(huì)解析name參數(shù)來(lái)獲取相關(guān)數(shù)據(jù)庫(kù)的連接信息坎怪，解析完成后，它將使用此信息來(lái)初始化一個(gè)Conn并返回它廓握。

3搅窿、driver.Conn

Conn是一個(gè)數(shù)據(jù)庫(kù)連接的接口定義，他定義了一系列方法隙券，這個(gè)Conn只能應(yīng)用在一個(gè)goroutine里面男应，不能使用在多個(gè)goroutine里面，詳情請(qǐng)參考上面的說(shuō)明是尔。

type Conn interface {

????Prepare(query string) (Stmt, error)

????Close() error

????Begin() (Tx, error)

}

Prepare函數(shù)返回與當(dāng)前連接相關(guān)的執(zhí)行Sql語(yǔ)句的準(zhǔn)備狀態(tài)，可以進(jìn)行查詢(xún)开仰、刪除等操作拟枚。

Close函數(shù)關(guān)閉當(dāng)前的連接薪铜，執(zhí)行釋放連接擁有的資源等清理工作。因?yàn)轵?qū)動(dòng)實(shí)現(xiàn)了database/sql里面建議的conn pool恩溅，所以你不用再去實(shí)現(xiàn)緩存conn之類(lèi)的隔箍，這樣會(huì)容易引起問(wèn)題。

Begin函數(shù)返回一個(gè)代表事務(wù)處理的Tx脚乡，通過(guò)它你可以進(jìn)行查詢(xún)蜒滩、更新等操作，或者對(duì)事務(wù)進(jìn)行回滾奶稠、遞交俯艰。

4、driver.Stmt

Stmt是一種準(zhǔn)備好的狀態(tài)锌订，和Conn相關(guān)聯(lián)竹握，而且只能應(yīng)用于一個(gè)goroutine中，不能應(yīng)用于多個(gè)goroutine辆飘。

type Stmt interface {

????Close() error

????NumInput() int

????Exec(args []Value) (Result, error)

????Query(args []Value) (Rows, error)

}

Close函數(shù)關(guān)閉當(dāng)前的鏈接狀態(tài)啦辐，但是如果當(dāng)前正在執(zhí)行query，query還是有效返回rows數(shù)據(jù)蜈项。

NumInput函數(shù)返回當(dāng)前預(yù)留參數(shù)的個(gè)數(shù)芹关，當(dāng)返回>=0時(shí)數(shù)據(jù)庫(kù)驅(qū)動(dòng)就會(huì)智能檢查調(diào)用者的參數(shù)。當(dāng)數(shù)據(jù)庫(kù)驅(qū)動(dòng)包不知道預(yù)留參數(shù)的時(shí)候紧卒，返回-1侥衬。

Exec函數(shù)執(zhí)行Prepare準(zhǔn)備好的sql，傳入?yún)?shù)執(zhí)行update/insert等操作常侦，返回Result數(shù)據(jù)浇冰。

Query函數(shù)執(zhí)行Prepare準(zhǔn)備好的sql，傳入需要的參數(shù)執(zhí)行select操作聋亡，返回Rows結(jié)果集肘习。

5、driver.Tx

事務(wù)處理一般就兩個(gè)過(guò)程坡倔，遞交或者回滾漂佩。數(shù)據(jù)庫(kù)驅(qū)動(dòng)里面也只需要實(shí)現(xiàn)這兩個(gè)函數(shù)就可以

type Tx interface {

????Commit() error ? ?// 提交

????Rollback() error ? ?// 回滾

}

這兩個(gè)函數(shù)一個(gè)用來(lái)遞交一個(gè)事務(wù)，一個(gè)用來(lái)回滾事務(wù)罪塔。

6投蝉、driver.Execer

這是一個(gè)Conn可選擇實(shí)現(xiàn)的接口

type Execer interface {

????Exec(query string, args []Value) (Result, error)

}

如果這個(gè)接口沒(méi)有定義，那么在調(diào)用DB.Exec征堪，就會(huì)首先調(diào)用Prepare返回Stmt瘩缆，然后執(zhí)行Stmt的Exec，然后關(guān)閉Stmt佃蚜。

7庸娱、driver.Result

這個(gè)是執(zhí)行Update/Insert等操作返回的結(jié)果接口定義

type Result interface {

????LastInsertId() (int64, error)

????RowsAffected() (int64, error)

}

LastInsertId函數(shù)返回由數(shù)據(jù)庫(kù)執(zhí)行插入操作得到的自增ID號(hào)着绊。

RowsAffected函數(shù)返回query操作影響的數(shù)據(jù)條目數(shù)。

8熟尉、driver.Rows

Rows是執(zhí)行查詢(xún)返回的結(jié)果集接口定義

type Rows interface {

????Columns() []string

????Close() error

????Next(dest []Value) error

}

Columns函數(shù)返回查詢(xún)數(shù)據(jù)庫(kù)表的字段信息归露，這個(gè)返回的slice和sql查詢(xún)的字段一一對(duì)應(yīng)，而不是返回整個(gè)表的所有字段斤儿。

Close函數(shù)用來(lái)關(guān)閉Rows迭代器剧包。

Next函數(shù)用來(lái)返回下一條數(shù)據(jù)，把數(shù)據(jù)賦值給dest往果。dest里面的元素必須是driver.Value的值除了string疆液，返回的數(shù)據(jù)里面所有的string都必須要轉(zhuǎn)換成[]byte。如果最后沒(méi)數(shù)據(jù)了棚放，Next函數(shù)最后返回io.EOF枚粘。

9、driver.RowsAffected

RowsAffected其實(shí)就是一個(gè)int64的別名飘蚯，但是他實(shí)現(xiàn)了Result接口馍迄，用來(lái)底層實(shí)現(xiàn)Result的表示方式

type RowsAffected int64

func (RowsAffected) LastInsertId() (int64, error)

func (v RowsAffected) RowsAffected() (int64, error)

10、driver.Value

Value其實(shí)就是一個(gè)空接口局骤，他可以容納任何的數(shù)據(jù)

type Value interface{}

drive的Value是驅(qū)動(dòng)必須能夠操作的Value攀圈，Value要么是nil，要么是下面的任意一種：

int64

float64

bool

[]byte

string? [*]除了Rows.Next返回的不能是string.

time.Time

11峦甩、driver.ValueConverter

ValueConverter接口定義了如何把一個(gè)普通的值轉(zhuǎn)化成driver.Value的接口

type ValueConverter interface {

????ConvertValue(v interface{}) (Value, error)

}

在開(kāi)發(fā)的數(shù)據(jù)庫(kù)驅(qū)動(dòng)包里面實(shí)現(xiàn)這個(gè)接口的函數(shù)在很多地方會(huì)使用到赘来，這個(gè)ValueConverter有很多好處：

* 轉(zhuǎn)化driver.value到數(shù)據(jù)庫(kù)表相應(yīng)的字段，例如int64的數(shù)據(jù)如何轉(zhuǎn)化成數(shù)據(jù)庫(kù)表uint16字段

* 把數(shù)據(jù)庫(kù)查詢(xún)結(jié)果轉(zhuǎn)化成driver.Value值

* 在scan函數(shù)里面如何把driver.Value值轉(zhuǎn)化成用戶(hù)定義的值

12凯傲、driver.Valuer

Valuer接口定義了返回一個(gè)driver.Value的方式

type Valuer interface {

????Value() (Value, error)

}

很多類(lèi)型都實(shí)現(xiàn)了這個(gè)Value方法犬辰，用來(lái)自身與driver.Value的轉(zhuǎn)化。

13冰单、database/sql

database/sql在database/sql/driver提供的接口基礎(chǔ)上定義了一些更高階的方法幌缝，用以簡(jiǎn)化數(shù)據(jù)庫(kù)操作，同時(shí)內(nèi)部還建議性地實(shí)現(xiàn)一個(gè)conn pool诫欠。

type DB struct {

????driver driver.Driver

????dsn? ? string

????mu? ? ? sync.Mutex // protects freeConn and closed

????freeConn []driver.Conn

????closed? bool

}

我們可以看到Open函數(shù)返回的是DB對(duì)象涵卵，里面有一個(gè)freeConn，它就是那個(gè)簡(jiǎn)易的連接池荒叼。它的實(shí)現(xiàn)相當(dāng)簡(jiǎn)單或者說(shuō)簡(jiǎn)陋轿偎，就是當(dāng)執(zhí)行db.prepare -> db.prepareDC的時(shí)候會(huì)defer dc.releaseConn，然后調(diào)用db.putConn被廓，也就是把這個(gè)連接放入連接池坏晦，每次調(diào)用db.conn的時(shí)候會(huì)先判斷freeConn的長(zhǎng)度是否大于0，大于0說(shuō)明有可以復(fù)用的conn，直接拿出來(lái)用就是了昆婿，如果不大于0间护，則創(chuàng)建一個(gè)conn，然后再返回之挖诸。

總結(jié)

一個(gè)驅(qū)動(dòng)只要實(shí)現(xiàn)了這些接口就能完成增刪查改等基本操作了，剩下的就是與相應(yīng)的數(shù)據(jù)庫(kù)進(jìn)行數(shù)據(jù)交互等細(xì)節(jié)問(wèn)題了法精。