attempt to discover zookeeper hostname

This commit is contained in:
Jeff Carr 2025-09-22 18:54:25 -05:00
parent 7f20ecc438
commit dd4ef5b61f
1 changed files with 53 additions and 5 deletions

58
main.go
View File

@ -5,10 +5,12 @@ package main
import ( import (
"embed" "embed"
"fmt"
"os" "os"
"time" "time"
"go.wit.com/dev/alexflint/arg" "go.wit.com/dev/alexflint/arg"
"go.wit.com/lib/hostname"
"go.wit.com/lib/protobuf/zoopb" "go.wit.com/lib/protobuf/zoopb"
"go.wit.com/log" "go.wit.com/log"
) )
@ -37,12 +39,12 @@ func main() {
} }
me.urlbase = "http://zookeeper.chap.wit.com:8080" if err := testZoo(); err != nil {
if argv.URL != "" { log.Info("FAILED TO CONNECT TO ZOOKEEPER: ", err)
log.Info("USING ARGV URL:", argv.URL) log.Info("sleeping for 3 minutes")
me.urlbase = argv.URL time.Sleep(3 * time.Minute)
os.Exit(0)
} }
me.pollDelay = 3 * time.Second me.pollDelay = 3 * time.Second
me.failcountmax = 20 // die every minute if zookeeper can't be found me.failcountmax = 20 // die every minute if zookeeper can't be found
@ -54,6 +56,52 @@ func main() {
zood() // talks to zookeeper 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 { func testURL(urlbase string, pb *zoopb.Machine) error {
newpb, wsPB, err := pb.HttpPost(urlbase, "test") newpb, wsPB, err := pb.HttpPost(urlbase, "test")
if err != nil { if err != nil {