From 02cc4a209790baf7ef7c0e0a0e683cfbad898f80 Mon Sep 17 00:00:00 2001
From: skiffer-git <44203734@qq.com>
Date: Sat, 11 May 2024 11:40:44 +0800
Subject: [PATCH] add etcd

---
 pkg/common/discoveryregister/etcd/etcd.go | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/pkg/common/discoveryregister/etcd/etcd.go b/pkg/common/discoveryregister/etcd/etcd.go
index e627dfedb..cc94f259a 100644
--- a/pkg/common/discoveryregister/etcd/etcd.go
+++ b/pkg/common/discoveryregister/etcd/etcd.go
@@ -95,8 +95,26 @@ func (r *SvcDiscoveryRegistryImpl) Register(serviceName, host string, port int,
 		return err
 	}
 
-	_, kaErr := r.client.KeepAlive(context.Background(), r.leaseID)
-	return kaErr
+	go r.keepAliveLease(r.leaseID)
+
+	return nil
+}
+
+func (r *SvcDiscoveryRegistryImpl) keepAliveLease(leaseID clientv3.LeaseID) {
+	ch, err := r.client.KeepAlive(context.Background(), leaseID)
+	if err != nil {
+		log.Printf("Failed to keep lease alive: %v", err)
+		return
+	}
+
+	for ka := range ch {
+		if ka != nil {
+			fmt.Printf("Received lease keep-alive response: %v", ka)
+		} else {
+			fmt.Printf("Lease keep-alive response channel closed")
+			break
+		}
+	}
 }
 
 func (r *SvcDiscoveryRegistryImpl) UnRegister() error {