bgpcommander.go

package main

import (
	"bufio"
	"flag"
	"github.com/coreos/go-etcd/etcd"
	"github.com/crazed/bgpcommander/node_state"
	"os"
	"os/signal"
	"strings"
	"syscall"
)

func readStdin(state *node_state.NodeState) {
	scanner := bufio.NewScanner(os.Stdin)
	for scanner.Scan() {
		state.ProcessExaBGPOutput(scanner.Bytes())
	}
}

func handleSignal(signals chan os.Signal, state *node_state.NodeState) {
	<-signals
	state.Shutdown()
	os.Exit(0)
}

type machines []string

func (m *machines) String() string {
	return strings.Join(*m, ", ")
}

func (m *machines) Set(value string) error {
	for _, machine := range strings.Split(value, ",") {
		*m = append(*m, strings.TrimSpace(machine))
	}
	return nil
}

func main() {
	cluster := machines{"http://127.0.0.1:4001"}
	healthcheckScriptPath := "/tmp"
	hostname, _ := os.Hostname()

	flag.Var(&cluster, "c", "Comma separated list of etcd cluster members")
	flag.StringVar(&healthcheckScriptPath, "p", healthcheckScriptPath, "Path to store healthcheck scripts")
	flag.StringVar(&hostname, "n", hostname, "Override node's name")
	flag.Parse()

	client := etcd.NewClient(cluster)

	state := node_state.NewNodeState(hostname, client, true, healthcheckScriptPath)
	state.Logger.Println("Using etcd servers:", cluster.String())
	state.Logger.Println("Using node name:", hostname)
	state.Logger.Println("Using healthcheck script path:", healthcheckScriptPath)

	signals := make(chan os.Signal, 1)
	signal.Notify(signals, syscall.SIGTERM, syscall.SIGINT)
	go readStdin(state)
	go handleSignal(signals, state)
	state.WatchKeys()

	// goroutines are doing the work, so block here
	select {}
}