perform gorm upserts inside transactions

apigee · Sep 29, 2020 · 8acdf98 · 8acdf98
1 parent 5ef75fd
commit 8acdf98
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/server/gorm/client.go b/server/gorm/client.go
@@ -211,13 +211,17 @@ func (c *Client) Put(ctx context.Context, k storage.Key, v interface{}) (storage
 	case *models.Label:
 		r.Key = k.(*Key).Name
 	}
-	rowsAffected := c.db.Model(v).Where("key = ?", k.(*Key).Name).Updates(v).RowsAffected
-	if rowsAffected == 0 {
-		err := c.db.Create(v).Error
-		if err != nil {
-			log.Printf("CREATE ERROR %s", err.Error())
-		}
-	}
+	c.db.Transaction(
+		func(tx *gorm.DB) error {
+			rowsAffected := tx.Model(v).Where("key = ?", k.(*Key).Name).Updates(v).RowsAffected
+			if rowsAffected == 0 {
+				err := tx.Create(v).Error
+				if err != nil {
+					log.Printf("CREATE ERROR %s", err.Error())
+				}
+			}
+			return nil
+		})
 	return k, nil
 }