Go: Implement PfMerge command

CHANGELOG.md

@@ -22,6 +22,7 @@
 * Go: Add `BITPOS` ([#3407](https://github.com/valkey-io/valkey-glide/pull/3407))
 * Go: Add `FLUSHALL` ([#3117](https://github.com/valkey-io/valkey-glide/pull/3117))
 * Go: Add `FLUSHDB` ([#3117](https://github.com/valkey-io/valkey-glide/pull/3117))
+* Go: Add `PFMERGE` ([#3082](https://github.com/valkey-io/valkey-glide/pull/3082))
 * Go: Add password update api ([#3346](https://github.com/valkey-io/valkey-glide/pull/3346))
 * Go: Add `BITOP` ([#3384](https://github.com/valkey-io/valkey-glide/pull/3384))
 * Go: Add `GeoHash` ([#3439](https://github.com/valkey-io/valkey-glide/pull/3439))

go/api/base_client.go

@@ -3228,6 +3228,33 @@ func (client *baseClient) PfCount(keys []string) (int64, error) {
 	return handleIntResponse(result)
 }
 
+// PfMerge merges multiple HyperLogLog values into a unique value.
+// If the destination variable exists, it is treated as one of the source HyperLogLog data sets,
+// otherwise a new HyperLogLog is created.
+//
+// Note:
+//
+//	When in cluster mode, `sourceKeys` and `destination` must map to the same hash slot.
+//
+// Parameters:
+//
+//	destination - The key of the destination HyperLogLog where the merged data sets will be stored.
+//	sourceKeys - An array of sourceKeys of the HyperLogLog structures to be merged.
+//
+// Return value:
+//
+//	If the HyperLogLog values is successfully merged  it returns "OK".
+//
+// [valkey.io]: https://valkey.io/commands/pfmerge/
+func (client *baseClient) PfMerge(destination string, sourceKeys []string) (string, error) {
+	result, err := client.executeCommand(C.PfMerge, append([]string{destination}, sourceKeys...))
+	if err != nil {
+		return DefaultStringResponse, err
+	}
+
+	return handleStringResponse(result)
+}
+
 // Unlink (delete) multiple keys from the database. A key is ignored if it does not exist.
 // This command, similar to Del However, this command does not block the server
 //

go/api/hyperloglog_commands.go

@@ -11,4 +11,6 @@ type HyperLogLogCommands interface {
 	PfAdd(key string, elements []string) (int64, error)
 
 	PfCount(keys []string) (int64, error)
+
+	PfMerge(destination string, sourceKeys []string) (string, error)
 }

go/api/hyperloglog_commands_test.go

@@ -61,3 +61,67 @@ func ExampleGlideClusterClient_PfCount() {
 	// 1
 	// 3
 }
+
+func ExampleGlideClient_PfMerge() {
+	var client *GlideClient = getExampleGlideClient() // example helper function
+
+	// Create source keys with some values
+	sourceKey1 := uuid.New().String() + "{group}"
+	sourceKey2 := uuid.New().String() + "{group}"
+
+	// Add values to source keys
+	_, err := client.PfAdd(sourceKey1, []string{"value1", "value2"})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	_, err = client.PfAdd(sourceKey2, []string{"value2", "value3", "value4"})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	// Merge the source keys into a destination key
+	destKey := uuid.New().String() + "{group}"
+	result, err := client.PfMerge(destKey, []string{sourceKey1, sourceKey2})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	fmt.Println(result)
+	// Output: OK
+}
+
+func ExampleGlideClusterClient_PfMerge() {
+	var client *GlideClusterClient = getExampleGlideClusterClient() // example helper function
+
+	// Create source keys with some values
+	sourceKey1 := uuid.New().String() + "{group}"
+	sourceKey2 := uuid.New().String() + "{group}"
+
+	// Add values to source keys
+	_, err := client.PfAdd(sourceKey1, []string{"value1", "value2"})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	_, err = client.PfAdd(sourceKey2, []string{"value2", "value3", "value4"})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	// Merge the source keys into a destination key
+	destKey := uuid.New().String() + "{group}"
+	result, err := client.PfMerge(destKey, []string{sourceKey1, sourceKey2})
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+		return
+	}
+
+	fmt.Println(result)
+	// Output: OK
+}

go/integTest/shared_commands_test.go

@@ -3819,6 +3819,64 @@ func (suite *GlideTestSuite) TestPfCount_NoExistingKeys() {
 	})
 }
 
+func (suite *GlideTestSuite) TestPfMerge() {
+	suite.runWithDefaultClients(func(client api.BaseClient) {
+		source1 := uuid.New().String() + "{group}"
+		source2 := uuid.New().String() + "{group}"
+		destination := uuid.New().String() + "{group}"
+
+		res, err := client.PfAdd(source1, []string{"a", "b", "c"})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(1), res)
+
+		res, err = client.PfAdd(source2, []string{"c", "d", "e"})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(1), res)
+
+		result, err := client.PfMerge(destination, []string{source1, source2})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), "OK", result)
+
+		count, err := client.PfCount([]string{destination})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(5), count)
+	})
+}
+
+func (suite *GlideTestSuite) TestPfMerge_SingleSource() {
+	suite.runWithDefaultClients(func(client api.BaseClient) {
+		source := uuid.New().String() + "{group}"
+		destination := uuid.New().String() + "{group}"
+
+		res, err := client.PfAdd(source, []string{"a", "b", "c"})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(1), res)
+
+		result, err := client.PfMerge(destination, []string{source})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), "OK", result)
+
+		count, err := client.PfCount([]string{destination})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(3), count)
+	})
+}
+
+func (suite *GlideTestSuite) TestPfMerge_NonExistentSource() {
+	suite.runWithDefaultClients(func(client api.BaseClient) {
+		nonExistentKey := uuid.New().String() + "{group}"
+		destination := uuid.New().String() + "{group}"
+
+		result, err := client.PfMerge(destination, []string{nonExistentKey})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), "OK", result)
+
+		count, err := client.PfCount([]string{destination})
+		assert.Nil(suite.T(), err)
+		assert.Equal(suite.T(), int64(0), count)
+	})
+}
+
 func (suite *GlideTestSuite) TestSortWithOptions_AscendingOrder() {
 	suite.runWithDefaultClients(func(client api.BaseClient) {
 		key := uuid.New().String()