Examples with Future org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future used on opensource projects

Example 96 with Future

use of org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future in project arthas by alibaba.

the class TunnelSocketFrameHandler method agentRegister.

private void agentRegister(ChannelHandlerContext ctx, HandshakeComplete handshake, String requestUri) throws URISyntaxException {
    QueryStringDecoder queryDecoder = new QueryStringDecoder(requestUri);
    Map<String, List<String>> parameters = queryDecoder.parameters();
    String appName = null;
    List<String> appNameList = parameters.get(URIConstans.APP_NAME);
    if (appNameList != null && !appNameList.isEmpty()) {
        appName = appNameList.get(0);
    }
    // generate a random agent id
    String id = null;
    if (appName != null) {
        // 如果有传 app name，则生成带 app name前缀的id，方便管理
        id = appName + "_" + RandomStringUtils.random(20, true, true).toUpperCase();
    } else {
        id = RandomStringUtils.random(20, true, true).toUpperCase();
    }
    // agent传过来，则优先用 agent的
    List<String> idList = parameters.get(URIConstans.ID);
    if (idList != null && !idList.isEmpty()) {
        id = idList.get(0);
    }
    String arthasVersion = null;
    List<String> arthasVersionList = parameters.get(URIConstans.ARTHAS_VERSION);
    if (arthasVersionList != null && !arthasVersionList.isEmpty()) {
        arthasVersion = arthasVersionList.get(0);
    }
    final String finalId = id;
    // URI responseUri = new URI("response", null, "/", "method=" + MethodConstants.AGENT_REGISTER + "&id=" + id, null);
    URI responseUri = UriComponentsBuilder.newInstance().scheme(URIConstans.RESPONSE).path("/").queryParam(URIConstans.METHOD, MethodConstants.AGENT_REGISTER).queryParam(URIConstans.ID, id).build().encode().toUri();
    AgentInfo info = new AgentInfo();
    // 前面可能有nginx代理
    HttpHeaders headers = handshake.requestHeaders();
    String host = HttpUtils.findClientIP(headers);
    if (host == null) {
        SocketAddress remoteAddress = ctx.channel().remoteAddress();
        if (remoteAddress instanceof InetSocketAddress) {
            InetSocketAddress inetSocketAddress = (InetSocketAddress) remoteAddress;
            info.setHost(inetSocketAddress.getHostString());
            info.setPort(inetSocketAddress.getPort());
        }
    } else {
        info.setHost(host);
        Integer port = HttpUtils.findClientPort(headers);
        if (port != null) {
            info.setPort(port);
        }
    }
    info.setChannelHandlerContext(ctx);
    if (arthasVersion != null) {
        info.setArthasVersion(arthasVersion);
    }
    tunnelServer.addAgent(id, info);
    ctx.channel().closeFuture().addListener(new GenericFutureListener<Future<? super Void>>() {

        @Override
        public void operationComplete(Future<? super Void> future) throws Exception {
            tunnelServer.removeAgent(finalId);
        }
    });
    ctx.channel().writeAndFlush(new TextWebSocketFrame(responseUri.toString()));
}

Example 97 with Future

use of org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future in project PaperDev by Kamillaova.

the class ServerConnection method c.

public void c() {
    List list = this.h;
    synchronized (this.h) {
        // Spigot Start
        // Paper
        addPending();
        // This prevents players from 'gaming' the server, and strategically relogging to increase their position in the tick order
        if (org.spigotmc.SpigotConfig.playerShuffle > 0 && MinecraftServer.currentTick % org.spigotmc.SpigotConfig.playerShuffle == 0) {
            Collections.shuffle(this.h);
        }
        // Spigot End
        Iterator iterator = this.h.iterator();
        while (iterator.hasNext()) {
            final NetworkManager networkmanager = (NetworkManager) iterator.next();
            if (!networkmanager.h()) {
                if (networkmanager.isConnected()) {
                    try {
                        networkmanager.a();
                    } catch (Exception exception) {
                        if (networkmanager.isLocal()) {
                            CrashReport crashreport = CrashReport.a(exception, "Ticking memory connection");
                            CrashReportSystemDetails crashreportsystemdetails = crashreport.a("Ticking connection");
                            crashreportsystemdetails.a("Connection", new CrashReportCallable() {

                                public String a() throws Exception {
                                    return networkmanager.toString();
                                }

                                public Object call() throws Exception {
                                    return this.a();
                                }
                            });
                            throw new ReportedException(crashreport);
                        }
                        ServerConnection.e.warn("Failed to handle packet for {}", networkmanager.getSocketAddress(), exception);
                        final ChatComponentText chatcomponenttext = new ChatComponentText("Internal server error");
                        networkmanager.sendPacket(new PacketPlayOutKickDisconnect(chatcomponenttext), new GenericFutureListener() {

                            public void operationComplete(Future future) throws Exception {
                                networkmanager.close(chatcomponenttext);
                            }
                        }, new GenericFutureListener[0]);
                        networkmanager.stopReading();
                    }
                } else {
                    // Fix a race condition where a NetworkManager could be unregistered just before connection.
                    if (networkmanager.preparing)
                        continue;
                    // Spigot End
                    iterator.remove();
                    networkmanager.handleDisconnection();
                }
            }
        }
    }
}

