com.dyuproject.protostuff.runtime.RuntimeUnsafeFieldFactory Maven / Gradle / Ivy

Go to download
//========================================================================
//Copyright 2007-2011 David Yu [email protected]
//------------------------------------------------------------------------
//Licensed 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 com.dyuproject.protostuff.runtime;

import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BIGDECIMAL;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BIGINTEGER;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BOOL;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BYTE;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BYTES;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_BYTE_ARRAY;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_CHAR;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_DATE;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_DELEGATE;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_DOUBLE;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_ENUM;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_FLOAT;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_INT32;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_INT64;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_OBJECT;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_POJO;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_SHORT;
import static com.dyuproject.protostuff.runtime.RuntimeFieldFactory.ID_STRING;

import java.io.IOException;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.Date;

import com.dyuproject.protostuff.ByteString;
import com.dyuproject.protostuff.GraphInput;
import com.dyuproject.protostuff.Input;
import com.dyuproject.protostuff.Morph;
import com.dyuproject.protostuff.Output;
import com.dyuproject.protostuff.Pipe;
import com.dyuproject.protostuff.Schema;
import com.dyuproject.protostuff.Tag;
import com.dyuproject.protostuff.WireFormat.FieldType;
import com.dyuproject.protostuff.runtime.MappedSchema.Field;

/**
 * Field factory via sun.misc.Unsafe.
 *
 * @author David Yu
 * @created Jul 7, 2011
 */
public final class RuntimeUnsafeFieldFactory
{
    
    static final sun.misc.Unsafe us = initUnsafe();
    
    private static sun.misc.Unsafe initUnsafe()
    {
        try
        {
            java.lang.reflect.Field f = 
                sun.misc.Unsafe.class.getDeclaredField("theUnsafe");
            
            f.setAccessible(true);
            
            return (sun.misc.Unsafe)f.get(null);
        }
        catch(Exception e)
        {
            // ignore
            
            /* android 3.x
            try
            {
                java.lang.reflect.Field f = 
                        sun.misc.Unsafe.class.getDeclaredField("THE_ONE");
                    
                f.setAccessible(true);
                
                return (sun.misc.Unsafe)f.get(null);
            }
            catch(Exception e1)
            {
                // ignore
            }*/
        }
        
        return sun.misc.Unsafe.getUnsafe();
    }
    
    private RuntimeUnsafeFieldFactory() {}
    
