-
Notifications
You must be signed in to change notification settings - Fork 3.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
xds: Parsing xDS Cluster Metadata (#11741)
- Loading branch information
1 parent
4222f77
commit 1edc4d8
Showing
6 changed files
with
470 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,71 @@ | ||
| /* | ||
| * Copyright 2024 The gRPC Authors | ||
| * | ||
| * 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 io.grpc.xds; | ||
|
|
||
| import com.google.common.annotations.VisibleForTesting; | ||
| import com.google.protobuf.Any; | ||
| import com.google.protobuf.InvalidProtocolBufferException; | ||
| import io.grpc.xds.GcpAuthenticationFilter.AudienceMetadataParser; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
|
|
||
| /** | ||
| * Registry for parsing cluster metadata values. | ||
| * | ||
| * <p>This class maintains a mapping of type URLs to {@link MetadataValueParser} instances, | ||
| * allowing for the parsing of different metadata types. | ||
| */ | ||
| final class MetadataRegistry { | ||
| private static final MetadataRegistry INSTANCE = new MetadataRegistry(); | ||
|
|
||
| private final Map<String, MetadataValueParser> supportedParsers = new HashMap<>(); | ||
|
|
||
| private MetadataRegistry() { | ||
| registerParser(new AudienceMetadataParser()); | ||
| } | ||
|
|
||
| static MetadataRegistry getInstance() { | ||
| return INSTANCE; | ||
| } | ||
|
|
||
| MetadataValueParser findParser(String typeUrl) { | ||
| return supportedParsers.get(typeUrl); | ||
| } | ||
|
|
||
| @VisibleForTesting | ||
| void registerParser(MetadataValueParser parser) { | ||
| supportedParsers.put(parser.getTypeUrl(), parser); | ||
| } | ||
|
|
||
| void removeParser(MetadataValueParser parser) { | ||
| supportedParsers.remove(parser.getTypeUrl()); | ||
| } | ||
|
|
||
| interface MetadataValueParser { | ||
|
|
||
| String getTypeUrl(); | ||
|
|
||
| /** | ||
| * Parses the given {@link Any} object into a specific metadata value. | ||
| * | ||
| * @param any the {@link Any} object to parse. | ||
| * @return the parsed metadata value. | ||
| * @throws InvalidProtocolBufferException if the parsing fails. | ||
| */ | ||
| Object parse(Any any) throws InvalidProtocolBufferException; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
61 changes: 61 additions & 0 deletions
61
xds/src/main/java/io/grpc/xds/internal/ProtobufJsonConverter.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| /* | ||
| * Copyright 2024 The gRPC Authors | ||
| * | ||
| * 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 io.grpc.xds.internal; | ||
|
|
||
| import com.google.protobuf.Struct; | ||
| import com.google.protobuf.Value; | ||
| import io.grpc.Internal; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| /** | ||
| * Converter for Protobuf {@link Struct} to JSON-like {@link Map}. | ||
| */ | ||
| @Internal | ||
| public final class ProtobufJsonConverter { | ||
| private ProtobufJsonConverter() {} | ||
|
|
||
| public static Map<String, Object> convertToJson(Struct struct) { | ||
| Map<String, Object> result = new HashMap<>(); | ||
| for (Map.Entry<String, Value> entry : struct.getFieldsMap().entrySet()) { | ||
| result.put(entry.getKey(), convertValue(entry.getValue())); | ||
| } | ||
| return result; | ||
| } | ||
|
|
||
| private static Object convertValue(Value value) { | ||
| switch (value.getKindCase()) { | ||
| case STRUCT_VALUE: | ||
| return convertToJson(value.getStructValue()); | ||
| case LIST_VALUE: | ||
| return value.getListValue().getValuesList().stream() | ||
| .map(ProtobufJsonConverter::convertValue) | ||
| .collect(Collectors.toList()); | ||
| case NUMBER_VALUE: | ||
| return value.getNumberValue(); | ||
| case STRING_VALUE: | ||
| return value.getStringValue(); | ||
| case BOOL_VALUE: | ||
| return value.getBoolValue(); | ||
| case NULL_VALUE: | ||
| return null; | ||
| default: | ||
| throw new IllegalArgumentException("Unknown Value type: " + value.getKindCase()); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.