Search in sources :

Example 1 with OriginManager

use of com.netflix.zuul.origins.OriginManager in project zuul by Netflix.

the class ProxyEndpoint method getOrigin.

/**
 * Get the implementing origin.
 *
 * Note: this method gets called in the constructor so if overloading it or any methods called within, you cannot
 * rely on your own constructor parameters.
 */
@Nullable
protected NettyOrigin getOrigin(HttpRequestMessage request) {
    SessionContext context = request.getContext();
    OriginManager<NettyOrigin> originManager = (OriginManager<NettyOrigin>) context.get(CommonContextKeys.ORIGIN_MANAGER);
    if (Debug.debugRequest(context)) {
        ImmutableList.Builder<String> routingLogEntries = context.get(CommonContextKeys.ROUTING_LOG);
        if (routingLogEntries != null) {
            for (String entry : routingLogEntries.build()) {
                Debug.addRequestDebug(context, "RoutingLog: " + entry);
            }
        }
    }
    String primaryRoute = context.getRouteVIP();
    if (Strings.isNullOrEmpty(primaryRoute)) {
        // If no vip selected, leave origin null, then later the handleNoOriginSelected() method will be invoked.
        return null;
    }
    // make sure the restClientName will never be a raw VIP in cases where it's the fallback for another route assignment
    String restClientVIP = primaryRoute;
    boolean useFullName = context.getBoolean(CommonContextKeys.USE_FULL_VIP_NAME);
    String restClientName = useFullName ? restClientVIP : VipUtils.getVIPPrefix(restClientVIP);
    NettyOrigin origin = null;
    // allow implementors to override the origin with custom injection logic
    OriginName overrideOriginName = injectCustomOriginName(request);
    if (overrideOriginName != null) {
        // Use the custom vip instead if one has been provided.
        origin = getOrCreateOrigin(originManager, overrideOriginName, request.reconstructURI(), context);
    } else if (restClientName != null) {
        // This is the normal flow - that a RoutingFilter has assigned a route
        OriginName originName = OriginName.fromVip(restClientVIP, restClientName);
        origin = getOrCreateOrigin(originManager, originName, request.reconstructURI(), context);
    }
    verifyOrigin(context, request, restClientName, origin);
    // Update the routeVip on context to show the actual raw VIP from the clientConfig of the chosen Origin.
    if (origin != null) {
        context.set(CommonContextKeys.ACTUAL_VIP, origin.getClientConfig().get(IClientConfigKey.Keys.DeploymentContextBasedVipAddresses));
        context.set(CommonContextKeys.ORIGIN_VIP_SECURE, origin.getClientConfig().get(IClientConfigKey.Keys.IsSecure));
    }
    return origin;
}
Also used : OriginName(com.netflix.zuul.origins.OriginName) NettyOrigin(com.netflix.zuul.origins.NettyOrigin) ImmutableList(com.google.common.collect.ImmutableList) SessionContext(com.netflix.zuul.context.SessionContext) OriginManager(com.netflix.zuul.origins.OriginManager) Nullable(javax.annotation.Nullable)

Aggregations

ImmutableList (com.google.common.collect.ImmutableList)1 SessionContext (com.netflix.zuul.context.SessionContext)1 NettyOrigin (com.netflix.zuul.origins.NettyOrigin)1 OriginManager (com.netflix.zuul.origins.OriginManager)1 OriginName (com.netflix.zuul.origins.OriginName)1 Nullable (javax.annotation.Nullable)1