From 66a21a4d45d9ca7701f674610c5364c6bca10223 Mon Sep 17 00:00:00 2001
From: gaowei <gaowei3@qq.com>
Date: Wed, 21 Nov 2018 21:06:39 +0800
Subject: [PATCH 1/3] add etcd2 backend, support etcd cluster and tls

---
 etcd2/etcd2.go | 152 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 etcd2/etcd2.go

diff --git a/etcd2/etcd2.go b/etcd2/etcd2.go
new file mode 100644
index 000000000..37b4cc1be
--- /dev/null
+++ b/etcd2/etcd2.go
@@ -0,0 +1,152 @@
+package etcd
+
+import (
+	"log"
+	"net"
+	"net/http"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+	"context"
+	"net/url"
+
+	//"github.com/coreos/go-etcd/etcd"
+        "go.etcd.io/etcd/client"
+	"github.com/docker/go-connections/tlsconfig"
+	"github.com/gliderlabs/registrator/bridge"
+	"crypto/tls"
+)
+
+func init() {
+	bridge.Register(new(Factory), "etcd2")
+}
+
+type Factory struct{}
+
+func CreateEndpoints(addrs []string, scheme string) (entries []string) {
+	for _, addr := range addrs {
+		entries = append(entries, scheme+"://"+addr)
+	}
+	return entries
+}
+
+func setTLS(cfg *client.Config, tls *tls.Config, addrs []string) {
+	entries := CreateEndpoints(addrs, "https")
+	cfg.Endpoints = entries
+
+	// Set transport
+	t := http.Transport{
+		Dial: (&net.Dialer{
+			Timeout:   30 * time.Second,
+			KeepAlive: 30 * time.Second,
+		}).Dial,
+		TLSHandshakeTimeout: 10 * time.Second,
+		TLSClientConfig:     tls,
+	}
+
+	cfg.Transport = &t
+}
+
+func (f *Factory) New(uri *url.URL) bridge.RegistryAdapter {
+	cert := os.Getenv("ETCD_CERT_FILE")
+	key := os.Getenv("ETCD_KEY_FILE")
+	cacert := os.Getenv("ETCD_CA_CERT_FILE")
+	
+
+	var (
+		addrs = strings.Split(uri.Host, ",")
+		path = uri.Path
+                entries []string
+	)
+        entries = CreateEndpoints(addrs, "http")
+	cfg := &client.Config{
+		Endpoints:               entries,
+		Transport:               client.DefaultTransport,
+		// set timeout per request to fail fast when the target endpoint is unavailable
+		HeaderTimeoutPerRequest: time.Second,
+	}
+
+	// Assuming https
+	if cacert != "" {
+		if key != "" && cert != "" {
+			tlsConfig, err := tlsconfig.Client(tlsconfig.Options{
+				CAFile:   cacert,
+				CertFile: cert,
+				KeyFile:  key,
+			})
+			if err != nil {
+				log.Fatalf("etcd: tls config: %s", err)
+			}
+			setTLS(cfg, tlsConfig, addrs)
+		}
+	}
+	c, err := client.New(*cfg)
+	if err != nil {
+		log.Fatalf("etcd: failure to connect: %s", err)
+	}
+	kapi := client.NewKeysAPI(c)
+
+	return &EtcdAdapter{client: kapi, path: path}
+}
+
+type EtcdAdapter struct {
+	client client.KeysAPI
+
+	path string
+}
+
+func (r *EtcdAdapter) Ping() error {
+	var err error
+
+	getOpts := &client.GetOptions{
+		Quorum: true,
+		Sort: true, 
+		Recursive: true,
+	}
+
+	_, err = r.client.Get(context.Background(), "/", getOpts)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (r *EtcdAdapter) Register(service *bridge.Service) error {
+	setOpts := &client.SetOptions{}
+	path := r.path + "/" + service.Name + "/" + service.ID
+	port := strconv.Itoa(service.Port)
+	addr := net.JoinHostPort(service.IP, port)
+
+	var err error
+	setOpts.TTL = time.Duration(service.TTL) * time.Second	
+	_, err = r.client.Set(context.Background(), path, addr, setOpts)
+
+	if err != nil {
+		log.Println("etcd: failed to register service:", err)
+	}
+	return err
+}
+
+func (r *EtcdAdapter) Deregister(service *bridge.Service) error {
+	opts := &client.DeleteOptions{
+		Recursive: false,
+	}
+	path := r.path + "/" + service.Name + "/" + service.ID
+
+	var err error
+	_, err = r.client.Delete(context.Background(), path, opts)
+
+	if err != nil {
+		log.Println("etcd: failed to deregister service:", err)
+	}
+	return err
+}
+
+func (r *EtcdAdapter) Refresh(service *bridge.Service) error {
+	return r.Register(service)
+}
+
+func (r *EtcdAdapter) Services() ([]*bridge.Service, error) {
+	return []*bridge.Service{}, nil
+}

From 1f6ea14a82f08aaa26772187b897f6dd21c80ebf Mon Sep 17 00:00:00 2001
From: Gawain <gaowei3@qq.com>
Date: Wed, 21 Nov 2018 21:14:32 +0800
Subject: [PATCH 2/3] Update etcd2.go

remove space
---
 etcd2/etcd2.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/etcd2/etcd2.go b/etcd2/etcd2.go
index 37b4cc1be..1bb2eb735 100644
--- a/etcd2/etcd2.go
+++ b/etcd2/etcd2.go
@@ -12,7 +12,7 @@ import (
 	"net/url"
 
 	//"github.com/coreos/go-etcd/etcd"
-        "go.etcd.io/etcd/client"
+	"go.etcd.io/etcd/client"
 	"github.com/docker/go-connections/tlsconfig"
 	"github.com/gliderlabs/registrator/bridge"
 	"crypto/tls"

From 81b641196a7407527d49f144bc66e4640c667e5b Mon Sep 17 00:00:00 2001
From: Gawain <gaowei3@qq.com>
Date: Wed, 21 Nov 2018 21:18:36 +0800
Subject: [PATCH 3/3] Update modules.go

---
 modules.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modules.go b/modules.go
index 4ac74c313..6d35d94c6 100644
--- a/modules.go
+++ b/modules.go
@@ -4,6 +4,7 @@ import (
 	_ "github.com/gliderlabs/registrator/consul"
 	_ "github.com/gliderlabs/registrator/consulkv"
 	_ "github.com/gliderlabs/registrator/etcd"
+	_ "github.com/gliderlabs/registrator/etcd2"
 	_ "github.com/gliderlabs/registrator/skydns2"
 	_ "github.com/gliderlabs/registrator/zookeeper"
 )