andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Simplified stream I/O in HttpTunnelingServlet using dynamic channel buffer

Browse Source
This commit is contained in:
Trustin Lee 2009-03-12 06:42:39 +00:00
parent 35b08df70a
commit 937ee06d9d
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/org/jboss/netty/channel/socket/http/HttpTunnelingServlet.java
View File

@ -22,6 +22,7 @@
package org.jboss.netty.channel.socket.http;
import java.io.IOException;
import java.io.InputStream;
import java.io.PushbackInputStream;
import java.util.List;
@ -136,16 +137,24 @@ public class HttpTunnelingServlet extends HttpServlet {
HttpServletRequest request,
HttpServletResponse response, HttpSession session,
HttpTunnelingChannelHandler handler) throws IOException {
int length = request.getContentLength();
if (length > 0) {
byte[] bytes = new byte[length];
request.getInputStream().read(bytes);
ChannelBuffer cb = ChannelBuffers.copiedBuffer(bytes);
channel.write(cb);
InputStream in = request.getInputStream();
if (in != null) {
ChannelBuffer requestContent = ChannelBuffers.dynamicBuffer();
for (;;) {
int writtenBytes = requestContent.writeBytes(in, 4096);
if (writtenBytes < 0) {
break;
}
}
if (requestContent.readable()) {
channel.write(requestContent);
}
}
handler.setOutputStream(response.getOutputStream());
List<MessageEvent> buffers = handler.getAwaitingEvents();
length = 0;
int length = 0;
if (buffers.size() > 0) {
for (MessageEvent buffer: buffers) {
length += ((ChannelBuffer) buffer.getMessage()).readableBytes();