make grpc client's dial-timeout configurable

Signed-off-by: Jun Kimura <[email protected]>

proto/relayer/provers/lcp/config/config.proto

@@ -11,13 +11,15 @@ message ProverConfig {
     google.protobuf.Any origin_prover = 1;
     // hex string
     string lcp_service_address = 2;
+    // unit: seconds
+    uint64 lcp_service_dial_timeout = 3;
     // hex string
-    string mrenclave = 3;
-    repeated string allowed_quote_statuses = 4;
-    repeated string allowed_advisory_ids = 5;
+    string mrenclave = 4;
+    repeated string allowed_quote_statuses = 5;
+    repeated string allowed_advisory_ids = 6;
     // unit: seconds
-    uint64 key_expiration = 6;
-    string elc_client_id = 7;
-    bool message_aggregation = 8;
-    uint64 message_aggregation_batch_size = 9;
+    uint64 key_expiration = 7;
+    string elc_client_id = 8;
+    bool message_aggregation = 9;
+    uint64 message_aggregation_batch_size = 10;
 }

relay/config.go

@@ -4,13 +4,17 @@ import (
 	"encoding/hex"
 	"fmt"
 	"strings"
+	"time"
 
 	codectypes "github.com/cosmos/cosmos-sdk/codec/types"
 	lcptypes "github.com/datachainlab/lcp-go/light-clients/lcp/types"
 	"github.com/hyperledger-labs/yui-relayer/core"
 )
 
-const DefaultMessageAggregationBatchSize = 8
+const (
+	DefaultDialTimeout                 = 20 // seconds
+	DefaultMessageAggregationBatchSize = 8
+)
 
 var _ core.ProverConfig = (*ProverConfig)(nil)
 
@@ -37,6 +41,14 @@ func (pc ProverConfig) Build(chain core.Chain) (core.Prover, error) {
 	return NewProver(pc, chain, prover)
 }
 
+func (pc ProverConfig) GetDialTimeout() time.Duration {
+	if pc.LcpServiceDialTimeout == 0 {
+		return DefaultDialTimeout * time.Second
+	} else {
+		return time.Duration(pc.LcpServiceDialTimeout) * time.Second
+	}
+}
+
 func (pc ProverConfig) GetMrenclave() []byte {
 	mrenclave, err := decodeMrenclaveHex(pc.Mrenclave)
 	if err != nil {

relay/prover.go

@@ -48,9 +48,10 @@ func NewProver(config ProverConfig, originChain core.Chain, originProver core.Pr
 		config.LcpServiceAddress,
 		grpc.WithTransportCredentials(insecure.NewCredentials()),
 		grpc.WithBlock(),
+		grpc.WithTimeout(config.GetDialTimeout()),
 	)
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to connect to LCP service: %w", err)
 	}
 	return &Prover{config: config, originChain: originChain, originProver: originProver, lcpServiceClient: NewLCPServiceClient(conn)}, nil
 }