2013-06-09 21:42:34 +04:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2014-07-09 00:49:22 +04:00
|
|
|
"crypto/tls"
|
2014-07-07 07:33:48 +04:00
|
|
|
"fmt"
|
2014-07-06 21:19:23 +04:00
|
|
|
"log"
|
2014-07-09 00:49:22 +04:00
|
|
|
"net"
|
2014-07-06 21:19:23 +04:00
|
|
|
"net/http"
|
2014-07-07 07:33:48 +04:00
|
|
|
"os"
|
2014-08-04 01:21:19 +04:00
|
|
|
"time"
|
2014-01-18 22:09:54 +04:00
|
|
|
|
2014-08-14 04:01:59 +04:00
|
|
|
"github.com/coreos/etcd/conf"
|
2014-08-23 03:46:56 +04:00
|
|
|
"github.com/coreos/etcd/etcdserver"
|
2014-07-06 21:19:23 +04:00
|
|
|
)
|
|
|
|
|
2013-06-09 21:42:34 +04:00
|
|
|
func main() {
|
2014-08-14 04:04:29 +04:00
|
|
|
var cfg = conf.New()
|
|
|
|
if err := cfg.Load(os.Args[1:]); err != nil {
|
2014-08-23 03:46:56 +04:00
|
|
|
fmt.Println(etcdserver.Usage() + "\n")
|
2014-07-07 07:33:48 +04:00
|
|
|
fmt.Println(err.Error(), "\n")
|
|
|
|
os.Exit(1)
|
2014-08-14 04:04:29 +04:00
|
|
|
} else if cfg.ShowVersion {
|
2014-07-07 07:33:48 +04:00
|
|
|
fmt.Println("0.5")
|
|
|
|
os.Exit(0)
|
2014-08-14 04:04:29 +04:00
|
|
|
} else if cfg.ShowHelp {
|
2014-07-07 07:33:48 +04:00
|
|
|
os.Exit(0)
|
2013-10-10 01:26:49 +04:00
|
|
|
}
|
|
|
|
|
2014-08-23 03:46:56 +04:00
|
|
|
e, err := etcdserver.New(cfg)
|
2014-08-04 01:21:19 +04:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal("etcd:", err)
|
|
|
|
}
|
2014-07-07 07:33:48 +04:00
|
|
|
go e.Run()
|
2014-07-06 21:19:23 +04:00
|
|
|
|
2014-08-14 04:04:29 +04:00
|
|
|
corsInfo, err := newCORSInfo(cfg.CorsOrigins)
|
2014-07-09 20:57:05 +04:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal("cors:", err)
|
|
|
|
}
|
|
|
|
|
2014-08-14 04:04:29 +04:00
|
|
|
readTimeout := time.Duration(cfg.HTTPReadTimeout) * time.Second
|
|
|
|
writeTimeout := time.Duration(cfg.HTTPWriteTimeout) * time.Second
|
2014-07-08 09:27:40 +04:00
|
|
|
go func() {
|
2014-08-14 04:04:29 +04:00
|
|
|
serve("raft", cfg.Peer.BindAddr, cfg.PeerTLSInfo(), corsInfo, e.RaftHandler(), readTimeout, writeTimeout)
|
2014-07-08 09:27:40 +04:00
|
|
|
}()
|
2014-08-14 04:04:29 +04:00
|
|
|
serve("etcd", cfg.BindAddr, cfg.EtcdTLSInfo(), corsInfo, e, readTimeout, writeTimeout)
|
2014-07-06 21:19:23 +04:00
|
|
|
}
|
|
|
|
|
2014-08-14 04:01:59 +04:00
|
|
|
func serve(who string, addr string, tinfo *conf.TLSInfo, cinfo *CORSInfo, handler http.Handler, readTimeout, writeTimeout time.Duration) {
|
2014-07-09 20:57:05 +04:00
|
|
|
t, terr := tinfo.ServerConfig()
|
2014-07-09 01:07:25 +04:00
|
|
|
l, err := net.Listen("tcp", addr)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
log.Printf("%v server starts listening on %v\n", who, addr)
|
|
|
|
|
2014-07-09 20:57:05 +04:00
|
|
|
switch tinfo.Scheme() {
|
2014-07-09 01:07:25 +04:00
|
|
|
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:
|
2014-07-09 20:57:05 +04:00
|
|
|
log.Fatal("unsupported http scheme", tinfo.Scheme())
|
2014-07-09 01:07:25 +04:00
|
|
|
}
|
|
|
|
|
2014-08-14 01:09:19 +04:00
|
|
|
h := &CORSHandler{handler, cinfo}
|
2014-08-04 01:21:19 +04:00
|
|
|
s := &http.Server{Handler: h, ReadTimeout: readTimeout, WriteTimeout: writeTimeout}
|
|
|
|
log.Fatal(s.Serve(l))
|
2014-07-09 01:07:25 +04:00
|
|
|
}
|