console/main.go

102 lines
2.9 KiB
Go

package main
import (
"errors"
_ "expvar"
"fmt"
"infini.sh/framework"
"infini.sh/framework/core/elastic"
"infini.sh/framework/core/env"
"infini.sh/framework/core/global"
"infini.sh/framework/core/module"
"infini.sh/framework/core/orm"
pipe "infini.sh/framework/core/pipeline"
"infini.sh/framework/modules"
"infini.sh/framework/modules/metrics"
"infini.sh/framework/plugins/elastic/json_indexing"
"infini.sh/search-center/config"
"infini.sh/search-center/model"
"infini.sh/search-center/model/alerting"
alertSrv "infini.sh/search-center/service/alerting"
)
var appConfig *config.AppConfig
var appUI *UI
func main() {
terminalHeader := (" ___ _\n")
terminalHeader += (" / __\\___ _ __ ___ ___ | | ___\n")
terminalHeader += (" / / / _ \\| '_ \\/ __|/ _ \\| |/ _ \\\n")
terminalHeader += ("/ /__| (_) | | | \\__ \\ (_) | | __/\n")
terminalHeader += ("\\____/\\___/|_| |_|___/\\___/|_|\\___|\n\n")
terminalFooter := ""
app := framework.NewApp("console", "the easiest way to operate your own elasticsearch platform.",
config.Version, config.LastCommitLog, config.BuildDate,config.EOLDate, terminalHeader, terminalFooter)
app.Init(nil)
defer app.Shutdown()
if app.Setup(func() {
//load core modules first
modules.Register()
pipe.RegisterProcessorPlugin("json_indexing", json_indexing.New)
module.RegisterUserPlugin(&metrics.MetricsModule{})
appConfig = &config.AppConfig{
Elasticsearch: "default",
UI: config.UIConfig{
LocalPath: ".public",
VFSEnabled: true,
LocalEnabled: true,
},
}
ok, err := env.ParseConfig("web", appConfig)
if err != nil {
panic(err)
}
if !ok {
panic(errors.New("config not exists"))
}
//load web UI files
appUI = &UI{Config: appConfig}
appUI.InitUI()
//uiConfig := ui.UIConfig{}
//env.ParseConfig("web", &uiConfig)
//
//if len(global.Env().SystemConfig.APIConfig.CrossDomain.AllowedOrigins)==0{
// global.Env().SystemConfig.APIConfig.CrossDomain.AllowedOrigins=
// append(global.Env().SystemConfig.APIConfig.CrossDomain.AllowedOrigins,uiConfig.NetworkConfig.GetBindingAddr())
//}
apiConfig := global.Env().SystemConfig.APIConfig
if len(apiConfig.CrossDomain.AllowedOrigins) == 0 {
apiConfig.CrossDomain.AllowedOrigins = []string{
fmt.Sprintf("%s://%s", appConfig.GetSchema(), apiConfig.NetworkConfig.GetPublishAddr()),
}
}
//start each module, with enabled provider
module.Start()
}, func() {
orm.RegisterSchemaWithIndexName(model.Dict{}, "dict")
orm.RegisterSchemaWithIndexName(model.Reindex{}, "reindex")
orm.RegisterSchemaWithIndexName(elastic.View{}, "view")
orm.RegisterSchemaWithIndexName(alerting.Config{}, "alerting-config")
orm.RegisterSchemaWithIndexName(alerting.Alert{}, "alerting-alerts")
orm.RegisterSchemaWithIndexName(alerting.AlertingHistory{}, "alerting-alert-history")
orm.RegisterSchema(elastic.CommonCommand{})
alertSrv.GetScheduler().Start()
},nil){
app.Run()
}
}