68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
|
|
"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 := etcd.New(config)
|
|
go e.Run()
|
|
|
|
corsInfo, err := ehttp.NewCORSInfo(config.CorsOrigins)
|
|
if err != nil {
|
|
log.Fatal("cors:", err)
|
|
}
|
|
|
|
go func() {
|
|
serve("raft", config.Peer.BindAddr, config.PeerTLSInfo(), corsInfo, e.RaftHandler())
|
|
}()
|
|
serve("etcd", config.BindAddr, config.EtcdTLSInfo(), corsInfo, e)
|
|
}
|
|
|
|
func serve(who string, addr string, tinfo *config.TLSInfo, cinfo *ehttp.CORSInfo, handler http.Handler) {
|
|
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}
|
|
log.Fatal(http.Serve(l, h))
|
|
}
|