Added deregister_critical_service_after=5s opt

consul/discovery.go

@@ -64,7 +64,8 @@ func (s *Service) Register() error {
 		Port:    s.port,
 		Tags:    s.getTags(),
 		Check: &consul.AgentServiceCheck{
-			TTL: s.ttl.String(),
+			TTL:                            s.ttl.String(),
+			DeregisterCriticalServiceAfter: "5s",
 		},
 	}
 
@@ -72,7 +73,7 @@ func (s *Service) Register() error {
 		return err
 	}
 
-	go func() { // startup register
+	go func(s *Service) { // startup register
 		ticker := time.NewTicker(time.Millisecond * 100)
 		for range ticker.C {
 			ok, _ := s.healthCheck()
@@ -80,9 +81,9 @@ func (s *Service) Register() error {
 				ticker.Stop()
 			}
 		}
-	}()
+	}(s)
 
-	go func() { // TTL
+	go func(s *Service) { // TTL
 		interval := s.ttl - time.Second*2
 		ticker := time.NewTicker(interval)
 		for range ticker.C {
@@ -91,7 +92,7 @@ func (s *Service) Register() error {
 				fmt.Printf("TTL Error: %v\n", err)
 			}
 		}
-	}()
+	}(s)
 
 	return nil
 }