fix(backend): add MaxCallRecvMsgSize(math.MaxInt32) to proxy server (…

…#4402)
kubeflow · Sep 2, 2020 · cd9c9ff · cd9c9ff
1 parent 32c9c2a
commit cd9c9ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/apiserver/main.go b/backend/src/apiserver/main.go
@@ -154,7 +154,7 @@ func startHttpProxy(resourceManager *resource.ResourceManager) {
 
 func registerHttpHandlerFromEndpoint(handler RegisterHttpHandlerFromEndpoint, serviceName string, ctx context.Context, mux *runtime.ServeMux) {
 	endpoint := "localhost" + *rpcPortFlag
-	opts := []grpc.DialOption{grpc.WithInsecure()}
+	opts := []grpc.DialOption{grpc.WithInsecure(), grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(math.MaxInt32))}
 
 	if err := handler(ctx, mux, endpoint, opts); err != nil {
 		glog.Fatalf("Failed to register %v handler: %v", serviceName, err)