开发

架构 {#a}

Gorm使用可链接的API,*gorm.DB是链的桥梁,对于每个链API,它将创建一个新的关系。

  1. db, err := gorm.Open("postgres", "user=gorm dbname=gorm sslmode=disable")
  2. // 创建新关系
  3. db = db.Where("name = ?", "jinzhu")
  4. // 过滤更多
  5. if SomeCondition {
  6. db = db.Where("age = ?", 20)
  7. } else {
  8. db = db.Where("age = ?", 30)
  9. }
  10. if YetAnotherCondition {
  11. db = db.Where("active = ?", 1)
  12. }

当我们开始执行任何操作时,GORM将基于当前的*gorm.DB创建一个新的*gorm.Scope实例

  1. // 执行查询操作
  2. db.First(&user)

并且基于当前操作的类型,它将调用注册的creating, updating, querying, deletingrow_querying回调来运行操作。

对于上面的例子,将调用querying,参考查询回调

写插件 {#w}

GORM本身由Callbacks提供支持,因此您可以根据需要完全自定义GORM

注册新的callback

  1. func updateCreated(scope *Scope) {
  2. if scope.HasColumn("Created") {
  3. scope.SetColumn("Created", NowFunc())
  4. }
  5. }
  6. db.Callback().Create().Register("update_created_at", updateCreated)
  7. // 注册Create进程的回调

删除现有的callback

  1. db.Callback().Create().Remove("gorm:create")
  2. // 从Create回调中删除`gorm:create`回调

替换现有的callback

  1. db.Callback().Create().Replace("gorm:create", newCreateFunction)
  2. // 使用新函数`newCreateFunction`替换回调`gorm:create`用于创建过程

注册callback顺序

  1. db.Callback().Create().Before("gorm:create").Register("update_created_at", updateCreated)
  2. db.Callback().Create().After("gorm:create").Register("update_created_at", updateCreated)
  3. db.Callback().Query().After("gorm:query").Register("my_plugin:after_query", afterQuery)
  4. db.Callback().Delete().After("gorm:delete").Register("my_plugin:after_delete", afterDelete)
  5. db.Callback().Update().Before("gorm:update").Register("my_plugin:before_update", beforeUpdate)
  6. db.Callback().Create().Before("gorm:create").After("gorm:before_create").Register("my_plugin:before_create", beforeCreate)

预定义回调

GORM定义了回调以执行其CRUD操作,在开始编写插件之前检查它们。

  1. func updateTableName(scope *gorm.Scope) {
  2. scope.Search.Table(scope.TableName() + "_draft") // append `_draft` to table name
  3. }
  4. db.Callback().RowQuery().Register("publish:update_table_name", updateTableName)