oc-schedulerd/daemons/execution_manager.go

60 lines
1.7 KiB
Go
Raw Normal View History

2024-07-11 18:25:40 +02:00
package daemons
2024-07-23 12:16:20 +02:00
import (
"oc-schedulerd/conf"
2024-07-29 15:45:32 +02:00
"os"
2024-07-23 12:16:20 +02:00
"time"
oclib "cloud.o-forge.io/core/oc-lib"
2024-07-23 12:16:20 +02:00
)
2024-07-11 18:25:40 +02:00
2024-08-20 15:24:46 +02:00
var Bookings = ScheduledBooking{Bookings: []Booking{}}
type ExecutionManager struct{}
2024-07-23 12:16:20 +02:00
// 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() {
logger := oclib.GetLogger()
for {
logger.Debug().Msg("New loop")
2024-08-20 15:24:46 +02:00
Bookings.Mu.Lock()
if len(Bookings.Bookings) > 0 {
for i := len(Bookings.Bookings) - 1; i >= 0; i-- {
logger.Debug().Msg("It should start at " + Bookings.Bookings[i].Start.String() + " and it is now " + time.Now().UTC().String())
if Bookings.Bookings[i].Start.Before(time.Now().UTC()) {
logger.Info().Msg("Will execute " + Bookings.Bookings[i].Workflow + " soon")
go em.executeBooking(Bookings.Bookings[i])
Bookings.Bookings = append(Bookings.Bookings[:i], Bookings.Bookings[i+1:]...)
2024-07-23 12:16:20 +02:00
}
}
}
2024-08-20 15:24:46 +02:00
Bookings.Mu.Unlock()
2024-07-23 12:16:20 +02:00
time.Sleep(time.Second)
}
}
func (em *ExecutionManager) executeBooking(booking Booking) {
// 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")
logger := oclib.GetLogger()
if exec_method == "k8s" {
logger.Error().Msg("TODO : executing oc-monitor in a k8s")
} else {
logger.Debug().Msg("Executing oc-monitor localy")
2024-08-20 16:14:10 +02:00
duration := 0
if booking.Stop != nil && booking.Start != nil {
duration = int(booking.Stop.Sub(*booking.Start).Seconds())
}
monitor := LocalMonitor{
Logger: logger,
2024-08-20 16:14:10 +02:00
Duration: duration,
LokiURL: conf.GetConfig().LokiUrl,
KubeURL: "localhost",
2024-08-20 15:24:46 +02:00
WorkflowName: booking.Workflow,
}
2024-08-09 18:44:33 +02:00
monitor.LaunchLocalMonitor()
2024-07-29 15:45:32 +02:00
}
}