Add statistics route for execution environments

* Add statistics route for execution environments

* Add maximum to port api definition

Co-authored-by: Sebastian Serth <MrSerth@users.noreply.github.com>

internal/runner/execution_environment_mock.go

@@ -66,6 +66,20 @@ func (_m *ExecutionEnvironmentMock) ID() dto.EnvironmentID {
 	return r0
 }
 
+// IdleRunnerCount provides a mock function with given fields:
+func (_m *ExecutionEnvironmentMock) IdleRunnerCount() int {
+	ret := _m.Called()
+
+	var r0 int
+	if rf, ok := ret.Get(0).(func() int); ok {
+		r0 = rf()
+	} else {
+		r0 = ret.Get(0).(int)
+	}
+
+	return r0
+}
+
 // Image provides a mock function with given fields:
 func (_m *ExecutionEnvironmentMock) Image() string {
 	ret := _m.Called()

internal/runner/manager.go

@@ -62,6 +62,8 @@ type ExecutionEnvironment interface {
 	AddRunner(r Runner)
 	// DeleteRunner removes an idle runner from the environment.
 	DeleteRunner(id string)
+	// IdleRunnerCount returns the number of idle runners of the environment.
+	IdleRunnerCount() int
 }
 
 // Manager keeps track of the used and unused runners of all execution environments in order to provide unused
@@ -82,6 +84,9 @@ type Manager interface {
 	// It does nothing if the specified environment can not be found.
 	DeleteEnvironment(id dto.EnvironmentID)
 
+	// EnvironmentStatistics returns statistical data for each execution environment.
+	EnvironmentStatistics() map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData
+
 	// Claim returns a new runner. The runner is deleted after duration seconds if duration is not 0.
 	// It makes sure that the runner is not in use yet and returns an error if no runner could be provided.
 	Claim(id dto.EnvironmentID, duration int) (Runner, error)
@@ -136,6 +141,27 @@ func (m *NomadRunnerManager) DeleteEnvironment(id dto.EnvironmentID) {
 	m.environments.Delete(id)
 }
 
+func (m *NomadRunnerManager) EnvironmentStatistics() map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData {
+	environments := make(map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData)
+	for _, e := range m.environments.List() {
+		environments[e.ID()] = &dto.StatisticalExecutionEnvironmentData{
+			ID:                 int(e.ID()),
+			PrewarmingPoolSize: e.PrewarmingPoolSize(),
+			IdleRunners:        uint(e.IdleRunnerCount()),
+			UsedRunners:        0,
+		}
+	}
+
+	for _, r := range m.usedRunners.List() {
+		id, err := nomad.EnvironmentIDFromRunnerID(r.ID())
+		if err != nil {
+			log.WithError(err).Error("Stored runners must have correct IDs")
+		}
+		environments[id].UsedRunners++
+	}
+	return environments
+}
+
 func (m *NomadRunnerManager) Claim(environmentID dto.EnvironmentID, duration int) (Runner, error) {
 	environment, ok := m.environments.Get(environmentID)
 	if !ok {

internal/runner/manager_mock.go

@@ -40,6 +40,22 @@ func (_m *ManagerMock) DeleteEnvironment(id dto.EnvironmentID) {
 	_m.Called(id)
 }
 
+// EnvironmentStatistics provides a mock function with given fields:
+func (_m *ManagerMock) EnvironmentStatistics() map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData {
+	ret := _m.Called()
+
+	var r0 map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData
+	if rf, ok := ret.Get(0).(func() map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(map[dto.EnvironmentID]*dto.StatisticalExecutionEnvironmentData)
+		}
+	}
+
+	return r0
+}
+
 // Get provides a mock function with given fields: runnerID
 func (_m *ManagerMock) Get(runnerID string) (Runner, error) {
 	ret := _m.Called(runnerID)

internal/runner/storage.go

@@ -6,6 +6,9 @@ import (
 
 // Storage is an interface for storing runners.
 type Storage interface {
+	// List returns all runners from the storage.
+	List() []Runner
+
 	// Add adds a runner to the storage.
 	// It overwrites the old runner if one with the same id was already stored.
 	Add(Runner)
@@ -41,6 +44,15 @@ func NewLocalRunnerStorage() *localRunnerStorage {
 	}
 }
 
+func (s *localRunnerStorage) List() (r []Runner) {
+	s.RLock()
+	defer s.RUnlock()
+	for _, value := range s.runners {
+		r = append(r, value)
+	}
+	return r
+}
+
 func (s *localRunnerStorage) Add(r Runner) {
 	s.Lock()
 	defer s.Unlock()