60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package daemons
|
|
|
|
import (
|
|
"oc-schedulerd/conf"
|
|
"os"
|
|
"time"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
workflow_execution "cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
|
)
|
|
|
|
var Bookings = ScheduledBooking{Bookings: map[string]*workflow_execution.WorkflowExecution{}}
|
|
|
|
type ExecutionManager struct{}
|
|
|
|
// 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")
|
|
Bookings.Mu.Lock()
|
|
if len(Bookings.Bookings) > 0 {
|
|
for k, v := range Bookings.Bookings {
|
|
if v.ExecDate.Before(time.Now().UTC()) {
|
|
logger.Info().Msg("Will execute " + k + " soon")
|
|
go em.executeBooking(v)
|
|
delete(Bookings.Bookings, k)
|
|
}
|
|
}
|
|
}
|
|
Bookings.Mu.Unlock()
|
|
time.Sleep(time.Second)
|
|
}
|
|
}
|
|
|
|
func (em *ExecutionManager) executeBooking(booking *workflow_execution.WorkflowExecution) {
|
|
// start execution
|
|
// create the yaml that describes the pod : filename, path/url to Loki
|
|
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")
|
|
duration := 0
|
|
if booking.EndDate != nil && booking.ExecDate != nil {
|
|
duration = int(booking.EndDate.Sub(*booking.ExecDate).Seconds())
|
|
}
|
|
monitor := LocalMonitor{
|
|
Logger: logger,
|
|
Duration: duration,
|
|
LokiURL: conf.GetConfig().LokiUrl,
|
|
KubeURL: "localhost",
|
|
ExecutionID: booking.UUID,
|
|
}
|
|
monitor.LaunchLocalMonitor()
|
|
}
|
|
}
|