mirror/prysm-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/prysm-pulse.git synced 2025-01-03 08:37:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Do Not Ignore Max Gateway Size gRPC (#6232)

Browse Source 
* do not ignore max message size in gateway
* fix build
This commit is contained in:
Raul Jordan 2020-06-12 15:02:20 -05:00 committed by GitHub
parent 87ca73d605
commit de45a54991
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
beacon-chain/gateway/gateway.go
View File

@ -151,7 +151,7 @@ func (g *Gateway) dial(ctx context.Context, network, addr string) (*grpc.ClientC
case "tcp": case "tcp":
return g.dialTCP(ctx, addr) return g.dialTCP(ctx, addr)
case "unix": case "unix":
return dialUnix(ctx, addr) return g.dialUnix(ctx, addr)
default: default:
return nil, fmt.Errorf("unsupported network type %q", network) return nil, fmt.Errorf("unsupported network type %q", network)
} }
@ -160,10 +160,9 @@ func (g *Gateway) dial(ctx context.Context, network, addr string) (*grpc.ClientC
// dialTCP creates a client connection via TCP. // dialTCP creates a client connection via TCP.
// "addr" must be a valid TCP address with a port number. // "addr" must be a valid TCP address with a port number.
func (g *Gateway) dialTCP(ctx context.Context, addr string) (*grpc.ClientConn, error) { func (g *Gateway) dialTCP(ctx context.Context, addr string) (*grpc.ClientConn, error) {
opts := []grpc.DialOption{grpc.WithInsecure()} opts := []grpc.DialOption{
grpc.WithInsecure(),
if g.enableDebugRPCEndpoints { grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(int(g.maxCallRecvMsgSize))),
opts = append(opts, grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(int(g.maxCallRecvMsgSize))))
} }
return grpc.DialContext( return grpc.DialContext(
@ -175,9 +174,14 @@ func (g *Gateway) dialTCP(ctx context.Context, addr string) (*grpc.ClientConn, e
// dialUnix creates a client connection via a unix domain socket. // dialUnix creates a client connection via a unix domain socket.
// "addr" must be a valid path to the socket. // "addr" must be a valid path to the socket.
func dialUnix(ctx context.Context, addr string) (*grpc.ClientConn, error) { func (g *Gateway) dialUnix(ctx context.Context, addr string) (*grpc.ClientConn, error) {
d := func(addr string, timeout time.Duration) (net.Conn, error) { d := func(addr string, timeout time.Duration) (net.Conn, error) {
return net.DialTimeout("unix", addr, timeout) return net.DialTimeout("unix", addr, timeout)
} }
return grpc.DialContext(ctx, addr, grpc.WithInsecure(), grpc.WithDialer(d)) opts := []grpc.DialOption{
grpc.WithInsecure(),
grpc.WithDialer(d),
grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(int(g.maxCallRecvMsgSize))),
}
return grpc.DialContext(ctx, addr, opts...)
} }