From 5182d46e22d2458b16f1f2fb0358f6b5f3e18b5d Mon Sep 17 00:00:00 2001 From: Aleksey Kosov Date: Wed, 21 May 2025 17:57:39 +0300 Subject: Added middleware for processing request_id grpc and http requests (#6805) --- weed/server/volume_server.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'weed/server/volume_server.go') diff --git a/weed/server/volume_server.go b/weed/server/volume_server.go index a3f072eb7..5c5ebc49a 100644 --- a/weed/server/volume_server.go +++ b/weed/server/volume_server.go @@ -115,22 +115,22 @@ func NewVolumeServer(adminMux, publicMux *http.ServeMux, ip string, vs.guard = security.NewGuard(whiteList, signingKey, expiresAfterSec, readSigningKey, readExpiresAfterSec) handleStaticResources(adminMux) - adminMux.HandleFunc("/status", vs.statusHandler) - adminMux.HandleFunc("/healthz", vs.healthzHandler) + adminMux.HandleFunc("/status", requestIDMiddleware(vs.statusHandler)) + adminMux.HandleFunc("/healthz", requestIDMiddleware(vs.healthzHandler)) if signingKey == "" || enableUiAccess { // only expose the volume server details for safe environments - adminMux.HandleFunc("/ui/index.html", vs.uiStatusHandler) + adminMux.HandleFunc("/ui/index.html", requestIDMiddleware(vs.uiStatusHandler)) /* adminMux.HandleFunc("/stats/counter", vs.guard.WhiteList(statsCounterHandler)) adminMux.HandleFunc("/stats/memory", vs.guard.WhiteList(statsMemoryHandler)) adminMux.HandleFunc("/stats/disk", vs.guard.WhiteList(vs.statsDiskHandler)) */ } - adminMux.HandleFunc("/", vs.privateStoreHandler) + adminMux.HandleFunc("/", requestIDMiddleware(vs.privateStoreHandler)) if publicMux != adminMux { // separated admin and public port handleStaticResources(publicMux) - publicMux.HandleFunc("/", vs.publicReadOnlyHandler) + publicMux.HandleFunc("/", requestIDMiddleware(vs.publicReadOnlyHandler)) } go vs.heartbeat() -- cgit v1.2.3