    public static final RuntimeFieldFactory CHAR = new RuntimeFieldFactory(ID_CHAR)
    {
        
        
        public  Field create(int number, String name, final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.UINT32, number, name, 
                    f.getAnnotation(Tag.class))
            {                 
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putChar(message, offset, (char)input.readUInt32());
                    else
                        us.putObject(message, offset, Character.valueOf((char)input.readUInt32()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeUInt32(number, us.getChar(message, offset), false);
                    else
                    {
                        Character value = (Character)us.getObject(message, offset);
                        if(value!=null)
                            output.writeUInt32(number, value.charValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeUInt32(number, input.readUInt32(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeUInt32(number, input.readUInt32(), repeated);
        }
        public Character readFrom(Input input) throws IOException
        {
            return Character.valueOf((char)input.readUInt32());
        }
        public void writeTo(Output output, int number, Character value, boolean repeated) 
        throws IOException
        {
            output.writeUInt32(number, value.charValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.UINT32;
        }
        public Class typeClass()
        {
            return Character.class;
        }
    };
    
    public static final RuntimeFieldFactory SHORT = new RuntimeFieldFactory(ID_SHORT)
    {
        public  Field create(int number, String name, final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.UINT32, number, name, 
                    f.getAnnotation(Tag.class))
            {
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putShort(message, offset, (short)input.readUInt32());
                    else
                        us.putObject(message, offset, Short.valueOf((short)input.readUInt32()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeUInt32(number, us.getShort(message, offset), false);
                    else
                    {
                        Short value = (Short)us.getObject(message, offset);
                        if(value!=null)
                            output.writeUInt32(number, value.shortValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeUInt32(number, input.readUInt32(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeUInt32(number, input.readUInt32(), repeated);
        }
        public Short readFrom(Input input) throws IOException
        {
            return Short.valueOf((short)input.readUInt32());
        }
        public void writeTo(Output output, int number, Short value, boolean repeated) 
        throws IOException
        {
            output.writeUInt32(number, value.shortValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.UINT32;
        }
        public Class typeClass()
        {
            return Short.class;
        }
    };
    
    public static final RuntimeFieldFactory BYTE = new RuntimeFieldFactory(ID_BYTE)
    {
        public  Field create(int number, String name, final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.UINT32, number, name, 
                    f.getAnnotation(Tag.class))
            {                  
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putByte(message, offset, (byte)input.readUInt32());
                    else
                        us.putObject(message, offset, Byte.valueOf((byte)input.readUInt32()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeUInt32(number, us.getByte(message, offset), false);
                    else
                    {
                        Byte value = (Byte)us.getObject(message, offset);
                        if(value!=null)
                            output.writeUInt32(number, value.byteValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeUInt32(number, input.readUInt32(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeUInt32(number, input.readUInt32(), repeated);
        }
        public Byte readFrom(Input input) throws IOException
        {
            return Byte.valueOf((byte)input.readUInt32());
        }
        public void writeTo(Output output, int number, Byte value, boolean repeated) 
        throws IOException
        {
            output.writeUInt32(number, value.byteValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.UINT32;
        }
        public Class typeClass()
        {
            return Byte.class;
        }
    };
    
    public static final RuntimeFieldFactory INT32 = new RuntimeFieldFactory(ID_INT32)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.INT32, number, name, 
                    f.getAnnotation(Tag.class))
            {
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putInt(message, offset, input.readInt32());
                    else
                        us.putObject(message, offset, Integer.valueOf(input.readInt32()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeInt32(number, us.getInt(message, offset), false);
                    else
                    {
                        Integer value = (Integer)us.getObject(message, offset);
                        if(value!=null)
                            output.writeInt32(number, value.intValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeInt32(number, input.readInt32(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeInt32(number, input.readInt32(), repeated);
        }
        public Integer readFrom(Input input) throws IOException
        {
            return Integer.valueOf(input.readInt32());
        }
        public void writeTo(Output output, int number, Integer value, boolean repeated) 
        throws IOException
        {
            output.writeInt32(number, value.intValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.INT32;
        }
        public Class typeClass()
        {
            return Integer.class;
        }
    };
    
    public static final RuntimeFieldFactory INT64 = new RuntimeFieldFactory(ID_INT64)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.INT64, number, name, 
                    f.getAnnotation(Tag.class))
            {                  
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putLong(message, offset, input.readInt64());
                    else
                        us.putObject(message, offset, Long.valueOf(input.readInt64()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeInt64(number, us.getLong(message, offset), false);
                    else
                    {
                        Long value = (Long)us.getObject(message, offset);
                        if(value!=null)
                            output.writeInt64(number, value.longValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeInt64(number, input.readInt64(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeInt64(number, input.readInt64(), repeated);
        }
        public Long readFrom(Input input) throws IOException
        {
            return Long.valueOf(input.readInt64());
        }
        public void writeTo(Output output, int number, Long value, boolean repeated) 
        throws IOException
        {
            output.writeInt64(number, value.longValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.INT64;
        }
        public Class typeClass()
        {
            return Long.class;
        }
    };
    
    public static final RuntimeFieldFactory FLOAT = new RuntimeFieldFactory(ID_FLOAT)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.FLOAT, number, name, 
                    f.getAnnotation(Tag.class))
            {                   
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putFloat(message, offset, input.readFloat());
                    else
                        us.putObject(message, offset, new Float(input.readFloat()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeFloat(number, us.getFloat(message, offset), false);
                    else
                    {
                        Float value = (Float)us.getObject(message, offset);
                        if(value!=null)
                            output.writeFloat(number, value.floatValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeFloat(number, input.readFloat(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeFloat(number, input.readFloat(), repeated);
        }
        public Float readFrom(Input input) throws IOException
        {
            return new Float(input.readFloat());
        }
        public void writeTo(Output output, int number, Float value, boolean repeated) 
        throws IOException
        {
            output.writeFloat(number, value.floatValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.FLOAT;
        }
        public Class typeClass()
        {
            return Float.class;
        }
    };
    
    public static final RuntimeFieldFactory DOUBLE = new RuntimeFieldFactory(ID_DOUBLE)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.DOUBLE, number, name, 
                    f.getAnnotation(Tag.class))
            {                   
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putDouble(message, offset, input.readDouble());
                    else
                        us.putObject(message, offset, new Double(input.readDouble()));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeDouble(number, us.getDouble(message, offset), false);
                    else
                    {
                        Double value = (Double)us.getObject(message, offset);
                        if(value!=null)
                            output.writeDouble(number, value.doubleValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeDouble(number, input.readDouble(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeDouble(number, input.readDouble(), repeated);
        }
        public Double readFrom(Input input) throws IOException
        {
            return new Double(input.readDouble());
        }
        public void writeTo(Output output, int number, Double value, boolean repeated) 
        throws IOException
        {
            output.writeDouble(number, value.doubleValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.DOUBLE;
        }
        public Class typeClass()
        {
            return Double.class;
        }
    };
    
    public static final RuntimeFieldFactory BOOL = new RuntimeFieldFactory(ID_BOOL)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final boolean primitive = f.getType().isPrimitive();
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.BOOL, number, name, 
                    f.getAnnotation(Tag.class))
            {              
                public void mergeFrom(Input input, T message) throws IOException
                {
                    if(primitive)
                        us.putBoolean(message, offset, input.readBool());
                    else
                        us.putObject(message, offset, input.readBool() ? Boolean.TRUE : Boolean.FALSE);
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    if(primitive)
                        output.writeBool(number, us.getBoolean(message, offset), false);
                    else
                    {
                        Boolean value = (Boolean)us.getObject(message, offset);
                        if(value!=null)
                            output.writeBool(number, value.booleanValue(), false);
                    }
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    output.writeBool(number, input.readBool(), repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            output.writeBool(number, input.readBool(), repeated);
        }
        public Boolean readFrom(Input input) throws IOException
        {
            return input.readBool() ? Boolean.TRUE : Boolean.FALSE;
        }
        public void writeTo(Output output, int number, Boolean value, boolean repeated) 
        throws IOException
        {
            output.writeBool(number, value.booleanValue(), repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.BOOL;
        }
        public Class typeClass()
        {
            return Boolean.class;
        }
    };
    
    public static final RuntimeFieldFactory STRING = new RuntimeFieldFactory(ID_STRING)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.STRING, number, name, 
                    f.getAnnotation(Tag.class))
            {                  
                public void mergeFrom(Input input, T message) throws IOException
                {
                    us.putObject(message, offset, input.readString());
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    String value = (String)us.getObject(message, offset);
                    if(value!=null)
                        output.writeString(number, value, false);
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    input.transferByteRangeTo(output, true, number, repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            input.transferByteRangeTo(output, true, number, repeated);
        }
        public String readFrom(Input input) throws IOException
        {
            return input.readString();
        }
        public void writeTo(Output output, int number, String value, boolean repeated) 
        throws IOException
        {
            output.writeString(number, value, repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.STRING;
        }
        public Class typeClass()
        {
            return String.class;
        }
    };
    
    public static final RuntimeFieldFactory BYTES = new RuntimeFieldFactory(ID_BYTES)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.BYTES, number, name, 
                    f.getAnnotation(Tag.class))
            {                  
                public void mergeFrom(Input input, T message) throws IOException
                {
                    us.putObject(message, offset, input.readBytes());
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    ByteString bs = (ByteString)us.getObject(message, offset);
                    if(bs!=null)
                        output.writeBytes(number, bs, false);
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    input.transferByteRangeTo(output, false, number, repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            input.transferByteRangeTo(output, false, number, repeated);
        }
        public ByteString readFrom(Input input) throws IOException
        {
            return input.readBytes();
        }
        public void writeTo(Output output, int number, ByteString value, boolean repeated) 
        throws IOException
        {
            output.writeBytes(number, value, repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.BYTES;
        }
        public Class typeClass()
        {
            return ByteString.class;
        }
    };
    
    public static final RuntimeFieldFactory BYTE_ARRAY = new RuntimeFieldFactory(ID_BYTE_ARRAY)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.BYTES, number, name, 
                    f.getAnnotation(Tag.class))
            {                 
                public void mergeFrom(Input input, T message) throws IOException
                {
                    us.putObject(message, offset, input.readByteArray());
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    byte[] array = (byte[])us.getObject(message, offset);
                    if(array!=null)
                        output.writeByteArray(number, array, false);
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    input.transferByteRangeTo(output, false, number, repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            input.transferByteRangeTo(output, false, number, repeated);
        }
        public byte[] readFrom(Input input) throws IOException
        {
            return input.readByteArray();
        }
        public void writeTo(Output output, int number, byte[] value, boolean repeated) 
        throws IOException
        {
            output.writeByteArray(number, value, repeated);
        }
        public FieldType getFieldType()
        {
            return FieldType.BYTES;
        }
        public Class typeClass()
        {
            return byte[].class;
        }
    };
    
    public static final RuntimeFieldFactory ENUM = new RuntimeFieldFactory(ID_ENUM)
    {
        public  Field create(int number, java.lang.String name, 
                final java.lang.reflect.Field f, IdStrategy strategy)
        {
            final EnumIO> eio = strategy.getEnumIO(f.getType());
            final long offset = us.objectFieldOffset(f);
            return new Field(FieldType.ENUM, number, name, 
                    f.getAnnotation(Tag.class))
            { 
                public void mergeFrom(Input input, T message) throws IOException
                {
                    us.putObject(message, offset, eio.readFrom(input));
                }
                public void writeTo(Output output, T message) throws IOException
                {
                    final Enum existing = (Enum)us.getObject(message, offset);
                    if(existing != null)
                        EnumIO.writeTo(output, number, repeated, existing);
                }
                public void transfer(Pipe pipe, Input input, Output output, 
                        boolean repeated) throws IOException
                {
                    EnumIO.transfer(pipe, input, output, number, repeated);
                }
            };
        }
        public void transfer(Pipe pipe, Input input, Output output, int number, 
                boolean repeated) throws IOException
        {
            throw new UnsupportedOperationException();
        }
        public Integer readFrom(Input input) throws IOException
        {
            throw new UnsupportedOperationException();
        }
        public void writeTo(Output output, int number, Integer value, boolean repeated) throws IOException
        {
            throw new UnsupportedOperationException();
        }
        public FieldType getFieldType()
        {
            throw new UnsupportedOperationException();
        }
        public Class typeClass()
        {
            throw new UnsupportedOperationException();
        }
    };
    
    // NON-INLINE VALUES
    
    static final RuntimeFieldFactory