[PRIV-158] Add RemoveHandler (#1514)

Author: cedric-cordenier

pkg/loop/internal/core/services/gateway/gateway_connector.go

@@ -47,6 +47,13 @@ func (c GatewayConnectorClient) AddHandler(ctx context.Context, methods []string
 	return nil
 }
 
+func (c GatewayConnectorClient) RemoveHandler(ctx context.Context, methods []string) error {
+	_, err := c.grpc.RemoveHandler(ctx, &pb.RemoveHandlerRequest{
+		Methods: methods,
+	})
+	return err
+}
+
 func (c GatewayConnectorClient) GatewayIDs(ctx context.Context) ([]string, error) {
 	resp, err := c.grpc.GatewayIDs(ctx, &emptypb.Empty{})
 	if err != nil {

pkg/loop/internal/pb/gatewayconnector/gateway_connector.pb.go

@@ -32,11 +32,16 @@ message AddHandlerRequest {
   uint32 handler_id = 2;
 }
 
+message RemoveHandlerRequest {
+  repeated string methods = 1;
+}
+
 service GatewayConnector {
   rpc AddHandler(AddHandlerRequest) returns (google.protobuf.Empty);
+  rpc RemoveHandler(RemoveHandlerRequest) returns (google.protobuf.Empty);
   rpc SendToGateway(SendMessageRequest) returns (google.protobuf.Empty);
   rpc SignMessage(SignMessageRequest) returns (SignMessageReply);
   rpc GatewayIDs(google.protobuf.Empty) returns (GatewayIDsReply);
   rpc DonID(google.protobuf.Empty) returns (DonIDReply);
   rpc AwaitConnection(GatewayIDRequest) returns (google.protobuf.Empty);
-}
+}

pkg/loop/internal/pb/gatewayconnector/gateway_connector.proto

@@ -3,7 +3,7 @@ package core
 import (
 	"context"
 	"encoding/json"
-	"fmt"
+	"errors"
 
 	jsonrpc "github.com/smartcontractkit/chainlink-common/pkg/jsonrpc2"
 )
@@ -12,6 +12,7 @@ import (
 type GatewayConnector interface {
 	// AddHandler adds a handler to the GatewayConnector
 	AddHandler(ctx context.Context, methods []string, handler GatewayConnectorHandler) error
+	RemoveHandler(ctx context.Context, methods []string) error
 	// SendToGateway takes a signed message as argument and sends it to the specified gateway
 	SendToGateway(ctx context.Context, gatewayID string, resp *jsonrpc.Response[json.RawMessage]) error
 	// Sign the given message and return signature
@@ -37,25 +38,29 @@ var _ GatewayConnector = (*UnimplementedGatewayConnector)(nil)
 type UnimplementedGatewayConnector struct{}
 
 func (u *UnimplementedGatewayConnector) AddHandler(ctx context.Context, methods []string, handler GatewayConnectorHandler) error {
-	return fmt.Errorf("not implemented")
+	return errors.New("not implemented")
+}
+
+func (u *UnimplementedGatewayConnector) RemoveHandler(ctx context.Context, methods []string) error {
+	return errors.New("not implemented")
 }
 
 func (u *UnimplementedGatewayConnector) SendToGateway(ctx context.Context, gatewayID string, resp *jsonrpc.Response[json.RawMessage]) error {
-	return fmt.Errorf("not implemented")
+	return errors.New("not implemented")
 }
 
 func (u *UnimplementedGatewayConnector) SignMessage(ctx context.Context, msg []byte) ([]byte, error) {
-	return nil, fmt.Errorf("not implemented")
+	return nil, errors.New("not implemented")
 }
 
 func (u *UnimplementedGatewayConnector) GatewayIDs(ctx context.Context) ([]string, error) {
-	return nil, fmt.Errorf("not implemented")
+	return nil, errors.New("not implemented")
 }
 
 func (u *UnimplementedGatewayConnector) DonID(ctx context.Context) (string, error) {
-	return "", fmt.Errorf("not implemented")
+	return "", errors.New("not implemented")
 }
 
 func (u *UnimplementedGatewayConnector) AwaitConnection(ctx context.Context, gatewayID string) error {
-	return fmt.Errorf("not implemented")
+	return errors.New("not implemented")
 }