Go: Implement Config Rewrite

CHANGELOG.md

@@ -23,6 +23,7 @@
 * Go: Add password update api ([#3346](https://github.com/valkey-io/valkey-glide/pull/3346))
 * Go: Add `GeoHash` ([#3439](https://github.com/valkey-io/valkey-glide/pull/3439))
 * Go/Core: Move FFI to a dedicated folder for reusability ([#3372](https://github.com/valkey-io/valkey-glide/pull/3372))
+* Go: Implement Rewrite ([#3156](https://github.com/valkey-io/valkey-glide/pull/3156))
 
 #### Breaking Changes
 

go/api/glide_client.go

@@ -314,3 +314,18 @@ func (client *GlideClient) FlushDBWithOptions(mode options.FlushMode) (string, e
 	}
 	return handleStringResponse(result)
 }
+
+// Rewrites the configuration file with the current configuration.
+//
+// Return value:
+//
+//	"OK" when the configuration was rewritten properly, otherwise an error is thrown.
+//
+// [valkey.io]: https://valkey.io/commands/config-rewrite/
+func (client *GlideClient) ConfigRewrite() (string, error) {
+	response, err := client.executeCommand(C.ConfigRewrite, []string{})
+	if err != nil {
+		return DefaultStringResponse, err
+	}
+	return handleStringResponse(response)
+}

go/api/glide_cluster_client.go

@@ -544,3 +544,39 @@ func (client *GlideClusterClient) ScanWithOptions(
 	nextCursor, keys, err := handleScanResponse(response)
 	return *options.NewClusterScanCursorWithId(nextCursor), keys, err
 }
+
+// Rewrites the configuration file with the current configuration.
+// The command will be routed a random node.
+//
+// Return value:
+//
+//	"OK" when the configuration was rewritten properly, otherwise an error is thrown.
+//
+// [valkey.io]: https://valkey.io/commands/config-rewrite/
+func (client *GlideClusterClient) ConfigRewrite() (string, error) {
+	response, err := client.executeCommand(C.ConfigRewrite, []string{})
+	if err != nil {
+		return DefaultStringResponse, err
+	}
+	return handleStringResponse(response)
+}
+
+// Rewrites the configuration file with the current configuration.
+//
+// Parameters:
+//
+//	opts - Specifies the routing configuration for the command. The client will route the
+//	        command to the nodes defined by route.
+//
+// Return value:
+//
+//	"OK" when the configuration was rewritten properly, otherwise an error is thrown.
+//
+// [valkey.io]: https://valkey.io/commands/config-rewrite/
+func (client *GlideClusterClient) ConfigRewriteWithOptions(opts options.RouteOption) (string, error) {
+	response, err := client.executeCommandWithRoute(C.ConfigRewrite, []string{}, opts.Route)
+	if err != nil {
+		return DefaultStringResponse, err
+	}
+	return handleStringResponse(response)
+}

go/api/server_management_cluster_commands.go

@@ -25,4 +25,8 @@ type ServerManagementClusterCommands interface {
 	FlushDB() (string, error)
 
 	FlushDBWithOptions(options options.FlushClusterOptions) (string, error)
+
+	ConfigRewrite() (string, error)
+
+	ConfigRewriteWithOptions(routeOption options.RouteOption) (string, error)
 }

go/api/server_management_cluster_commands_test.go

@@ -77,10 +77,10 @@ func ExampleGlideClusterClient_DBSizeWithOptions() {
 	if err != nil {
 		fmt.Println("Glide example failed with an error: ", err)
 	}
-
 	fmt.Println(result)
 
-	// Output: 0
+	// Output:
+	// 0
 }
 
 func ExampleGlideClusterClient_FlushAll() {
@@ -152,3 +152,141 @@ func ExampleGlideClusterClient_FlushDBWithOptions() {
 
 	// Output: OK
 }
+
+func ExampleGlideClusterClient_ConfigRewrite() {
+	var client *GlideClusterClient = getExampleGlideClusterClient() // example helper function
+	var resultRewrite string
+	opts := options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: []options.Section{options.Server}},
+	}
+	res, err := client.InfoWithOptions(opts)
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+	}
+	for _, data := range res.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			if err != nil {
+				fmt.Println("Glide example failed with an error: ", err)
+			}
+			resultRewrite = responseRewrite
+			break
+		} else {
+			resultRewrite = "OK"
+		}
+
+	}
+	fmt.Println(resultRewrite)
+
+	// Output:
+	// OK
+}
+
+func ExampleGlideClusterClient_ConfigRewriteWithOptions() {
+	var client *GlideClusterClient = getExampleGlideClusterClient() // example helper function
+	sections := []options.Section{options.Server}
+
+	// info with option or with multiple options without route
+	var runResultNilRoute string
+	opts := options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: sections},
+		RouteOption: nil,
+	}
+	response, err := client.InfoWithOptions(opts)
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+	}
+
+	for _, data := range response.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			if err != nil {
+				fmt.Println("Glide example failed with an error: ", err)
+			}
+			runResultNilRoute = responseRewrite
+			break
+		}
+		runResultNilRoute = "OK"
+	}
+
+	// same sections with random route
+	var runResultRandomRoute string
+	opts = options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: sections},
+		RouteOption: &options.RouteOption{Route: config.RandomRoute},
+	}
+	response, err = client.InfoWithOptions(opts)
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+	}
+	lines := strings.Split(response.SingleValue(), "\n")
+	var configFile string
+	for _, line := range lines {
+		if strings.HasPrefix(line, "config_file:") {
+			configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+			break
+		}
+	}
+	if len(configFile) > 0 {
+		responseRewrite, err := client.ConfigRewrite()
+		if err != nil {
+			fmt.Println("Glide example failed with an error: ", err)
+		}
+		runResultRandomRoute = responseRewrite
+	}
+	runResultRandomRoute = "OK"
+
+	// default sections, multi node route
+	var runResultMultiNodeRoute string
+	opts = options.ClusterInfoOptions{
+		InfoOptions: nil,
+		RouteOption: &options.RouteOption{Route: config.AllPrimaries},
+	}
+	response, err = client.InfoWithOptions(opts)
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+	}
+	for _, data := range response.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			if err != nil {
+				fmt.Println("Glide example failed with an error: ", err)
+			}
+			runResultMultiNodeRoute = responseRewrite
+			break
+		}
+		runResultMultiNodeRoute = "OK"
+	}
+	fmt.Println("Multiple options without route result:", runResultNilRoute)
+	fmt.Println("Random route result:", runResultRandomRoute)
+	fmt.Println("Multi node route result:", runResultMultiNodeRoute)
+
+	// Output:
+	// Multiple options without route result: OK
+	// Random route result: OK
+	// Multi node route result: OK
+}

