Examples with ChannelHandlerContext org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext used on opensource projects

Example 76 with ChannelHandlerContext

use of org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext in project apiRecord by tobecoder2015.

the class ApirecordApplication method main.

public static void main(String[] args) {
    SpringApplication.run(ApirecordApplication.class, args);
    System.setProperty("proxySet", "true");
    System.setProperty("http.proxyHost", "127.0.0.1");
    System.setProperty("http.proxyPort", "9999");
    log.info("本地代理服务器ip:" + System.getProperty("http.proxyHost"));
    log.info("本地代理服务器port:" + System.getProperty("http.proxyPort"));
    // 配置文件
    RecordQueue.getInstance().start();
    FilterChain.addFilter(new ContentTypeFilter(FilterService.contentType));
    FilterChain.addFilter(new UrlFilter(FilterService.urls));
    FilterChain.addFilter(new MethodFilter(FilterService.methods));
    FileService.setSavePath(null);
    new NettyHttpProxyServer().initProxyInterceptFactory(() -> new HttpProxyIntercept() {

        @Override
        public boolean beforeRequest(Channel clientChannel, HttpRequest httpRequest) {
            log.debug("request " + clientChannel.id().asShortText() + "   " + httpRequest.toString());
            Request request = TransferUtil.toRequest(httpRequest);
            Message message = new Message(clientChannel.id().asShortText(), Message.REQUEST_HEAD, request);
            FilterChain filterChain = FilterChain.getFilterChain();
            if (filterChain.doFilter(message, filterChain)) {
            // 
            } else {
                RecordQueue.getInstance().add(message);
            }
            return true;
        }

        @Override
        public boolean beforeRequest(Channel clientChannel, HttpContent httpContent) {
            log.debug("request httpContent:" + clientChannel.id().asShortText() + "   " + clientChannel.id().asShortText() + "  " + httpContent.content().toString(Charset.defaultCharset()));
            if (!RecordMap.containKey(clientChannel.id().asShortText()))
                return true;
            Message message = new Message(clientChannel.id().asShortText(), Message.REQUEST_BODY, httpContent.content().toString(io.netty.util.CharsetUtil.UTF_8));
            FilterChain filterChain = FilterChain.getFilterChain();
            if (filterChain.doFilter(message, filterChain)) {
            // 
            } else {
                RecordQueue.getInstance().add(message);
            }
            return true;
        }

        @Override
        public boolean afterResponse(Channel clientChannel, Channel proxyChannel, HttpResponse httpResponse) {
            log.debug("httpResponse:" + clientChannel.id().asShortText() + "  " + httpResponse.toString());
            if (!RecordMap.containKey(clientChannel.id().asShortText()))
                return true;
            Response response = TransferUtil.toReponse(httpResponse);
            Message message = new Message(clientChannel.id().asShortText(), Message.RESPONSE_HEAD, response);
            FilterChain filterChain = FilterChain.getFilterChain();
            if (filterChain.doFilter(message, filterChain)) {
                message.setFilter(true);
                RecordQueue.getInstance().add(message);
            } else {
                RecordQueue.getInstance().add(message);
            }
            return true;
        }

        @Override
        public boolean afterResponse(Channel clientChannel, Channel proxyChannel, HttpContent httpContent) {
            log.debug("httpContent:" + clientChannel.id().asShortText() + "  " + httpContent.content().toString(Charset.defaultCharset()));
            if (!RecordMap.containKey(clientChannel.id().asShortText()))
                return true;
            ByteBuf buf = httpContent.content().copy();
            byte[] bytes = new byte[buf.readableBytes()];
            buf.readBytes(bytes);
            Message message = new Message(clientChannel.id().asShortText(), Message.RESPONSE_BODY, bytes);
            buf.writeBytes(bytes);
            if (httpContent instanceof LastHttpContent)
                message.setFinish(true);
            FilterChain filterChain = FilterChain.getFilterChain();
            if (filterChain.doFilter(message, filterChain)) {
            // 
            } else {
                RecordQueue.getInstance().add(message);
            }
            return true;
        }

        @Override
        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
            ctx.fireExceptionCaught(cause);
        // log.error("异常",cause);
        }
    }).start(9999);
}

Example 77 with ChannelHandlerContext

use of org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext in project jackrabbit-oak by apache.

the class ForwardHandler method channelActive.

