org.opendaylight.groupbasedpolicy.resolver.PolicyInfo.java Source code

Introduction

Here is the source code for org.opendaylight.groupbasedpolicy.resolver.PolicyInfo.java

Source

/*
 * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */

package org.opendaylight.groupbasedpolicy.resolver;

import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.annotation.concurrent.Immutable;

import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ConditionName;

import com.google.common.collect.Sets;
import com.google.common.collect.Table;

/**
 * Represent the current policy snapshot for the set of tenants that are
 * in scope
 * @author readams
 */
@Immutable
public class PolicyInfo {
    final Table<EgKey, EgKey, Policy> policyMap;
    final Map<EgKey, Set<ConditionSet>> egConditions;

    public PolicyInfo(Table<EgKey, EgKey, Policy> policyMap, Map<EgKey, Set<ConditionSet>> egConditions) {
        super();
        this.policyMap = policyMap;
        this.egConditions = egConditions;
    }

    public Table<EgKey, EgKey, Policy> getPolicyMap() {
        return policyMap;
    }

    /**
     * Get the policy that currently applies to traffic flowing out of 
     * the specified originating endpoint group into the specified destination
     * endpoint group.  Note that there will be policy only for one of the two
     * possible directions.
     * 
     * @param fromGroup the endpoint group for the originating endpoint
     * @param toGroup the endpoint group for the destination endpoint
     * @return the {@link Policy} that applies.  Cannot be null
     */
    public Policy getPolicy(EgKey fromGroup, EgKey toGroup) {
        Policy p = policyMap.get(fromGroup, toGroup);
        if (p == null)
            return Policy.EMPTY;
        return p;
    }

    /**
     * Get the condition sets for a particular endpoint group
     * @param eg the endpoint group
     * @return the set of condition sets that could apply to an endpoint
     * in that endpoint group
     */
    public Set<ConditionSet> getEgConditions(EgKey eg) {
        return Collections.unmodifiableSet(egConditions.get(eg));
    }

    /**
     * Get the condition group as it applies to the given list of conditions
     * @param eg
     * @param conditions
     * @return
     */
    public ConditionGroup getEgCondGroup(EgKey eg, List<ConditionName> conditions) {
        Set<ConditionSet> egconds = egConditions.get(eg);
        if (egconds == null)
            return ConditionGroup.EMPTY;
        Set<ConditionSet> matching = null;
        for (ConditionSet cs : egconds) {
            if (cs.matches(conditions)) {
                if (matching == null)
                    matching = new HashSet<>();
                matching.add(cs);
            }
        }
        if (matching == null)
            return ConditionGroup.EMPTY;
        return new ConditionGroup(matching);
    }

    /**
     * Get the set of endpoint groups that are peers for the given endpoint
     * group
     * @param eg the endpoint group
     * @return the set of endpoint groups
     */
    public Set<EgKey> getPeers(EgKey eg) {
        return Sets.union(policyMap.row(eg).keySet(), policyMap.column(eg).keySet());
    }
}