180 lines
4.2 KiB
Go
180 lines
4.2 KiB
Go
package orm
|
|
|
|
import "reflect"
|
|
|
|
const (
|
|
AfterQueryHookFlag = uint8(1) << iota
|
|
AfterSelectHookFlag
|
|
BeforeInsertHookFlag
|
|
AfterInsertHookFlag
|
|
BeforeUpdateHookFlag
|
|
AfterUpdateHookFlag
|
|
BeforeDeleteHookFlag
|
|
AfterDeleteHookFlag
|
|
)
|
|
|
|
type hookStubs struct{}
|
|
|
|
func (hookStubs) Init() error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) AfterQuery(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) AfterSelect(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) BeforeInsert(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) AfterInsert(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) BeforeUpdate(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) AfterUpdate(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) BeforeDelete(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func (hookStubs) AfterDelete(_ DB) error {
|
|
return nil
|
|
}
|
|
|
|
func callHookSlice(slice reflect.Value, ptr bool, db DB, hook func(reflect.Value, DB) error) error {
|
|
var firstErr error
|
|
for i := 0; i < slice.Len(); i++ {
|
|
var err error
|
|
if ptr {
|
|
err = hook(slice.Index(i), db)
|
|
} else {
|
|
err = hook(slice.Index(i).Addr(), db)
|
|
}
|
|
if err != nil && firstErr == nil {
|
|
firstErr = err
|
|
}
|
|
}
|
|
return firstErr
|
|
}
|
|
|
|
type afterQueryHook interface {
|
|
AfterQuery(db DB) error
|
|
}
|
|
|
|
var afterQueryHookType = reflect.TypeOf((*afterQueryHook)(nil)).Elem()
|
|
|
|
func callAfterQueryHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(afterQueryHook).AfterQuery(db)
|
|
}
|
|
|
|
func callAfterQueryHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callAfterQueryHook)
|
|
}
|
|
|
|
type afterSelectHook interface {
|
|
AfterSelect(db DB) error
|
|
}
|
|
|
|
var afterSelectHookType = reflect.TypeOf((*afterSelectHook)(nil)).Elem()
|
|
|
|
func callAfterSelectHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(afterSelectHook).AfterSelect(db)
|
|
}
|
|
|
|
func callAfterSelectHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callAfterSelectHook)
|
|
}
|
|
|
|
type beforeInsertHook interface {
|
|
BeforeInsert(db DB) error
|
|
}
|
|
|
|
var beforeInsertHookType = reflect.TypeOf((*beforeInsertHook)(nil)).Elem()
|
|
|
|
func callBeforeInsertHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(beforeInsertHook).BeforeInsert(db)
|
|
}
|
|
|
|
func callBeforeInsertHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callBeforeInsertHook)
|
|
}
|
|
|
|
type afterInsertHook interface {
|
|
AfterInsert(db DB) error
|
|
}
|
|
|
|
var afterInsertHookType = reflect.TypeOf((*afterInsertHook)(nil)).Elem()
|
|
|
|
func callAfterInsertHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(afterInsertHook).AfterInsert(db)
|
|
}
|
|
|
|
func callAfterInsertHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callAfterInsertHook)
|
|
}
|
|
|
|
type beforeUpdateHook interface {
|
|
BeforeUpdate(db DB) error
|
|
}
|
|
|
|
var beforeUpdateHookType = reflect.TypeOf((*beforeUpdateHook)(nil)).Elem()
|
|
|
|
func callBeforeUpdateHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(beforeUpdateHook).BeforeUpdate(db)
|
|
}
|
|
|
|
func callBeforeUpdateHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callBeforeUpdateHook)
|
|
}
|
|
|
|
type afterUpdateHook interface {
|
|
AfterUpdate(db DB) error
|
|
}
|
|
|
|
var afterUpdateHookType = reflect.TypeOf((*afterUpdateHook)(nil)).Elem()
|
|
|
|
func callAfterUpdateHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(afterUpdateHook).AfterUpdate(db)
|
|
}
|
|
|
|
func callAfterUpdateHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callAfterUpdateHook)
|
|
}
|
|
|
|
type beforeDeleteHook interface {
|
|
BeforeDelete(db DB) error
|
|
}
|
|
|
|
var beforeDeleteHookType = reflect.TypeOf((*beforeDeleteHook)(nil)).Elem()
|
|
|
|
func callBeforeDeleteHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(beforeDeleteHook).BeforeDelete(db)
|
|
}
|
|
|
|
func callBeforeDeleteHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callBeforeDeleteHook)
|
|
}
|
|
|
|
type afterDeleteHook interface {
|
|
AfterDelete(db DB) error
|
|
}
|
|
|
|
var afterDeleteHookType = reflect.TypeOf((*afterDeleteHook)(nil)).Elem()
|
|
|
|
func callAfterDeleteHook(v reflect.Value, db DB) error {
|
|
return v.Interface().(afterDeleteHook).AfterDelete(db)
|
|
}
|
|
|
|
func callAfterDeleteHookSlice(slice reflect.Value, ptr bool, db DB) error {
|
|
return callHookSlice(slice, ptr, db, callAfterDeleteHook)
|
|
}
|