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.
com.dyuproject.protostuff.runtime.RuntimeRepeatedFieldFactory Maven / Gradle / Ivy
//========================================================================
//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 java.io.IOException;
import java.util.Collection;
import java.util.EnumSet;
import com.dyuproject.protostuff.CollectionSchema.MessageFactory;
import com.dyuproject.protostuff.GraphInput;
import com.dyuproject.protostuff.Input;
import com.dyuproject.protostuff.Morph;
import com.dyuproject.protostuff.Output;
import com.dyuproject.protostuff.Message;
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;
/**
* Static utility for creating runtime repeated (list/collection) fields.
*
* @author David Yu
* @created Jan 23, 2011
*/
final class RuntimeRepeatedFieldFactory
{
private RuntimeRepeatedFieldFactory() {}
/**
* For lazy initialization called by {@link RuntimeFieldFactory}.
*/
static RuntimeFieldFactory> getFactory()
{
return REPEATED;
}
private static Field createCollectionInlineV(int number, String name,
final java.lang.reflect.Field f, final MessageFactory messageFactory,
final Delegate inline)
{
return new Field(inline.getFieldType(), number, name, true,
f.getAnnotation(Tag.class))
{
{
f.setAccessible(true);
}
@SuppressWarnings("unchecked")
protected void mergeFrom(Input input, T message) throws IOException
{
final Object value = inline.readFrom(input);
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
@SuppressWarnings("unchecked")
protected void writeTo(Output output, T message) throws IOException
{
final Collection collection;
try
{
collection = (Collection)f.get(message);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
if(collection != null && !collection.isEmpty())
{
for(Object o : collection)
{
if(o != null)
inline.writeTo(output, number, o, true);
}
}
}
protected void transfer(Pipe pipe, Input input, Output output,
boolean repeated) throws IOException
{
inline.transfer(pipe, input, output, number, repeated);
}
};
}
private static Field createCollectionEnumV(int number, String name,
final java.lang.reflect.Field f, final MessageFactory messageFactory,
final Class genericType, IdStrategy strategy)
{
final EnumIO eio = strategy.getEnumIO(genericType);
return new Field(FieldType.ENUM, number, name, true,
f.getAnnotation(Tag.class))
{
{
f.setAccessible(true);
}
@SuppressWarnings("unchecked")
protected void mergeFrom(Input input, T message) throws IOException
{
final Enum value = eio.readFrom(input);
try
{
final Collection> existing = (Collection>)f.get(message);
if(existing == null)
{
final Collection> collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
@SuppressWarnings("unchecked")
protected void writeTo(Output output, T message) throws IOException
{
final Collection> collection;
try
{
collection = (Collection>)f.get(message);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
if(collection != null && !collection.isEmpty())
{
for(Enum en : collection)
EnumIO.writeTo(output, number, true, en);
}
}
protected void transfer(Pipe pipe, Input input, Output output,
boolean repeated) throws IOException
{
EnumIO.transfer(pipe, input, output, number, repeated);
}
};
}
private static Field createCollectionPojoV(int number, String name,
final java.lang.reflect.Field f, final MessageFactory messageFactory,
final Class genericType, IdStrategy strategy)
{
return new RuntimeMessageField(
genericType, strategy.getSchemaWrapper(genericType, true),
FieldType.MESSAGE, number, name, true,
f.getAnnotation(Tag.class))
{
{
f.setAccessible(true);
}
@SuppressWarnings("unchecked")
protected void mergeFrom(Input input, T message) throws IOException
{
final Object value = input.mergeObject(null, getSchema());
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
@SuppressWarnings("unchecked")
protected void writeTo(Output output, T message) throws IOException
{
final Collection collection;
try
{
collection = (Collection)f.get(message);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
if(collection != null && !collection.isEmpty())
{
final Schema schema = getSchema();
for(Object o : collection)
{
if(o != null)
output.writeObject(number, o, schema, true);
}
}
}
protected void transfer(Pipe pipe, Input input, Output output,
boolean repeated) throws IOException
{
output.writeObject(number, pipe, getPipeSchema(), repeated);
}
};
}
private static Field createCollectionPolymorphicV(int number, String name,
final java.lang.reflect.Field f, final MessageFactory messageFactory,
final Class genericType, IdStrategy strategy)
{
return new RuntimeDerivativeField(
genericType, FieldType.MESSAGE, number, name, true,
f.getAnnotation(Tag.class),
strategy)
{
{
f.setAccessible(true);
}
@SuppressWarnings("unchecked")
protected void mergeFrom(Input input, T message) throws IOException
{
final Object value = input.mergeObject(message, schema);
if(input instanceof GraphInput &&
((GraphInput)input).isCurrentMessageReference())
{
// a reference from polymorphic+cyclic graph deser
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
}
@SuppressWarnings("unchecked")
protected void writeTo(Output output, T message) throws IOException
{
final Collection existing;
try
{
existing = (Collection)f.get(message);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
if(existing != null && !existing.isEmpty())
{
for(Object o : existing)
{
if(o != null)
output.writeObject(number, o, schema, true);
}
}
}
protected void transfer(Pipe pipe, Input input, Output output,
boolean repeated) throws IOException
{
output.writeObject(number, pipe, schema.pipeSchema, repeated);
}
@SuppressWarnings("unchecked")
protected void doMergeFrom(Input input, Schema schema,
Object message) throws IOException
{
final Object value = schema.newMessage();
if(input instanceof GraphInput)
{
// update the actual reference.
((GraphInput)input).updateLast(value, message);
}
schema.mergeFrom(input, value);
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
};
}
private static Field createCollectionObjectV(int number, String name,
final java.lang.reflect.Field f, final MessageFactory messageFactory,
PolymorphicSchema.Factory factory, IdStrategy strategy)
{
return new RuntimeObjectField(
FieldType.MESSAGE, number, name, true,
f.getAnnotation(Tag.class),
factory, strategy)
{
{
f.setAccessible(true);
}
@SuppressWarnings("unchecked")
protected void mergeFrom(Input input, T message) throws IOException
{
final Object value = input.mergeObject(message, schema);
if(input instanceof GraphInput &&
((GraphInput)input).isCurrentMessageReference())
{
// a reference from polymorphic+cyclic graph deser
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
}
@SuppressWarnings("unchecked")
protected void writeTo(Output output, T message) throws IOException
{
final Collection existing;
try
{
existing = (Collection)f.get(message);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
if(existing != null && !existing.isEmpty())
{
for(Object o : existing)
{
if(o != null)
output.writeObject(number, o, schema, true);
}
}
}
protected void transfer(Pipe pipe, Input input, Output output,
boolean repeated) throws IOException
{
output.writeObject(number, pipe, schema.getPipeSchema(), repeated);
}
@SuppressWarnings("unchecked")
public void setValue(Object value, Object message)
{
try
{
final Collection existing = (Collection)f.get(message);
if(existing == null)
{
final Collection collection = messageFactory.newMessage();
collection.add(value);
f.set(message, collection);
}
else
existing.add(value);
}
catch (IllegalArgumentException e)
{
throw new RuntimeException(e);
}
catch (IllegalAccessException e)
{
throw new RuntimeException(e);
}
}
};
}
private static final RuntimeFieldFactory> REPEATED = new RuntimeFieldFactory>(RuntimeFieldFactory.ID_COLLECTION)
{
@SuppressWarnings("unchecked")
public Field create(int number, String name, final java.lang.reflect.Field f, IdStrategy strategy)
{
if(null != f.getAnnotation(Morph.class))
{
// can be used to override the configured system property:
// RuntimeEnv.COLLECTION_SCHEMA_ON_REPEATED_FIELDS
// In this context, Morph annotation will force using a collection
// schema only for this particular field.
return RuntimeCollectionFieldFactory.getFactory().create(number,
name, f, strategy);
}
if(EnumSet.class.isAssignableFrom(f.getType()))
{
final Class enumType = (Class)getGenericType(f, 0);
if(enumType == null)
{
// still handle the serialization of EnumSets even without generics
return RuntimeFieldFactory.OBJECT.create(number, name, f, strategy);
}
return createCollectionEnumV(number, name, f,
strategy.getEnumIO(enumType).getEnumSetFactory(), enumType, strategy);
}
final MessageFactory messageFactory = strategy.getCollectionFactory(f.getType());
final Class genericType = (Class)getGenericType(f, 0);
if(genericType == null)
{
// the value is not a simple parameterized type.
return createCollectionObjectV(number, name, f, messageFactory, PolymorphicSchemaFactories.OBJECT, strategy);
}
final Delegate inline = getDelegateOrInline(genericType, strategy);
if(inline != null)
return createCollectionInlineV(number, name, f, messageFactory, inline);
if(Message.class.isAssignableFrom(genericType))
return createCollectionPojoV(number, name, f, messageFactory, genericType, strategy);
if(genericType.isEnum())
return createCollectionEnumV(number, name, f, messageFactory, genericType, strategy);
final PolymorphicSchema.Factory factory =
PolymorphicSchemaFactories.getFactoryFromRepeatedValueGenericType(genericType);
if(factory != null)
return createCollectionObjectV(number, name, f, messageFactory, factory, strategy);
if(pojo(genericType, f.getAnnotation(Morph.class), strategy))
return createCollectionPojoV(number, name, f, messageFactory, genericType, strategy);
if(genericType.isInterface())
return createCollectionObjectV(number, name, f, messageFactory, PolymorphicSchemaFactories.OBJECT, strategy);
return createCollectionPolymorphicV(number, name, f, messageFactory, genericType, strategy);
}
public void transfer(Pipe pipe, Input input, Output output, int number,
boolean repeated) throws IOException
{
throw new UnsupportedOperationException();
}
public Collection readFrom(Input input) throws IOException
{
throw new UnsupportedOperationException();
}
public void writeTo(Output output, int number, Collection value,
boolean repeated) throws IOException
{
throw new UnsupportedOperationException();
}
public FieldType getFieldType()
{
throw new UnsupportedOperationException();
}
public Class typeClass()
{
throw new UnsupportedOperationException();
}
};
}
