crons_monitor_data.go

package main

import (
	"sync"

	"github.com/getsentry/sentry-go"
	batchv1 "k8s.io/api/batch/v1"
)

// Struct associated with a job
type CronsJobData struct {
	CheckinID sentry.EventID
}

// Constructor for cronsMonitorData
func NewCronsJobData(checkinId sentry.EventID) *CronsJobData {
	return &CronsJobData{
		CheckinID: checkinId,
	}
}

func (j *CronsJobData) getCheckinID() sentry.EventID {
	return j.CheckinID
}

// Struct associated with a cronJob
type CronsMonitorData struct {
	mutex               sync.RWMutex
	MonitorSlug         string
	monitorConfig       *sentry.MonitorConfig
	JobDatas            map[string]*CronsJobData
	requiredCompletions int32
}

// Constructor for cronsMonitorData
func NewCronsMonitorData(monitorSlug string, schedule string, completions *int32) *CronsMonitorData {
	// Get required number of pods to complete
	var requiredCompletions int32
	if completions == nil {
		// If not set, any pod success is enough
		requiredCompletions = 1
	} else {
		requiredCompletions = *completions
	}
	monitorSchedule := sentry.CrontabSchedule(schedule)
	return &CronsMonitorData{
		mutex:       sync.RWMutex{},
		MonitorSlug: monitorSlug,
		monitorConfig: &sentry.MonitorConfig{
			Schedule: monitorSchedule,
		},
		JobDatas:            make(map[string]*CronsJobData),
		requiredCompletions: requiredCompletions,
	}
}

// Add a job to the crons monitor
func (c *CronsMonitorData) addJob(job *batchv1.Job, checkinID sentry.EventID) error {
	c.mutex.Lock()
	defer c.mutex.Unlock()
	c.JobDatas[job.Name] = NewCronsJobData(checkinID)
	return nil
}

// Wrapper struct over crons monitor map that
// handles synchronization
type CronsMetaData struct {
	mutex               *sync.RWMutex
	cronsMonitorDataMap map[string]*CronsMonitorData
}

func NewCronsMetaData() *CronsMetaData {
	return &CronsMetaData{
		mutex:               &sync.RWMutex{},
		cronsMonitorDataMap: make(map[string]*CronsMonitorData),
	}
}

func (c *CronsMetaData) addCronsMonitorData(cronjobName string, newCronsMonitorData *CronsMonitorData) {
	c.mutex.Lock()
	defer c.mutex.Unlock()
	c.cronsMonitorDataMap[cronjobName] = newCronsMonitorData
}

func (c *CronsMetaData) deleteCronsMonitorData(cronjobName string) {
	c.mutex.Lock()
	defer c.mutex.Unlock()
	delete(c.cronsMonitorDataMap, cronjobName)
}

func (c *CronsMetaData) getCronsMonitorData(cronjobName string) (*CronsMonitorData, bool) {
	c.mutex.RLock()
	defer c.mutex.RUnlock()
	cronsMonitorData, ok := c.cronsMonitorDataMap[cronjobName]
	return cronsMonitorData, ok
}