go/api/server_management_commands.go

@@ -33,4 +33,6 @@ type ServerManagementCommands interface {
 	FlushDB() (string, error)
 
 	FlushDBWithOptions(mode options.FlushMode) (string, error)
+
+	ConfigRewrite() (string, error)
 }

go/api/server_management_commands_test.go

@@ -5,6 +5,7 @@ package api
 import (
 	"fmt"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/valkey-io/valkey-glide/go/api/options"
@@ -161,3 +162,34 @@ func ExampleGlideClient_FlushDBWithOptions() {
 
 	// Output: OK
 }
+
+func ExampleGlideClient_ConfigRewrite() {
+	var client *GlideClient = getExampleGlideClient() // example helper function
+	opts := options.InfoOptions{Sections: []options.Section{options.Server}}
+	var resultRewrite string
+	response, err := client.InfoWithOptions(opts)
+	if err != nil {
+		fmt.Println("Glide example failed with an error: ", err)
+	}
+	lines := strings.Split(response, "\n")
+	var configFile string
+	for _, line := range lines {
+		if strings.HasPrefix(line, "config_file:") {
+			configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+			break
+		}
+	}
+	if len(configFile) > 0 {
+		response, err = client.ConfigRewrite()
+		if err != nil {
+			fmt.Println("Glide example failed with an error: ", err)
+		}
+		resultRewrite = response
+	} else {
+		resultRewrite = "OK"
+	}
+	fmt.Println(resultRewrite)
+
+	// Output:
+	// OK
+}

go/integTest/cluster_commands_test.go

@@ -1099,3 +1099,103 @@ func (suite *GlideTestSuite) TestUpdateConnectionPasswordCluster_ImmediateAuthWr
 	_, err = adminClient.CustomCommand([]string{"CONFIG", "SET", "requirepass", ""})
 	assert.NoError(suite.T(), err)
 }
+
+func (suite *GlideTestSuite) TestConfigRewriteCluster() {
+	client := suite.defaultClusterClient()
+	t := suite.T()
+	opts := options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: []options.Section{options.Server}},
+	}
+	res, err := client.InfoWithOptions(opts)
+	assert.NoError(t, err)
+	for _, data := range res.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			assert.NoError(t, err)
+			assert.Equal(t, "OK", responseRewrite)
+		}
+	}
+}
+
+func (suite *GlideTestSuite) TestConfigRewriteWithOptions() {
+	client := suite.defaultClusterClient()
+	t := suite.T()
+	sections := []options.Section{options.Server}
+
+	// info with option or with multiple options without route
+	opts := options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: sections},
+		RouteOption: nil,
+	}
+	response, err := client.InfoWithOptions(opts)
+	assert.NoError(t, err)
+	for _, data := range response.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			assert.NoError(t, err)
+			assert.Equal(t, "OK", responseRewrite)
+			break
+		}
+	}
+
+	// same sections with random route
+	opts = options.ClusterInfoOptions{
+		InfoOptions: &options.InfoOptions{Sections: sections},
+		RouteOption: &options.RouteOption{Route: config.RandomRoute},
+	}
+	response, err = client.InfoWithOptions(opts)
+	assert.NoError(t, err)
+	lines := strings.Split(response.SingleValue(), "\n")
+	var configFile string
+	for _, line := range lines {
+		if strings.HasPrefix(line, "config_file:") {
+			configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+			break
+		}
+	}
+	if len(configFile) > 0 {
+		responseRewrite, err := client.ConfigRewrite()
+		assert.NoError(t, err)
+		assert.Equal(t, "OK", responseRewrite)
+	}
+
+	// default sections, multi node route
+	opts = options.ClusterInfoOptions{
+		InfoOptions: nil,
+		RouteOption: &options.RouteOption{Route: config.AllPrimaries},
+	}
+	response, err = client.InfoWithOptions(opts)
+	assert.NoError(t, err)
+	for _, data := range response.MultiValue() {
+		lines := strings.Split(data, "\n")
+		var configFile string
+		for _, line := range lines {
+			if strings.HasPrefix(line, "config_file:") {
+				configFile = strings.TrimSpace(strings.TrimPrefix(line, "config_file:"))
+				break
+			}
+		}
+		if len(configFile) > 0 {
+			responseRewrite, err := client.ConfigRewrite()
+			assert.NoError(t, err)
+			assert.Equal(t, "OK", responseRewrite)
+			break
+		}
+	}
+}