fix: customize plugin can not update old data (#7389) (#7392)

* fix: customize plugin can not update old data * fix: recover pipeline_plan * fix: e2e test Co-authored-by: abeizn <[email protected]>
apache · Apr 26, 2024 · 4a0a0d1 · 4a0a0d1
1 parent f74c0e6
commit 4a0a0d1
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/backend/impls/dalgorm/dalgorm.go b/backend/impls/dalgorm/dalgorm.go
@@ -236,6 +236,20 @@ func (d *Dalgorm) Create(entity interface{}, clauses ...dal.Clause) errors.Error
 // CreateWithMap insert record to database
 func (d *Dalgorm) CreateWithMap(entity interface{}, record map[string]interface{}) errors.Error {
 	d.unwrapDynamic(&entity, nil)
+	if record != nil {
+		if id, ok := record["id"]; ok && id != nil {
+			var columns []string
+			for column := range record {
+				columns = append(columns, column)
+			}
+			return d.convertGormError(buildTx(d.db, nil).Model(entity).Clauses(clause.OnConflict{
+				Columns:   []clause.Column{{Name: "id"}},
+				DoUpdates: clause.AssignmentColumns(columns),
+			}).Create(record).Error)
+		} else {
+			return d.convertGormError(buildTx(d.db, nil).Model(entity).Clauses(clause.OnConflict{UpdateAll: true}).Create(record).Error)
+		}
+	}
 	return d.convertGormError(buildTx(d.db, nil).Model(entity).Clauses(clause.OnConflict{UpdateAll: true}).Create(record).Error)
 }