You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
634 B
23 lines
634 B
package callbacks |
|
|
|
import ( |
|
"reflect" |
|
|
|
"gorm.io/gorm" |
|
) |
|
|
|
func callMethod(db *gorm.DB, fc func(value interface{}, tx *gorm.DB) bool) { |
|
tx := db.Session(&gorm.Session{NewDB: true}) |
|
if called := fc(db.Statement.ReflectValue.Interface(), tx); !called { |
|
switch db.Statement.ReflectValue.Kind() { |
|
case reflect.Slice, reflect.Array: |
|
db.Statement.CurDestIndex = 0 |
|
for i := 0; i < db.Statement.ReflectValue.Len(); i++ { |
|
fc(reflect.Indirect(db.Statement.ReflectValue.Index(i)).Addr().Interface(), tx) |
|
db.Statement.CurDestIndex++ |
|
} |
|
case reflect.Struct: |
|
fc(db.Statement.ReflectValue.Addr().Interface(), tx) |
|
} |
|
} |
|
}
|
|
|