upgrade main.go (oc-lib refactor)
This commit is contained in:
		
							
								
								
									
										38
									
								
								main.go
									
									
									
									
									
								
							
							
						
						
									
										38
									
								
								main.go
									
									
									
									
									
								
							@@ -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()
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user