116 lines
2.3 KiB
Go
116 lines
2.3 KiB
Go
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
|
|
// Use of this source code is governed by the GPL 3.0
|
|
|
|
package main
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"go.wit.com/dev/alexflint/arg"
|
|
"go.wit.com/lib/hostname"
|
|
"go.wit.com/lib/protobuf/zoopb"
|
|
"go.wit.com/log"
|
|
)
|
|
|
|
var VERSION string
|
|
var BUILDTIME string
|
|
|
|
//go:embed resources/*
|
|
var resources embed.FS
|
|
|
|
func main() {
|
|
me = new(zoodStruct)
|
|
me.pp = arg.MustParse(&argv)
|
|
|
|
if me.pp == nil {
|
|
me.pp.WriteHelp(os.Stdout)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if argv.Daemon {
|
|
// turn off timestamps for STDOUT (systemd adds them)
|
|
log.DaemonMode(true)
|
|
me.machine, me.fullpath = zoopb.InitDaemon()
|
|
} else {
|
|
me.machine, me.fullpath = zoopb.InitMachine()
|
|
|
|
}
|
|
|
|
if err := testZoo(); err != nil {
|
|
log.Info("FAILED TO CONNECT TO ZOOKEEPER: ", err)
|
|
log.Info("sleeping for 3 minutes")
|
|
time.Sleep(3 * time.Minute)
|
|
os.Exit(0)
|
|
}
|
|
me.pollDelay = 3 * time.Second
|
|
me.failcountmax = 20 // die every minute if zookeeper can't be found
|
|
|
|
if argv.Test != nil {
|
|
testURL(me.urlbase, me.machine)
|
|
os.Exit(0)
|
|
}
|
|
|
|
zood() // talks to zookeeper
|
|
}
|
|
|
|
func testZoo() error {
|
|
var err error
|
|
if argv.URL != "" {
|
|
log.Info("USING ARGV URL:", argv.URL)
|
|
if err = testURL(argv.URL, me.machine); err == nil {
|
|
me.urlbase = argv.URL
|
|
return nil
|
|
}
|
|
}
|
|
|
|
zooname := findZookeeper()
|
|
|
|
url := fmt.Sprintf("http://%s:8080/", zooname)
|
|
if err = testURL(url, me.machine); err == nil {
|
|
me.urlbase = url
|
|
return nil
|
|
}
|
|
url = fmt.Sprintf("https://%s/", zooname)
|
|
if err = testURL(url, me.machine); err == nil {
|
|
me.urlbase = url
|
|
return nil
|
|
}
|
|
url = "http://zookeeper.wit.com:8080/"
|
|
if err = testURL(url, me.machine); err == nil {
|
|
me.urlbase = url
|
|
return nil
|
|
}
|
|
url = "https://zookeeper.wit.com/"
|
|
if err = testURL(url, me.machine); err == nil {
|
|
me.urlbase = url
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
func findZookeeper() string {
|
|
hname, dname, err := hostname.GetDomainname()
|
|
if err != nil {
|
|
log.Info("error with os.Hostname()", hname, dname, err)
|
|
}
|
|
if dname == "" {
|
|
return "zookeeper.wit.com"
|
|
}
|
|
return hostname.Join("zookeeper", dname)
|
|
}
|
|
|
|
func testURL(urlbase string, pb *zoopb.Machine) error {
|
|
newpb, wsPB, err := pb.HttpPost(urlbase, "test")
|
|
if err != nil {
|
|
log.Info("got error:", err)
|
|
return err
|
|
}
|
|
wsPB.DumpClient()
|
|
newpb.Dump()
|
|
log.Info("got error:", err)
|
|
return err
|
|
}
|