54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package daemons
|
|
|
|
import (
|
|
"fmt"
|
|
"oc-schedulerd/conf"
|
|
"os/exec"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type LocalMonitor struct {
|
|
LokiURL string
|
|
KubeURL string
|
|
ExecutionID string
|
|
PeerID string
|
|
Duration int
|
|
Logger zerolog.Logger
|
|
}
|
|
|
|
func (lm *LocalMonitor) LaunchLocalMonitor() {
|
|
if lm.LokiURL == "" || lm.KubeURL == "" || lm.ExecutionID == "" {
|
|
lm.Logger.Error().Msg("Missing parameter in LocalMonitor")
|
|
}
|
|
|
|
// For dev purposes, in prod KubeURL must be a kube API's URL
|
|
if lm.KubeURL != "localhost" {
|
|
lm.execRemoteKube()
|
|
} else {
|
|
lm.execLocalKube()
|
|
}
|
|
}
|
|
|
|
func (lm *LocalMonitor) execLocalKube() {
|
|
args := []string{"-e", lm.ExecutionID, "-p", lm.PeerID, "-u", lm.LokiURL, "-m", conf.GetConfig().MongoUrl,
|
|
"-d", conf.GetConfig().DBName}
|
|
|
|
if conf.GetConfig().Mode == "kubernetes" {
|
|
args = append(args, []string{"-M", conf.GetConfig().Mode, "-H", conf.GetConfig().KubeHost, "-P", conf.GetConfig().KubePort,
|
|
"-C", conf.GetConfig().KubeCert, "-D", conf.GetConfig().KubeData, "-c", conf.GetConfig().KubeCA}...)
|
|
}
|
|
if lm.Duration > 0 {
|
|
args = append(args, "-t", fmt.Sprintf("%d", lm.Duration))
|
|
}
|
|
cmd := exec.Command(conf.GetConfig().MonitorPath, args...)
|
|
fmt.Printf("Command : %v\n", cmd)
|
|
err := cmd.Start()
|
|
if err != nil {
|
|
lm.Logger.Error().Msg("Could not start oc-monitor for " + lm.ExecutionID + " : " + err.Error())
|
|
}
|
|
}
|
|
|
|
// TODO : implement this
|
|
func (lm *LocalMonitor) execRemoteKube() {}
|