bgp/openconfig-rp-impl/src/main/java/org/opendaylight/protocol/bgp/openconfig/routing/policy/impl/BGPRibPolicyImpl.java

   1 /*
   2  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.protocol.bgp.openconfig.routing.policy.impl;
  10
  11 import static java.util.Objects.requireNonNull;
  12 import static org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.registry.RouteAttributeContainer.routeAttributeContainerFalse;
  13
  14 import com.google.common.cache.CacheBuilder;
  15 import com.google.common.cache.CacheLoader;
  16 import com.google.common.cache.LoadingCache;
  17 import java.util.Collections;
  18 import java.util.List;
  19 import java.util.Optional;
  20 import java.util.concurrent.ExecutionException;
  21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  22 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
  25 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.registry.RouteAttributeContainer;
  26 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.registry.StatementRegistryConsumer;
  27 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRibRoutingPolicy;
  28 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
  29 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
  30 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.DefaultPolicyType;
  31 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.apply.policy.group.apply.policy.Config;
  32 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.RoutingPolicy;
  33 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.PolicyDefinitions;
  34 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.policy.definitions.PolicyDefinition;
  35 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.policy.definitions.PolicyDefinitionKey;
  36 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.policy.definitions.policy.definition.Statements;
  37 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.policy.definitions.policy.definition.statements.Statement;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.Attributes;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
  41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  42
  43 final class BGPRibPolicyImpl implements BGPRibRoutingPolicy {
  44     private static final InstanceIdentifier<RoutingPolicy> ROUTING_POLICY_IID
  45             = InstanceIdentifier.create(RoutingPolicy.class);
  46     private static final List<String> DEFAULT_IMPORT_POLICY = Collections.singletonList("default-odl-import-policy");
  47     private static final List<String> DEFAULT_EXPORT_POLICY = Collections.singletonList("default-odl-export-policy");
  48     private final DefaultPolicyType defaultExportPolicy;
  49     private final DefaultPolicyType defaultImportPolicy;
  50     private final List<String> exportPolicy;
  51     private final List<String> importPolicy;
  52     private final StatementRegistryConsumer policyRegistry;
  53     private final RouteEntryBaseAttributes ribBaseParameters;
  54     private final DataBroker databroker;
  55     private final LoadingCache<String, List<Statement>> statements = CacheBuilder.newBuilder()
  56             .build(new CacheLoader<String, List<Statement>>() {
  57                 @Override
  58                 public List<Statement> load(final String key) throws ExecutionException, InterruptedException {
  59                     return loadStatements(key);
  60                 }
  61             });
  62
  63     BGPRibPolicyImpl(final DataBroker databroker, final StatementRegistryConsumer policyRegistry,
  64             final long localAs, final Ipv4Address originatorId, final ClusterIdentifier clusterId,
  65             final Config policyConfig) {
  66         this.policyRegistry = requireNonNull(policyRegistry);
  67         this.databroker = requireNonNull(databroker);
  68         requireNonNull(policyConfig);
  69
  70
  71         List<String> epolicy = policyConfig.getExportPolicy();
  72         if (epolicy == null) {
  73             epolicy = DEFAULT_EXPORT_POLICY;
  74         }
  75         List<String> ipolicy = policyConfig.getImportPolicy();
  76         if (ipolicy == null) {
  77             ipolicy = DEFAULT_IMPORT_POLICY;
  78         }
  79
  80         this.defaultExportPolicy = requireNonNull(policyConfig.getDefaultExportPolicy());
  81         this.defaultImportPolicy = requireNonNull(policyConfig.getDefaultImportPolicy());
  82         this.exportPolicy = requireNonNull(epolicy);
  83         this.importPolicy = requireNonNull(ipolicy);
  84         this.ribBaseParameters = new PolicyRIBBaseParametersImpl(localAs, originatorId, clusterId);
  85     }
  86
  87     private List<Statement> loadStatements(final String key) throws ExecutionException, InterruptedException {
  88         final ReadOnlyTransaction tr = this.databroker.newReadOnlyTransaction();
  89         final com.google.common.base.Optional<Statements> result =
  90                 tr.read(LogicalDatastoreType.CONFIGURATION, ROUTING_POLICY_IID.child(PolicyDefinitions.class)
  91                         .child(PolicyDefinition.class, new PolicyDefinitionKey(key)).child(Statements.class)).get();
  92         if (!result.isPresent()) {
  93             return Collections.emptyList();
  94         }
  95         return result.get().getStatement();
  96     }
  97
  98     @Override
  99     public Optional<Attributes> applyImportPolicies(final BGPRouteEntryImportParameters policyParameters,
 100             final Attributes attributes) {
 101         RouteAttributeContainer currentAttributes = routeAttributeContainerFalse(attributes);
 102         for (final String policyName : this.importPolicy) {
 103             for (final Statement statement : this.statements.getUnchecked(policyName)) {
 104                 currentAttributes = this.policyRegistry
 105                         .applyImportStatement(this.ribBaseParameters, policyParameters, currentAttributes, statement);
 106             }
 107         }
 108         if (!currentAttributes.anyConditionSatisfied()) {
 109             if (DefaultPolicyType.REJECTROUTE.equals(this.defaultImportPolicy)) {
 110                 return Optional.empty();
 111             }
 112         }
 113         return Optional.ofNullable(currentAttributes.getAttributes());
 114     }
 115
 116     @Override
 117     public Optional<Attributes> applyExportPolicies(final BGPRouteEntryExportParameters policyParameters,
 118             final Attributes attributes) {
 119         RouteAttributeContainer currentAttributes = routeAttributeContainerFalse(attributes);
 120         for (final String policyName : this.exportPolicy) {
 121             for (final Statement statement : this.statements.getUnchecked(policyName)) {
 122                 currentAttributes = this.policyRegistry.applyExportStatement(
 123                         this.ribBaseParameters, policyParameters, currentAttributes, statement);
 124             }
 125         }
 126         if (!currentAttributes.anyConditionSatisfied()) {
 127             if (DefaultPolicyType.REJECTROUTE.equals(this.defaultExportPolicy)) {
 128                 return Optional.empty();
 129             }
 130         }
 131
 132         return Optional.ofNullable(currentAttributes.getAttributes());
 133     }
 134 }