Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.qpid.proton.codec.EncoderImpl 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.ByteBuffer;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import org.apache.qpid.proton.amqp.Binary;
import org.apache.qpid.proton.amqp.Decimal128;
import org.apache.qpid.proton.amqp.Decimal32;
import org.apache.qpid.proton.amqp.Decimal64;
import org.apache.qpid.proton.amqp.DescribedType;
import org.apache.qpid.proton.amqp.Symbol;
import org.apache.qpid.proton.amqp.UnsignedByte;
import org.apache.qpid.proton.amqp.UnsignedInteger;
import org.apache.qpid.proton.amqp.UnsignedLong;
import org.apache.qpid.proton.amqp.UnsignedShort;
public final class EncoderImpl implements ByteBufferEncoder
{
private static final byte DESCRIBED_TYPE_OP = (byte)0;
private WritableBuffer _buffer;
private final DecoderImpl _decoder;
private final Map _typeRegistry = new HashMap();
private Map _describedDescriptorRegistry = new HashMap();
private Map _describedTypesClassRegistry = new HashMap();
private final NullType _nullType;
private final BooleanType _booleanType;
private final ByteType _byteType;
private final UnsignedByteType _unsignedByteType;
private final ShortType _shortType;
private final UnsignedShortType _unsignedShortType;
private final IntegerType _integerType;
private final UnsignedIntegerType _unsignedIntegerType;
private final LongType _longType;
private final UnsignedLongType _unsignedLongType;
private final BigIntegerType _bigIntegerType;
private final CharacterType _characterType;
private final FloatType _floatType;
private final DoubleType _doubleType;
private final TimestampType _timestampType;
private final UUIDType _uuidType;
private final Decimal32Type _decimal32Type;
private final Decimal64Type _decimal64Type;
private final Decimal128Type _decimal128Type;
private final BinaryType _binaryType;
private final SymbolType _symbolType;
private final StringType _stringType;
private final ListType _listType;
private final MapType _mapType;
private final ArrayType _arrayType;
EncoderImpl(ByteBuffer buffer, DecoderImpl decoder)
{
this(decoder);
setByteBuffer(buffer);
}
public EncoderImpl(DecoderImpl decoder)
{
_decoder = decoder;
_nullType = new NullType(this, decoder);
_booleanType = new BooleanType(this, decoder);
_byteType = new ByteType(this, decoder);
_unsignedByteType = new UnsignedByteType(this, decoder);
_shortType = new ShortType(this, decoder);
_unsignedShortType = new UnsignedShortType(this, decoder);
_integerType = new IntegerType(this, decoder);
_unsignedIntegerType = new UnsignedIntegerType(this, decoder);
_longType = new LongType(this, decoder);
_unsignedLongType = new UnsignedLongType(this, decoder);
_bigIntegerType = new BigIntegerType(this, decoder);
_characterType = new CharacterType(this, decoder);
_floatType = new FloatType(this, decoder);
_doubleType = new DoubleType(this, decoder);
_timestampType = new TimestampType(this, decoder);
_uuidType = new UUIDType(this, decoder);
_decimal32Type = new Decimal32Type(this, decoder);
_decimal64Type = new Decimal64Type(this, decoder);
_decimal128Type = new Decimal128Type(this, decoder);
_binaryType = new BinaryType(this, decoder);
_symbolType = new SymbolType(this, decoder);
_stringType = new StringType(this, decoder);
_listType = new ListType(this, decoder);
_mapType = new MapType(this, decoder);
_arrayType = new ArrayType(this,
decoder,
_booleanType,
_byteType,
_shortType,
_integerType,
_longType,
_floatType,
_doubleType,
_characterType);
}
public void setByteBuffer(final ByteBuffer buf)
{
_buffer = new WritableBuffer.ByteBufferWrapper(buf);
}
public void setByteBuffer(final WritableBuffer buf)
{
_buffer = buf;
}
public WritableBuffer getBuffer()
{
return _buffer;
}
public DecoderImpl getDecoder()
{
return _decoder;
}
@Override
public AMQPType getType(final Object element)
{
return getTypeFromClass(element == null ? Void.class : element.getClass(), element);
}
public AMQPType getTypeFromClass(final Class clazz)
{
return getTypeFromClass(clazz, null);
}
private AMQPType getTypeFromClass(final Class clazz, Object instance)
{
AMQPType amqpType = _typeRegistry.get(clazz);
if(amqpType == null)
{
if(clazz.isArray())
{
amqpType = _arrayType;
}
else
{
if(List.class.isAssignableFrom(clazz))
{
amqpType = _listType;
}
else if(Map.class.isAssignableFrom(clazz))
{
amqpType = _mapType;
}
else if(DescribedType.class.isAssignableFrom(clazz))
{
amqpType = _describedTypesClassRegistry.get(clazz);
if(amqpType == null && instance != null)
{
Object descriptor = ((DescribedType) instance).getDescriptor();
amqpType = _describedDescriptorRegistry.get(descriptor);
if(amqpType == null)
{
amqpType = new DynamicDescribedType(this, descriptor);
_describedDescriptorRegistry.put(descriptor, amqpType);
}
}
return amqpType;
}
}
_typeRegistry.put(clazz, amqpType);
}
return amqpType;
}
@Override
public void register(AMQPType type)
{
register(type.getTypeClass(), type);
}
void register(Class clazz, AMQPType type)
{
_typeRegistry.put(clazz, type);
}
public void registerDescribedType(Class clazz, Object descriptor)
{
AMQPType type = _describedDescriptorRegistry.get(descriptor);
if(type == null)
{
type = new DynamicDescribedType(this, descriptor);
_describedDescriptorRegistry.put(descriptor, type);
}
_describedTypesClassRegistry.put(clazz, type);
}
public void writeNull()
{
_buffer.put(EncodingCodes.NULL);
}
public void writeBoolean(final boolean bool)
{
if (bool)
{
_buffer.put(EncodingCodes.BOOLEAN_TRUE);
}
else
{
_buffer.put(EncodingCodes.BOOLEAN_FALSE);
}
}
public void writeBoolean(final Boolean bool)
{
if(bool == null)
{
writeNull();
}
else if (Boolean.TRUE.equals(bool))
{
_buffer.put(EncodingCodes.BOOLEAN_TRUE);
}
else
{
_buffer.put(EncodingCodes.BOOLEAN_FALSE);
}
}
public void writeUnsignedByte(final UnsignedByte ubyte)
{
if(ubyte == null)
{
writeNull();
}
else
{
_unsignedByteType.fastWrite(this, ubyte);
}
}
public void writeUnsignedShort(final UnsignedShort ushort)
{
if(ushort == null)
{
writeNull();
}
else
{
_unsignedShortType.fastWrite(this, ushort);
}
}
public void writeUnsignedInteger(final UnsignedInteger uint)
{
if(uint == null)
{
writeNull();
}
else
{
_unsignedIntegerType.fastWrite(this, uint);
}
}
public void writeUnsignedLong(final UnsignedLong ulong)
{
if(ulong == null)
{
writeNull();
}
else
{
_unsignedLongType.fastWrite(this, ulong);
}
}
public void writeByte(final byte b)
{
_byteType.write(b);
}
public void writeByte(final Byte b)
{
if(b == null)
{
writeNull();
}
else
{
writeByte(b.byteValue());
}
}
public void writeShort(final short s)
{
_shortType.write(s);
}
public void writeShort(final Short s)
{
if(s == null)
{
writeNull();
}
else
{
writeShort(s.shortValue());
}
}
public void writeInteger(final int i)
{
_integerType.write(i);
}
public void writeInteger(final Integer i)
{
if(i == null)
{
writeNull();
}
else
{
writeInteger(i.intValue());
}
}
public void writeLong(final long l)
{
_longType.write(l);
}
public void writeLong(final Long l)
{
if(l == null)
{
writeNull();
}
else
{
writeLong(l.longValue());
}
}
public void writeFloat(final float f)
{
_floatType.write(f);
}
public void writeFloat(final Float f)
{
if(f == null)
{
writeNull();
}
else
{
writeFloat(f.floatValue());
}
}
public void writeDouble(final double d)
{
_doubleType.write(d);
}
public void writeDouble(final Double d)
{
if(d == null)
{
writeNull();
}
else
{
writeDouble(d.doubleValue());
}
}
public void writeDecimal32(final Decimal32 d)
{
if(d == null)
{
writeNull();
}
else
{
_decimal32Type.write(d);
}
}
public void writeDecimal64(final Decimal64 d)
{
if(d == null)
{
writeNull();
}
else
{
_decimal64Type.write(d);
}
}
public void writeDecimal128(final Decimal128 d)
{
if(d == null)
{
writeNull();
}
else
{
_decimal128Type.write(d);
}
}
public void writeCharacter(final char c)
{
// TODO - java character may be half of a pair, should probably throw exception then
_characterType.write(c);
}
public void writeCharacter(final Character c)
{
if(c == null)
{
writeNull();
}
else
{
writeCharacter(c.charValue());
}
}
public void writeTimestamp(final long timestamp)
{
_timestampType.fastWrite(this, timestamp);
}
public void writeTimestamp(final Date d)
{
if(d == null)
{
writeNull();
}
else
{
_timestampType.fastWrite(this, d.getTime());
}
}
public void writeUUID(final UUID uuid)
{
if(uuid == null)
{
writeNull();
}
else
{
_uuidType.fastWrite(this, uuid);
}
}
public void writeBinary(final Binary b)
{
if(b == null)
{
writeNull();
}
else
{
_binaryType.fastWrite(this, b);
}
}
public void writeString(final String s)
{
if(s == null)
{
writeNull();
}
else
{
_stringType.write(s);
}
}
public void writeSymbol(final Symbol s)
{
if(s == null)
{
writeNull();
}
else
{
_symbolType.fastWrite(this, s);
}
}
public void writeList(final List l)
{
if(l == null)
{
writeNull();
}
else
{
_listType.write(l);
}
}
public void writeMap(final Map m)
{
if(m == null)
{
writeNull();
}
else
{
_mapType.write(m);
}
}
public void writeDescribedType(final DescribedType d)
{
if(d == null)
{
writeNull();
}
else
{
_buffer.put(DESCRIBED_TYPE_OP);
writeObject(d.getDescriptor());
writeObject(d.getDescribed());
}
}
public void writeArray(final boolean[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final byte[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final short[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final int[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final long[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final float[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final double[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final char[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeArray(final Object[] a)
{
if(a == null)
{
writeNull();
}
else
{
_arrayType.write(a);
}
}
public void writeObject(final Object o)
{
if (o == null)
{
getBuffer().put(EncodingCodes.NULL);
return;
}
AMQPType type = _typeRegistry.get(o.getClass());
if(type == null)
{
if(o.getClass().isArray())
{
writeArrayType(o);
}
else if(o instanceof List)
{
writeList((List)o);
}
else if(o instanceof Map)
{
writeMap((Map)o);
}
else if(o instanceof DescribedType)
{
writeDescribedType((DescribedType)o);
}
else
{
throw new IllegalArgumentException(
"Do not know how to write Objects of class " + o.getClass().getName());
}
}
else
{
type.write(o);
}
}
private void writeArrayType(Object array) {
Class componentType = array.getClass().getComponentType();
if(componentType.isPrimitive())
{
if(componentType == Boolean.TYPE)
{
writeArray((boolean[])array);
}
else if(componentType == Byte.TYPE)
{
writeArray((byte[])array);
}
else if(componentType == Short.TYPE)
{
writeArray((short[])array);
}
else if(componentType == Integer.TYPE)
{
writeArray((int[])array);
}
else if(componentType == Long.TYPE)
{
writeArray((long[])array);
}
else if(componentType == Float.TYPE)
{
writeArray((float[])array);
}
else if(componentType == Double.TYPE)
{
writeArray((double[])array);
}
else if(componentType == Character.TYPE)
{
writeArray((char[])array);
}
else
{
throw new IllegalArgumentException("Cannot write arrays of type " + componentType.getName());
}
}
else
{
writeArray((Object[]) array);
}
}
public void writeRaw(final byte b)
{
_buffer.put(b);
}
void writeRaw(final short s)
{
_buffer.putShort(s);
}
void writeRaw(final int i)
{
_buffer.putInt(i);
}
void writeRaw(final long l)
{
_buffer.putLong(l);
}
void writeRaw(final float f)
{
_buffer.putFloat(f);
}
void writeRaw(final double d)
{
_buffer.putDouble(d);
}
void writeRaw(byte[] src, int offset, int length)
{
_buffer.put(src, offset, length);
}
void writeRaw(String string)
{
final int length = string.length();
int c;
for (int i = 0; i < length; i++)
{
c = string.charAt(i);
if ((c & 0xFF80) == 0) /* U+0000..U+007F */
{
_buffer.put((byte) c);
}
else if ((c & 0xF800) == 0) /* U+0080..U+07FF */
{
_buffer.put((byte)(0xC0 | ((c >> 6) & 0x1F)));
_buffer.put((byte)(0x80 | (c & 0x3F)));
}
else if ((c & 0xD800) != 0xD800 || (c > 0xDBFF)) /* U+0800..U+FFFF - excluding surrogate pairs */
{
_buffer.put((byte)(0xE0 | ((c >> 12) & 0x0F)));
_buffer.put((byte)(0x80 | ((c >> 6) & 0x3F)));
_buffer.put((byte)(0x80 | (c & 0x3F)));
}
else
{
int low;
if((++i == length) || ((low = string.charAt(i)) & 0xDC00) != 0xDC00)
{
throw new IllegalArgumentException("String contains invalid Unicode code points");
}
c = 0x010000 + ((c & 0x03FF) << 10) + (low & 0x03FF);
_buffer.put((byte)(0xF0 | ((c >> 18) & 0x07)));
_buffer.put((byte)(0x80 | ((c >> 12) & 0x3F)));
_buffer.put((byte)(0x80 | ((c >> 6) & 0x3F)));
_buffer.put((byte)(0x80 | (c & 0x3F)));
}
}
}
AMQPType getNullTypeEncoder()
{
return _nullType;
}
}
