intercept kubelet get node request in order to reduce the traffic

openyurtio · May 6, 2024 · 18b6383 · 18b6383
1 parent 37addb5
commit 18b6383
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/pkg/yurthub/proxy/proxy.go b/pkg/yurthub/proxy/proxy.go
@@ -191,6 +191,12 @@ func (p *yurtReverseProxy) ServeHTTP(rw http.ResponseWriter, req *http.Request)
 	switch {
 	case util.IsKubeletLeaseReq(req):
 		p.handleKubeletLease(rw, req)
+	case util.IsKubeletGetNodeReq(req):
+		if p.localProxy != nil {
+			p.localProxy.ServeHTTP(rw, req)
+		} else {
+			p.loadBalancer.ServeHTTP(rw, req)
+		}
 	case util.IsEventCreateRequest(req):
 		p.eventHandler(rw, req)
 	case util.IsPoolScopedResouceListWatchRequest(req):

diff --git a/pkg/yurthub/proxy/util/util.go b/pkg/yurthub/proxy/util/util.go
@@ -415,6 +415,18 @@ func IsKubeletLeaseReq(req *http.Request) bool {
 	return true
 }
 
+// IsKubeletGetNodeReq judge whether the request is a get node request from kubelet
+func IsKubeletGetNodeReq(req *http.Request) bool {
+	ctx := req.Context()
+	if comp, ok := util.ClientComponentFrom(ctx); !ok || comp != "kubelet" {
+		return false
+	}
+	if info, ok := apirequest.RequestInfoFrom(ctx); !ok || info.Resource != "nodes" || info.Verb != "get" {
+		return false
+	}
+	return true
+}
+
 // WriteObject write object to response writer
 func WriteObject(statusCode int, obj runtime.Object, w http.ResponseWriter, req *http.Request) error {
 	ctx := req.Context()