From 9c5d719f388edf949fff1bfcf5631aef89a8d810 Mon Sep 17 00:00:00 2001 From: Ori Newman Date: Fri, 9 Sep 2022 15:34:36 +0300 Subject: [PATCH] Increase grpc server send message size --- cmd/kaspawallet/daemon/client/client.go | 3 ++- cmd/kaspawallet/daemon/server/server.go | 6 +++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/cmd/kaspawallet/daemon/client/client.go b/cmd/kaspawallet/daemon/client/client.go index 6cfd38a32..bc25753ed 100644 --- a/cmd/kaspawallet/daemon/client/client.go +++ b/cmd/kaspawallet/daemon/client/client.go @@ -2,6 +2,7 @@ package client import ( "context" + "github.com/kaspanet/kaspad/cmd/kaspawallet/daemon/server" "time" "github.com/pkg/errors" @@ -16,7 +17,7 @@ func Connect(address string) (pb.KaspawalletdClient, func(), error) { ctx, cancel := context.WithTimeout(context.Background(), time.Second) defer cancel() - conn, err := grpc.DialContext(ctx, address, grpc.WithInsecure(), grpc.WithBlock()) + conn, err := grpc.DialContext(ctx, address, grpc.WithInsecure(), grpc.WithBlock(), grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(server.MaxDaemonSendMsgSize))) if err != nil { if errors.Is(err, context.DeadlineExceeded) { return nil, nil, errors.New("kaspawallet daemon is not running, start it with `kaspawallet start-daemon`") diff --git a/cmd/kaspawallet/daemon/server/server.go b/cmd/kaspawallet/daemon/server/server.go index afa140319..0640cff4e 100644 --- a/cmd/kaspawallet/daemon/server/server.go +++ b/cmd/kaspawallet/daemon/server/server.go @@ -44,6 +44,10 @@ type server struct { maxProcessedAddressesForLog uint32 } +// MaxDaemonSendMsgSize is the max send message size used for the daemon server. +// Currently, set to 100MB +const MaxDaemonSendMsgSize = 100_000_000 + // Start starts the kaspawalletd server func Start(params *dagconfig.Params, listen, rpcServer string, keysFilePath string, profile string, timeout uint32) error { initLog(defaultLogFile, defaultErrLogFile) @@ -96,7 +100,7 @@ func Start(params *dagconfig.Params, listen, rpcServer string, keysFilePath stri } }) - grpcServer := grpc.NewServer() + grpcServer := grpc.NewServer(grpc.MaxSendMsgSize(MaxDaemonSendMsgSize)) pb.RegisterKaspawalletdServer(grpcServer, serverInstance) spawn("grpcServer.Serve", func() {