logos-messaging/go-libp2p-rendezvous
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-rendezvous.git synced 2026-08-09 15:13:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

user parameter for ttl in Register

Browse Source
This commit is contained in:
vyzo 2018-04-23 14:50:46 +03:00
parent 4e3eaa7ba0
commit aa3f46c897
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
client.go
View File

@ -73,27 +73,31 @@ func (cli *client) Register(ctx context.Context, ns string, ttl int) error {
return nil
}
func Register(ctx context.Context, rz Rendezvous, ns string) error {
err := rz.Register(ctx, ns, DefaultTTL)
func Register(ctx context.Context, rz Rendezvous, ns string, ttl int) error {
if ttl < 120 {
return fmt.Errorf("registration TTL is too short")
}
err := rz.Register(ctx, ns, ttl)
if err != nil {
return err
}
go registerRefresh(ctx, rz, ns)
go registerRefresh(ctx, rz, ns, ttl)
return nil
}
func registerRefresh(ctx context.Context, rz Rendezvous, ns string) {
const refresh = DefaultTTL - 30
func registerRefresh(ctx context.Context, rz Rendezvous, ns string, ttl int) {
refresh := time.Duration(ttl-30) * time.Second
for {
select {
case <-time.After(refresh * time.Second):
case <-time.After(refresh):
case <-ctx.Done():
return
}
err := rz.Register(ctx, ns, DefaultTTL)
err := rz.Register(ctx, ns, ttl)
if err != nil {
log.Errorf("Error registering [%s]: %s", ns, err.Error())
}