support to get fib service in vpn manager
[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
43     @Override
44     public void onSessionInitiated(ProviderContext session) {
45         LOG.info("VpnserviceProvider Session Initiated");
46         try {
47             final  DataBroker dataBroker = session.getSALService(DataBroker.class);
48             vpnManager = new VpnManager(dataBroker, bgpManager);
49             vpnManager.setIdManager(idManager);
50             vpnInterfaceManager = new VpnInterfaceManager(dataBroker, bgpManager);
51             vpnInterfaceManager.setMdsalManager(mdsalManager);
52             vpnInterfaceManager.setInterfaceManager(interfaceManager);
53             vpnInterfaceManager.setIdManager(idManager);
54             createIdPool();
55         } catch (Exception e) {
56             LOG.error("Error initializing services", e);
57         }
58     }
59
60     public void setBgpManager(IBgpManager bgpManager) {
61         LOG.debug("BGP Manager reference initialized");
62         this.bgpManager = bgpManager;
63     }
64
65     public void setMdsalManager(IMdsalApiManager mdsalManager) {
66         this.mdsalManager = mdsalManager;
67     }
68
69     public void setFibManager(IFibManager fibManager) {
70         this.fibManager = fibManager;
71     }
72
73     public void setInterfaceManager(IInterfaceManager interfaceManager) {
74         this.interfaceManager = interfaceManager;
75     }
76
77     public void setIdManager(IdManagerService idManager) {
78         this.idManager = idManager;
79     }
80
81     private void createIdPool() {
82         CreateIdPoolInput createPool = new CreateIdPoolInputBuilder()
83             .setPoolName(VpnConstants.VPN_IDPOOL_NAME)
84             .setIdStart(VpnConstants.VPN_IDPOOL_START)
85             .setPoolSize(new BigInteger(VpnConstants.VPN_IDPOOL_SIZE))
86             .build();
87         try {
88            Future<RpcResult<Void>> result = idManager.createIdPool(createPool);
89            if ((result != null) && (result.get().isSuccessful())) {
90                 LOG.debug("Created IdPool for VPN Service");
91             }
92         } catch (InterruptedException | ExecutionException e) {
93             LOG.error("Failed to create idPool for VPN Service",e);
94         }
95     }
96
97     @Override
98     public void close() throws Exception {
99         vpnManager.close();
100         vpnInterfaceManager.close();
101     }
102
103     @Override
104     public Collection<Long> getDpnsForVpn(long vpnId) {
105         return vpnInterfaceManager.getDpnsForVpn(vpnId);
106     }
107
108     @Override
109     public void setFibService(IFibManager fibManager) {
110         LOG.debug("Fib service reference is initialized in VPN Manager");
111         this.fibManager = fibManager;
112         vpnInterfaceManager.setFibManager(fibManager);
113     }
114 }