@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception {
    group = new NioEventLoopGroup(0, r -> {
        return new Thread(r, String.format("forward-handler-%d", threadNumber.getAndIncrement()));
    });
    Bootstrap b = new Bootstrap().group(group).channel(NioSocketChannel.class).handler(new ChannelInitializer<SocketChannel>() {

        @Override
        public void initChannel(SocketChannel ch) throws Exception {
            if (flipPosition >= 0) {
                ch.pipeline().addLast(new FlipHandler(flipPosition));
            }
            if (skipBytes > 0) {
                ch.pipeline().addLast(new SkipHandler(skipPosition, skipBytes));
            }
            ch.pipeline().addLast(new BackwardHandler(ctx.channel()));
        }
    });
    remote = b.connect(targetHost, targetPort).sync().channel();
}

Example 78 with ChannelHandlerContext

use of org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext in project jocean-http by isdom.

the class ChannelTestCase method testChannelInactiveForInactiveChannel.

@Test
public final void testChannelInactiveForInactiveChannel() throws InterruptedException {
    final EmbeddedChannel channel = new EmbeddedChannel();
    channel.disconnect().syncUninterruptibly();
    assertTrue(!channel.isActive());
    final AtomicBoolean isInactiveCalled = new AtomicBoolean(false);
    channel.pipeline().addLast(new ChannelInboundHandlerAdapter() {

        @Override
        public void channelInactive(final ChannelHandlerContext ctx) throws Exception {
            isInactiveCalled.set(true);
        }
    });
    Thread.currentThread().sleep(1000);
    assertFalse(isInactiveCalled.get());
}

Example 79 with ChannelHandlerContext

use of org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext in project ratpack by ratpack.

the class ContentAggregatingRequestAction method addResponseHandlers.

@Override
protected void addResponseHandlers(ChannelPipeline p, Downstream<? super ReceivedResponse> downstream) {
    p.addLast(AGGREGATOR_HANDLER_NAME, new NoContentLengthOnNoBodyHttpObjectAggregator(requestConfig.maxContentLength));
    p.addLast(RESPONSE_HANDLER_NAME, new SimpleChannelInboundHandler<FullHttpResponse>(false) {

        @Override
        protected void channelRead0(ChannelHandlerContext ctx, FullHttpResponse response) throws Exception {
            response.touch();
            dispose(ctx.pipeline(), response);
            ByteBuf content = new ByteBufRef(response.content());
            execution.onComplete(() -> {
                if (content.refCnt() > 0) {
                    content.release();
                }
            });
            success(downstream, toReceivedResponse(response, content));
        }

        @Override
        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
            cause = decorateException(cause);
            error(downstream, cause);
            forceDispose(ctx.pipeline());
        }
    });
}

Example 80 with ChannelHandlerContext

use of org.apache.flink.shaded.netty4.io.netty.channel.ChannelHandlerContext in project cdap by caskdata.

the class NettyRouterPipelineTest method testHttpPipelining.

@Test
public void testHttpPipelining() throws Exception {
    final BlockingQueue<HttpResponseStatus> responseStatuses = new LinkedBlockingQueue<>();
    EventLoopGroup eventGroup = new NioEventLoopGroup();
    Bootstrap bootstrap = new Bootstrap().channel(NioSocketChannel.class).group(eventGroup).handler(new ChannelInitializer<SocketChannel>() {

        @Override
        protected void initChannel(SocketChannel ch) throws Exception {
            ChannelPipeline pipeline = ch.pipeline();
            pipeline.addLast("codec", new HttpClientCodec());
            pipeline.addLast("aggregator", new HttpObjectAggregator(1048576));
            pipeline.addLast("handler", new ChannelInboundHandlerAdapter() {

                @Override
                public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                    if (msg instanceof HttpResponse) {
                        responseStatuses.add(((HttpResponse) msg).status());
                    }
                    ReferenceCountUtil.release(msg);
                }
            });
        }
    });
    // Create a connection and make five consecutive HTTP call without waiting for the first to respond
    Channel channel = bootstrap.connect(HOSTNAME, ROUTER.getServiceMap().get(GATEWAY_NAME)).sync().channel();
    for (int i = 0; i < 5; i++) {
        HttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET, "/v1/sleep?sleepMillis=3000");
        request.headers().set(HttpHeaderNames.HOST, HOSTNAME);
        channel.writeAndFlush(request);
    }
    // Should get the first response as normal one
    HttpResponseStatus status = responseStatuses.poll(5, TimeUnit.SECONDS);
    Assert.assertEquals(HttpResponseStatus.OK, status);
    // The rest four should be failure responses
    for (int i = 0; i < 4; i++) {
        Assert.assertEquals(HttpResponseStatus.NOT_IMPLEMENTED, responseStatuses.poll(1, TimeUnit.SECONDS));
    }
    eventGroup.shutdownGracefully();
    channel.close();
    Assert.assertTrue(responseStatuses.isEmpty());
}