NETVIRT-1630 migrate to md-sal APIs
[netvirt.git] / cloud-servicechain / impl / src / main / java / org / opendaylight / netvirt / cloudservicechain / utils / VpnPseudoPortCache.java
1 /*
2  * Copyright (c) 2016 Ericsson India Global Services Pvt Ltd. 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.netvirt.cloudservicechain.utils;
9
10 import java.util.List;
11 import java.util.concurrent.ConcurrentHashMap;
12 import java.util.concurrent.ConcurrentMap;
13 import javax.annotation.PostConstruct;
14 import javax.inject.Inject;
15 import javax.inject.Singleton;
16 import org.eclipse.jdt.annotation.NonNull;
17 import org.eclipse.jdt.annotation.Nullable;
18 import org.opendaylight.mdsal.binding.api.DataBroker;
19 import org.opendaylight.mdsal.common.api.ReadFailedException;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.cloud.servicechain.state.rev160711.vpn.to.pseudo.port.list.VpnToPseudoPortData;
21 import org.slf4j.Logger;
22 import org.slf4j.LoggerFactory;
23
24 /**
25  * Manages a per-blade cache, which is fed by a clustered data change listener.
26  */
27 @Singleton
28 public class VpnPseudoPortCache {
29     private static final Logger LOG = LoggerFactory.getLogger(VpnPseudoPortCache.class);
30
31     private final DataBroker broker;
32     private final ConcurrentMap<String, Long> cache =  new ConcurrentHashMap<>();
33
34     @Inject
35     public VpnPseudoPortCache(DataBroker broker) {
36         this.broker = broker;
37     }
38
39     @PostConstruct
40     public void init() {
41         LOG.info("Initial read of Vpn to VpnPseudoPort map from Datastore");
42         try {
43             List<VpnToPseudoPortData> allVpnToPseudoPortData = VpnServiceChainUtils.getAllVpnToPseudoPortData(broker);
44             for (VpnToPseudoPortData vpnToPseudoPort : allVpnToPseudoPortData) {
45                 add(vpnToPseudoPort.getVrfId(), vpnToPseudoPort.getVpnLportTag());
46             }
47         } catch (InterruptedException | ExecutionException e) {
48             LOG.error("Error reading VPN to pseudo-port map", e);
49         }
50     }
51
52     public void add(@NonNull String vrfId, long vpnPseudoLportTag) {
53         LOG.debug("Adding vpn {} and vpnPseudoLportTag {} to VpnPseudoPortCache", vrfId, vpnPseudoLportTag);
54         cache.put(vrfId, Long.valueOf(vpnPseudoLportTag));
55     }
56
57     @Nullable
58     public Long get(@NonNull String vrfId) {
59         return cache.get(vrfId);
60     }
61
62     @Nullable
63     public Long remove(@NonNull String vrfId) {
64         LOG.debug("Removing vpn {} from VpnPseudoPortCache", vrfId);
65         return cache.remove(vrfId);
66     }
67 }