org.apache.qpid.proton.codec.WritableBuffer Maven / Gradle / Ivy
/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
package org.apache.qpid.proton.codec;
import java.nio.BufferOverflowException;
import java.nio.ByteBuffer;
public interface WritableBuffer {
void put(byte b);
void putFloat(float f);
void putDouble(double d);
void put(byte[] src, int offset, int length);
void putShort(short s);
void putInt(int i);
void putLong(long l);
boolean hasRemaining();
int remaining();
int position();
void position(int position);
void put(ByteBuffer payload);
void put(ReadableBuffer payload);
default void put(final String value) {
final int length = value.length();
for (int i = 0; i < length; i++) {
int c = value.charAt(i);
if ((c & 0xFF80) == 0) {
// U+0000..U+007F
put((byte) c);
} else if ((c & 0xF800) == 0) {
// U+0080..U+07FF
put((byte) (0xC0 | ((c >> 6) & 0x1F)));
put((byte) (0x80 | (c & 0x3F)));
} else if ((c & 0xD800) != 0xD800 || (c > 0xDBFF)) {
// U+0800..U+FFFF - excluding surrogate pairs
put((byte) (0xE0 | ((c >> 12) & 0x0F)));
put((byte) (0x80 | ((c >> 6) & 0x3F)));
put((byte) (0x80 | (c & 0x3F)));
} else {
int low;
if ((++i == length) || ((low = value.charAt(i)) & 0xDC00) != 0xDC00) {
throw new IllegalArgumentException("String contains invalid Unicode code points");
}
c = 0x010000 + ((c & 0x03FF) << 10) + (low & 0x03FF);
put((byte) (0xF0 | ((c >> 18) & 0x07)));
put((byte) (0x80 | ((c >> 12) & 0x3F)));
put((byte) (0x80 | ((c >> 6) & 0x3F)));
put((byte) (0x80 | (c & 0x3F)));
}
}
}
int limit();
class ByteBufferWrapper implements WritableBuffer {
private final ByteBuffer _buf;
public ByteBufferWrapper(ByteBuffer buf) {
_buf = buf;
}
@Override
public void put(byte b) {
_buf.put(b);
}
@Override
public void putFloat(float f) {
_buf.putFloat(f);
}
@Override
public void putDouble(double d) {
_buf.putDouble(d);
}
@Override
public void put(byte[] src, int offset, int length) {
_buf.put(src, offset, length);
}
@Override
public void putShort(short s) {
_buf.putShort(s);
}
@Override
public void putInt(int i) {
_buf.putInt(i);
}
@Override
public void putLong(long l) {
_buf.putLong(l);
}
@Override
public boolean hasRemaining() {
return _buf.hasRemaining();
}
@Override
public int remaining() {
return _buf.remaining();
}
@Override
public int position() {
return _buf.position();
}
@Override
public void position(int position) {
_buf.position(position);
}
@Override
public void put(ByteBuffer src) {
_buf.put(src);
}
@Override
public void put(ReadableBuffer src) {
src.get(this);
}
@Override
public void put(final String value) {
final int length = value.length();
int pos = _buf.position();
for (int i = 0; i < length; i++) {
int c = value.charAt(i);
try {
if ((c & 0xFF80) == 0) {
// U+0000..U+007F
put(pos++, (byte) c);
} else if ((c & 0xF800) == 0) {
// U+0080..U+07FF
put(pos++, (byte) (0xC0 | ((c >> 6) & 0x1F)));
put(pos++, (byte) (0x80 | (c & 0x3F)));
} else if ((c & 0xD800) != 0xD800 || (c > 0xDBFF)) {
// U+0800..U+FFFF - excluding surrogate pairs
put(pos++, (byte) (0xE0 | ((c >> 12) & 0x0F)));
put(pos++, (byte) (0x80 | ((c >> 6) & 0x3F)));
put(pos++, (byte) (0x80 | (c & 0x3F)));
} else {
int low;
if ((++i == length) || ((low = value.charAt(i)) & 0xDC00) != 0xDC00) {
throw new IllegalArgumentException("String contains invalid Unicode code points");
}
c = 0x010000 + ((c & 0x03FF) << 10) + (low & 0x03FF);
put(pos++, (byte) (0xF0 | ((c >> 18) & 0x07)));
put(pos++, (byte) (0x80 | ((c >> 12) & 0x3F)));
put(pos++, (byte) (0x80 | ((c >> 6) & 0x3F)));
put(pos++, (byte) (0x80 | (c & 0x3F)));
}
}
catch(IndexOutOfBoundsException ioobe) {
throw new BufferOverflowException();
}
}
// Now move the buffer position to reflect the work done here
_buf.position(pos);
}
@Override
public int limit() {
return _buf.limit();
}
public ByteBuffer byteBuffer() {
return _buf;
}
public ReadableBuffer toReadableBuffer() {
return ReadableBuffer.ByteBufferReader.wrap((ByteBuffer) _buf.duplicate().flip());
}
@Override
public String toString() {
return String.format("[pos: %d, limit: %d, remaining:%d]", _buf.position(), _buf.limit(), _buf.remaining());
}
public static ByteBufferWrapper allocate(int size) {
ByteBuffer allocated = ByteBuffer.allocate(size);
return new ByteBufferWrapper(allocated);
}
public static ByteBufferWrapper wrap(ByteBuffer buffer) {
return new ByteBufferWrapper(buffer);
}
public static ByteBufferWrapper wrap(byte[] bytes) {
return new ByteBufferWrapper(ByteBuffer.wrap(bytes));
}
private void put(int index, byte value) {
if (_buf.hasArray()) {
_buf.array()[_buf.arrayOffset() + index] = value;
} else {
_buf.put(index, value);
}
}
}
}