Examples with HashPolicy io.grpc.xds.VirtualHost.Route.RouteAction.HashPolicy used on opensource projects

Example 16 with HashPolicy

use of io.grpc.xds.VirtualHost.Route.RouteAction.HashPolicy 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());
    }
}