Compare commits
9 Commits
v0.1.58
...
sh_24-04-1
| Author | SHA1 | Date | |
|---|---|---|---|
| 359a89ba62 | |||
| 3b73091cfb | |||
| c8beab76f1 | |||
| 599eaa3712 | |||
| 1de38fdfdf | |||
| 4d1f3644b0 | |||
| 71b3f13284 | |||
| c64bc3fe6b | |||
| 016bddb5c4 |
@@ -61,7 +61,12 @@ func (s *Service) GetID() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) GetFullAddr() string {
|
func (s *Service) GetFullAddr() string {
|
||||||
return fmt.Sprintf("https://%s:%d/", s.domain, s.port)
|
isTLS := s.port == 443
|
||||||
|
proto := "http"
|
||||||
|
if isTLS {
|
||||||
|
proto = "https"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%s://%s:%d/", proto, s.domain, s.port)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) Register() error {
|
func (s *Service) Register() error {
|
||||||
@@ -72,12 +77,7 @@ func (s *Service) Register() error {
|
|||||||
Address: s.Address,
|
Address: s.Address,
|
||||||
Port: s.port,
|
Port: s.port,
|
||||||
Tags: s.getTags(),
|
Tags: s.getTags(),
|
||||||
Connect: &consul.AgentServiceConnect{
|
// Connect: &consul.AgentServiceConnect{Native: true},
|
||||||
Native: true,
|
|
||||||
// SidecarService: &consul.AgentServiceRegistration{
|
|
||||||
// Port: s.port,
|
|
||||||
// },
|
|
||||||
},
|
|
||||||
// Proxy: &consul.AgentServiceConnectProxyConfig{
|
// Proxy: &consul.AgentServiceConnectProxyConfig{
|
||||||
// DestinationServiceName: s.Name,
|
// DestinationServiceName: s.Name,
|
||||||
// },
|
// },
|
||||||
@@ -92,38 +92,42 @@ func (s *Service) Register() error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
go func(s *Service) { // startup register
|
|
||||||
ticker := time.NewTicker(time.Second * 1)
|
|
||||||
for range ticker.C {
|
|
||||||
if ok, _ := s.healthCheck(); ok {
|
|
||||||
ticker.Stop()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}(s)
|
|
||||||
|
|
||||||
go func(s *Service) { // TTL
|
|
||||||
interval := s.ttl - time.Second*2 // 2 seconds overhead
|
|
||||||
ticker := time.NewTicker(interval)
|
|
||||||
for range ticker.C {
|
|
||||||
if _, err := s.healthCheck(); err != nil {
|
|
||||||
fmt.Printf("TTL Error #: %v\n", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}(s)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (s *Service) Unregister() error {
|
func (s *Service) Unregister() error {
|
||||||
return s.agent.ServiceDeregister(s.GetID())
|
return s.agent.ServiceDeregister(s.GetID())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *Service) RegisterHealthChecks() {
|
||||||
|
go func() { // startup register
|
||||||
|
ticker := time.NewTicker(time.Second * 1)
|
||||||
|
for range ticker.C {
|
||||||
|
if ok, _ := s.healthCheck(); ok {
|
||||||
|
ticker.Stop()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
go func() { // TTL
|
||||||
|
interval := s.ttl - (time.Second * 2) // 2 seconds overhead
|
||||||
|
ticker := time.NewTicker(interval)
|
||||||
|
for range ticker.C {
|
||||||
|
if _, err := s.healthCheck(); err != nil {
|
||||||
|
fmt.Printf("TTL Error #: %v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
func (s *Service) Connect() (*connect.Service, error) {
|
func (s *Service) Connect() (*connect.Service, error) {
|
||||||
// srvName := s.Name
|
// l := hclog.New(&hclog.LoggerOptions{
|
||||||
srvName := s.Name
|
// Name: "consul-registry",
|
||||||
svc, err := connect.NewService(srvName, s.client)
|
// Level: hclog.Trace,
|
||||||
|
// })
|
||||||
|
svc, err := connect.NewService(s.Name, s.client)
|
||||||
s.connect = svc
|
s.connect = svc
|
||||||
cnf := svc.ServerTLSConfig()
|
cnf := svc.ServerTLSConfig()
|
||||||
fmt.Printf("CONNECT SERVER:: %s CONFIG:: %v\n", srvName, cnf)
|
fmt.Printf("CONNECT SERVER:: %s CONFIG:: %v\n", s.Name, cnf)
|
||||||
for k, c := range cnf.Certificates {
|
for k, c := range cnf.Certificates {
|
||||||
fmt.Printf("CONNECT CERT %d: %v", k, c)
|
fmt.Printf("CONNECT CERT %d: %v", k, c)
|
||||||
}
|
}
|
||||||
@@ -145,17 +149,14 @@ func (s *Service) healthCheck() (bool, error) {
|
|||||||
}
|
}
|
||||||
req.Header.Set("User-Agent", "service/internal")
|
req.Header.Set("User-Agent", "service/internal")
|
||||||
|
|
||||||
fmt.Printf("Sending HEALTH CHECK request to: %s\n", healthUrl)
|
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Printf("Sending HEALTH CHECK request error: %v\n", err)
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
var body []byte
|
var body []byte
|
||||||
resp.Body.Read(body)
|
resp.Body.Read(body)
|
||||||
fmt.Printf("HEALTH CHECK response to: %v -- %v\n", resp, body)
|
|
||||||
|
|
||||||
return resp.StatusCode == http.StatusOK
|
return resp.StatusCode == http.StatusOK
|
||||||
}()
|
}()
|
||||||
|
|||||||
Reference in New Issue
Block a user