Examples with OriginName com.netflix.zuul.origins.OriginName used on opensource projects

Example 1 with OriginName

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

the class DefaultClientChannelManagerTest method initializeAndShutdown.

@Test
public void initializeAndShutdown() throws Exception {
    final String appName = "app-" + UUID.randomUUID();
    final ServerSocket serverSocket = new ServerSocket(0);
    final InetSocketAddress serverSocketAddress = (InetSocketAddress) serverSocket.getLocalSocketAddress();
    final String serverHostname = serverSocketAddress.getHostName();
    final int serverPort = serverSocketAddress.getPort();
    final OriginName originName = OriginName.fromVipAndApp("vip", appName);
    final DefaultClientConfigImpl clientConfig = new DefaultClientConfigImpl();
    Server.defaultOutboundChannelType.set(NioSocketChannel.class);
    final InstanceInfo instanceInfo = Builder.newBuilder().setAppName(appName).setHostName(serverHostname).setPort(serverPort).build();
    DiscoveryResult discoveryResult = DiscoveryResult.from(instanceInfo, true);
    final DynamicServerResolver resolver = mock(DynamicServerResolver.class);
    when(resolver.resolve(any())).thenReturn(discoveryResult);
    when(resolver.hasServers()).thenReturn(true);
    final Registry registry = new DefaultRegistry();
    final DefaultClientChannelManager clientChannelManager = new DefaultClientChannelManager(originName, clientConfig, resolver, registry);
    final NioEventLoopGroup eventLoopGroup = new NioEventLoopGroup(10);
    final EventLoop eventLoop = eventLoopGroup.next();
    clientChannelManager.init();
    Truth.assertThat(clientChannelManager.getConnsInUse()).isEqualTo(0);
    final Promise<PooledConnection> promiseConn = clientChannelManager.acquire(eventLoop);
    promiseConn.await(200, TimeUnit.MILLISECONDS);
    assertTrue(promiseConn.isDone());
    assertTrue(promiseConn.isSuccess());
    final PooledConnection connection = promiseConn.get();
    assertTrue(connection.isActive());
    assertFalse(connection.isInPool());
    Truth.assertThat(clientChannelManager.getConnsInUse()).isEqualTo(1);
    final boolean releaseResult = clientChannelManager.release(connection);
    assertTrue(releaseResult);
    assertTrue(connection.isInPool());
    Truth.assertThat(clientChannelManager.getConnsInUse()).isEqualTo(0);
    clientChannelManager.shutdown();
    serverSocket.close();
}

Example 2 with OriginName

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

the class DefaultClientChannelManagerTest method updateServerRefOnValidDiscoveryResult.

@Test
public void updateServerRefOnValidDiscoveryResult() {
    OriginName originName = OriginName.fromVip("vip", "test");
    final DefaultClientConfigImpl clientConfig = new DefaultClientConfigImpl();
    final DynamicServerResolver resolver = mock(DynamicServerResolver.class);
    final InstanceInfo instanceInfo = Builder.newBuilder().setAppName("server-equality").setHostName("server-equality").setPort(7777).build();
    final DiscoveryResult discoveryResult = DiscoveryResult.from(instanceInfo, false);
    when(resolver.resolve(any())).thenReturn(discoveryResult);
    final DefaultClientChannelManager clientChannelManager = new DefaultClientChannelManager(originName, clientConfig, resolver, new DefaultRegistry());
    final AtomicReference<DiscoveryResult> serverRef = new AtomicReference<>();
    // TODO(argha-c) capture and assert on the promise once we have a dummy with ServerStats initialized
    clientChannelManager.acquire(new DefaultEventLoop(), null, CurrentPassport.create(), serverRef, new AtomicReference<>());
    Truth.assertThat(serverRef.get()).isSameInstanceAs(discoveryResult);
}

Example 3 with OriginName

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

the class DefaultClientChannelManagerTest method updateServerRefOnEmptyDiscoveryResult.

@Test
public void updateServerRefOnEmptyDiscoveryResult() {
    OriginName originName = OriginName.fromVip("vip", "test");
    final DefaultClientConfigImpl clientConfig = new DefaultClientConfigImpl();
    final DynamicServerResolver resolver = mock(DynamicServerResolver.class);
    when(resolver.resolve(any())).thenReturn(DiscoveryResult.EMPTY);
    final DefaultClientChannelManager clientChannelManager = new DefaultClientChannelManager(originName, clientConfig, resolver, new DefaultRegistry());
    final AtomicReference<DiscoveryResult> serverRef = new AtomicReference<>();
    final Promise<PooledConnection> promise = clientChannelManager.acquire(new DefaultEventLoop(), null, CurrentPassport.create(), serverRef, new AtomicReference<>());
    Truth.assertThat(promise.isSuccess()).isFalse();
    Truth.assertThat(serverRef.get()).isSameInstanceAs(DiscoveryResult.EMPTY);
}

Example 4 with OriginName

use of com.netflix.zuul.origins.OriginName 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;
}