Example 98 with Future

use of org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future in project zookeeper by apache.

the class NettyServerCnxn method close.

public void close() {
    closingChannel = true;
    LOG.debug("close called for session id: 0x{}", Long.toHexString(sessionId));
    setStale();
    // ZOOKEEPER-2743:
    // Always unregister connection upon close to prevent
    // connection bean leak under certain race conditions.
    factory.unregisterConnection(this);
    // if this is not in cnxns then it's already closed
    if (!factory.cnxns.remove(this)) {
        LOG.debug("cnxns size:{}", factory.cnxns.size());
        if (channel.isOpen()) {
            channel.close();
        }
        return;
    }
    LOG.debug("close in progress for session id: 0x{}", Long.toHexString(sessionId));
    factory.removeCnxnFromSessionMap(this);
    factory.removeCnxnFromIpMap(this, ((InetSocketAddress) channel.remoteAddress()).getAddress());
    if (zkServer != null) {
        zkServer.removeCnxn(this);
    }
    if (channel.isOpen()) {
        // Since we don't check on the futures created by write calls to the channel complete we need to make sure
        // that all writes have been completed before closing the channel or we risk data loss
        // See: http://lists.jboss.org/pipermail/netty-users/2009-August/001122.html
        channel.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(new ChannelFutureListener() {

            @Override
            public void operationComplete(ChannelFuture future) {
                future.channel().close().addListener(f -> releaseQueuedBuffer());
            }
        });
    } else {
        ServerMetrics.getMetrics().CONNECTION_DROP_COUNT.add(1);
        channel.eventLoop().execute(this::releaseQueuedBuffer);
    }
}

Example 99 with Future

use of org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future in project vert.x by eclipse.

the class HttpServerWorker method handle.

@Override
public void handle(Channel ch) {
    if (HAProxyMessageCompletionHandler.canUseProxyProtocol(options.isUseProxyProtocol())) {
        IdleStateHandler idle;
        io.netty.util.concurrent.Promise<Channel> p = ch.eventLoop().newPromise();
        ch.pipeline().addLast(new HAProxyMessageDecoder());
        if (options.getProxyProtocolTimeout() > 0) {
            ch.pipeline().addLast("idle", idle = new IdleStateHandler(0, 0, options.getProxyProtocolTimeout(), options.getProxyProtocolTimeoutUnit()));
        } else {
            idle = null;
        }
        ch.pipeline().addLast(new HAProxyMessageCompletionHandler(p));
        p.addListener((GenericFutureListener<Future<Channel>>) future -> {
            if (future.isSuccess()) {
                if (idle != null) {
                    ch.pipeline().remove(idle);
                }
                configurePipeline(future.getNow());
            } else {
                handleException(future.cause());
            }
        });
    } else {
        configurePipeline(ch);
    }
}

Example 100 with Future

use of org.apache.flink.shaded.netty4.io.netty.util.concurrent.Future in project grpc-java by grpc.

the class NettyServerTest method testBindScheduleFailure.

@Test
@SuppressWarnings("unchecked")
public void testBindScheduleFailure() throws Exception {
    when(bindFuture.awaitUninterruptibly()).thenReturn(bindFuture);
    when(bindFuture.isSuccess()).thenReturn(false);
    when(bindFuture.getNow()).thenReturn(null);
    Throwable mockCause = mock(Throwable.class);
    when(bindFuture.cause()).thenReturn(mockCause);
    Future<Void> mockFuture = (Future<Void>) mock(Future.class);
    doReturn(mockFuture).when(mockEventLoopGroup).submit(any(Runnable.class));
    SocketAddress addr = new InetSocketAddress(0);
    verifyServerNotStart(Collections.singletonList(addr), mockEventLoopGroup, IOException.class, "Failed to bind to addresses " + Arrays.asList(addr));
}