bufferbuilder.java

银行项目为后台ｓｏｃｋｅｔ通信写的程序

/****************************************************************************
 * Package		: com.ecSolutions.ecAppServer.server.buffer
 * File			: BufferBuilder.java
 * Create Date  : 2007-7-20
 * Author		: Steven Chen
 * 
 * Copyright(C) 2006 ecSolutions(shanghai) Co.,Limited.All Rights Reserved.
 *			
 ***************************************************************************/
package com.ecSolutions.ecAppServer.server.buffer;

import java.nio.ByteBuffer;

import com.ecSolutions.ecAppServer.server.Buffer;

/**
 * Buffer builder.
 * 
 * @author Steven Chen
 * @version $id$
 */
public class BufferBuilder {

    private Buffer buffer;

    public BufferBuilder() {
        this(0);
    }

    public BufferBuilder(int capacity) {
        if (capacity <= 0)
            capacity = 512;
        buffer = BufferFactory.allocate(capacity);
    }

    private void resize(int size) {
        if (buffer.remaining() < size) {
            int capacity = Math.max(buffer.position() + size, (buffer
                    .capacity() + 1) * 2);
            Buffer newBuffer = BufferFactory.allocate(capacity).put(
                    buffer.flip());
            buffer.release();
            buffer = newBuffer;
        }
    }

    public Buffer toBuffer() {
        Buffer result = BufferFactory.allocate(buffer.position());
        buffer.get(0, result);
        return result.flip();
    }

    public void release() {
        buffer.release();
    }

    public BufferBuilder append(byte b) {
        resize(1);
        buffer.put(b);
        return this;
    }

    public BufferBuilder append(short s) {
        resize(2);
        buffer.putShort(s);
        return this;
    }

    public BufferBuilder append(int i) {
        resize(4);
        buffer.putInt(i);
        return this;
    }

    public BufferBuilder append(long l) {
        resize(8);
        buffer.putLong(l);
        return this;
    }

    public BufferBuilder append(float f) {
        resize(4);
        buffer.putFloat(f);
        return this;
    }

    public BufferBuilder append(double d) {
        resize(8);
        buffer.putDouble(d);
        return this;
    }

    public BufferBuilder append(byte[] b) {
        resize(b.length);
        buffer.put(b);
        return this;
    }

    public BufferBuilder append(ByteBuffer buffer) {
        resize(buffer.remaining());
        this.buffer.put(buffer);
        return this;
    }

    public BufferBuilder append(Buffer buffer) {
        resize(buffer.remaining());
        this.buffer.put(buffer);
        return this;
    }

}