73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package callbacks
|
|
|
|
import (
|
|
"database/sql"
|
|
"reflect"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/jinzhu/gorm/clause"
|
|
)
|
|
|
|
func Query(db *gorm.DB) {
|
|
if db.Statement.SQL.String() == "" {
|
|
db.Statement.AddClauseIfNotExists(clause.Select{})
|
|
db.Statement.AddClauseIfNotExists(clause.From{})
|
|
db.Statement.Build("SELECT", "FROM", "WHERE", "GROUP BY", "ORDER BY", "LIMIT", "FOR")
|
|
}
|
|
|
|
rows, err := db.DB.QueryContext(db.Context, db.Statement.SQL.String(), db.Statement.Vars...)
|
|
if err != nil {
|
|
db.AddError(err)
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
|
|
columns, _ := rows.Columns()
|
|
values := make([]interface{}, len(columns))
|
|
|
|
for idx, column := range columns {
|
|
if field, ok := db.Statement.Schema.FieldsByDBName[column]; ok {
|
|
values[idx] = field.ReflectValueOf(db.Statement.ReflectValue).Addr().Interface()
|
|
} else {
|
|
values[idx] = sql.RawBytes{}
|
|
}
|
|
}
|
|
|
|
for rows.Next() {
|
|
db.RowsAffected++
|
|
rows.Scan(values...)
|
|
}
|
|
|
|
if db.RowsAffected == 0 && db.Statement.RaiseErrorOnNotFound {
|
|
db.AddError(gorm.ErrRecordNotFound)
|
|
}
|
|
}
|
|
|
|
func Preload(db *gorm.DB) {
|
|
}
|
|
|
|
func AfterQuery(db *gorm.DB) {
|
|
if db.Statement.Schema != nil && db.Statement.Schema.AfterFind {
|
|
callMethod := func(value interface{}) bool {
|
|
if db.Statement.Schema.AfterFind {
|
|
if i, ok := value.(gorm.AfterFindInterface); ok {
|
|
i.AfterFind(db)
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
if ok := callMethod(db.Statement.Dest); !ok {
|
|
switch db.Statement.ReflectValue.Kind() {
|
|
case reflect.Slice, reflect.Array:
|
|
for i := 0; i <= db.Statement.ReflectValue.Len(); i++ {
|
|
callMethod(db.Statement.ReflectValue.Index(i).Interface())
|
|
}
|
|
case reflect.Struct:
|
|
callMethod(db.Statement.ReflectValue.Interface())
|
|
}
|
|
}
|
|
}
|
|
}
|