From e445beab8974ba5482c5fa4f34cd4a8eda83831b Mon Sep 17 00:00:00 2001 From: Sam Stuewe Date: Wed, 7 Aug 2024 16:10:19 -0500 Subject: [PATCH] feat: add support for loadgen log-levels Signed-off-by: Sam Stuewe --- common/testrun.go | 1 + coordinator/testruns/roleconfig.go | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/common/testrun.go b/common/testrun.go index aefe447..6e4f786 100644 --- 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 index 2648a5f..2269fd1 100644 --- 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.SystemRolePhaseTwoGen: + fallthrough + case common.SystemRoleTwoPhaseGen: + loglevel = tr.LoadGenLogLevel } return loglevel }