Merge pull request #88149 from yue9944882/feat/flow-control-ob

Flowcontrol Obserbiliity: Add FS/PL UIDs into response headers

Kubernetes-commit: afa72e93cd55ec4f873b8d218054fc8b5f060e02
This commit is contained in:
Kubernetes Publisher 2020-02-20 12:31:58 -08:00
commit ab5dd25a00
1 changed files with 7 additions and 0 deletions

View File

@ -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}