Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 6537d79b19 | |||
| e55a2f42fe | |||
| 82d68e91b8 | |||
| 767eb5688c | |||
| 37fa05402e | |||
| 615281ae9a |
@@ -23,6 +23,9 @@ type Service struct {
|
|||||||
agent *consul.Agent
|
agent *consul.Agent
|
||||||
connect *connect.Service
|
connect *connect.Service
|
||||||
kv *consul.KV
|
kv *consul.KV
|
||||||
|
|
||||||
|
hcTicker *time.Ticker
|
||||||
|
ttlTicker *time.Ticker
|
||||||
}
|
}
|
||||||
|
|
||||||
var ErrServiceUnavailable = fmt.Errorf("Service is unavailable")
|
var ErrServiceUnavailable = fmt.Errorf("Service is unavailable")
|
||||||
@@ -99,30 +102,33 @@ func (s *Service) Register() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (s *Service) Unregister() error {
|
func (s *Service) Unregister() error {
|
||||||
// s.client.Catalog().Deregister(&consul.CatalogDeregistration{
|
// s.ttlTicker.Stop()
|
||||||
// Address: s.Address,
|
// s.hcTicker.Stop()
|
||||||
// ServiceID: s.GetID(),
|
|
||||||
// }, nil)
|
s.client.Catalog().Deregister(&consul.CatalogDeregistration{
|
||||||
|
Address: s.Address,
|
||||||
|
ServiceID: s.GetID(),
|
||||||
|
}, nil)
|
||||||
|
|
||||||
return s.agent.ServiceDeregister(s.GetID())
|
return s.agent.ServiceDeregister(s.GetID())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) RegisterHealthChecks() {
|
func (s *Service) RegisterHealthChecks() {
|
||||||
go func() { // startup register
|
go func() { // startup register
|
||||||
ticker := time.NewTicker(time.Second * 1)
|
s.hcTicker = time.NewTicker(time.Second)
|
||||||
for range ticker.C {
|
for range s.hcTicker.C {
|
||||||
if ok, _ := s.healthCheck(); ok {
|
if ok, _ := s.healthCheck(); ok {
|
||||||
ticker.Stop()
|
s.hcTicker.Stop()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
go func() { // TTL
|
go func() { // TTL
|
||||||
interval := s.ttl - (time.Second * 2) // 2 seconds overhead
|
s.ttlTicker = time.NewTicker(s.ttl)
|
||||||
ticker := time.NewTicker(interval)
|
for range s.ttlTicker.C {
|
||||||
for range ticker.C {
|
|
||||||
if _, err := s.healthCheck(); err != nil {
|
if _, err := s.healthCheck(); err != nil {
|
||||||
fmt.Printf("HealthCheck endpoint not available (%s)#: %v\n", s.GetFullAddr(), err)
|
fmt.Printf("HealthCheck endpoint not available (%s)#: %v\n", s.GetFullAddr(), err)
|
||||||
|
s.ttlTicker.Stop()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|||||||
Reference in New Issue
Block a user