message size

Signed-off-by: Yashash H L <yashash_hl@intuit.com>

pkg/function/client/client.go

@@ -21,7 +21,8 @@ type client struct {
 func New(inputOptions ...Option) (*client, error) {
 
 	var opts = &options{
-		sockAddr: function.Addr,
+		sockAddr:       function.Addr,
+		maxMessageSize: function.DefaultMaxMessageSize,
 	}
 
 	for _, inputOption := range inputOptions {
@@ -30,7 +31,10 @@ func New(inputOptions ...Option) (*client, error) {
 
 	c := new(client)
 	sockAddr := fmt.Sprintf("%s:%s", function.Protocol, opts.sockAddr)
-	conn, err := grpc.Dial(sockAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+
+	conn, err := grpc.Dial(sockAddr, grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(opts.maxMessageSize), grpc.MaxCallSendMsgSize(opts.maxMessageSize)))
+
 	if err != nil {
 		return nil, fmt.Errorf("failed to execute grpc.Dial(%q): %w", sockAddr, err)
 	}

pkg/function/client/options.go

@@ -1,7 +1,8 @@
 package client
 
 type options struct {
-	sockAddr string
+	sockAddr       string
+	maxMessageSize int
 }
 
 // Option is the interface to apply options.
@@ -13,3 +14,10 @@ func WithSockAddr(addr string) Option {
 		opts.sockAddr = addr
 	}
 }
+
+// WithMaxMessageSize sets the server max receive message size and the server max send message size to the given size.
+func WithMaxMessageSize(size int) Option {
+	return func(opts *options) {
+		opts.maxMessageSize = size
+	}
+}