cloudflared-mirror/overwatch/app_manager.go

package overwatch

// ServiceCallback is a service notify it's runloop finished.
// the first parameter is the service type
// the second parameter is the service name
// the third parameter is an optional error if the service failed
type ServiceCallback func(string, string, error)

// AppManager is the default implementation of overwatch service management
type AppManager struct {
	services map[string]Service
	callback ServiceCallback
}

// NewAppManager creates a new overwatch manager
func NewAppManager(callback ServiceCallback) Manager {
	return &AppManager{services: make(map[string]Service), callback: callback}
}

// Add takes in a new service to manage.
// It stops the service if it already exist in the manager and is running
// It then starts the newly added service
func (m *AppManager) Add(service Service) {
	// check for existing service
	if currentService, ok := m.services[service.Name()]; ok {
		if currentService.Hash() == service.Hash() {
			return // the exact same service, no changes, so move along
		}
		currentService.Shutdown() //shutdown the listener since a new one is starting
	}
	m.services[service.Name()] = service

	//start the service!
	go m.serviceRun(service)
}

// Remove shutdowns the service by name and removes it from its current management list
func (m *AppManager) Remove(name string) {
	if currentService, ok := m.services[name]; ok {
		currentService.Shutdown()
	}
	delete(m.services, name)
}

// Services returns all the current Services being managed
func (m *AppManager) Services() []Service {
	values := []Service{}
	for _, value := range m.services {
		values = append(values, value)
	}
	return values
}

func (m *AppManager) serviceRun(service Service) {
	err := service.Run()
	if m.callback != nil {
		m.callback(service.Type(), service.Name(), err)
	}
}