# DBResolver
DBResolver 為 GORM 提供了多個數據庫支持,支持以下功能:
- 支持多個 sources、replicas
- 讀寫分離
- 根據工作表、struct 自動切換連接
- 手動切換連接
- Sources/Replicas 負載均衡
- 適用于原生 SQL
https://github.com/go-gorm/dbresolver
## 用法
```go
import (
"gorm.io/gorm"
"gorm.io/plugin/dbresolver"
"gorm.io/driver/mysql"
)
DB, err := gorm.Open(mysql.Open("db1_dsn"), &gorm.Config{})
DB.Use(dbresolver.Register(dbresolver.Config{
// `db2` 作為 sources,`db3`、`db4` 作為 replicas
Sources: []gorm.Dialector{mysql.Open("db2_dsn")},
Replicas: []gorm.Dialector{mysql.Open("db3_dsn"), mysql.Open("db4_dsn")},
// sources/replicas 負載均衡策略
Policy: dbresolver.RandomPolicy{},
}).Register(dbresolver.Config{
// `db1` 作為 sources(DB 的默認連接),對于 `User`、`Address` 使用 `db5` 作為 replicas
Replicas: []gorm.Dialector{mysql.Open("db5_dsn")},
}, &User{}, &Address{}).Register(dbresolver.Config{
// `db6`、`db7` 作為 sources,對于 `orders`、`Product` 使用 `db8` 作為 replicas
Sources: []gorm.Dialector{mysql.Open("db6_dsn"), mysql.Open("db7_dsn")},
Replicas: []gorm.Dialector{mysql.Open("db8_dsn")},
}, "orders", &Product{}, "secondary"))
```
## 事務
使用 transaction 時,DBResolver 也會使用一個事務,且不會切換 sources/replicas 連接
## 自動切換連接
DBResolver 會根據工作表、struct 自動切換連接
對于原生 SQL,DBResolver 會從 SQL 中提取表名以匹配 Resolver,除非 SQL 開頭為 `SELECT`,否則 DBResolver 總是會使用 `sources`,例如:
```go
// `User` Resolver 示例
DB.Table("users").Rows() // replicas `db5`
DB.Model(&User{}).Find(&AdvancedUser{}) // replicas `db5`
DB.Exec("update users set name = ?", "jinzhu") // sources `db1`
DB.Raw("select name from users").Row().Scan(&name) // replicas `db5`
DB.Create(&user) // sources `db1`
DB.Delete(&User{}, "name = ?", "jinzhu") // sources `db1`
DB.Table("users").Update("name", "jinzhu") // sources `db1`
// 全局 Resolver 示例
DB.Find(&Pet{}) // replicas `db3`/`db4`
DB.Save(&Pet{}) // sources `db2`
// Orders Resolver 示例
DB.Find(&Order{}) // replicas `db8`
DB.Table("orders").Find(&Report{}) // replicas `db8`
```
## 讀寫分離
DBResolver 的讀寫分離目前是基于 [GORM callback](https://gorm.io/docs/write_plugins.html) 實現的。
對于 `Query`、`Row` callback,如果手動指定為 `Write` 模式,此時會使用 `sources`,否則使用 `replicas`。 對于 `Raw` callback,如果 SQL 是以 `SELECT` 開頭,語句會被認為是只讀的,會使用 `replicas`,否則會使用 `sources`。
## 手動切換連接
```go
// 使用 Write 模式:從 sources db `db1` 讀取 user
DB.Clauses(dbresolver.Write).First(&user)
// 指定 Resolver:從 `secondary` 的 replicas db `db8` 讀取 user
DB.Clauses(dbresolver.Use("secondary")).First(&user)
// 指定 Resolver 和 Write 模式:從 `secondary` 的 sources db `db6` 或 `db7` 讀取 user
DB.Clauses(dbresolver.Use("secondary"), dbresolver.Write).First(&user)
```
## 負載均衡
GORM 支持基于策略的 sources/replicas 負載均衡,自定義策略需實現以下接口:
```go
type Policy interface {
Resolve([]gorm.ConnPool) gorm.ConnPool
}
```
當前只實現了一個 `RandomPolicy` 策略,如果沒有指定策略,它就是默認策略。
## 連接池
```go
DB.Use(
dbresolver.Register(dbresolver.Config{ /* xxx */ }).
SetConnMaxIdleTime(time.Hour).
SetConnMaxLifetime(24 * time.Hour).
SetMaxIdleConns(100).
SetMaxOpenConns(200)
)
```