gorm是如何保证协程安全的 |
您所在的位置：网站首页 › 如何保证保姆的安全 › gorm是如何保证协程安全的 |
|
Gorm 官方文档提供了如何正确使用链式调用的例子以及会引起协程不安全的反例，知道了如何正确使用，也要知道原理才能用的更安心。下面以文档示例和源码切入，浅析 Gorm 在链式调用时时如何保证协程安全的？ 源码分析下面是一段 gorm 常见的使用代码，先初始化连接，然后根据链式调用进行增删改。 db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{}) if err != nil { panic("failed to connect database") } tx := db.Where("age = 22").Where("name = '小明'").Find(&user)通过 gorm.Open() 初始化连接，将拿到一个 *gorm.DB 结构体指针 db。结构体的 clone 属性为 1。 type DB struct { *Config Error error RowsAffected int64 Statement *Statement clone int } func Open(dialector Dialector, opts ...Option) (db *DB, err error) { // …… db = &DB{Config: config, clone: 1} // …… }通过 db.Where() 进行链式查询，Where() 方法也将返回一个*gorm.DB结构体指针 tx。 // Where add conditions func (db *DB) Where(query interface{}, args ...interface{}) (tx *DB) { tx = db.getInstance() if conds := tx.Statement.BuildCondition(query, args...); len(conds) > 0 { tx.Statement.AddClause(clause.Where{Exprs: conds}) } return }查看源码可以发现，像Where()、Select()、Limit() 等每个链式方法中，都要先去获取 *gorm.DB结构体指针 tx，也就是 tx = db.getInstance()。 func (db *DB) getInstance() *DB { if db.clone > 0 { tx := &DB{Config: db.Config, Error: db.Error} if db.clone == 1 { // clone with new statement tx.Statement = &Statement{ DB: tx, ConnPool: db.Statement.ConnPool, Context: db.Statement.Context, Clauses: map[string]clause.Clause{}, Vars: make([]interface{}, 0, 8), } } else { // with clone statement tx.Statement = db.Statement.clone() tx.Statement.DB = tx } return tx } return db }当传入的 *gorm.DB指针指向的结构体属性 clone 为 1时，将会克隆一份 DB 结构体，并返回一个新的指针指向这个 DB 结构体，也即是创建一个新的会话。clone 为 0 时，返回原来的指针指向地址。 db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{}) if err != nil { panic("failed to connect database") } db.Where().Where() // 等同于 tx1 = db.Where() tx2 = tx.Where()基于上面的分析，在此例代码中，由于 db 指针指向 gorm.DB 结构体的 clone 属性为1。tx1 将指向一个复制的新的 gorm.DB 结构体，它的 clone 属性为 0。所以，tx.Where() 时将不会复制一个新的结构体（不会创建新会话），即 tx2 与 tx1 都指向同一个 gorm.DB 结构体，与 db 指向不同。 例子经过上面的剖析，再看官方文档的例子，就能理解怎样使用是协程安全的链式调用。 例子1 db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{}) // 安全的使用新初始化的 *gorm.DB for i := 0; i |
今日新闻 |
推荐新闻 |
| CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |