From 0cb1a630253b0126bfc448e20a855540dea75cb1 Mon Sep 17 00:00:00 2001
From: Boming Zhang <bomingzh@sjtu.edu.cn>
Date: Wed, 6 Nov 2024 07:14:53 -0500
Subject: [PATCH] feat(executor): debug log total protobuf cmds size

---
 internal/executor/sandbox/executor.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/executor/sandbox/executor.go b/internal/executor/sandbox/executor.go
index 01311ee..7a3a63f 100644
--- a/internal/executor/sandbox/executor.go
+++ b/internal/executor/sandbox/executor.go
@@ -38,11 +38,14 @@ func (e *Sandbox) Run(cmds []stage.Cmd) ([]stage.ExecutorResult, error) {
 		}
 	}
 	pbCmds := convertPBCmd(cmds)
+	totalPbCmdSize := 0
 	for i, pbCmd := range pbCmds {
+		pbCmdSize := proto.Size(pbCmd)
+		totalPbCmdSize += pbCmdSize
 		slog.Debug(
 			"sandbox execute",
 			"index", i,
-			"protobuf cmd size", proto.Size(pbCmd),
+			"protobuf cmd size", pbCmdSize,
 			"protobuf cmd stdin size", proto.Size(pbCmd.Files[0]),
 			"protobuf cmd stdout size", proto.Size(pbCmd.Files[1]),
 			"protobuf cmd stderr size", proto.Size(pbCmd.Files[2]),
@@ -56,6 +59,7 @@ func (e *Sandbox) Run(cmds []stage.Cmd) ([]stage.ExecutorResult, error) {
 			)
 		}
 	}
+	slog.Debug("sandbox execute", "total protobuf cmd size", totalPbCmdSize)
 	pbReq := &pb.Request{Cmd: pbCmds}
 	pbRet, err := e.execClient.Exec(context.TODO(), pbReq)
 	if err != nil {