59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"oc-discovery/conf"
|
|
"oc-discovery/daemons/node"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
)
|
|
|
|
const appname = "oc-discovery"
|
|
|
|
func main() {
|
|
// Init the oc-lib
|
|
oclib.InitDaemon(appname)
|
|
// get the right config file
|
|
|
|
o := oclib.GetConfLoader(appname)
|
|
|
|
conf.GetConfig().Name = o.GetStringDefault("NAME", "opencloud-demo")
|
|
conf.GetConfig().Hostname = o.GetStringDefault("HOSTNAME", "127.0.0.1")
|
|
conf.GetConfig().PSKPath = o.GetStringDefault("PSK_PATH", "./psk/psk.key")
|
|
conf.GetConfig().NodeEndpointPort = o.GetInt64Default("NODE_ENDPOINT_PORT", 4001)
|
|
conf.GetConfig().IndexerAddresses = o.GetStringDefault("INDEXER_ADDRESSES", "")
|
|
conf.GetConfig().NativeIndexerAddresses = o.GetStringDefault("NATIVE_INDEXER_ADDRESSES", "")
|
|
|
|
conf.GetConfig().PeerIDS = o.GetStringDefault("PEER_IDS", "")
|
|
|
|
conf.GetConfig().NodeMode = o.GetStringDefault("NODE_MODE", "node")
|
|
|
|
conf.GetConfig().MinIndexer = o.GetIntDefault("MIN_INDEXER", 1)
|
|
conf.GetConfig().MaxIndexer = o.GetIntDefault("MAX_INDEXER", 5)
|
|
|
|
ctx, stop := signal.NotifyContext(
|
|
context.Background(),
|
|
os.Interrupt,
|
|
syscall.SIGTERM,
|
|
)
|
|
defer stop()
|
|
fmt.Println(conf.GetConfig().NodeMode)
|
|
isNode := strings.Contains(conf.GetConfig().NodeMode, "node")
|
|
isIndexer := strings.Contains(conf.GetConfig().NodeMode, "indexer")
|
|
isNativeIndexer := strings.Contains(conf.GetConfig().NodeMode, "native-indexer")
|
|
|
|
if n, err := node.InitNode(isNode, isIndexer, isNativeIndexer); err != nil {
|
|
panic(err)
|
|
} else {
|
|
<-ctx.Done() // the only blocking point
|
|
log.Println("shutting down")
|
|
n.Close()
|
|
}
|
|
}
|