Hein
19 lines
502 B
Go
19 lines
502 B
Go
package sqlserver
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/callbacks"
|
|
)
|
|
|
|
var updateFunc = callbacks.Update(&callbacks.Config{
|
|
UpdateClauses: []string{"UPDATE", "SET", "RETURNING", "FROM", "WHERE"},
|
|
})
|
|
|
|
func Update(db *gorm.DB) {
|
|
if db.Statement.Schema != nil && db.Statement.Schema.PrioritizedPrimaryField != nil && db.Statement.Schema.PrioritizedPrimaryField.AutoIncrement {
|
|
db.Statement.Omits = append(db.Statement.Omits, db.Statement.Schema.PrioritizedPrimaryField.DBName)
|
|
}
|
|
|
|
updateFunc(db)
|
|
}
|