组件分享之后端组件——对其GORM增加多数据库支持功能的组件dbresolver

组件分享之后端组件——对其GORM增加多数据库支持功能的组件dbresolver

背景

近期正在探索前端、后端、系统端各类常用组件与工具,对其一些常见的组件进行再次整理一下,形成标准化组件专题,后续该专题将包含各类语言中的一些常用组件。欢迎大家进行持续关注。

组件基本信息

内容

本节分享一个对其GORM增加多数据库支持功能的组件dbresolver,它支持以下功能:

  • 多个来源,副本
  • 读/写分离
  • 基于工作表/结构的自动连接切换
  • 手动连接切换
  • 源/副本负载均衡
  • 适用于原始 SQL
  • 事务

具体使用如下:

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{
  // use `db2` as sources, `db3`, `db4` as replicas
  Sources:  []gorm.Dialector{mysql.Open("db2_dsn")},
  Replicas: []gorm.Dialector{mysql.Open("db3_dsn"), mysql.Open("db4_dsn")},
  // sources/replicas load balancing policy
  Policy: dbresolver.RandomPolicy{},
}).Register(dbresolver.Config{
  // use `db1` as sources (DB's default connection), `db5` as replicas for `User`, `Address`
  Replicas: []gorm.Dialector{mysql.Open("db5_dsn")},
}, &User{}, &Address{}).Register(dbresolver.Config{
  // use `db6`, `db7` as sources, `db8` as replicas for `orders`, `Product`
  Sources:  []gorm.Dialector{mysql.Open("db6_dsn"), mysql.Open("db7_dsn")},
  Replicas: []gorm.Dialector{mysql.Open("db8_dsn")},
}, "orders", &Product{}, "secondary"))

自动切换

// `User` Resolver Examples
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`

// Global Resolver Examples
DB.Find(&Pet{}) // replicas `db3`/`db4`
DB.Save(&Pet{}) // sources `db2`

// Orders Resolver Examples
DB.Find(&Order{}) // replicas `db8`
DB.Table("orders").Find(&Report{}) // replicas `db8`

手动连接开关

// Use Write Mode: read user from sources `db1`
DB.Clauses(dbresolver.Write).First(&user)

// Specify Resolver: read user from `secondary`'s replicas: db8
DB.Clauses(dbresolver.Use("secondary")).First(&user)

// Specify Resolver and Write Mode: read user from `secondary`'s sources: db6 or db7
DB.Clauses(dbresolver.Use("secondary"), dbresolver.Write).First(&user)

事务

// Start transaction based on default replicas db
tx := DB.Clauses(dbresolver.Read).Begin()

// Start transaction based on default sources db
tx := DB.Clauses(dbresolver.Write).Begin()

// Start transaction based on `secondary`'s sources
tx := DB.Clauses(dbresolver.Use("secondary"), dbresolver.Write).Begin()

负载平衡

GORM支持基于策略的源/副本负载均衡,策略是一个接口,实现如下接口:

type Policy interface {
    Resolve([]gorm.ConnPool) gorm.ConnPool
}

目前只执行RandomPolicy,如果没有指定策略,则为默认选项。

连接池

DB.Use(
  dbresolver.Register(dbresolver.Config{ /* xxx */ }).
  SetConnMaxIdleTime(time.Hour).
  SetConnMaxLifetime(24 * time.Hour).
  SetMaxIdleConns(100).
  SetMaxOpenConns(200)
)

更多前后端组件,可以持续关注我,我将持续给大家分享各种各样的组件包。

本文声明:
88x31.png
知识共享许可协议
本作品由 cn華少 采用 知识共享署名-非商业性使用 4.0 国际许可协议 进行许可。
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
gorm是一个非常流行的Go语言的ORM库,它可以轻松地与多种数据库进行交互,包括MySQL、PostgreSQLSQLite等。 在实现多租户功能时,一般需要将每个租户的数据隔离开来,避免不同租户之间的数据混淆。具体实现方式可以有多种,下面介绍一种基于gorm的实现方式。 首先,我们需要为每个租户创建一个独立的数据库数据库名可以根据租户的标识来命名。例如,我们可以为租户A创建一个名为“tenant_a”的数据库,为租户B创建一个名为“tenant_b”的数据库,以此类推。 接下来,我们需要在gorm中配置多个数据库连接,每个连接对应一个租户的数据库。可以使用gorm的`Open`方法打开一个新的数据库连接,例如: ```go dbA, err := gorm.Open(mysql.Open("user:password@tcp(localhost:3306)/tenant_a"), &gorm.Config{}) dbB, err := gorm.Open(mysql.Open("user:password@tcp(localhost:3306)/tenant_b"), &gorm.Config{}) ``` 然后,我们可以通过gorm的`With`方法指定当前操作的数据库连接。例如: ```go // 获取租户A的用户列表 dbA.WithContext(ctx).Find(&users) // 获取租户B的用户列表 dbB.WithContext(ctx).Find(&users) ``` 最后,我们需要在gorm的模型定义中添加一个字段来表示所属的租户,例如: ```go type User struct { gorm.Model Name string Age int TenantID string // 表示所属租户的标识 } ``` 在查询数据时,我们可以通过`Where`方法指定当前操作的租户。例如: ```go // 获取租户A的年龄大于18的用户列表 dbA.WithContext(ctx).Where("age > ?", 18).Find(&users) // 获取租户B的年龄大于18的用户列表 dbB.WithContext(ctx).Where("age > ?", 18).Find(&users) ``` 通过上述方式,我们可以轻松地实现基于gorm的独立数据库多租户功能
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

CN華少

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值