com.sun.xml.messaging.saaj.util.ByteOutputStream Maven / Gradle / Ivy
/*
* Copyright (c) 1997, 2020 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Distribution License v. 1.0, which is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
package com.sun.xml.messaging.saaj.util;
import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.InputStream;
import java.io.ByteArrayInputStream;
/**
* Customized {@link BufferedOutputStream}.
*
*
* Compared to {@link BufferedOutputStream},
* this class:
*
*
* - doesn't do synchronization
*
- allows access to the raw buffer
*
- almost no parameter check
*
*/
public final class ByteOutputStream extends OutputStream {
/**
* The buffer where data is stored.
*/
protected byte[] buf;
/**
* The number of valid bytes in the buffer.
*/
protected int count = 0;
public ByteOutputStream() {
this(1024);
}
public ByteOutputStream(int size) {
buf = new byte[size];
}
/**
* Copies all the bytes from this input into this buffer.
*
* @param in input stream.
* @exception IOException in case of an I/O error.
*/
public void write(InputStream in) throws IOException {
if (in instanceof ByteArrayInputStream) {
int size = in.available();
ensureCapacity(size);
count += in.read(buf,count,size);
return;
}
while(true) {
int cap = buf.length-count;
int sz = in.read(buf,count,cap);
if(sz<0) return; // hit EOS
count += sz;
if(cap==sz)
// the buffer filled up. double the buffer
ensureCapacity(count);
}
}
@Override
public void write(int b) {
ensureCapacity(1);
buf[count] = (byte) b;
count++;
}
/**
* Ensure that the buffer has at least this much space.
*/
private void ensureCapacity(int space) {
int newcount = space + count;
if (newcount > buf.length) {
byte[] newbuf = new byte[Math.max(buf.length << 1, newcount)];
System.arraycopy(buf, 0, newbuf, 0, count);
buf = newbuf;
}
}
@Override
public void write(byte[] b, int off, int len) {
ensureCapacity(len);
System.arraycopy(b, off, buf, count, len);
count += len;
}
@Override
public void write(byte[] b) {
write(b, 0, b.length);
}
/**
* Writes a string as ASCII string.
*
* @param s string to write.
*/
public void writeAsAscii(String s) {
int len = s.length();
ensureCapacity(len);
int ptr = count;
for( int i=0; i