e7ccefcf92
* Add standalone job controller * Add makefile * Add limit on the current implementation * Move job controllers to app package * Add TLS flags
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"k8s.io/client-go/tools/cache"
|
|
|
|
"github.com/grafana/grafana-app-sdk/logging"
|
|
informer "github.com/grafana/grafana/apps/provisioning/pkg/generated/informers/externalversions/provisioning/v0alpha1"
|
|
)
|
|
|
|
const (
|
|
jobControllerLoggerName = "provisioning-job-controller"
|
|
)
|
|
|
|
// JobController manages job create notifications.
|
|
type JobController struct {
|
|
jobSynced cache.InformerSynced
|
|
logger logging.Logger
|
|
|
|
// notification channel for job create events (replaces InsertNotifications)
|
|
notifications chan struct{}
|
|
}
|
|
|
|
// NewJobController creates a new JobController.
|
|
func NewJobController(
|
|
jobInformer informer.JobInformer,
|
|
) (*JobController, error) {
|
|
jc := &JobController{
|
|
jobSynced: jobInformer.Informer().HasSynced,
|
|
logger: logging.DefaultLogger.With("logger", jobControllerLoggerName),
|
|
notifications: make(chan struct{}, 1),
|
|
}
|
|
|
|
_, err := jobInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
|
|
AddFunc: func(obj interface{}) {
|
|
// Send notification for job create events (replaces InsertNotifications)
|
|
jc.sendNotification()
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return jc, nil
|
|
}
|
|
|
|
// InsertNotifications returns a channel that receives notifications when jobs are created.
|
|
// This replaces the InsertNotifications method from persistentstore.go.
|
|
func (jc *JobController) InsertNotifications() chan struct{} {
|
|
return jc.notifications
|
|
}
|
|
|
|
func (jc *JobController) sendNotification() {
|
|
select {
|
|
case jc.notifications <- struct{}{}:
|
|
default:
|
|
// Don't block if there's already a notification waiting
|
|
}
|
|
}
|