创建 Engine 组

创建 Engine 组

通过创建引擎组 EngineGroup 来实现对从数据库 (Master/Slave) 读写分离的支持。在创建引擎章节中，我们已经介绍过了，在 xorm 里面，可以同时存在多个 Orm 引擎，一个 Orm 引擎称为 Engine，一个 Engine 一般只对应一个数据库，而 EngineGroup 一般则对应一组数据库。EngineGroup 通过调用 xorm.NewEngineGroup 生成，如：

  1. import (
  2.     _ "github.com/lib/pq"
  3.     "xorm.io/xorm"
  4. )
  6. var eg *xorm.EngineGroup
  8. func main() {
  9.     conns := []string{
  10.         "postgres://postgres:[email protected]:5432/test?sslmode=disable;", // 第一个默认是master
  11.         "postgres://postgres:[email protected]:5432/test1?sslmode=disable;", // 第二个开始都是slave
  12.         "postgres://postgres:[email protected]:5432/test2?sslmode=disable",
  13.     }
  15.     var err error
  16.     eg, err = xorm.NewEngineGroup("postgres", conns)
  17. }

或者

  1. import (
  2.     _ "github.com/lib/pq"
  3.     "xorm.io/xorm"
  4. )
  6. var eg *xorm.EngineGroup
  8. func main() {
  9.     var err error
  10.     master, err := xorm.NewEngine("postgres", "postgres://postgres:[email protected]:5432/test?sslmode=disable")
  11.     if err != nil {
  12.         return
  13.     }
  15.     slave1, err := xorm.NewEngine("postgres", "postgres://postgres:[email protected]:5432/test1?sslmode=disable")
  16.     if err != nil {
  17.         return
  18.     }
  20.     slave2, err := xorm.NewEngine("postgres", "postgres://postgres:[email protected]:5432/test2?sslmode=disable")
  21.     if err != nil {
  22.         return
  23.     }
  25.        slaves := []*xorm.Engine{slave1, slave2}
  26.     eg, err = xorm.NewEngineGroup(master, slaves)
  27. }

创建完成 EngineGroup 之后，并没有立即连接数据库，此时可以通过 eg.Ping() 来进行数据库的连接测试是否可以连接到数据库，该方法会依次调用引擎组中每个Engine的Ping方法。另外对于某些数据库有连接超时设置的，可以通过起一个定期Ping的Go程来保持连接鲜活。EngineGroup 可以通过 eg.Close() 来手动关闭，但是一般情况下可以不用关闭，在程序退出时会自动关闭。

  • NewEngineGroup方法
  1. func NewEngineGroup(args1 interface{}, args2 interface{}, policies ...GroupPolicy) (*EngineGroup, error)

前两个参数的使用示例如上，有两种模式。

  • 模式一：通过给定 DriverName，DataSourceName 来创建引擎组，每个引擎使用相同的Driver。每个引擎的 DataSourceNames 是 []string 类型，第一个元素是 Master 的 DataSourceName，之后的元素是 Slave的DataSourceName 。

  • 模式二：通过给定*xorm.Engine，[]*xorm.Engine 来创建引擎组，每个引擎可以使用不同的 Driver。第一个参数为 Master 的 *xorm.Engine，第二个参数为 Slave 的 []*xorm.Engine。 NewEngineGroup 方法，第三个参数为 policies，为 Slave 设定负载策略，该参数将在负载策略章节详细介绍，如示例中未指定，则默认为轮询负载策略。

  • Master方法

  1. func (eg *EngineGroup) Master() *Engine

返回Master数据库引擎

  • Slave方法
  1. func (eg *EngineGroup) Slave() *Engine

依据给定的负载策略返回一个Slave数据库引擎

  • Slaves方法
  1. func (eg *EngineGroup) Slaves() []*Engine

返回所以 Slave 数据库引擎

  • SetPolicy方法
  1. func (eg *EngineGroup) SetPolicy(policy GroupPolicy) *EngineGroup

设置引擎组负载策略