Examples with PluginConfig io.grpc.xds.ClusterSpecifierPlugin.PluginConfig used on opensource projects

Example 1 with PluginConfig

use of io.grpc.xds.ClusterSpecifierPlugin.PluginConfig in project grpc-java by grpc.

the class ClientXdsClientDataTest method parseClusterSpecifierPlugin_typedStructInTypedExtension.

@Test
public void parseClusterSpecifierPlugin_typedStructInTypedExtension() throws Exception {
    class TestPluginConfig implements PluginConfig {

        @Override
        public String typeUrl() {
            return "type.googleapis.com/google.protobuf.Empty";
        }
    }
    ClusterSpecifierPluginRegistry registry = ClusterSpecifierPluginRegistry.newRegistry();
    registry.register(new ClusterSpecifierPlugin() {

        @Override
        public String[] typeUrls() {
            return new String[] { "type.googleapis.com/google.protobuf.Empty" };
        }

        @Override
        public ConfigOrError<? extends PluginConfig> parsePlugin(Message rawProtoMessage) {
            return ConfigOrError.fromConfig(new TestPluginConfig());
        }
    });
    TypedStruct typedStruct = TypedStruct.newBuilder().setTypeUrl("type.googleapis.com/google.protobuf.Empty").setValue(Struct.newBuilder()).build();
    io.envoyproxy.envoy.config.route.v3.ClusterSpecifierPlugin pluginProto = io.envoyproxy.envoy.config.route.v3.ClusterSpecifierPlugin.newBuilder().setExtension(TypedExtensionConfig.newBuilder().setTypedConfig(Any.pack(typedStruct))).build();
    PluginConfig pluginConfig = ClientXdsClient.parseClusterSpecifierPlugin(pluginProto, registry);
    assertThat(pluginConfig).isInstanceOf(TestPluginConfig.class);
}

Example 2 with PluginConfig

use of io.grpc.xds.ClusterSpecifierPlugin.PluginConfig in project grpc-java by grpc.

the class ClientXdsClientDataTest method parseClusterSpecifierPlugin_v3TypedStructInTypedExtension.

@Test
public void parseClusterSpecifierPlugin_v3TypedStructInTypedExtension() throws Exception {
    class TestPluginConfig implements PluginConfig {

        @Override
        public String typeUrl() {
            return "type.googleapis.com/google.protobuf.Empty";
        }
    }
    ClusterSpecifierPluginRegistry registry = ClusterSpecifierPluginRegistry.newRegistry();
    registry.register(new ClusterSpecifierPlugin() {

        @Override
        public String[] typeUrls() {
            return new String[] { "type.googleapis.com/google.protobuf.Empty" };
        }

        @Override
        public ConfigOrError<? extends PluginConfig> parsePlugin(Message rawProtoMessage) {
            return ConfigOrError.fromConfig(new TestPluginConfig());
        }
    });
    com.github.xds.type.v3.TypedStruct typedStruct = com.github.xds.type.v3.TypedStruct.newBuilder().setTypeUrl("type.googleapis.com/google.protobuf.Empty").setValue(Struct.newBuilder()).build();
    io.envoyproxy.envoy.config.route.v3.ClusterSpecifierPlugin pluginProto = io.envoyproxy.envoy.config.route.v3.ClusterSpecifierPlugin.newBuilder().setExtension(TypedExtensionConfig.newBuilder().setTypedConfig(Any.pack(typedStruct))).build();
    PluginConfig pluginConfig = ClientXdsClient.parseClusterSpecifierPlugin(pluginProto, registry);
    assertThat(pluginConfig).isInstanceOf(TestPluginConfig.class);
}

Example 3 with PluginConfig

use of io.grpc.xds.ClusterSpecifierPlugin.PluginConfig in project grpc-java by grpc.

the class ClientXdsClient method extractVirtualHosts.

private static List<VirtualHost> extractVirtualHosts(RouteConfiguration routeConfig, FilterRegistry filterRegistry, boolean parseHttpFilter) throws ResourceInvalidException {
    Map<String, PluginConfig> pluginConfigMap = new HashMap<>();
    if (enableRouteLookup) {
        List<ClusterSpecifierPlugin> plugins = routeConfig.getClusterSpecifierPluginsList();
        for (ClusterSpecifierPlugin plugin : plugins) {
            PluginConfig existing = pluginConfigMap.put(plugin.getExtension().getName(), parseClusterSpecifierPlugin(plugin));
            if (existing != null) {
                throw new ResourceInvalidException("Multiple ClusterSpecifierPlugins with the same name: " + plugin.getExtension().getName());
            }
        }
    }
    List<VirtualHost> virtualHosts = new ArrayList<>(routeConfig.getVirtualHostsCount());
    for (io.envoyproxy.envoy.config.route.v3.VirtualHost virtualHostProto : routeConfig.getVirtualHostsList()) {
        StructOrError<VirtualHost> virtualHost = parseVirtualHost(virtualHostProto, filterRegistry, parseHttpFilter, pluginConfigMap);
        if (virtualHost.getErrorDetail() != null) {
            throw new ResourceInvalidException("RouteConfiguration contains invalid virtual host: " + virtualHost.getErrorDetail());
        }
        virtualHosts.add(virtualHost.getStruct());
    }
    return virtualHosts;
}

Example 4 with PluginConfig

use of io.grpc.xds.ClusterSpecifierPlugin.PluginConfig in project grpc-java by grpc.

