go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-01-14 02:33:03 +02:00
Code Issues Releases Activity

fix(registry/consul): use health check option to control custom checks (#2391)

Browse Source
This commit is contained in:
包子 2022-09-21 15:21:56 +08:00 committed by GitHub
parent 2d325fd386
commit 7866ff75fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
contrib/registry/consul/client.go
View File

@ -132,6 +132,8 @@ func (c *Client) Register(_ context.Context, svc *registry.ServiceInstance, enab
Timeout: "5s",
})
}
// custom checks
asr.Checks = append(asr.Checks, c.serviceChecks...)
}
if c.heartbeat {
asr.Checks = append(asr.Checks, &api.AgentServiceCheck{
@ -141,9 +143,6 @@ func (c *Client) Register(_ context.Context, svc *registry.ServiceInstance, enab
})
}
// custom checks
asr.Checks = append(asr.Checks, c.serviceChecks...)
err := c.cli.Agent().ServiceRegister(asr)
if err != nil {
return err