Cherry Pick to Master - Fix for Bug3644
[vpnservice.git] / vpnmanager / vpnmanager-impl / src / main / java / org / opendaylight / vpnservice / VpnserviceProvider.java
1 /*
2  * Copyright (c) 2015 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.vpnservice;
9
10 import java.math.BigInteger;
11 import java.util.Collection;
12 import java.util.Collections;
13 import java.util.concurrent.ExecutionException;
14 import java.util.concurrent.Future;
15
16 import org.opendaylight.bgpmanager.api.IBgpManager;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
19 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
20 import org.opendaylight.fibmanager.api.IFibManager;
21 import org.opendaylight.vpnmanager.api.IVpnManager;
22 import org.opendaylight.vpnservice.interfacemgr.interfaces.IInterfaceManager;
23 import org.opendaylight.vpnservice.mdsalutil.interfaces.IMdsalApiManager;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.CreateIdPoolInput;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.CreateIdPoolInputBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.IdManagerService;
27 import org.opendaylight.yangtools.yang.common.RpcResult;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 public class VpnserviceProvider implements BindingAwareProvider, IVpnManager,
32                                                        AutoCloseable {
33
34     private static final Logger LOG = LoggerFactory.getLogger(VpnserviceProvider.class);
35     private VpnInterfaceManager vpnInterfaceManager;
36     private VpnManager vpnManager;
37     private IBgpManager bgpManager;
38     private IFibManager fibManager;
39     private IMdsalApiManager mdsalManager;
40     private IInterfaceManager interfaceManager;
41     private IdManagerService idManager;
42     private InterfaceChangeListener interfaceListener;
43
44     @Override
45     public void onSessionInitiated(ProviderContext session) {
46         LOG.info("VpnserviceProvider Session Initiated");
47         try {
48             final  DataBroker dataBroker = session.getSALService(DataBroker.class);
49             vpnManager = new VpnManager(dataBroker, bgpManager);
50             vpnManager.setIdManager(idManager);
51             vpnInterfaceManager = new VpnInterfaceManager(dataBroker, bgpManager);
52             vpnInterfaceManager.setMdsalManager(mdsalManager);
53             vpnInterfaceManager.setInterfaceManager(interfaceManager);
54             vpnInterfaceManager.setIdManager(idManager);
55             vpnManager.setVpnInterfaceManager(vpnInterfaceManager);
56             interfaceListener = new InterfaceChangeListener(dataBroker, vpnInterfaceManager);
57             createIdPool();
58         } catch (Exception e) {
59             LOG.error("Error initializing services", e);
60         }
61     }
62
63     public void setBgpManager(IBgpManager bgpManager) {
64         LOG.debug("BGP Manager reference initialized");
65         this.bgpManager = bgpManager;
66     }
67
68     public void setMdsalManager(IMdsalApiManager mdsalManager) {
69         this.mdsalManager = mdsalManager;
70     }
71
72     public void setFibManager(IFibManager fibManager) {
73         this.fibManager = fibManager;
74     }
75
76     public void setInterfaceManager(IInterfaceManager interfaceManager) {
77         this.interfaceManager = interfaceManager;
78     }
79
80     public void setIdManager(IdManagerService idManager) {
81         this.idManager = idManager;
82     }
83
84     private void createIdPool() {
85         CreateIdPoolInput createPool = new CreateIdPoolInputBuilder()
86             .setPoolName(VpnConstants.VPN_IDPOOL_NAME)
87             .setIdStart(VpnConstants.VPN_IDPOOL_START)
88             .setPoolSize(new BigInteger(VpnConstants.VPN_IDPOOL_SIZE))
89             .build();
90         try {
91            Future<RpcResult<Void>> result = idManager.createIdPool(createPool);
92            if ((result != null) && (result.get().isSuccessful())) {
93                 LOG.debug("Created IdPool for VPN Service");
94             }
95         } catch (InterruptedException | ExecutionException e) {
96             LOG.error("Failed to create idPool for VPN Service",e);
97         }
98     }
99
100     @Override
101     public void close() throws Exception {
102         vpnManager.close();
103         vpnInterfaceManager.close();
104         interfaceListener.close();
105     }
106
107     @Override
108     public Collection<BigInteger> getDpnsForVpn(long vpnId) {
109         return vpnInterfaceManager.getDpnsForVpn(vpnId);
110     }
111
112     @Override
113     public void setFibService(IFibManager fibManager) {
114         LOG.debug("Fib service reference is initialized in VPN Manager");
115         this.fibManager = fibManager;
116         vpnInterfaceManager.setFibManager(fibManager);
117     }
118 }