Add support for replace operation to the agency

Author: informalict

agency/operation.go

@@ -74,6 +74,12 @@ type keyArrayErase struct {
 	value interface{}
 }
 
+type keyArrayReplace struct {
+	keyCommon
+	newValue interface{}
+	oldValue interface{}
+}
+
 // NewKeyDelete returns a new key operation which must be removed from the agency
 func NewKeyDelete(key []string) KeyChanger {
 	return &keyDelete{
@@ -127,6 +133,17 @@ func NewKeyArrayErase(key []string, value interface{}) KeyChanger {
 	}
 }
 
+// NewKeyArrayReplace returns a new key operation for replacing element in the array.
+func NewKeyArrayReplace(key []string, oldValue, newValue interface{}) KeyChanger {
+	return &keyArrayReplace{
+		keyCommon: keyCommon{
+			key: key,
+		},
+		newValue: newValue,
+		oldValue: oldValue,
+	}
+}
+
 func (k *keyDelete) GetOperation() string {
 	return "delete"
 }
@@ -229,3 +246,23 @@ func (k *keyArrayErase) GetURL() string {
 func (k *keyArrayErase) GetVal() interface{} {
 	return k.value
 }
+
+func (k *keyArrayReplace) GetOperation() string {
+	return "replace"
+}
+
+func (k *keyArrayReplace) GetTTL() time.Duration {
+	return 0
+}
+
+func (k *keyArrayReplace) GetNew() interface{} {
+	return k.newValue
+}
+
+func (k *keyArrayReplace) GetURL() string {
+	return ""
+}
+
+func (k *keyArrayReplace) GetVal() interface{} {
+	return k.oldValue
+}

test/agency_test.go

@@ -815,6 +815,67 @@ func writeTransaction(t *testing.T, transient bool) {
 				},
 			},
 		},
+		{
+			name: "Replace element in array",
+			requests: []Request{
+				{
+					transaction: TransactionTest{
+						keys: []agency.KeyChanger{
+							agency.NewKeyArrayPush([]string{rootKeyAgency, "test", "array"}, map[string]interface{}{
+								"database":   "db",
+								"collection": "col",
+								"shard":      0,
+							}),
+						},
+					},
+				},
+				{
+					transaction: TransactionTest{
+						keys: []agency.KeyChanger{
+							agency.NewKeyArrayPush([]string{rootKeyAgency, "test", "array"}, map[string]interface{}{
+								"database":   "db",
+								"collection": "col",
+								"shard":      1,
+							}),
+						},
+					},
+				},
+				{
+					transaction: TransactionTest{
+						keys: []agency.KeyChanger{
+							agency.NewKeyArrayReplace([]string{rootKeyAgency, "test", "array"},
+								map[string]interface{}{
+									"database":   "db",
+									"collection": "col",
+									"shard":      0,
+								}, map[string]interface{}{
+									"database":   "db",
+									"collection": "col",
+									"shard":      2,
+								}),
+						},
+					},
+				},
+			},
+			expectedResult: map[string]interface{}{
+				rootKeyAgency: map[string]interface{}{
+					"test": map[string]interface{}{
+						"array": []interface{}{
+							map[string]interface{}{
+								"database":   "db",
+								"collection": "col",
+								"shard":      float64(2),
+							},
+							map[string]interface{}{
+								"database":   "db",
+								"collection": "col",
+								"shard":      float64(1),
+							},
+						},
+					},
+				},
+			},
+		},
 	}
 
 	for _, testCase := range testCases {