feat: ajust if logic

go-gorm · Oct 9, 2021 · 6001a47 · 6001a47
1 parent 8f8809f
commit 6001a47
Showing 1 changed file with 27 additions and 25 deletions.
diff --git a/callbacks/update.go b/callbacks/update.go
@@ -51,39 +51,41 @@ func BeforeUpdate(db *gorm.DB) {
 }
 
 func Update(db *gorm.DB) {
-	if db.Error == nil {
-		if db.Statement.Schema != nil && !db.Statement.Unscoped {
-			for _, c := range db.Statement.Schema.UpdateClauses {
-				db.Statement.AddClause(c)
-			}
-		}
-
-		if db.Statement.SQL.String() == "" {
-			db.Statement.SQL.Grow(180)
-			db.Statement.AddClauseIfNotExists(clause.Update{})
-			set := ConvertToAssignments(db.Statement)
-			if len(set) == 0 {
-				return
-			}
+	if db.Error != nil {
+		return
+	}
 
-			db.Statement.AddClause(set)
-			db.Statement.Build(db.Statement.BuildClauses...)
+	if db.Statement.Schema != nil && !db.Statement.Unscoped {
+		for _, c := range db.Statement.Schema.UpdateClauses {
+			db.Statement.AddClause(c)
 		}
+	}
 
-		if _, ok := db.Statement.Clauses["WHERE"]; !db.AllowGlobalUpdate && !ok {
-			db.AddError(gorm.ErrMissingWhereClause)
+	if db.Statement.SQL.String() == "" {
+		db.Statement.SQL.Grow(180)
+		db.Statement.AddClauseIfNotExists(clause.Update{})
+		set := ConvertToAssignments(db.Statement)
+		if len(set) == 0 {
 			return
 		}
 
-		if !db.DryRun && db.Error == nil {
-			result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
-			if err != nil {
-				db.AddError(err)
-				return
-			}
+		db.Statement.AddClause(set)
+		db.Statement.Build(db.Statement.BuildClauses...)
+	}
+
+	if _, ok := db.Statement.Clauses["WHERE"]; !db.AllowGlobalUpdate && !ok {
+		db.AddError(gorm.ErrMissingWhereClause)
+		return
+	}
 
-			db.RowsAffected, _ = result.RowsAffected()
+	if !db.DryRun && db.Error == nil {
+		result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
+		if err != nil {
+			db.AddError(err)
+			return
 		}
+
+		db.RowsAffected, _ = result.RowsAffected()
 	}
 }