Add aws environments to the statistics

but only with the field usedRunners.
This commit is contained in:
Maximilian Paß
2022-04-03 18:07:08 +02:00
parent d7b1c2d691
commit 136f596dc2
7 changed files with 45 additions and 23 deletions

View File

@ -45,3 +45,22 @@ func (a AWSRunnerManager) Return(r Runner) error {
}
return nil
}
// EnvironmentStatistics returns only the used runner for each environment as the prewarming is handled
// by AWS transparently.
func (a AWSRunnerManager) EnvironmentStatistics() map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData {
environments := a.AbstractManager.EnvironmentStatistics()
for _, r := range a.usedRunners.List() {
workload, isAWSRunner := r.(*AWSFunctionWorkload)
if !isAWSRunner {
log.WithField("workload", workload).Error("Stored runners must be AWS runner")
continue
}
environmentID := workload.environment.ID()
environments[environmentID].UsedRunners++
}
return environments
}