Java Code Examples for io.protostuff.CollectionSchema#MessageFactory

@Override
protected CollectionSchema.MessageFactory resolveCollectionFrom(Input input) throws IOException {
	final String className = input.readString();
	CollectionSchema.MessageFactory factory = collectionMapping.get(className);
	if (factory == null) {
		if (className.indexOf('.') == -1) {
			factory = CollectionSchema.MessageFactories.valueOf(className);
		} else {
			factory = new RuntimeCollectionFactory(RuntimeEnv.loadClass(className));
			CollectionSchema.MessageFactory f = collectionMapping.putIfAbsent(className, factory);
			if (f != null)
				factory = f;
		}
	}

	return factory;
}
 

/**
 * Collection ids start at 1.
 */
@Override
public <T extends Collection<?>> Registry registerCollection(
        CollectionSchema.MessageFactory factory, int id)
{
    if (id < 1)
        throw new IllegalArgumentException("collection ids start at 1.");

    if (id >= strategy.collections.size())
        grow(strategy.collections, id + 1);
    else if (strategy.collections.get(id) != null)
    {
        throw new IllegalArgumentException("Duplicate id registration: " + id +
                " (" + factory.typeClass() + ")");
    }

    RegisteredCollectionFactory rf = new RegisteredCollectionFactory(id, factory);
    strategy.collections.set(id, rf);
    // just in case
    if (strategy.collectionMapping.put(factory.typeClass(), rf) != null)
        throw new IllegalArgumentException("Duplicate registration for: " + factory.typeClass());

    return this;
}
 

private static <E extends Enum<E>> CollectionSchema.MessageFactory newEnumSetFactory(
        final EnumIO<E> eio)
{
    return new CollectionSchema.MessageFactory()
    {
        @Override
        @SuppressWarnings("unchecked")
        public <V> Collection<V> newMessage()
        {
            return (Collection<V>) eio.newEnumSet();
        }

        @Override
        public Class<?> typeClass()
        {
            return EnumSet.class;
        }
    };
}
 

/**
 * Collection ids start at 1.
 */
@Override
public <T extends Collection<?>> Registry registerCollection(
        CollectionSchema.MessageFactory factory, int id)
{
    if (id < 1)
        throw new IllegalArgumentException("collection ids start at 1.");

    if (id >= strategy.collectionIdStart)
        throw new IllegalArgumentException("collection ids must be lesser than " + strategy.collectionIdStart);
    else if (strategy.collections.get(id) != null)
    {
        throw new IllegalArgumentException("Duplicate id registration: " + id +
                " (" + factory.typeClass() + ")");
    }

    RuntimeCollectionFactory rf = new RuntimeCollectionFactory();
    rf.id = id;
    rf.factory = factory;
    strategy.collections.set(id, rf);
    // just in case
    if (strategy.collectionMapping.put(factory.typeClass(), rf) != null)
        throw new IllegalArgumentException("Duplicate registration for: " + factory.typeClass());

    return this;
}
 

@Override
protected CollectionSchema.MessageFactory getCollectionFactory(
        Class<?> clazz)
{
    final String className = clazz.getName();
    CollectionSchema.MessageFactory factory = collectionMapping
            .get(className);
    if (factory == null)
    {
        if (className.startsWith("java.util") && CollectionSchema.MessageFactories.accept(clazz.getSimpleName()))
        {
            factory = CollectionSchema.MessageFactories.valueOf(clazz
                    .getSimpleName());
        }
        else
        {
            factory = new RuntimeCollectionFactory(clazz);
            CollectionSchema.MessageFactory f = collectionMapping
                    .putIfAbsent(className, factory);
            if (f != null)
                factory = f;
        }
    }

    return factory;
}
 

@Override
protected void writeCollectionIdTo(Output output, int fieldNumber,
        Class<?> clazz) throws IOException
{
    final CollectionSchema.MessageFactory factory = collectionMapping
            .get(clazz.getName());
    if (factory == null && clazz.getName().startsWith("java.util")
        && CollectionSchema.MessageFactories.accept(clazz.getSimpleName()))
    {
        // jdk collection
        // better not to register the jdk collection if using this strategy
        // as it saves space by not writing the full package
        output.writeString(fieldNumber, clazz.getSimpleName(), false);
    }
    else
    {
        output.writeString(fieldNumber, clazz.getName(), false);
    }
}
 

