Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-03-30 15:08:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NOD-1395] Increase the grpcclient's max message size to match the grpcserver's. (#932)

Browse Source
This commit is contained in:
stasatdaglabs 2020-09-17 16:53:33 +03:00 committed by GitHub
parent 75e539f4d2
commit 798abf2103
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
infrastructure/network/netadapter/server/grpcserver/grpc_server.go
View File

@ -17,12 +17,13 @@ type gRPCServer struct {
server *grpc.Server
}
const maxMessageSize = 1024 * 1024 * 10 // 10MB
// MaxMessageSize is the max size allowed for a message
const MaxMessageSize = 1024 * 1024 * 10 // 10MB
// newGRPCServer creates a gRPC server
func newGRPCServer(listeningAddresses []string) *gRPCServer {
return &gRPCServer{
server: grpc.NewServer(grpc.MaxRecvMsgSize(maxMessageSize), grpc.MaxSendMsgSize(maxMessageSize)),
server: grpc.NewServer(grpc.MaxRecvMsgSize(MaxMessageSize), grpc.MaxSendMsgSize(MaxMessageSize)),
listeningAddresses: listeningAddresses,
}
}
@ -39,7 +40,7 @@ func (s *gRPCServer) Start() error {
}
}
log.Debugf("Server started with maxMessageSize %d", maxMessageSize)
log.Debugf("Server started with MaxMessageSize %d", MaxMessageSize)
return nil
}

4
infrastructure/network/rpcclient/grpcclient/grpcclient.go
View File

@ -4,6 +4,7 @@ import (
"context"
"github.com/kaspanet/kaspad/app/appmessage"
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/server/grpcserver"
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/server/grpcserver/protowire"
"github.com/pkg/errors"
"google.golang.org/grpc"
@ -33,7 +34,8 @@ func Connect(address string) (*GRPCClient, error) {
}
grpcClient := protowire.NewRPCClient(gRPCConnection)
stream, err := grpcClient.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name))
stream, err := grpcClient.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name),
grpc.MaxCallRecvMsgSize(grpcserver.MaxMessageSize), grpc.MaxCallSendMsgSize(grpcserver.MaxMessageSize))
if err != nil {
return nil, errors.Wrapf(err, "error getting client stream for %s", address)
}