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-01-18 22:09:54 +04:00
|
|
|
|
2014-07-07 07:33:48 +04:00
|
|
|
"github.com/coreos/etcd/config"
|
2014-03-24 21:55:37 +04:00
|
|
|
"github.com/coreos/etcd/etcd"
|
2014-07-09 20:57:05 +04:00
|
|
|
ehttp "github.com/coreos/etcd/http"
|
2014-07-06 21:19:23 +04:00
|
|
|
)
|
|
|
|
|
2013-06-09 21:42:34 +04:00
|
|
|
func main() {
|
2014-07-07 07:33:48 +04:00
|
|
|
var config = config.New()
|
|
|
|
if err := config.Load(os.Args[1:]); err != nil {
|
2014-07-11 23:42:08 +04:00
|
|
|
fmt.Println(etcd.Usage() + "\n")
|
2014-07-07 07:33:48 +04:00
|
|
|
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)
|
2013-10-10 01:26:49 +04:00
|
|
|
}
|
|
|
|
|
2014-07-22 03:45:03 +04:00
|
|
|
e := etcd.New(config)
|
2014-07-07 07:33:48 +04:00
|
|
|
go e.Run()
|
2014-07-06 21:19:23 +04:00
|
|
|
|
2014-07-09 20:57:05 +04:00
|
|
|
corsInfo, err := ehttp.NewCORSInfo(config.CorsOrigins)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("cors:", err)
|
|
|
|
}
|
|
|
|
|
2014-07-08 09:27:40 +04:00
|
|
|
go func() {
|
2014-07-09 20:57:05 +04:00
|
|
|
serve("raft", config.Peer.BindAddr, config.PeerTLSInfo(), corsInfo, e.RaftHandler())
|
2014-07-08 09:27:40 +04:00
|
|
|
}()
|
2014-07-09 20:57:05 +04:00
|
|
|
serve("etcd", config.BindAddr, config.EtcdTLSInfo(), corsInfo, e)
|
2014-07-06 21:19:23 +04:00
|
|
|
}
|
|
|
|
|
2014-07-09 20:57:05 +04:00
|
|
|
func serve(who string, addr string, tinfo *config.TLSInfo, cinfo *ehttp.CORSInfo, handler http.Handler) {
|
|
|
|
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-07-09 20:57:05 +04:00
|
|
|
h := &ehttp.CORSHandler{handler, cinfo}
|
|
|
|
log.Fatal(http.Serve(l, h))
|
2014-07-09 01:07:25 +04:00
|
|
|
}
|