akvorado/orchestrator/root.go

// SPDX-FileCopyrightText: 2022 Free Mobile
// SPDX-License-Identifier: AGPL-3.0-only

// Package orchestrator synchronizes the different internal services.
package orchestrator

import (
	"sync"

	"akvorado/common/httpserver"
	"akvorado/common/reporter"
)

// Component represents the broker.
type Component struct {
	r      *reporter.Reporter
	d      *Dependencies
	config Configuration

	serviceLock           sync.Mutex
	serviceConfigurations map[ServiceType][]any
}

// Dependencies define the dependencies of the broker.
type Dependencies struct {
	HTTP *httpserver.Component
}

// ServiceType describes the different internal services
type ServiceType string

var (
	// InletService represents the inlet service type
	InletService ServiceType = "inlet"
	// OutletService represents the outlet service type
	OutletService ServiceType = "outlet"
	// OrchestratorService represents the orchestrator service type
	OrchestratorService ServiceType = "orchestrator"
	// ConsoleService represents the console service type
	ConsoleService ServiceType = "console"
	// DemoExporterService represents the demo exporter service type
	DemoExporterService ServiceType = "demo-exporter"
)

// New creates a new broker component.
func New(r *reporter.Reporter, configuration Configuration, dependencies Dependencies) (*Component, error) {
	c := Component{
		r:      r,
		d:      &dependencies,
		config: configuration,

		serviceConfigurations: map[ServiceType][]any{},
	}

	c.d.HTTP.GinRouter.GET("/api/v0/orchestrator/configuration/:service", c.configurationHandlerFunc)
	c.d.HTTP.GinRouter.GET("/api/v0/orchestrator/configuration/:service/:index", c.configurationHandlerFunc)

	return &c, nil
}

// RegisterConfiguration registers the configuration for a service.
func (c *Component) RegisterConfiguration(service ServiceType, configuration any) {
	c.serviceLock.Lock()
	if _, ok := c.serviceConfigurations[service]; !ok {
		c.serviceConfigurations[service] = []any{}
	}
	c.serviceConfigurations[service] = append(c.serviceConfigurations[service], configuration)
	c.serviceLock.Unlock()
}