sxp-integration/sxp-ep-provider/src/main/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/dao/EPForwardingTemplateDaoImpl.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  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 package org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.dao;
   9
  10 import com.google.common.base.Function;
  11 import com.google.common.base.Optional;
  12 import com.google.common.util.concurrent.CheckedFuture;
  13 import com.google.common.util.concurrent.Futures;
  14 import com.google.common.util.concurrent.ListenableFuture;
  15 import com.google.common.util.concurrent.MoreExecutors;
  16
  17 import java.util.List;
  18 import javax.annotation.Nonnull;
  19 import javax.annotation.Nullable;
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  21 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  24 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.DSAsyncDao;
  25 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.EPTemplateListener;
  26 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.SimpleCachedDao;
  27 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl.util.SxpListenerUtil;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.SxpEpMapper;
  30 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.EndpointForwardingTemplateBySubnet;
  31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  32
  33 /**
  34  * Purpose: general dao for EndPoint templates
  35  */
  36 public class EPForwardingTemplateDaoImpl implements DSAsyncDao<IpPrefix, EndpointForwardingTemplateBySubnet> {
  37
  38     private static final ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> READ_FUTURE_ABSENT = Futures.immediateFuture(Optional.absent());
  39     private final DataBroker dataBroker;
  40     private final SimpleCachedDao<IpPrefix, EndpointForwardingTemplateBySubnet> cachedDao;
  41
  42     public EPForwardingTemplateDaoImpl(final DataBroker dataBroker,
  43                                        final SimpleCachedDao<IpPrefix, EndpointForwardingTemplateBySubnet> cachedDao) {
  44         this.dataBroker = dataBroker;
  45         this.cachedDao = cachedDao;
  46     }
  47
  48     @Override
  49     public ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> read(@Nonnull final IpPrefix key) {
  50         final Optional<EndpointForwardingTemplateBySubnet> value = lookup(cachedDao, key);
  51         final ListenableFuture<Optional<EndpointForwardingTemplateBySubnet>> readResult;
  52         if (value.isPresent()) {
  53             readResult = Futures.immediateFuture(value);
  54         } else if (!cachedDao.isEmpty()) {
  55             return READ_FUTURE_ABSENT;
  56         } else {
  57             final ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
  58             final CheckedFuture<Optional<SxpEpMapper>, ReadFailedException> read =
  59                     rTx.read(LogicalDatastoreType.CONFIGURATION, buildReadPath(key));
  60
  61             Futures.addCallback(read, SxpListenerUtil.createTxCloseCallback(rTx), MoreExecutors.directExecutor());
  62
  63             readResult = Futures.transform(read, new Function<Optional<SxpEpMapper>, Optional<EndpointForwardingTemplateBySubnet>>() {
  64                 @Nullable
  65                 @Override
  66                 public Optional<EndpointForwardingTemplateBySubnet> apply(@Nullable final Optional<SxpEpMapper> input) {
  67                     if (input!=null && input.isPresent()) {
  68                         // clean cache
  69                         cachedDao.invalidateCache();
  70
  71                         // iterate through all template entries and update cachedDao
  72                         final List<EndpointForwardingTemplateBySubnet> templateLot = input.get().getEndpointForwardingTemplateBySubnet();
  73                         if (templateLot != null) {
  74                             for (EndpointForwardingTemplateBySubnet template : templateLot) {
  75                                 cachedDao.update(template.getIpPrefix(), template);
  76                             }
  77                         }
  78                         return lookup(cachedDao, key);
  79                     } else {
  80                         return Optional.absent();
  81                     }
  82                 }
  83             }, MoreExecutors.directExecutor());
  84         }
  85         return readResult;
  86     }
  87
  88     protected InstanceIdentifier<SxpEpMapper> buildReadPath(final IpPrefix key) {
  89         return EPTemplateListener.SXP_MAPPER_TEMPLATE_PARENT_PATH;
  90     }
  91
  92     private Optional<EndpointForwardingTemplateBySubnet> lookup(final SimpleCachedDao<IpPrefix, EndpointForwardingTemplateBySubnet> cachedDao, final IpPrefix key) {
  93         return cachedDao.find(key);
  94     }
  95
  96 }