2024-07-11 18:25:40 +02:00
|
|
|
package daemons
|
|
|
|
|
2024-07-23 12:16:20 +02:00
|
|
|
import (
|
2024-07-29 15:45:32 +02:00
|
|
|
"oc-scheduler/conf"
|
2024-07-23 12:16:20 +02:00
|
|
|
"oc-scheduler/logger"
|
|
|
|
"oc-scheduler/models"
|
2024-07-29 15:45:32 +02:00
|
|
|
"os"
|
2024-07-23 12:16:20 +02:00
|
|
|
"time"
|
|
|
|
)
|
2024-07-11 18:25:40 +02:00
|
|
|
|
|
|
|
type ExecutionManager struct {
|
2024-07-23 12:16:20 +02:00
|
|
|
bookings *models.ScheduledBooking
|
|
|
|
executions []models.Booking
|
2024-07-11 18:25:40 +02:00
|
|
|
}
|
|
|
|
|
2024-08-09 18:44:33 +02:00
|
|
|
|
2024-07-29 15:45:32 +02:00
|
|
|
|
2024-07-23 12:16:20 +02:00
|
|
|
func (em *ExecutionManager) SetBookings(b *models.ScheduledBooking){
|
|
|
|
em.bookings = b
|
|
|
|
}
|
|
|
|
|
|
|
|
// Loop every second on the booking's list and move the booking that must start to a new list
|
|
|
|
// that will be looped over to start them
|
|
|
|
func (em *ExecutionManager) RetrieveNextExecutions(){
|
|
|
|
|
2024-07-29 15:45:32 +02:00
|
|
|
|
2024-07-23 12:16:20 +02:00
|
|
|
if(em.bookings == nil){
|
2024-08-09 18:44:33 +02:00
|
|
|
logger.Logger.Error().Msg("booking has not been set in the exection manager")
|
|
|
|
return
|
2024-07-23 12:16:20 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
for(true){
|
|
|
|
logger.Logger.Debug().Msg("New loop")
|
|
|
|
em.bookings.Mu.Lock()
|
|
|
|
bookings := em.bookings.Bookings
|
|
|
|
if (len(bookings) > 0){
|
|
|
|
for i := len( bookings) - 1 ; i >= 0 ; i--{
|
|
|
|
logger.Logger.Debug().Msg("It should start at " + bookings[i].Start.String() + " and it is now " + time.Now().UTC() .String())
|
|
|
|
if (bookings[i].Start.Before(time.Now().UTC())){
|
|
|
|
logger.Logger.Info().Msg("Will execute " + bookings[i].Workflow + " soon")
|
|
|
|
go em.executeBooking(bookings[i])
|
|
|
|
bookings = append(bookings[:i], bookings[i+1:]...)
|
|
|
|
em.bookings.Bookings = bookings
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
em.bookings.Mu.Unlock()
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (em *ExecutionManager) executeBooking(booking models.Booking){
|
2024-07-29 15:45:32 +02:00
|
|
|
|
2024-07-23 12:16:20 +02:00
|
|
|
|
|
|
|
// start execution
|
2024-07-29 15:45:32 +02:00
|
|
|
// create the yaml that describes the pod : filename, path/url to Loki
|
|
|
|
|
2024-08-09 18:44:33 +02:00
|
|
|
|
|
|
|
exec_method := os.Getenv("MONITOR_METHOD")
|
2024-07-29 15:45:32 +02:00
|
|
|
|
2024-08-09 18:44:33 +02:00
|
|
|
if exec_method == "local"{
|
|
|
|
logger.Logger.Debug().Msg("Executing oc-monitor localy")
|
2024-08-12 16:09:30 +02:00
|
|
|
monitor := LocalMonitor{LokiURL: conf.GetConfig().LokiUrl,KubeURL: "localhost",WorkflowName: booking.Workflow,}
|
2024-08-09 18:44:33 +02:00
|
|
|
monitor.LaunchLocalMonitor()
|
|
|
|
}else{
|
|
|
|
logger.Logger.Error().Msg("TODO : executing oc-monitor in a k8s")
|
2024-07-29 15:45:32 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2024-08-09 18:44:33 +02:00
|
|
|
|
2024-07-29 15:45:32 +02:00
|
|
|
}
|
|
|
|
|