upgrade main.go (oc-lib refactor)

This commit is contained in:
ycc
2024-09-04 17:23:11 +02:00
parent 163f4555a1
commit a34e04781e
8 changed files with 61 additions and 728 deletions

38
main.go
View File

@@ -4,37 +4,33 @@ import (
_ "oc-scheduler/routers"
oclib "cloud.o-forge.io/core/oc-lib"
"cloud.o-forge.io/core/oc-lib/tools"
beego "github.com/beego/beego/v2/server/web"
"github.com/goraz/onion"
)
const defaultConfigFile = "/etc/oc/scheduler.json"
const appname = "oc-scheduler"
func main() {
configFile := ""
var o *onion.Onion
l3 := onion.NewEnvLayerPrefix("_", "OCSCHEDULER_")
l2, err := onion.NewFileLayer(defaultConfigFile, nil)
if err == nil {
configFile = defaultConfigFile
}
if configFile == "" || l2 == nil {
o = onion.New(l3)
} else {
o = onion.New(l2, l3)
}
tools.SetConfig(
// Init the oc-lib
oclib.Init(appname)
// Load the right config file
o := oclib.GetConfLoader()
// feed the library with the loaded config
oclib.SetConfig(
o.GetStringDefault("MONGO_URL", "mongodb://127.0.0.1:27017"),
o.GetStringDefault("MONGO_DATABASE", "DC_myDC"),
"",
o.GetStringDefault("lokiurl", ""),
o.GetStringDefault("loglevel", "info"),
)
// Init OC with hostname and port for discovery purpose
oclib.Init("oc-scheduler", o.GetStringDefault("HOSTNAME", "localhost"), o.GetStringDefault("PORT", "8090"))
// Normal beego init
//if beego.BConfig.RunMode == "dev" {
// Beego init
beego.BConfig.AppName = appname
beego.BConfig.Listen.HTTPPort = o.GetIntDefault("port", 8080)
beego.BConfig.WebConfig.DirectoryIndex = true
beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
//}
beego.Run()
}