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: zookeeper add auto re-register (#2235)

Browse Source 
* feat(registry): contrib/registry/zookeeper add re-register support

* feat(registry): contrib/registry/zookeeper add re-register support fix with lint
This commit is contained in:
林晓炜 2022-07-27 10:50:12 +08:00 committed by GitHub
parent 92b3c8f94a
commit 246d8d9c28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
contrib/registry/zookeeper/register.go
View File

@ -3,6 +3,7 @@ package zookeeper
import (
"context"
"path"
"time"
"github.com/go-zookeeper/zk"
"golang.org/x/sync/singleflight"
@ -77,6 +78,7 @@ func (r *Registry) Register(ctx context.Context, service *registry.ServiceInstan
if err = r.ensureName(servicePath, data, zk.FlagEphemeral); err != nil {
return err
}
go r.reRegister(servicePath, data)
return nil
}
@ -150,3 +152,21 @@ func (r *Registry) ensureName(path string, data []byte, flags int32) error {
}
return nil
}
// reRegister re-register data node info when bad connection recovered
func (r *Registry) reRegister(path string, data []byte) {
sessionID := r.conn.SessionID()
ticker := time.NewTicker(time.Second)
defer ticker.Stop()
for range ticker.C {
cur := r.conn.SessionID()
// sessionID changed
if cur > 0 && sessionID != cur {
// re-ensureName
if err := r.ensureName(path, data, zk.FlagEphemeral); err != nil {
return
}
sessionID = cur
}
}
}