BUG-6650: ep-ip/sgt, implement and wire template provider

[groupbasedpolicy.git] / sxp-integration / sxp-ep-provider / src / main / java / org / opendaylight / groupbasedpolicy / sxp / ep / provider / impl / dao / EPPolicyTemplateDaoFacadeImpl.java

diff --git a/sxp-integration/sxp-ep-provider/src/main/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/dao/EPPolicyTemplateDaoFacadeImpl.java b/sxp-integration/sxp-ep-provider/src/main/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/dao/EPPolicyTemplateDaoFacadeImpl.java
new file mode 100644 (file)
index 0000000..e23064c
--- /dev/null
+++ b/sxp-integration/sxp-ep-provider/src/main/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/dao/EPPolicyTemplateDaoFacadeImpl.java
@@ -0,0 +1,126 @@
+/*
+ * Copyright (c) 2016 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.sxp.ep.provider.impl.dao;
+
+import com.google.common.base.Function;
+import com.google.common.base.Optional;
+import com.google.common.util.concurrent.AsyncFunction;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import java.util.Collection;
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.EPPolicyTemplateDaoFacade;
+import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.EPPolicyTemplateProvider;
+import org.opendaylight.groupbasedpolicy.util.IidFactory;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.Description;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.Name;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.EndpointGroup;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.EndpointGroupBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.SxpEpMapper;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointPolicyTemplateBySgt;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointPolicyTemplateBySgtKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Purpose: add template provider fallback to {@link EPPolicyTemplateDaoImpl}
+ */
+public class EPPolicyTemplateDaoFacadeImpl implements EPPolicyTemplateDaoFacade {
+
+    private static final Logger LOG = LoggerFactory.getLogger(EPPolicyTemplateDaoFacadeImpl.class);
+
+    private final EPPolicyTemplateDaoImpl epPolicyTemplateDao;
+    private final DataBroker dataBroker;
+
+    private EPPolicyTemplateProvider templateProvider;
+
+    public EPPolicyTemplateDaoFacadeImpl(final DataBroker dataBroker, final EPPolicyTemplateDaoImpl epPolicyTemplateDao) {
+        this.dataBroker = dataBroker;
+        this.epPolicyTemplateDao = epPolicyTemplateDao;
+    }
+
+    @Override
+    public void setTemplateProvider(final EPPolicyTemplateProvider templateProvider) {
+        this.templateProvider = templateProvider;
+    }
+
+
+    @Override
+    public ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> read(@Nonnull final Sgt key) {
+        // read from delegate
+        final ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> templateFu = epPolicyTemplateDao.read(key);
+
+        // involve fallback if template is absent
+        return Futures.transform(templateFu, new AsyncFunction<Optional<EndpointPolicyTemplateBySgt>, Optional<EndpointPolicyTemplateBySgt>>() {
+            @Override
+            public ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> apply(
+                    @Nonnull final Optional<EndpointPolicyTemplateBySgt> templateOpt) throws Exception {
+
+                return templateOpt.transform(template -> Futures.immediateFuture(templateOpt))
+                        // failed to read template -> invoke fallback if available
+                        .or(() -> java.util.Optional.ofNullable(templateProvider)
+                                .flatMap(provider -> templateProvider.provideTemplate(key))
+                                .map(template -> storeTemplateAndEpg(template))
+                                .orElse(Futures.immediateFuture(Optional.absent()))
+                        );
+            }
+        });
+    }
+
+    private ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> storeTemplateAndEpg(final EndpointPolicyTemplateBySgt template) {
+        // store EPG (presume that it does not exist)
+        final Sgt sgtValue = template.getSgt();
+        LOG.trace("storing EPGs for generated epPolicyTemplate: {} [{}]",
+                sgtValue.getValue(), template.getEndpointGroups().size());
+        final WriteTransaction wTx = dataBroker.newWriteOnlyTransaction();
+
+        boolean createParent = true;
+        for (EndpointGroupId epgId : template.getEndpointGroups()) {
+            final InstanceIdentifier<EndpointGroup> epgPath = IidFactory.endpointGroupIid(template.getTenant(), epgId);
+            final EndpointGroup epg = new EndpointGroupBuilder()
+                    .setId(epgId)
+                    .setDescription(new Description("imported from ISE for sgt=" + sgtValue.getValue()))
+                    .setName(new Name(String.format("%s_ISE_SGT_%d", epgId.getValue(), sgtValue.getValue())))
+                    .build();
+            wTx.put(LogicalDatastoreType.CONFIGURATION, epgPath, epg, createParent);
+            createParent = false;
+        }
+
+        // store ep-policy-template
+        LOG.trace("storing generated epPolicyTemplate: {}", sgtValue.getValue());
+        final InstanceIdentifier<EndpointPolicyTemplateBySgt> epPolicyTemplatePath = InstanceIdentifier
+                .create(SxpEpMapper.class)
+                .child(EndpointPolicyTemplateBySgt.class, new EndpointPolicyTemplateBySgtKey(sgtValue));
+        wTx.put(LogicalDatastoreType.CONFIGURATION, epPolicyTemplatePath, template, true);
+
+        return Futures.transform(wTx.submit(), createStoreOutcomeHandler(template));
+    }
+
+    private Function<Void, Optional<EndpointPolicyTemplateBySgt>> createStoreOutcomeHandler(final EndpointPolicyTemplateBySgt template) {
+        return new Function<Void, Optional<EndpointPolicyTemplateBySgt>>() {
+            @Nullable
+            @Override
+            public Optional<EndpointPolicyTemplateBySgt> apply(@Nullable final Void aVoid) {
+                return Optional.of(template);
+            }
+        };
+    }
+
+    @Override
+    public Collection<EndpointPolicyTemplateBySgt> readBy(@Nonnull final EpPolicyTemplateValueKey specialKey) {
+        return epPolicyTemplateDao.readBy(specialKey);
+    }
+}