1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.util;
21
22 import java.io.OutputStream;
23 import java.nio.ByteBuffer;
24
25
26
27
28
29
30
31
32
33
34
35 public class ByteBufferOutputStream extends OutputStream {
36 private ByteBuffer buffer;
37
38 private boolean elastic = false;
39
40 public ByteBufferOutputStream() {
41 this(1024);
42 }
43
44 private ByteBufferOutputStream(ByteBuffer buffer) {
45 this.buffer = buffer;
46 }
47
48 public ByteBufferOutputStream(int initialSize) {
49 this(ByteBuffer.allocate(initialSize));
50 }
51
52 public ByteBuffer getByteBuffer() {
53 ByteBuffer out = buffer.asReadOnlyBuffer();
54 out.limit(out.position());
55 out.position(0);
56
57 return out;
58 }
59
60 public boolean isElastic() {
61 return elastic;
62 }
63
64 private void needSpace(int len) {
65 if (elastic && buffer.capacity() - buffer.position() < len) {
66 ByteBuffer newBuffer = ByteBuffer.allocate(Math.max(buffer.capacity() * 2, buffer.position() + len));
67 buffer.limit(buffer.position());
68 newBuffer.put(buffer);
69 buffer = newBuffer;
70 }
71 }
72
73 public void setElastic(boolean elastic) {
74 this.elastic = elastic;
75 }
76
77 @Override
78 public void write(byte[] b, int off, int len) {
79 needSpace(len);
80 buffer.put(b, off, len);
81 }
82
83 @Override
84 public void write(int b) {
85 needSpace(1);
86 buffer.put((byte) b);
87 }
88 }