@Override
protected CollectionSchema.MessageFactory resolveCollectionFrom(Input input)
        throws IOException
{
    final String className = input.readString();
    CollectionSchema.MessageFactory factory = collectionMapping
            .get(className);
    if (factory == null)
    {
        if (className.indexOf('.') == -1 && CollectionSchema.MessageFactories.accept(className))
        {
            factory = CollectionSchema.MessageFactories.valueOf(className);
        }
        else
        {
            factory = new RuntimeCollectionFactory(
                    RuntimeEnv.loadClass(className));
            CollectionSchema.MessageFactory f = collectionMapping
                    .putIfAbsent(className, factory);
            if (f != null)
                factory = f;
        }
    }

    return factory;
}
 

@Override
protected CollectionSchema.MessageFactory resolveCollectionFrom(Input input)
        throws IOException
{
    final int id = input.readUInt32();

    final CollectionSchema.MessageFactory factory = id < collections.size() ?
            collections.get(id) : null;
    if (factory == null)
        throw new UnknownTypeException("collection id: " + id + " (Outdated registry)");

    return factory;
}
 

@Override
protected void writeCollectionIdTo(Output output, int fieldNumber, Class<?> clazz) throws IOException {
	@SuppressWarnings("unlikely-arg-type")
	final CollectionSchema.MessageFactory factory = collectionMapping.get(clazz);
	if (factory == null && clazz.getName().startsWith("java.util")) {
		// jdk collection
		// better not to register the jdk collection if using this strategy
		// as it saves space by not writing the full package
		output.writeString(fieldNumber, clazz.getSimpleName(), false);
	} else {
		output.writeString(fieldNumber, clazz.getName(), false);
	}
}
 

/**
 * Returns the factory for an EnumSet (lazy).
 */
public CollectionSchema.MessageFactory getEnumSetFactory()
{
    CollectionSchema.MessageFactory enumSetFactory = this.enumSetFactory;
    if (enumSetFactory == null)
    {
        synchronized (this)
        {
            if ((enumSetFactory = this.enumSetFactory) == null)
                this.enumSetFactory = enumSetFactory = newEnumSetFactory(this);
        }
    }
    return enumSetFactory;
}
 

@Override
protected CollectionSchema.MessageFactory resolveCollectionFrom(Input input)
        throws IOException
{
    final int id = input.readUInt32();

    final RuntimeCollectionFactory factory = id < collections.size() ?
            collections.get(id) : null;
    if (factory == null)
        throw new UnknownTypeException("Unknown collection id: " + id);

    return factory;
}
 

/**
 * Registers a collection. Returns true if registration is successful.
 */
public boolean registerCollection(CollectionSchema.MessageFactory factory)
{
    return null == collectionMapping.putIfAbsent(factory.typeClass()
            .getName(), factory);
}
 

public RegisteredCollectionFactory(int id, CollectionSchema.MessageFactory factory)
{
    this.id = id;
    this.factory = factory;
}
 

@Override
protected CollectionSchema.MessageFactory getCollectionFactory(Class<?> clazz)
{
    return getRuntimeCollectionFactory(clazz);
}
 

/**
 * Registers a collection. Returns true if registration is successful.
 */
public boolean registerCollection(CollectionSchema.MessageFactory factory) {
	return null == collectionMapping.putIfAbsent(factory.typeClass().getName(), factory);
}
 

protected abstract CollectionSchema.MessageFactory resolveCollectionFrom(
Input input) throws IOException;
 

protected abstract CollectionSchema.MessageFactory resolveCollectionFrom(
Input input) throws IOException;
 

/**
 * Returns the {@link CollectionSchema.MessageFactory}. The callers (internal field factories) are responsible that
 * the class provided implements {@link Collection}.
 */
protected abstract CollectionSchema.MessageFactory getCollectionFactory(
        Class<?> clazz);
 

/**
 * Returns the {@link CollectionSchema.MessageFactory}. The callers (internal field factories) are responsible that
 * the class provided implements {@link Collection}.
 */
protected abstract CollectionSchema.MessageFactory getCollectionFactory(
        Class<?> clazz);
 

/**
 * Collection ids start at 1.
 */
public <T extends Collection<?>> Registry registerCollection(
        CollectionSchema.MessageFactory factory, int id);