server: allow configuration default send/recv message sizes

Signed-off-by: Stephen J Day <stephen.day@docker.com>

server/config.go

@@ -49,9 +49,11 @@ type Config struct {
 
 // GRPCConfig provides GRPC configuration for the socket
 type GRPCConfig struct {
-	Address string `toml:"address"`
-	UID     int    `toml:"uid"`
-	GID     int    `toml:"gid"`
+	Address        string `toml:"address"`
+	UID            int    `toml:"uid"`
+	GID            int    `toml:"gid"`
+	MaxRecvMsgSize int    `toml:"max_recv_message_size"`
+	MaxSendMsgSize int    `toml:"max_send_message_size"`
 }
 
 // Debug provides debug configuration

server/server.go

@@ -67,6 +67,8 @@ func New(ctx context.Context, config *Config) (*Server, error) {
 		return nil, err
 	}
 	rpc := grpc.NewServer(
+		grpc.MaxRecvMsgSize(config.GRPC.MaxRecvMsgSize),
+		grpc.MaxSendMsgSize(config.GRPC.MaxSendMsgSize),
 		grpc.UnaryInterceptor(grpc_prometheus.UnaryServerInterceptor),
 		grpc.StreamInterceptor(grpc_prometheus.StreamServerInterceptor),
 	)