feat: add support for loadgen log-levels

Signed-off-by: Sam Stuewe <stuewe@mit.edu>
mit-dci · Aug 8, 2024 · 185f430 · 185f430
1 parent 0a78595
commit 185f430
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/common/testrun.go b/common/testrun.go
@@ -61,6 +61,7 @@ type TestRun struct {
 	TicketerLogLevel          string             `json:"ticketerLogLevel"          feFieldTitle:"Ticketer Log Level"              feFieldType:"loglevel"`
 	CoordinatorLogLevel       string             `json:"coordinatorLogLevel"       feFieldTitle:"Coordinator Log Level"           feFieldType:"loglevel"`
 	WatchtowerLogLevel        string             `json:"watchtowerLogLevel"        feFieldTitle:"Watchtower Log Level"            feFieldType:"loglevel"`
+	LoadGenLogLevel           string             `json:"loadGenLogLevel"           feFieldTitle:"Loadgen Log Level"               feFieldType:"loglevel"`
 	AtomizerTelemetryLevel    string             `json:"atomizerTelemetryLevel"    feFieldTitle:"Atomizer Telemetry Level"        feFieldType:"tellevel"`
 	ArchiverTelemetryLevel    string             `json:"archiverTelemetryLevel"    feFieldTitle:"Archiver Telemetry Level"        feFieldType:"tellevel"`
 	SentinelTelemetryLevel    string             `json:"sentinelTelemetryLevel"    feFieldTitle:"Sentinel Telemetry Level"        feFieldType:"tellevel"`

diff --git a/coordinator/testruns/roleconfig.go b/coordinator/testruns/roleconfig.go
@@ -321,6 +321,12 @@ func (t *TestRunManager) RoleLogLevel(
 		loglevel = tr.WatchtowerLogLevel
 	case common.SystemRoleCoordinator:
 		loglevel = tr.CoordinatorLogLevel
+	case common.SystemRoleAtomizerCliWatchtower:
+		fallthrough
+	case common.SystemRoleParsecGen:
+		fallthrough
+	case common.SystemRoleTwoPhaseGen:
+		loglevel = tr.LoadGenLogLevel
 	}
 	return loglevel
 }