com.xxdb.io.AbstractExtendedDataOutputStream Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of api-java Show documentation
Show all versions of api-java Show documentation
The messaging and data conversion protocol between Java and DolphinDB server
package com.xxdb.io;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UTFDataFormatException;
public abstract class AbstractExtendedDataOutputStream extends FilterOutputStream implements ExtendedDataOutput{
private static final int UTF8_STRING_LIMIT = 65535;
protected static final int BUF_SIZE = 4096;
protected byte[] buf;
protected static final int longBufSize = BUF_SIZE / 8;
protected static final int intBufSize = BUF_SIZE / 4;
protected int[] intBuf;
protected long[] longBuf;
public AbstractExtendedDataOutputStream(OutputStream out) {
super(out);
}
public void flush() throws IOException{
out.flush();
}
@Override
public void writeBoolean(boolean v) throws IOException {
write(v?1:0);
}
@Override
public void writeByte(int v) throws IOException {
write(v & 0xff);
}
@Override
public void writeChar(int v) throws IOException {
writeShort(v);
}
@Override
public void writeFloat(float v) throws IOException {
writeInt(Float.floatToIntBits (v));
}
@Override
public void writeDouble(double v) throws IOException {
writeLong(Double.doubleToLongBits (v));
}
@Override
public void writeBytes(String s) throws IOException {
int len = s.length();
int i = 0;
int pos = 0;
if (buf == null)
buf = new byte[BUF_SIZE];
do {
while (i < len && pos < buf.length - 4){
char c = s.charAt(i++);
if (c >= '\u0001' && c <= '\u007f')
buf[pos++] = (byte) c;
else if (c == '\u0000' || (c >= '\u0080' && c <= '\u07ff')){
buf[pos++] = (byte) (0xc0 | (0x1f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
else{
buf[pos++] = (byte) (0xe0 | (0x0f & (c >> 12)));
buf[pos++] = (byte) (0x80 | (0x3f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
}
write(buf, 0, pos);
pos = 0;
}while (i < len);
}
@Override
public void writeChars(String s) throws IOException {
int len = s.length();
for (int i = 0; i < len; ++i)
writeChar(s.charAt(i));
}
@Override
public void writeUTF(String value) throws IOException {
int len = value.length();
int i = 0;
int pos = 0;
boolean lengthWritten = false;
if (buf == null)
buf = new byte[BUF_SIZE];
do {
while (i < len && pos < buf.length - 3){
char c = value.charAt(i++);
if (c >= '\u0001' && c <= '\u007f')
buf[pos++] = (byte) c;
else if (c == '\u0000' || (c >= '\u0080' && c <= '\u07ff')){
buf[pos++] = (byte) (0xc0 | (0x1f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
else{
buf[pos++] = (byte) (0xe0 | (0x0f & (c >> 12)));
buf[pos++] = (byte) (0x80 | (0x3f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
}
if (! lengthWritten){
if (i == len)
writeShort(pos);
else
writeShort(getUTFlength(value, i, pos));
lengthWritten = true;
}
write(buf, 0, pos);
pos = 0;
}while (i < len);
}
@Override
public void writeString(String value) throws IOException {
int len = value.length();
int i = 0;
int pos = 0;
if (buf == null)
buf = new byte[BUF_SIZE];
do {
while (i < len && pos < buf.length - 4){
char c = value.charAt(i++);
if (c >= '\u0001' && c <= '\u007f')
buf[pos++] = (byte) c;
else if (c == '\u0000' || (c >= '\u0080' && c <= '\u07ff')){
buf[pos++] = (byte) (0xc0 | (0x1f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
else{
buf[pos++] = (byte) (0xe0 | (0x0f & (c >> 12)));
buf[pos++] = (byte) (0x80 | (0x3f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
}
if(i >= len)
buf[pos++] = 0;
write(buf, 0, pos);
pos = 0;
}while (i < len);
}
public static int getUTFlength(String value, int start, int sum) throws IOException {
int len = value.length();
for (int i = start; i < len && sum <= 65535; ++i){
char c = value.charAt(i);
if (c >= '\u0001' && c <= '\u007f')
sum += 1;
else if (c == '\u0000' || (c >= '\u0080' && c <= '\u07ff'))
sum += 2;
else
sum += 3;
}
if (sum > UTF8_STRING_LIMIT)
throw new UTFDataFormatException ();
return sum;
}
@Override
public void writeShortArray(short[] A) throws IOException {
writeShortArray(A, 0, A.length);
}
@Override
public void writeIntArray(int[] A) throws IOException {
writeIntArray(A, 0, A.length);
}
@Override
public void writeLongArray(long[] A) throws IOException {
writeLongArray(A, 0, A.length);
}
@Override
public void writeDoubleArray(double[] A) throws IOException {
writeDoubleArray(A, 0, A.length);
}
@Override
public void writeDoubleArray(double[] A, int startIdx, int len) throws IOException {
if (longBuf == null) {
longBuf = new long[longBufSize];
}
int end = startIdx + len;
int pos = 0;
for (int i = startIdx; i < end; ++i) {
if (pos >= longBufSize) {
writeLongArray(longBuf,0, pos);
pos = 0;
}
longBuf[pos++] = Double.doubleToLongBits(A[i]);
}
if (pos > 0)
writeLongArray(longBuf, 0, pos);
}
@Override
public void writeFloatArray(float[] A) throws IOException {
writeFloatArray(A, 0, A.length);
}
@Override
public void writeFloatArray(float[] A, int startIdx, int len) throws IOException {
if (intBuf == null) {
intBuf = new int[intBufSize];
}
int end = startIdx + len;
int pos = 0;
for (int i = startIdx; i < end; ++i) {
if (pos >= intBufSize) {
writeIntArray(intBuf,0, pos);
pos = 0;
}
intBuf[pos++] = Float.floatToIntBits(A[i]);
}
if (pos > 0)
writeIntArray(intBuf,0, pos);
}
@Override
public void writeStringArray(String[] A) throws IOException {
writeStringArray(A, 0, A.length);
}
@Override
public void writeStringArray(String[] A, int startIdx, int len) throws IOException {
if (buf == null)
buf = new byte[BUF_SIZE];
int end = startIdx + len;
int pos = 0;
for (int j = startIdx; j < end; ++j) {
String value = A[j];
int valueLen = value.length();
int i = 0;
do {
while (i < valueLen && pos < buf.length - 4){
char c = value.charAt(i++);
if (c >= '\u0001' && c <= '\u007f')
buf[pos++] = (byte) c;
else if (c == '\u0000' || (c >= '\u0080' && c <= '\u07ff')){
buf[pos++] = (byte) (0xc0 | (0x1f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
else{
buf[pos++] = (byte) (0xe0 | (0x0f & (c >> 12)));
buf[pos++] = (byte) (0x80 | (0x3f & (c >> 6)));
buf[pos++] = (byte) (0x80 | (0x3f & c));
}
}
if(i >= valueLen)
buf[pos++] = 0;
if (pos + 4 >= buf.length) {
write(buf, 0, pos);
pos = 0;
}
}while (i < valueLen);
}
if (pos > 0)
write(buf, 0, pos);
}
@Override
public void writeLong2Array(Long2[] A) throws IOException {
writeLong2Array(A, 0, A.length);
}
}