added tls support...

consul/discovery.go

@@ -2,7 +2,6 @@ package consul
 
 import (
 	"fmt"
-	"log"
 	"net/http"
 	"strconv"
 	"time"
@@ -82,9 +81,13 @@ func (s *Service) Register() error {
 	}()
 
 	go func() { // TTL
-		ticker := time.NewTicker(s.TTL)
+		interval := s.TTL - time.Second*2
+		ticker := time.NewTicker(interval)
 		for range ticker.C {
-			s.healthCheck()
+			_, err := s.healthCheck()
+			if err != nil {
+				fmt.Printf("TTL Error: %v\n", err)
+			}
 		}
 	}()
 
@@ -115,13 +118,13 @@ func (s *Service) healthCheck() (bool, error) {
 
 	if alive {
 		if err := s.ConsulAgent.PassTTL("service:"+s.GetID(), "OK"); err != nil {
-			log.Println("health-check pass err:" + err.Error())
+			return false, err
 		}
 		return true, nil
 	}
 
 	if err := s.ConsulAgent.FailTTL("service:"+s.GetID(), ErrServiceUnavailable.Error()); err != nil {
-		log.Println("health-check fail err:" + err.Error())
+		return false, err
 	}
 	return false, ErrServiceUnavailable
 }
@@ -129,7 +132,7 @@ func (s *Service) healthCheck() (bool, error) {
 func (s *Service) getTags() []string {
 	tags := []string{
 		"traefik.enable=true",
-		"traefik.http.routers." + s.Name + ".rule=Host(`" + s.Domain + "`)",
+		"traefik.http.routers." + s.Name + ".rule=Headers(`X-API-SERVICE`, `" + s.Name + "`)",
 		"traefik.http.routers." + s.Name + ".entryPoints=https",
 		"traefik.http.routers." + s.Name + ".tls=true",
 		"traefik.http.routers." + s.Name + ".service=" + s.Name,
@@ -138,6 +141,8 @@ func (s *Service) getTags() []string {
 		"traefik.http.services." + s.Name + ".loadbalancer.server.port=" + strconv.Itoa(s.Port),
 		"traefik.http.middlewares.compress.compress=true",
 		"traefik.http.middlewares.requestid.plugin.requestid.headerName=X-Request-ID",
+		"traefik.tls.certificates.certfile=/certs/client.cert",
+		"traefik.tls.certificates.keyfile=/certs/client.key",
 		// "traefik.http.services." + s.Name + ".loadbalancer.passhostheader=false",
 		// "traefik.http.services." + s.Name + ".loadbalancer.servers." + fullName + "=" + bFullAddr,
 		// "traefik.http.services." + s.Name + ".loadbalancer.servers." + fullName + ".url=" + bFullAddr,