signoz/pkg/query-service/app/opamp/model/coordinator.go

package model

import (
	"fmt"
	"sync"
)

// communicates with calling apis when config is applied or fails
var coordinator *Coordinator

func init() {
	subscribers := make(map[string][]OnChangeCallback, 0)
	coordinator = &Coordinator{
		subscribers: subscribers,
	}
}

type OnChangeCallback func(agentId string, hash string, err error)

// responsible for managing subscribers on config change
type Coordinator struct {
	mutex sync.Mutex

	// hash wise list of subscribers
	subscribers map[string][]OnChangeCallback
}

func onConfigSuccess(agentId string, hash string) {
	notifySubscribers(agentId, hash, nil)
}

func onConfigFailure(agentId string, hash string, errorMessage string) {
	notifySubscribers(agentId, hash, fmt.Errorf(errorMessage))
}

// OnSuccess listens to config changes and notifies subscribers
func notifySubscribers(agentId string, hash string, err error) {
	// this method currently does not handle multi-agent scenario.
	// as soon as a message is delivered, we release all the subscribers
	// for a given hash
	subs, ok := coordinator.subscribers[hash]
	if !ok {
		return
	}

	for _, s := range subs {
		s(agentId, hash, err)
	}

	// delete all subscribers for this hash, assume future
	// notifies will be disabled. the first response is processed
	delete(coordinator.subscribers, hash)
}

// callers subscribe to this function to listen on config change requests
func ListenToConfigUpdate(agentId string, hash string, ss OnChangeCallback) {
	coordinator.mutex.Lock()
	defer coordinator.mutex.Unlock()

	if subs, ok := coordinator.subscribers[hash]; ok {
		subs = append(subs, ss)
		coordinator.subscribers[hash] = subs
	} else {
		coordinator.subscribers[hash] = []OnChangeCallback{ss}
	}
}