Fix build

This commit is contained in:
Lunny Xiao 2021-06-08 17:34:13 +08:00
parent 4552e26d7b
commit c4f8c17d71
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
3 changed files with 26 additions and 23 deletions

View File

@ -1212,10 +1212,10 @@ func (engine *Engine) Update(bean interface{}, condiBeans ...interface{}) (int64
}
// Delete records, bean's non-empty fields are conditions
func (engine *Engine) Delete(bean interface{}) (int64, error) {
func (engine *Engine) Delete(beans ...interface{}) (int64, error) {
session := engine.NewSession()
defer session.Close()
return session.Delete(bean)
return session.Delete(beans...)
}
// Get retrieve one record from table, bean's non-empty fields

View File

@ -30,7 +30,7 @@ type Interface interface {
CreateUniques(bean interface{}) error
Decr(column string, arg ...interface{}) *Session
Desc(...string) *Session
Delete(interface{}) (int64, error)
Delete(...interface{}) (int64, error)
Distinct(columns ...string) *Session
DropIndexes(bean interface{}) error
Exec(sqlOrArgs ...interface{}) (sql.Result, error)

View File

@ -96,9 +96,10 @@ func (session *Session) Delete(beans ...interface{}) (int64, error) {
condSQL string
condArgs []interface{}
err error
bean interface{}
)
if len(beans) > 0 {
bean := beans[0]
bean = beans[0]
if err = session.statement.SetRefBean(bean); err != nil {
return 0, err
}
@ -231,6 +232,7 @@ func (session *Session) Delete(beans ...interface{}) (int64, error) {
return 0, err
}
if bean != nil {
// handle after delete processors
if session.isAutoCommit {
for _, closure := range session.afterClosures {
@ -255,6 +257,7 @@ func (session *Session) Delete(beans ...interface{}) (int64, error) {
}
}
}
}
cleanupProcessorsClosures(&session.afterClosures)
// --