| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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. |
| */ |
| |
| /* |
| * |
| * Copyright 2021 gRPC authors. |
| * |
| */ |
| |
| package xds |
| |
| import ( |
| "google.golang.org/grpc/attributes" |
| |
| "google.golang.org/grpc/resolver" |
| ) |
| |
| // handshakeClusterNameKey is the type used as the key to store cluster name in |
| // the Attributes field of resolver.Address. |
| type handshakeClusterNameKey struct{} |
| |
| // SetXDSHandshakeClusterName returns a copy of addr in which the Attributes field |
| // is updated with the cluster name. |
| func SetXDSHandshakeClusterName(addr resolver.Address, clusterName string) resolver.Address { |
| addr.Attributes = addr.Attributes.WithValue(handshakeClusterNameKey{}, clusterName) |
| return addr |
| } |
| |
| // GetXDSHandshakeClusterName returns cluster name stored in attr. |
| func GetXDSHandshakeClusterName(attr *attributes.Attributes) (string, bool) { |
| v := attr.Value(handshakeClusterNameKey{}) |
| name, ok := v.(string) |
| return name, ok |
| } |