Compare commits
No commits in common. "master" and "v0.0.91" have entirely different histories.
2
Makefile
2
Makefile
|
@ -9,7 +9,7 @@ REDOMOD = $(shell if [ -e go.sum ]; then echo go.sum exists; else GO111MODULE=
|
|||
|
||||
all: build
|
||||
./zood --version
|
||||
./zood test
|
||||
./zood
|
||||
|
||||
build: goimports
|
||||
GO111MODULE=off go build -v -x \
|
||||
|
|
2
README
2
README
|
@ -1,3 +1 @@
|
|||
This is the zookeeper daemon.
|
||||
|
||||
It is simple. It's just designed to keep packages up to date.
|
||||
|
|
4
argv.go
4
argv.go
|
@ -16,15 +16,11 @@ import (
|
|||
var argv args
|
||||
|
||||
type args struct {
|
||||
Test *EmptyCmd `arg:"subcommand:test" help:"New to forge? This is for you.'"`
|
||||
Daemon bool `arg:"--daemon" default:"false" help:"run in daemon mode"`
|
||||
Port int `arg:"--port" default:"2521" help:"port to run on"`
|
||||
URL string `arg:"--url" help:"url to use"`
|
||||
}
|
||||
|
||||
type EmptyCmd struct {
|
||||
}
|
||||
|
||||
func (args) Version() string {
|
||||
return "zood " + VERSION + " Built on: " + BUILDTIME
|
||||
}
|
||||
|
|
88
main.go
88
main.go
|
@ -5,12 +5,10 @@ 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"
|
||||
)
|
||||
|
@ -22,94 +20,30 @@ var BUILDTIME string
|
|||
var resources embed.FS
|
||||
|
||||
func main() {
|
||||
me = new(zoodStruct)
|
||||
me.pp = arg.MustParse(&argv)
|
||||
var pp *arg.Parser
|
||||
pp = arg.MustParse(&argv)
|
||||
|
||||
if me.pp == nil {
|
||||
me.pp.WriteHelp(os.Stdout)
|
||||
if pp == nil {
|
||||
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 = new(stuff)
|
||||
me.urlbase = "http://zookeeper.grid.wit.com:8080"
|
||||
if argv.URL != "" {
|
||||
log.Info("USING ARGV URL:", argv.URL)
|
||||
me.urlbase = argv.URL
|
||||
}
|
||||
|
||||
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)
|
||||
}
|
||||
me.machine = zoopb.InitMachine()
|
||||
|
||||
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
|
||||
}
|
||||
|
|
|
@ -6,14 +6,13 @@ package main
|
|||
import (
|
||||
"time"
|
||||
|
||||
"go.wit.com/dev/alexflint/arg"
|
||||
"go.wit.com/lib/protobuf/zoopb"
|
||||
)
|
||||
|
||||
var me *zoodStruct
|
||||
var me *stuff
|
||||
|
||||
// this app's variables
|
||||
type zoodStruct struct {
|
||||
type stuff struct {
|
||||
urlbase string // the dns name for the zookeeper
|
||||
hostname string // my hostname
|
||||
pollDelay time.Duration // how often to report our status
|
||||
|
@ -21,6 +20,4 @@ type zoodStruct struct {
|
|||
machine *zoopb.Machine // populated from protobuf based zoopb
|
||||
failcount int // how many times we've failed to contact the zookeeper
|
||||
failcountmax int // after this, exit and let systemd restart the daemon
|
||||
fullpath string // where to save the machine PB file
|
||||
pp *arg.Parser // from go-args
|
||||
}
|
||||
|
|
|
@ -16,7 +16,6 @@ func TimeFunction(f func()) time.Duration {
|
|||
return time.Since(startTime) // Calculate the elapsed time
|
||||
}
|
||||
|
||||
// sends updates to zookeeper
|
||||
func zood() {
|
||||
me.dog = time.NewTicker(me.pollDelay)
|
||||
defer me.dog.Stop()
|
Loading…
Reference in New Issue