func newProxy(cfg *config.Config) *route.Proxy {
if err := route.SetPickerStrategy(cfg.Proxy.Strategy); err != nil {
log.Fatal("[FATAL] ", err)
}
log.Printf("[INFO] Using routing strategy %q", cfg.Proxy.Strategy)
tr := &http.Transport{
ResponseHeaderTimeout: cfg.Proxy.ResponseHeaderTimeout,
MaxIdleConnsPerHost: cfg.Proxy.MaxConn,
Dial: (&net.Dialer{
Timeout: cfg.Proxy.DialTimeout,
KeepAlive: cfg.Proxy.KeepAliveTimeout,
}).Dial,
}
return route.NewProxy(tr, cfg.Proxy)
}
func TestGracefulShutdown(t *testing.T) {
req := func(url string) int {
resp, err := http.Get(url)
if err != nil {
t.Fatal(err)
}
defer resp.Body.Close()
return resp.StatusCode
}
// start a server which responds after the shutdown has been triggered.
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
<-quit // wait for shutdown signal
return
}))
defer srv.Close()
// load the routing table
tbl, err := route.ParseString("route add svc / " + srv.URL)
if err != nil {
t.Fatal(err)
}
route.SetTable(tbl)
// start proxy with graceful shutdown period long enough
// to complete one more request.
var wg sync.WaitGroup
l := config.Listen{Addr: "127.0.0.1:57777"}
wg.Add(1)
go func() {
defer wg.Done()
startListeners([]config.Listen{l}, 250*time.Millisecond, route.NewProxy(http.DefaultTransport, config.Proxy{}))
}()
// trigger shutdown after some time
shutdownDelay := 100 * time.Millisecond
go func() {
time.Sleep(shutdownDelay)
close(quit)
}()
// give proxy some time to start up
// needs to be done before shutdown is triggered
time.Sleep(shutdownDelay / 2)
// make 200 OK request
// start before and complete after shutdown was triggered
if got, want := req("http://"+l.Addr+"/"), 200; got != want {
t.Fatalf("request 1: got %v want %v", got, want)
}
// make 503 request
// start and complete after shutdown was triggered
if got, want := req("http://"+l.Addr+"/"), 503; got != want {
t.Fatalf("got %v want %v", got, want)
}
// wait for listen() to return
// note that the actual listeners have not returned yet
wg.Wait()
}
func main() {
var cfg string
var v bool
flag.StringVar(&cfg, "cfg", "", "path to config file")
flag.BoolVar(&v, "v", false, "show version")
flag.Parse()
if v {
fmt.Println(version)
return
}
log.Printf("[INFO] Version %s starting", version)
if cfg != "" {
if err := loadConfig(cfg); err != nil {
log.Fatal("[FATAL] ", err)
}
}
if err := metrics.Init(metricsTarget, metricsPrefix, metricsInterval, metricsGraphiteAddr); err != nil {
log.Fatal("[FATAL] ", err)
}
if os.Getenv("GOMAXPROCS") == "" {
log.Print("[INFO] Setting GOMAXPROCS=", gomaxprocs)
runtime.GOMAXPROCS(gomaxprocs)
} else {
log.Print("[INFO] Using GOMAXPROCS=", os.Getenv("GOMAXPROCS"), " from env")
}
if os.Getenv("GOGC") == "" {
log.Print("[INFO] Setting GOGC=", gogc)
debug.SetGCPercent(gogc)
} else {
log.Print("[INFO] Using GOGC=", os.Getenv("GOGC"), " from env")
}
if proxyRoutes == "" {
useDynamicRoutes()
} else {
useStaticRoutes()
}
if err := route.SetPickerStrategy(proxyStrategy); err != nil {
log.Fatal("[FATAL] ", err)
}
consul.Addr = consulAddr
consul.URL = consulURL
dc, err := consul.Datacenter()
if err != nil {
log.Fatal("[FATAL] ", err)
}
log.Printf("[INFO] Using routing strategy %q", proxyStrategy)
log.Printf("[INFO] Connecting to consul on %q in datacenter %q", consulAddr, dc)
log.Printf("[INFO] Consul can be reached via %q", consulURL)
log.Printf("[INFO] UI listening on %q", uiAddr)
go func() {
if err := ui.Start(uiAddr, consulKVPath); err != nil {
log.Fatal("[FATAL] ui: ", err)
}
}()
tr := &http.Transport{
ResponseHeaderTimeout: proxyTimeout,
MaxIdleConnsPerHost: proxyMaxConn,
Dial: (&net.Dialer{
Timeout: proxyDialTimeout,
KeepAlive: proxyTimeout,
}).Dial,
}
proxy := route.NewProxy(tr, proxyHeaderClientIP, proxyHeaderTLS, proxyHeaderTLSValue)
listen(proxyAddr, proxyShutdownWait, proxy)
}