the class ClientXdsClient method parseRouteAction.

/**
 * Parses the RouteAction config. The returned result may contain a (parsed form)
 * {@link RouteAction} or an error message. Returns {@code null} if the RouteAction
 * should be ignored.
 */
@VisibleForTesting
@Nullable
static StructOrError<RouteAction> parseRouteAction(io.envoyproxy.envoy.config.route.v3.RouteAction proto, FilterRegistry filterRegistry, boolean parseHttpFilter, Map<String, PluginConfig> pluginConfigMap) {
    Long timeoutNano = null;
    if (proto.hasMaxStreamDuration()) {
        io.envoyproxy.envoy.config.route.v3.RouteAction.MaxStreamDuration maxStreamDuration = proto.getMaxStreamDuration();
        if (maxStreamDuration.hasGrpcTimeoutHeaderMax()) {
            timeoutNano = Durations.toNanos(maxStreamDuration.getGrpcTimeoutHeaderMax());
        } else if (maxStreamDuration.hasMaxStreamDuration()) {
            timeoutNano = Durations.toNanos(maxStreamDuration.getMaxStreamDuration());
        }
    }
    RetryPolicy retryPolicy = null;
    if (enableRetry && proto.hasRetryPolicy()) {
        StructOrError<RetryPolicy> retryPolicyOrError = parseRetryPolicy(proto.getRetryPolicy());
        if (retryPolicyOrError != null) {
            if (retryPolicyOrError.errorDetail != null) {
                return StructOrError.fromError(retryPolicyOrError.errorDetail);
            }
            retryPolicy = retryPolicyOrError.struct;
        }
    }
    List<HashPolicy> hashPolicies = new ArrayList<>();
    for (io.envoyproxy.envoy.config.route.v3.RouteAction.HashPolicy config : proto.getHashPolicyList()) {
        HashPolicy policy = null;
        boolean terminal = config.getTerminal();
        switch(config.getPolicySpecifierCase()) {
            case HEADER:
                io.envoyproxy.envoy.config.route.v3.RouteAction.HashPolicy.Header headerCfg = config.getHeader();
                Pattern regEx = null;
                String regExSubstitute = null;
                if (headerCfg.hasRegexRewrite() && headerCfg.getRegexRewrite().hasPattern() && headerCfg.getRegexRewrite().getPattern().hasGoogleRe2()) {
                    regEx = Pattern.compile(headerCfg.getRegexRewrite().getPattern().getRegex());
                    regExSubstitute = headerCfg.getRegexRewrite().getSubstitution();
                }
                policy = HashPolicy.forHeader(terminal, headerCfg.getHeaderName(), regEx, regExSubstitute);
                break;
            case FILTER_STATE:
                if (config.getFilterState().getKey().equals(HASH_POLICY_FILTER_STATE_KEY)) {
                    policy = HashPolicy.forChannelId(terminal);
                }
                break;
            default:
        }
        if (policy != null) {
            hashPolicies.add(policy);
        }
    }
    switch(proto.getClusterSpecifierCase()) {
        case CLUSTER:
            return StructOrError.fromStruct(RouteAction.forCluster(proto.getCluster(), hashPolicies, timeoutNano, retryPolicy));
        case CLUSTER_HEADER:
            return null;
        case WEIGHTED_CLUSTERS:
            List<io.envoyproxy.envoy.config.route.v3.WeightedCluster.ClusterWeight> clusterWeights = proto.getWeightedClusters().getClustersList();
            if (clusterWeights.isEmpty()) {
                return StructOrError.fromError("No cluster found in weighted cluster list");
            }
            List<ClusterWeight> weightedClusters = new ArrayList<>();
            for (io.envoyproxy.envoy.config.route.v3.WeightedCluster.ClusterWeight clusterWeight : clusterWeights) {
                StructOrError<ClusterWeight> clusterWeightOrError = parseClusterWeight(clusterWeight, filterRegistry, parseHttpFilter);
                if (clusterWeightOrError.getErrorDetail() != null) {
                    return StructOrError.fromError("RouteAction contains invalid ClusterWeight: " + clusterWeightOrError.getErrorDetail());
                }
                weightedClusters.add(clusterWeightOrError.getStruct());
            }
            // TODO(chengyuanzhang): validate if the sum of weights equals to total weight.
            return StructOrError.fromStruct(RouteAction.forWeightedClusters(weightedClusters, hashPolicies, timeoutNano, retryPolicy));
        case CLUSTER_SPECIFIER_PLUGIN:
            if (enableRouteLookup) {
                String pluginName = proto.getClusterSpecifierPlugin();
                PluginConfig pluginConfig = pluginConfigMap.get(pluginName);
                if (pluginConfig == null) {
                    return StructOrError.fromError("ClusterSpecifierPlugin for [" + pluginName + "] not found");
                }
                NamedPluginConfig namedPluginConfig = NamedPluginConfig.create(pluginName, pluginConfig);
                return StructOrError.fromStruct(RouteAction.forClusterSpecifierPlugin(namedPluginConfig, hashPolicies, timeoutNano, retryPolicy));
            } else {
                return StructOrError.fromError("Support for ClusterSpecifierPlugin not enabled");
            }
        case CLUSTERSPECIFIER_NOT_SET:
        default:
            return StructOrError.fromError("Unknown cluster specifier: " + proto.getClusterSpecifierCase());
    }
}