context.go

// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this
// file except in compliance with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software distributed under
// the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
// ANY KIND, either express or implied. See the License for the specific language
// governing permissions and limitations under the License.

package okms

import "context"

type ctxHeaderKey struct{}

// WithContextHeader adds some custom headers to the request's context.
func WithContextHeader(ctx context.Context, key, value string) context.Context {
	hdrs, ok := ctx.Value(ctxHeaderKey{}).(map[string]string)
	if !ok || hdrs == nil {
		hdrs = make(map[string]string)
		ctx = context.WithValue(ctx, ctxHeaderKey{}, hdrs)
	}
	hdrs[key] = value
	return ctx
}

func getContextHeaders(ctx context.Context) map[string]string {
	hdrs, _ := ctx.Value(ctxHeaderKey{}).(map[string]string)
	return hdrs
}