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