etcd/main.go
Yicheng Qin 4f10917ce9 etcd: add a read/write timeout to server
The default is for connections to last forever[1]. This leads to fds
leaking. I set the timeout so high by default so that watches don't have
to keep retrying but perhaps we should set it slower.

Tested on a cluster with lots of clients and it seems to have relieved
the problem.

[1] https://groups.google.com/forum/#!msg/golang-nuts/JFhGwh1q9xU/heh4J8pul3QJ

Original Commit: 084dcb5596d52a3a1386d59acced24286137e385
From: philips <brandon@ifup.org>
2014-09-03 09:20:02 -07:00

75 lines
1.8 KiB
Go

package main
import (
"crypto/tls"
"fmt"
"log"
"net"
"net/http"
"os"
"time"
"github.com/coreos/etcd/config"
"github.com/coreos/etcd/etcd"
ehttp "github.com/coreos/etcd/http"
)
func main() {
var config = config.New()
if err := config.Load(os.Args[1:]); err != nil {
fmt.Println(etcd.Usage() + "\n")
fmt.Println(err.Error(), "\n")
os.Exit(1)
} else if config.ShowVersion {
fmt.Println("0.5")
os.Exit(0)
} else if config.ShowHelp {
os.Exit(0)
}
e, err := etcd.New(config)
if err != nil {
log.Fatal("etcd:", err)
}
go e.Run()
corsInfo, err := ehttp.NewCORSInfo(config.CorsOrigins)
if err != nil {
log.Fatal("cors:", err)
}
readTimeout := time.Duration(config.HTTPReadTimeout) * time.Second
writeTimeout := time.Duration(config.HTTPWriteTimeout) * time.Second
go func() {
serve("raft", config.Peer.BindAddr, config.PeerTLSInfo(), corsInfo, e.RaftHandler(), readTimeout, writeTimeout)
}()
serve("etcd", config.BindAddr, config.EtcdTLSInfo(), corsInfo, e, readTimeout, writeTimeout)
}
func serve(who string, addr string, tinfo *config.TLSInfo, cinfo *ehttp.CORSInfo, handler http.Handler, readTimeout, writeTimeout time.Duration) {
t, terr := tinfo.ServerConfig()
l, err := net.Listen("tcp", addr)
if err != nil {
log.Fatal(err)
}
log.Printf("%v server starts listening on %v\n", who, addr)
switch tinfo.Scheme() {
case "http":
log.Printf("%v server starts serving HTTP\n", who)
case "https":
if t == nil {
log.Fatalf("failed to create %v tls: %v\n", who, terr)
}
l = tls.NewListener(l, t)
log.Printf("%v server starts serving HTTPS\n", who)
default:
log.Fatal("unsupported http scheme", tinfo.Scheme())
}
h := &ehttp.CORSHandler{handler, cinfo}
s := &http.Server{Handler: h, ReadTimeout: readTimeout, WriteTimeout: writeTimeout}
log.Fatal(s.Serve(l))
}