Add subnet aware for vpp-mapper and vpp-renderer
[groupbasedpolicy.git] / renderers / vpp / src / main / java / org / opendaylight / groupbasedpolicy / renderer / vpp / util / MountedDataBrokerProvider.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.vpp.util;
10
11 import javax.annotation.Nonnull;
12 import java.util.concurrent.ExecutionException;
13 import java.util.concurrent.TimeUnit;
14 import java.util.concurrent.TimeoutException;
15 import com.google.common.base.Optional;
16 import com.google.common.base.Preconditions;
17 import com.google.common.util.concurrent.SettableFuture;
18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
19 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
20 import org.opendaylight.controller.md.sal.binding.api.MountPointService;
21 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
22 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
23 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
26
27 public class MountedDataBrokerProvider {
28
29     private static final Logger LOG = LoggerFactory.getLogger(MountedDataBrokerProvider.class);
30     private final MountPointService mountService;
31     private final DataBroker dataBroker;
32     @SuppressWarnings("FieldCanBeLocal")
33     private final byte NODE_CONNECTION_TIMER = 60; // seconds
34
35     public MountedDataBrokerProvider(@Nonnull MountPointService mountService, @Nonnull DataBroker dataBroker) {
36         this.mountService = Preconditions.checkNotNull(mountService);
37         this.dataBroker = Preconditions.checkNotNull(dataBroker);
38     }
39
40     public Optional<DataBroker> getDataBrokerForMountPoint(@Nonnull InstanceIdentifier<?> iidToMountPoint) {
41         try {
42             final SettableFuture<Boolean> futureNodeStatus = SettableFuture.create();
43             final NodeKey nodeKey = iidToMountPoint.firstKeyOf(Node.class);
44             new GbpVppNetconfConnectionProbe(nodeKey, futureNodeStatus, dataBroker);
45             if (futureNodeStatus.get(NODE_CONNECTION_TIMER, TimeUnit.SECONDS)) {
46                 LOG.debug("Node connected, mountpoint with iid {} available", iidToMountPoint);
47                 Optional<MountPoint> potentialMountPoint = mountService.getMountPoint(iidToMountPoint);
48                 if (potentialMountPoint.isPresent()) {
49                     return potentialMountPoint.get().getService(DataBroker.class);
50                 } else {
51                     LOG.warn("Mount point does not exist for {}", iidToMountPoint);
52                     return Optional.absent();
53                 }
54             } else {
55                 LOG.warn("Failed while connecting to node, Iid: {}", iidToMountPoint);
56                 return Optional.absent();
57             }
58         } catch (TimeoutException e) {
59             LOG.warn("Mountpoint not obtained within {} seconds. Iid: {}", NODE_CONNECTION_TIMER, iidToMountPoint, e);
60             return Optional.absent();
61         } catch (ExecutionException | InterruptedException e) {
62             LOG.warn("Error while getting mountpoint. Iid: {}", iidToMountPoint, e);
63             return Optional.absent();
64         }
65     }
66 }