subscribe.go

// SPDX-License-Identifier: Apache-2.0
// Copyright Authors of Cilium

package endpointmanager

import "github.com/cilium/cilium/pkg/endpoint"

// Subscribers may register via Subscribe() to be notified when events occur.
type Subscriber interface {
	// EndpointCreated is called at the end of endpoint creation.
	// Implementations must not attempt write operations on the
	// EndpointManager from this callback.
	EndpointCreated(ep *endpoint.Endpoint)

	// EndpointDeleted is called at the end of endpoint deletion.
	// Implementations must not attempt write operations on the
	// EndpointManager from this callback.
	EndpointDeleted(ep *endpoint.Endpoint, conf endpoint.DeleteConfig)
}

func (mgr *EndpointManager) Subscribe(s Subscriber) {
	mgr.mutex.Lock()
	defer mgr.mutex.Unlock()

	mgr.subscribers[s] = struct{}{}
}

func (mgr *EndpointManager) Unsubscribe(s Subscriber) {
	mgr.mutex.Lock()
	defer mgr.mutex.Unlock()
	delete(mgr.subscribers, s)
}