go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-03-17 21:07:54 +02:00
Code Issues Releases Activity

fix polaris registry bug: nodes is empty at startup (#3160)

Browse Source
This commit is contained in:
libi 2024-03-15 11:13:18 +08:00 committed by GitHub
parent 63344c3c97
commit 953a331de5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
contrib/registry/polaris/registry.go
View File

@ -332,6 +332,7 @@ type Watcher struct {
Cancel context.CancelFunc
Channel <-chan model.SubScribeEvent
ServiceInstances []*registry.ServiceInstance
first bool
}
func newWatcher(ctx context.Context, namespace string, serviceName string, consumer api.ConsumerAPI) (*Watcher, error) {
@ -350,6 +351,7 @@ func newWatcher(ctx context.Context, namespace string, serviceName string, consu
w := &Watcher{
Namespace: namespace,
ServiceName: serviceName,
first: true,
Channel: watchServiceResponse.EventChannel,
ServiceInstances: instancesToServiceInstances(watchServiceResponse.GetAllInstancesResp.GetInstances()),
}
@ -362,6 +364,10 @@ func newWatcher(ctx context.Context, namespace string, serviceName string, consu
// 2.any service instance changes found.
// if the above two conditions are not met, it will block until context deadline exceeded or canceled
func (w *Watcher) Next() ([]*registry.ServiceInstance, error) {
if w.first {
w.first = false
return w.ServiceInstances, nil
}
select {
case <-w.Ctx.Done():
return nil, w.Ctx.Err()