ru.yandex.qatools.camelot.common.builders.ReadonlyAggregatorRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of camelot-commons Show documentation

There is a newer version: 2.5.4

package ru.yandex.qatools.camelot.common.builders;

import org.apache.camel.CamelContext;
import org.apache.camel.Exchange;
import org.apache.camel.spi.AggregationRepository;
import ru.yandex.qatools.camelot.api.AggregatorRepository;
import ru.yandex.qatools.camelot.common.AggregationRepositoryWithLocalKeys;
import ru.yandex.qatools.camelot.common.AggregationRepositoryWithLocks;
import ru.yandex.qatools.camelot.config.Plugin;

import java.io.Serializable;
import java.util.Set;

/**
 * @author Ilya Sadykov (mailto: [email protected])
 */
public class ReadonlyAggregatorRepository implements AggregatorRepository {

    private final Plugin plugin;
    private final CamelContext camelContext;

    public ReadonlyAggregatorRepository(CamelContext camelContext, Plugin plugin) {
        this.plugin = plugin;
        this.camelContext = camelContext;
    }

    @Override
    @SuppressWarnings("unchecked")
    public T get(String key) {
        final AggregationRepository repo = plugin.getContext().getAggregationRepo();

        final Exchange exchange = (repo instanceof AggregationRepositoryWithLocks) ?
                ((AggregationRepositoryWithLocks) repo).getWithoutLock(camelContext, key) : repo.get(camelContext, key);
        ClassLoader realClassLoader = plugin.getContext().getClassLoader();
        final ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
        final ClassLoader thisClassLoader = getClass().getClassLoader();

        if (realClassLoader != thisClassLoader && contextClassLoader != thisClassLoader) {
            realClassLoader = contextClassLoader;
        }

        if (exchange != null) {
            return (T) plugin.getContext().getMessagesSerializer().deserialize(
                    exchange.getIn().getBody(), realClassLoader);
        }
        return null;
    }

    @Override
    public Set keys() {
        return plugin.getContext().getAggregationRepo().getKeys();
    }

    @Override
    public Set localKeys() {
        final AggregationRepository repo = plugin.getContext().getAggregationRepo();
        return (repo instanceof AggregationRepositoryWithLocalKeys) ?
                ((AggregationRepositoryWithLocalKeys) repo).localKeys() : repo.getKeys();
    }
}