Examples with ChannelPromise com.couchbase.client.core.deps.io.netty.channel.ChannelPromise used on opensource projects

Example 6 with ChannelPromise

use of com.couchbase.client.core.deps.io.netty.channel.ChannelPromise in project couchbase-jvm-clients by couchbase.

the class SaslListMechanismsHandler method connect.

/**
 * Intercepts the connect process inside the pipeline to only propagate either
 * success or failure if the hello process is completed either way.
 *
 * @param ctx           the {@link ChannelHandlerContext} for which the connect operation is made.
 * @param remoteAddress the {@link SocketAddress} to which it should connect.
 * @param localAddress  the {@link SocketAddress} which is used as source on connect.
 * @param promise       the {@link ChannelPromise} to notify once the operation completes.
 */
@Override
public void connect(final ChannelHandlerContext ctx, final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {
    interceptedConnectPromise = promise;
    ChannelPromise downstream = ctx.newPromise();
    downstream.addListener(f -> {
        if (!f.isSuccess() && !interceptedConnectPromise.isDone()) {
            ConnectTimings.record(ctx.channel(), this.getClass());
            interceptedConnectPromise.tryFailure(f.cause());
        }
    });
    ctx.connect(remoteAddress, localAddress, downstream);
}

Example 7 with ChannelPromise

use of com.couchbase.client.core.deps.io.netty.channel.ChannelPromise in project couchbase-jvm-clients by couchbase.

the class ErrorMapLoadingHandler method connect.

/**
 * Intercepts the connect process inside the pipeline to only propagate either
 * success or failure if the error map loading process is completed either way.
 *
 * @param ctx           the {@link ChannelHandlerContext} for which the connect operation is made.
 * @param remoteAddress the {@link SocketAddress} to which it should connect.
 * @param localAddress  the {@link SocketAddress} which is used as source on connect.
 * @param promise       the {@link ChannelPromise} to notify once the operation completes.
 */
@Override
public void connect(final ChannelHandlerContext ctx, final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {
    interceptedConnectPromise = promise;
    ChannelPromise downstream = ctx.newPromise();
    downstream.addListener(f -> {
        if (!f.isSuccess() && !interceptedConnectPromise.isDone()) {
            ConnectTimings.record(ctx.channel(), this.getClass());
            interceptedConnectPromise.tryFailure(f.cause());
        }
    });
    ctx.connect(remoteAddress, localAddress, downstream);
}

Example 8 with ChannelPromise

use of com.couchbase.client.core.deps.io.netty.channel.ChannelPromise in project couchbase-jvm-clients by couchbase.

the class SelectBucketHandlerTest method propagateConnectFailureFromDownstream.

/**
 * This test verifies that if a downstream promise fails that the error
 * is propagated through the captured promise.
 */
@Test
void propagateConnectFailureFromDownstream() {
    final Exception connectException = new Exception("I failed");
    ChannelDuplexHandler failingHandler = new ChannelDuplexHandler() {

        @Override
        public void connect(final ChannelHandlerContext ctx, final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {
            promise.setFailure(connectException);
        }
    };
    SelectBucketHandler handler = new SelectBucketHandler(endpointContext, "bucket");
    channel.pipeline().addLast(failingHandler).addLast(handler);
    ChannelFuture connect = channel.connect(new InetSocketAddress("1.2.3.4", 1234));
    assertEquals(connectException, connect.awaitUninterruptibly().cause());
}

Example 9 with ChannelPromise

use of com.couchbase.client.core.deps.io.netty.channel.ChannelPromise in project couchbase-jvm-clients by couchbase.

the class BaseEndpointTest method emitsEventOnFailedDisconnect.

/**
 * If the disconnect failed for some reason, make sure the proper warning event
 * is raised and captured.
 */
@Test
void emitsEventOnFailedDisconnect() {
    final Throwable expectedCause = new Exception("something failed");
    EmbeddedChannel channel = new EmbeddedChannel(new ChannelOutboundHandlerAdapter() {

        @Override
        public void close(ChannelHandlerContext ctx, ChannelPromise promise) {
            promise.tryFailure(expectedCause);
        }
    });
    InstrumentedEndpoint endpoint = connectSuccessfully(channel);
    endpoint.disconnect();
    waitUntilCondition(() -> endpoint.state() == EndpointState.DISCONNECTED);
    assertEquals(2, eventBus.publishedEvents().size());
    assertTrue(eventBus.publishedEvents().get(0) instanceof EndpointConnectedEvent);
    EndpointDisconnectionFailedEvent event = (EndpointDisconnectionFailedEvent) eventBus.publishedEvents().get(1);
    assertEquals(expectedCause, event.cause());
    assertEquals(Event.Severity.WARN, event.severity());
}

Example 10 with ChannelPromise

use of com.couchbase.client.core.deps.io.netty.channel.ChannelPromise in project couchbase-jvm-clients by couchbase.

the class FeatureNegotiatingHandlerTest method propagateConnectFailureFromDownstream.

/**
 * This test verifies that if a downstream promise fails that the error
 * is propagated through the captured promise.
 */
@Test
void propagateConnectFailureFromDownstream() {
    final Exception connectException = new Exception("I failed");
    ChannelDuplexHandler failingHandler = new ChannelDuplexHandler() {

        @Override
        public void connect(final ChannelHandlerContext ctx, final SocketAddress remoteAddress, final SocketAddress localAddress, final ChannelPromise promise) {
            promise.setFailure(connectException);
        }
    };
    FeatureNegotiatingHandler handler = new FeatureNegotiatingHandler(endpointContext, Collections.singleton(ServerFeature.TRACING));
    channel.pipeline().addLast(failingHandler).addLast(handler);
    ChannelFuture connect = channel.connect(new InetSocketAddress("1.2.3.4", 1234));
    assertEquals(connectException, connect.awaitUninterruptibly().cause());
}