package us.codecraft.netty_servlet.connector.netty; import org.jboss.netty.buffer.ChannelBuffer; import javax.servlet.ServletOutputStream; import java.io.DataOutput; import java.io.DataOutputStream; import java.io.IOException; /** * @author [email protected] */ public class ChannelBufferServletOutputStream extends ServletOutputStream implements DataOutput { private final ChannelBuffer buffer; private final int startIndex; private final DataOutputStream utf8out = new DataOutputStream(this); /** * Creates a new stream which writes data to the specified {@code buffer}. */ public ChannelBufferServletOutputStream(ChannelBuffer buffer) { if (buffer == null) { throw new NullPointerException("buffer"); } this.buffer = buffer; startIndex = buffer.writerIndex(); } /** * Returns the number of written bytes by this stream so far. */ public int writtenBytes() { return buffer.writerIndex() - startIndex; } @Override public void write(byte[] b, int off, int len) throws IOException { if (len == 0) { return; } buffer.writeBytes(b, off, len); } @Override public void write(byte[] b) throws IOException { buffer.writeBytes(b); } @Override public void write(int b) throws IOException { buffer.writeByte((byte) b); } public void writeBoolean(boolean v) throws IOException { write(v ? (byte) 1 : (byte) 0); } public void writeByte(int v) throws IOException { write(v); } public void writeBytes(String s) throws IOException { int len = s.length(); for (int i = 0; i < len; i++) { write((byte) s.charAt(i)); } } public void writeChar(int v) throws IOException { writeShort((short) v); } public void writeChars(String s) throws IOException { int len = s.length(); for (int i = 0; i < len; i++) { writeChar(s.charAt(i)); } } public void writeDouble(double v) throws IOException { writeLong(Double.doubleToLongBits(v)); } public void writeFloat(float v) throws IOException { writeInt(Float.floatToIntBits(v)); } public void writeInt(int v) throws IOException { buffer.writeInt(v); } public void writeLong(long v) throws IOException { buffer.writeLong(v); } public void writeShort(int v) throws IOException { buffer.writeShort((short) v); } public void writeUTF(String s) throws IOException { utf8out.writeUTF(s); } /** * Returns the buffer where this stream is writing data. */ public ChannelBuffer buffer() { return buffer; } }