From 5744bd10c233b753084616410eee4fedfc838418 Mon Sep 17 00:00:00 2001
From: askuy <askuygo@foxmail.com>
Date: Wed, 26 May 2021 17:28:21 +0800
Subject: [PATCH] optimize k8s endpoints watch

---
 ek8s/examples/kubegrpc/config.toml | 10 +++-------
 ek8s/watcher_app.go                | 10 +++++-----
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/ek8s/examples/kubegrpc/config.toml b/ek8s/examples/kubegrpc/config.toml
index 39bc308..83b0245 100644
--- a/ek8s/examples/kubegrpc/config.toml
+++ b/ek8s/examples/kubegrpc/config.toml
@@ -1,16 +1,12 @@
 [k8s]
-addr=""
-token=""
 namespaces=["default"]
 
-[registry]
-scheme = "k8s" # grpc resolver默认scheme为"k8s"，你可以自行修改
-
 [grpc.test]
 debug = true # 开启后并加上export EGO_DEBUG=true，可以看到每次grpc请求，配置名、地址、耗时、请求数据、响应数据
 addr = "k8s:///test:9090"
 #balancerName = "round_robin" # 默认值
 #dialTimeout = "1s" # 默认值
 #enableAccessInterceptor = true
-#enableAccessInterceptorReply = true
-#enableAccessInterceptorReq = true
\ No newline at end of file
+#enableAccessInterceptorRes = true
+#enableAccessInterceptorReq = true
+
diff --git a/ek8s/watcher_app.go b/ek8s/watcher_app.go
index 063f347..ac0ab4c 100644
--- a/ek8s/watcher_app.go
+++ b/ek8s/watcher_app.go
@@ -54,7 +54,7 @@ func (c *WatcherApp) watch(ctx context.Context, ns string) error {
 		)
 
 		informer := informersFactory.Core().V1().Pods()
-		c.logger.Debug("k8s watch prefix", zap.String("appname", c.appName), zap.String("kind", c.kind), zap.String("kind", c.kind))
+		c.logger.Debug("k8s watch pods", zap.String("appname", c.appName), zap.String("kind", c.kind), zap.String("kind", c.kind))
 		informer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
 			AddFunc:    c.addPod,
 			UpdateFunc: c.updatePod,
@@ -63,24 +63,24 @@ func (c *WatcherApp) watch(ctx context.Context, ns string) error {
 		// 启动该命名空间里监听
 		go informersFactory.Start(ctx.Done())
 	case KindEndpoints:
-		label, err := c.getServicesSelector(ns, c.appName)
+		endPoints, err := c.CoreV1().Endpoints(ns).Get(context.Background(), c.getDeploymentName(c.appName), metav1.GetOptions{})
 		if err != nil {
 			return err
 		}
-		c.logger.Debug("watch prefix label", zap.String("appname", c.appName), zap.String("label", label), zap.String("kind", c.kind))
+
+		c.logger.Info("k8s watch endpoints", zap.String("appname", c.appName), zap.String("namespace", endPoints.Namespace), zap.String("endPointName", endPoints.Name), zap.String("kind", c.kind))
 		informersFactory := informers.NewSharedInformerFactoryWithOptions(
 			c.Clientset,
 			defaultResync,
 			informers.WithNamespace(ns),
 			informers.WithTweakListOptions(func(options *metav1.ListOptions) {
-				options.LabelSelector = label
+				options.FieldSelector = "metadata.name=" + endPoints.Name
 				// todo
 				options.ResourceVersion = "0"
 			}),
 		)
 
 		informer := informersFactory.Core().V1().Endpoints()
-		c.logger.Debug("k8s watch prefix", zap.String("appname", c.appName), zap.String("kind", c.kind))
 		informer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
 			AddFunc:    c.addEndpoints,
 			UpdateFunc: c.updateEndpoints,