69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"oc-discovery/conf"
|
|
"oc-discovery/daemons/node"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
beego "github.com/beego/beego/v2/server/web"
|
|
"github.com/beego/beego/v2/server/web/filter/cors"
|
|
)
|
|
|
|
const appname = "oc-discovery"
|
|
|
|
func main() {
|
|
// Init the oc-lib
|
|
oclib.Init(appname)
|
|
// get the right config file
|
|
|
|
o := oclib.GetConfLoader()
|
|
|
|
conf.GetConfig().Name = o.GetStringDefault("NAME", "opencloud-demo")
|
|
conf.GetConfig().Hostname = o.GetStringDefault("HOSTNAME", "localhost")
|
|
conf.GetConfig().PSKPath = o.GetStringDefault("PSK_PATH", "./psk/psk.key")
|
|
conf.GetConfig().NodeEndpointPort = o.GetInt64Default("NODE_ENDPOINT_PORT", 4001)
|
|
conf.GetConfig().PublicKeyPath = o.GetStringDefault("PUBLIC_KEY_PATH", "./pem/public.pem")
|
|
conf.GetConfig().PrivateKeyPath = o.GetStringDefault("PRIVATE_KEY_PATH", "./pem/private.pem")
|
|
conf.GetConfig().IndexerAddresses = o.GetStringDefault("INDEXER_ADDRESSES", "")
|
|
|
|
conf.GetConfig().NodeMode = o.GetStringDefault("NODE_MODE", "node")
|
|
|
|
// Normal 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.InsertFilter("*", beego.BeforeRouter, cors.Allow(&cors.Options{
|
|
AllowAllOrigins: true,
|
|
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
|
AllowHeaders: []string{"Origin", "Authorization", "Content-Type"},
|
|
ExposeHeaders: []string{"Content-Length", "Content-Type"},
|
|
AllowCredentials: true,
|
|
}))
|
|
|
|
ctx, stop := signal.NotifyContext(
|
|
context.Background(),
|
|
os.Interrupt,
|
|
syscall.SIGTERM,
|
|
)
|
|
defer stop()
|
|
|
|
isNode := strings.Contains(conf.GetConfig().NodeMode, "node")
|
|
isIndexer := strings.Contains(conf.GetConfig().NodeMode, "indexer")
|
|
|
|
if n, err := node.InitNode(isNode, isIndexer); err != nil {
|
|
panic(err)
|
|
} else {
|
|
<-ctx.Done() // 👈 the only blocking point
|
|
log.Println("shutting down")
|
|
n.Close()
|
|
}
|
|
|
|
}
|