feat(executor): default 512 MB grpc msg size

2024-11-06 07:20:05 -05:00 · 2024-11-06 07:20:05 -05:00 · adb9dd2a9a
commit adb9dd2a9a
parent 0cb1a63025
2 changed files with 4 additions and 1 deletions
--- a/internal/executor/sandbox/executor.go
+++ b/internal/executor/sandbox/executor.go
@ -59,7 +59,7 @@ func (e *Sandbox) Run(cmds []stage.Cmd) ([]stage.ExecutorResult, error) {
 			)
 		}
 	}
-	slog.Debug("sandbox execute", "total protobuf cmd size", totalPbCmdSize)
+	slog.Info("sandbox execute", "total protobuf cmd size", totalPbCmdSize)
 	pbReq := &pb.Request{Cmd: pbCmds}
 	pbRet, err := e.execClient.Exec(context.TODO(), pbReq)
 	if err != nil {
--- a/internal/executor/sandbox/grpc.go
+++ b/internal/executor/sandbox/grpc.go
@ -23,6 +23,9 @@ func createExecClient(execServer, token string) (pb.ExecutorClient, error) {
 func createGRPCConnection(addr, token string) (*grpc.ClientConn, error) {
 	opts := []grpc.DialOption{
 		grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpc.WithDefaultCallOptions(
+			grpc.MaxCallRecvMsgSize(512 * 1024 * 1024), // 512 MB
+		),
 	}
 	if token != "" {
 		opts = append(opts, grpc.WithPerRPCCredentials(newTokenAuth(token)))