Adding IdManager/LockManager and related Changes
[vpnservice.git] / idmanager / idmanager-impl / src / main / java / org / opendaylight / idmanager / IdManagerServiceProvider.java
index ab016b1ef26322334da7a603544b827eba04fde6..b027da1cda4348e3422b046e8b43cba5b1ebcd2a 100644 (file)
@@ -15,6 +15,7 @@ import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
 
 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.IdManagerService;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.lockmanager.rev150819.LockManagerService;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -22,9 +23,10 @@ import org.slf4j.LoggerFactory;
 public class IdManagerServiceProvider implements BindingAwareProvider,
             AutoCloseable {
 
-        private static final Logger LOG = LoggerFactory.getLogger(IdManagerServiceProvider.class);
-        private IdManager idManager;
-        private RpcProviderRegistry rpcProviderRegistry;
+    private static final Logger LOG = LoggerFactory.getLogger(IdManagerServiceProvider.class);
+    private IdManager idManager;
+    private RpcProviderRegistry rpcProviderRegistry;
+    private LockManagerService lockManager;
 
     public RpcProviderRegistry getRpcProviderRegistry() {
         return rpcProviderRegistry;
@@ -34,12 +36,17 @@ public class IdManagerServiceProvider implements BindingAwareProvider,
         this.rpcProviderRegistry = rpcProviderRegistry;
     }
 
-        @Override
+    public void setLockManager(LockManagerService lockManager) {
+        this.lockManager = lockManager;
+    }
+
+    @Override
     public void onSessionInitiated(ProviderContext session){
         LOG.info("IDManagerserviceProvider Session Initiated");
         try {
             final  DataBroker dataBroker = session.getSALService(DataBroker.class);
             idManager = new IdManager(dataBroker);
+            idManager.setLockManager(lockManager);
             final BindingAwareBroker.RpcRegistration<IdManagerService> rpcRegistration = getRpcProviderRegistry().addRpcImplementation(IdManagerService.class, idManager);
         } catch (Exception e) {
             LOG.error("Error initializing services", e);