Use cloned request in heal registry (#1287)

Signed-off-by: Artem Glazychev <artem.glazychev@xored.com>
networkservicemesh · May 13, 2022 · 53df7f1 · 53df7f1
1 parent 4d9bebd
commit 53df7f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/registry/common/heal/ns_client.go b/pkg/registry/common/heal/ns_client.go
@@ -41,7 +41,7 @@ func NewNetworkServiceRegistryClient(ctx context.Context) registry.NetworkServic
 }
 
 func (c *healNSClient) Register(ctx context.Context, ns *registry.NetworkService, opts ...grpc.CallOption) (*registry.NetworkService, error) {
-	resp, err := next.NetworkServiceRegistryClient(ctx).Register(ctx, ns, opts...)
+	resp, err := next.NetworkServiceRegistryClient(ctx).Register(ctx, ns.Clone(), opts...)
 
 	if err != nil {
 		return nil, err

diff --git a/pkg/registry/common/heal/nse_client.go b/pkg/registry/common/heal/nse_client.go
@@ -41,7 +41,7 @@ func NewNetworkServiceEndpointRegistryClient(ctx context.Context) registry.Netwo
 }
 
 func (c *healNSEClient) Register(ctx context.Context, nse *registry.NetworkServiceEndpoint, opts ...grpc.CallOption) (*registry.NetworkServiceEndpoint, error) {
-	resp, err := next.NetworkServiceEndpointRegistryClient(ctx).Register(ctx, nse, opts...)
+	resp, err := next.NetworkServiceEndpointRegistryClient(ctx).Register(ctx, nse.Clone(), opts...)
 
 	if err != nil {
 		return nil, err