From c4f8c17d71ba02fb30cd3b4f987aa82f945d463b Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Tue, 8 Jun 2021 17:34:13 +0800
Subject: [PATCH] Fix build

---
 engine.go         |  4 ++--
 interface.go      |  2 +-
 session_delete.go | 43 +++++++++++++++++++++++--------------------
 3 files changed, 26 insertions(+), 23 deletions(-)

diff --git a/engine.go b/engine.go
index d49eea9a..748cc20c 100644
--- a/engine.go
+++ b/engine.go
@@ -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
diff --git a/interface.go b/interface.go
index 55162c8c..6898d9dc 100644
--- a/interface.go
+++ b/interface.go
@@ -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)
diff --git a/session_delete.go b/session_delete.go
index 0b341de4..a2f8999b 100644
--- a/session_delete.go
+++ b/session_delete.go
@@ -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,27 +232,29 @@ func (session *Session) Delete(beans ...interface{}) (int64, error) {
 		return 0, err
 	}
 
-	// handle after delete processors
-	if session.isAutoCommit {
-		for _, closure := range session.afterClosures {
-			closure(bean)
-		}
-		if processor, ok := interface{}(bean).(AfterDeleteProcessor); ok {
-			processor.AfterDelete()
-		}
-	} else {
-		lenAfterClosures := len(session.afterClosures)
-		if lenAfterClosures > 0 && len(beans) > 0 {
-			if value, has := session.afterDeleteBeans[beans[0]]; has && value != nil {
-				*value = append(*value, session.afterClosures...)
-			} else {
-				afterClosures := make([]func(interface{}), lenAfterClosures)
-				copy(afterClosures, session.afterClosures)
-				session.afterDeleteBeans[bean] = &afterClosures
+	if bean != nil {
+		// handle after delete processors
+		if session.isAutoCommit {
+			for _, closure := range session.afterClosures {
+				closure(bean)
+			}
+			if processor, ok := interface{}(bean).(AfterDeleteProcessor); ok {
+				processor.AfterDelete()
 			}
 		} else {
-			if _, ok := interface{}(bean).(AfterDeleteProcessor); ok {
-				session.afterDeleteBeans[bean] = nil
+			lenAfterClosures := len(session.afterClosures)
+			if lenAfterClosures > 0 && len(beans) > 0 {
+				if value, has := session.afterDeleteBeans[beans[0]]; has && value != nil {
+					*value = append(*value, session.afterClosures...)
+				} else {
+					afterClosures := make([]func(interface{}), lenAfterClosures)
+					copy(afterClosures, session.afterClosures)
+					session.afterDeleteBeans[bean] = &afterClosures
+				}
+			} else {
+				if _, ok := interface{}(bean).(AfterDeleteProcessor); ok {
+					session.afterDeleteBeans[bean] = nil
+				}
 			}
 		}
 	}