github.com/kyma-incubator/compass/components/director@v0.0.0-20230623144113-d764f56ff805/pkg/tenant/tenant.go (about) 1 /* 2 * Copyright 2020 The Compass Authors 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package tenant 18 19 import ( 20 "context" 21 22 "github.com/kyma-incubator/compass/components/director/pkg/apperrors" 23 ) 24 25 type contextKey string 26 27 // ContextKey missing godoc 28 const ContextKey contextKey = "TenantCtxKey" 29 30 // LoadFromContext retrieves the tenantID from the provided context or returns error if missing 31 func LoadFromContext(ctx context.Context) (string, error) { 32 tenantID, ok := ctx.Value(ContextKey).(string) 33 34 if !ok { 35 return "", apperrors.NewCannotReadTenantError() 36 } 37 38 if tenantID == "" { 39 return "", apperrors.NewTenantRequiredError() 40 } 41 42 return tenantID, nil 43 } 44 45 // SaveToContext saves the provided tenantID into the respective context 46 func SaveToContext(ctx context.Context, tenantID string) context.Context { 47 return context.WithValue(ctx, ContextKey, tenantID) 48 }