renderers/ios-xe/src/main/java/org/opendaylight/groupbasedpolicy/renderer/ios_xe_provider/impl/writer/NodeWriter.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
   9 package org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.writer;
  10
  11 import com.google.common.util.concurrent.CheckedFuture;
  12 import com.google.common.util.concurrent.Futures;
  13 import com.google.common.util.concurrent.ListenableFuture;
  14 import java.util.ArrayList;
  15 import java.util.List;
  16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  17 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  20 import org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.manager.PolicyManagerImpl;
  21 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.RendererName;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.Renderers;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.Renderer;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.RendererKey;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.RendererNodes;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.nodes.RendererNode;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.nodes.RendererNodeKey;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33 public class NodeWriter {
  34
  35     private static final Logger LOG = LoggerFactory.getLogger(NodeWriter.class);
  36     private final List<RendererNode> rendererNodesCache;
  37
  38     public NodeWriter() {
  39         rendererNodesCache = new ArrayList<>();
  40     }
  41
  42     public void cache(RendererNode node) {
  43         rendererNodesCache.add(node);
  44     }
  45
  46     /**
  47      * Put all cached items to data store
  48      *
  49      * @param dataBroker appropriate data provider
  50      */
  51     public ListenableFuture<Boolean> commitToDatastore(final DataBroker dataBroker) {
  52         if (rendererNodesCache.isEmpty()) {
  53             return Futures.immediateFuture(true);
  54         }
  55
  56         final WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
  57         final ArrayList<RendererNode> rendererNodes = new ArrayList<>(rendererNodesCache);
  58         // Clear cache
  59         rendererNodesCache.clear();
  60
  61         boolean createParents = true;
  62         for (RendererNode rendererNode : rendererNodes) {
  63             final InstanceIdentifier<RendererNode> iid = buildRendererNodeIid(rendererNode);
  64             writeTransaction.put(LogicalDatastoreType.OPERATIONAL, iid, rendererNode, createParents);
  65             createParents = false;
  66         }
  67         try {
  68             final boolean result = DataStoreHelper.submitToDs(writeTransaction);
  69             return Futures.immediateFuture(result);
  70         } catch (Exception e) {
  71             LOG.error("Failed to .. {}", e.getMessage());
  72         }
  73         return Futures.immediateFuture(false);
  74     }
  75
  76     /**
  77      * Removes all cached items from data store
  78      *
  79      * @param dataBroker appropriate data provider
  80      */
  81     public ListenableFuture<Boolean> removeFromDatastore(final DataBroker dataBroker) {
  82         boolean result = true;
  83         final WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
  84         for (RendererNode nodeToRemove : rendererNodesCache) {
  85             final InstanceIdentifier<RendererNode> iid = buildRendererNodeIid(nodeToRemove);
  86             try {
  87                 writeTransaction.delete(LogicalDatastoreType.OPERATIONAL, iid);
  88                 final CheckedFuture<Void, TransactionCommitFailedException> submitFuture = writeTransaction.submit();
  89                 submitFuture.checkedGet();
  90                 // Clear cache
  91             } catch (TransactionCommitFailedException e) {
  92                 LOG.error("Write transaction failed to {}", e.getMessage());
  93                 result = false;
  94             } catch (Exception e) {
  95                 LOG.error("Failed to .. {}", e.getMessage());
  96                 result = false;
  97             }
  98         }
  99         // Clear cache
 100         rendererNodesCache.clear();
 101         return Futures.immediateFuture(result);
 102     }
 103
 104     private InstanceIdentifier<RendererNode> buildRendererNodeIid(final RendererNode rendererNode) {
 105         return InstanceIdentifier.builder(Renderers.class)
 106                 .child(Renderer.class, new RendererKey(new RendererName(PolicyManagerImpl.IOS_XE_RENDERER)))
 107                 .child(RendererNodes.class)
 108                 .child(RendererNode.class, new RendererNodeKey(rendererNode.getNodePath()))
 109                 .build();
 110     }
 111 }