Browse Source

feat:修改

wangchuanjin 3 years ago
parent
commit
63ac3377a4
1 changed files with 10 additions and 8 deletions
  1. 10 8
      core/node/register.go

+ 10 - 8
core/node/register.go

@@ -1,12 +1,13 @@
 package node
 package node
 
 
 import (
 import (
-	"app.yhyue.com/moapp/jybase/iputil"
 	"context"
 	"context"
 	"fmt"
 	"fmt"
-	client "go.etcd.io/etcd/client/v3"
 	"log"
 	"log"
 	"time"
 	"time"
+
+	"app.yhyue.com/moapp/jybase/iputil"
+	client "go.etcd.io/etcd/client/v3"
 )
 )
 
 
 // Register 注册服务
 // Register 注册服务
@@ -41,15 +42,16 @@ func (n *Node) RegisterWithContext(ctx context.Context, serverCode string, serve
 	}
 	}
 	log.Printf("service registed gateway for %s success!\n", key)
 	log.Printf("service registed gateway for %s success!\n", key)
 	//续租约
 	//续租约
-	_, err = n.Client.KeepAlive(ctx, lease.ID)
+	keepAlive, err = n.Client.KeepAlive(ctx, lease.ID)
 	if err != nil {
 	if err != nil {
 		return nil, err
 		return nil, err
 	}
 	}
-	//go func() {
-	//	for {
-	//		<-keepAlive
-	//	}
-	//}()
+	go func() {
+		for {
+			d := <-keepAlive
+			log.Println("------", d)
+		}
+	}()
 	return func() {
 	return func() {
 		_, _ = n.Client.Revoke(ctx, lease.ID)
 		_, _ = n.Client.Revoke(ctx, lease.ID)
 	}, nil
 	}, nil