Fix SetRequestCallbacks/SetResponseCallbacks

client_test.go

@@ -5,6 +5,7 @@ package vault
 
 import (
 	"fmt"
+	"net/http"
 	"testing"
 	"time"
 
@@ -27,6 +28,12 @@ func Test_Client_Clone(t *testing.T) {
 
 	require.NoError(t, client.SetToken("test-token"))
 	require.NoError(t, client.SetNamespace("test-namespace"))
+	require.NoError(t, client.SetRequestCallbacks(func(req *http.Request) {
+		t.Log(req)
+	}))
+	require.NoError(t, client.SetResponseCallbacks(func(req *http.Request, resp *http.Response) {
+		t.Log(req, resp)
+	}))
 
 	clone := client.Clone()
 

generate/templates/client.handlebars

@@ -180,8 +180,8 @@ func (c *Client) cloneClientRequestModifiers() requestModifiers {
 
 	var clone requestModifiers
 
-	copy(clone.requestCallbacks, c.clientRequestModifiers.requestCallbacks)
-	copy(clone.responseCallbacks, c.clientRequestModifiers.responseCallbacks)
+	clone.requestCallbacks = append(clone.requestCallbacks, c.clientRequestModifiers.requestCallbacks...)
+	clone.responseCallbacks = append(clone.responseCallbacks, c.clientRequestModifiers.responseCallbacks...)
 
 	clone.headers = requestHeaders{
 		userAgent:                 c.clientRequestModifiers.headers.userAgent,
@@ -192,7 +192,7 @@ func (c *Client) cloneClientRequestModifiers() requestModifiers {
 		customHeaders:             c.clientRequestModifiers.headers.customHeaders.Clone(),
 	}
 
-	copy(clone.headers.mfaCredentials, c.clientRequestModifiers.headers.mfaCredentials)
+    clone.headers.mfaCredentials = append(clone.headers.mfaCredentials, c.clientRequestModifiers.headers.mfaCredentials...)
 
 	return clone
 }

replication_consistency.go

@@ -228,7 +228,7 @@ func (c *replicationStateCache) clone() replicationStateCache {
 	/* */ c.statesLock.RLock()
 	defer c.statesLock.RUnlock()
 
-	var cloned []string
+	cloned := make([]string, len(c.states))
 	copy(cloned, c.states)
 
 	return replicationStateCache{

request_modifiers.go

@@ -172,6 +172,7 @@ func (c *Client) ClearCustomHeaders() {
 // SetRequestCallbacks sets callbacks which will be invoked before each request.
 func (c *Client) SetRequestCallbacks(callbacks ...RequestCallback) error {
 	c.clientRequestModifiersLock.Lock()
+	c.clientRequestModifiers.requestCallbacks = make([]RequestCallback, len(callbacks))
 	copy(c.clientRequestModifiers.requestCallbacks, callbacks)
 	c.clientRequestModifiersLock.Unlock()
 
@@ -189,6 +190,7 @@ func (c *Client) ClearRequestCallbacks() {
 // successful response.
 func (c *Client) SetResponseCallbacks(callbacks ...ResponseCallback) error {
 	c.clientRequestModifiersLock.Lock()
+	c.clientRequestModifiers.responseCallbacks = make([]ResponseCallback, len(callbacks))
 	copy(c.clientRequestModifiers.responseCallbacks, callbacks)
 	c.clientRequestModifiersLock.Unlock()