egommerce/go-api-pkg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix in consul discovery

Browse Source
This commit is contained in:
Piotr Biernat 2024-07-19 16:39:11 +02:00
parent 37fa05402e
commit 767eb5688c
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
consul/discovery.go
View File

@ -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")
@ -104,25 +107,27 @@ func (s *Service) Unregister() error {
ServiceID: s.GetID(), ServiceID: s.GetID(),
}, nil) }, nil)
s.ttlTicker.Stop()
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) 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
ticker := time.NewTicker(s.ttl) s.ttlTicker = time.NewTicker(s.ttl)
for range ticker.C { for range s.ttlTicker.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)
ticker.Stop() // s.ttlTicker.Stop()
} }
} }
}() }()