1. 创建插件

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

1.1. 注册新的 callback

将 callback 注册进如 callbacks:

  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 进程的回调

1.2. 删除已有的 callback

从 callbacks 中删除一个 callback:

  1. db.Callback().Create().Remove("gorm:create")
  2. // delete callback `gorm:create` from Create callbacks

1.3. 替换 callback

替换拥有相同名字的 callback :

  1. db.Callback().Create().Replace("gorm:create", newCreateFunction)
  2. // replace callback `gorm:create` with new function `newCreateFunction` for Create process

1.4. 注册 callback 的顺序

在注册 callbacks 时设置顺序:

  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)

1.5. 自带的 Callbacks

GORM 在处理 CRUD 操作时自带了一些 Callback,建议你在写插件前先熟悉这些 Callback:

你可以用以下的方法来注册你的 Callback:

  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)

请前往查看所有的 API —— https://godoc.org/github.com/jinzhu/gorm