/** Wrap the HttpObject with the appropriate type and fire read on the next handler. */ private void wrapRequest(ChannelHandlerContext ctx, HttpObject msg) { Http1MessageSession session = setDefaultMessageSession(ctx); try { Request request; if (msg instanceof FullHttpRequest) { request = new FullHttp1Request((FullHttpRequest) msg); session.onRequest(request); } else if (msg instanceof HttpRequest) { request = new SegmentedHttp1Request((HttpRequest) msg); session.onRequest(request); } else if (msg instanceof HttpContent) { SegmentedData data = new Http1SegmentedData((HttpContent) msg); session.onRequestData(data); Request sessionRequest = session.currentRequest(); if (sessionRequest == null) { // We don't have a sessionRequest so we can't construct a SegmentedRequestData. // Don't log as session.onRequestData should have logged. return; } request = new SegmentedRequestData(sessionRequest, data); } else { log.error("Dropping unsupported http object: {}", msg); return; } ctx.fireChannelRead(request); } finally { session.flush(); } }
@Test public void testOnRequestData() { Request request = DefaultSegmentedRequest.builder() .headers(new DefaultHeaders()) .method(GET) .path("/") .build(); session.onRequest(request); assertFalse(session.initialRequest().requestFinished); assertFalse(session.initialRequest().responseFinished); assertFalse(session.closeConnection()); SegmentedData data = DefaultSegmentedData.builder() .content(Unpooled.EMPTY_BUFFER) .endOfMessage(true) .trailingHeaders(new DefaultHeaders()) .build(); session.onRequestData(data); assertTrue(session.initialRequest().requestFinished); }