com.opengamma.bbg.livedata.faketicks.BySchemeFakeSubscriptionSelector.java Source code

Introduction

Here is the source code for com.opengamma.bbg.livedata.faketicks.BySchemeFakeSubscriptionSelector.java

Source

/**
 * Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.bbg.livedata.faketicks;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import com.google.common.collect.Sets;
import com.opengamma.id.ExternalScheme;
import com.opengamma.livedata.LiveDataSpecification;
import com.opengamma.util.tuple.ObjectsPair;
import com.opengamma.util.tuple.Pair;

/**
 * Decides based of scheme which subscriptions to fake.
 * Fakes iff any of the identifiers in the bundle has one of these schemes
 */
public class BySchemeFakeSubscriptionSelector implements FakeSubscriptionSelector {

    private final Set<ExternalScheme> _toFake;

    public BySchemeFakeSubscriptionSelector(String... toFakes) {
        this(getSchemes(toFakes));
    }

    public BySchemeFakeSubscriptionSelector(ExternalScheme... toFakes) {
        this(Sets.newHashSet(toFakes));
    }

    public BySchemeFakeSubscriptionSelector(Set<ExternalScheme> toFake) {
        _toFake = toFake;
    }

    @Override
    public ObjectsPair<Collection<LiveDataSpecification>, Collection<LiveDataSpecification>> splitShouldFake(
            FakeSubscriptionBloombergLiveDataServer server, Collection<LiveDataSpecification> specs) {
        if (specs.isEmpty()) {
            return Pair.of((Collection<LiveDataSpecification>) new ArrayList<LiveDataSpecification>(),
                    (Collection<LiveDataSpecification>) new ArrayList<LiveDataSpecification>());
        }

        Set<LiveDataSpecification> fakes = new HashSet<LiveDataSpecification>();
        Set<LiveDataSpecification> underlyings = new HashSet<LiveDataSpecification>();

        for (LiveDataSpecification liveDataSpecification : specs) {
            if (shouldFake(liveDataSpecification)) {
                fakes.add(liveDataSpecification);
            } else {
                underlyings.add(liveDataSpecification);
            }
        }

        return Pair.of((Collection<LiveDataSpecification>) underlyings, (Collection<LiveDataSpecification>) fakes);
    }

    private boolean shouldFake(LiveDataSpecification liveDataSpecification) {
        for (ExternalScheme scheme : _toFake) {
            if (liveDataSpecification.getIdentifier(scheme) != null) {
                return true;
            }
        }
        return false;
    }

    private static Set<ExternalScheme> getSchemes(String[] schemes) {
        HashSet<ExternalScheme> ret = new HashSet<ExternalScheme>();
        for (String scheme : schemes) {
            ret.add(ExternalScheme.of(scheme));
        }
        return ret;
    }
}