From 1c89464ba6e26cd9b06bd6826dac1c64c57a6a20 Mon Sep 17 00:00:00 2001 From: yue9944882 <291271447@qq.com> Date: Mon, 17 Feb 2020 16:06:13 +0800 Subject: [PATCH] adding response headers Kubernetes-commit: da6f89217d357a6683ca4a7825c673d7c2af92f0 --- pkg/server/filters/priority-and-fairness.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pkg/server/filters/priority-and-fairness.go b/pkg/server/filters/priority-and-fairness.go index a815dc407..e0f4fe733 100644 --- a/pkg/server/filters/priority-and-fairness.go +++ b/pkg/server/filters/priority-and-fairness.go @@ -37,6 +37,11 @@ type priorityAndFairnessKeyType int const priorityAndFairnessKey priorityAndFairnessKeyType = iota +const ( + responseHeaderMatchedPriorityLevelConfigurationUID = "X-Kubernetes-PF-PriorityLevelUID" + responseHeaderMatchedFlowSchemaUID = "X-Kubernetes-PF-FlowSchemaUID" +) + // PriorityAndFairnessClassification identifies the results of // classification for API Priority and Fairness type PriorityAndFairnessClassification struct { @@ -97,6 +102,8 @@ func WithPriorityAndFairness( served = true innerCtx := context.WithValue(ctx, priorityAndFairnessKey, classification) innerReq := r.Clone(innerCtx) + w.Header().Set(responseHeaderMatchedPriorityLevelConfigurationUID, string(classification.PriorityLevelUID)) + w.Header().Set(responseHeaderMatchedFlowSchemaUID, string(classification.FlowSchemaUID)) handler.ServeHTTP(w, innerReq) } digest := utilflowcontrol.RequestDigest{requestInfo, user}