itm/itm-impl/src/main/java/org/opendaylight/vpnservice/itm/impl/ItmProvider.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.itm.impl;
   9
  10 import java.math.BigInteger;
  11 import java.util.List;
  12 import java.util.concurrent.ExecutionException;
  13 import java.util.concurrent.Future;
  14
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
  17 import org.opendaylight.controller.md.sal.binding.api.NotificationService;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
  20 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
  21 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
  22 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  23 import org.opendaylight.vpnservice.interfacemgr.interfaces.IInterfaceManager;
  24 import org.opendaylight.vpnservice.itm.api.IITMProvider;
  25 import org.opendaylight.vpnservice.itm.globals.ITMConstants;
  26 import org.opendaylight.vpnservice.itm.listeners.TransportZoneListener;
  27 import org.opendaylight.vpnservice.itm.rpc.ItmManagerRpcService;
  28 import org.opendaylight.vpnservice.mdsalutil.interfaces.IMdsalApiManager;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.CreateIdPoolInput;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.CreateIdPoolInputBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.IdManagerService;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rpcs.rev151003.OdlInterfaceRpcService;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rpcs.rev151217.ItmRpcService;
  34
  35 import org.opendaylight.yangtools.yang.common.RpcResult;
  36 import org.slf4j.Logger;
  37 import org.slf4j.LoggerFactory;
  38
  39 public class ItmProvider implements BindingAwareProvider, AutoCloseable, IITMProvider /*,ItmStateService */{
  40
  41     private static final Logger LOG = LoggerFactory.getLogger(ItmProvider.class);
  42     private IInterfaceManager interfaceManager;
  43     private ITMManager itmManager;
  44     private IMdsalApiManager mdsalManager;
  45     private DataBroker dataBroker;
  46     private NotificationPublishService notificationPublishService;
  47     private ItmManagerRpcService itmRpcService ;
  48     private IdManagerService idManager;
  49     private NotificationService notificationService;
  50     private TransportZoneListener tzChangeListener;
  51     private RpcProviderRegistry rpcProviderRegistry;
  52
  53     public void setRpcProviderRegistry(RpcProviderRegistry rpcProviderRegistry) {
  54         this.rpcProviderRegistry = rpcProviderRegistry;
  55     }
  56
  57     public RpcProviderRegistry getRpcProviderRegistry() {
  58         return this.rpcProviderRegistry;
  59     }
  60
  61     @Override
  62     public void onSessionInitiated(ProviderContext session) {
  63         LOG.info("ItmProvider Session Initiated");
  64         try {
  65             dataBroker = session.getSALService(DataBroker.class);
  66             idManager = getRpcProviderRegistry().getRpcService(IdManagerService.class);
  67
  68             itmManager = new ITMManager(dataBroker);
  69             tzChangeListener = new TransportZoneListener(dataBroker, idManager) ;
  70             itmRpcService = new ItmManagerRpcService(dataBroker, idManager);
  71             final BindingAwareBroker.RpcRegistration<ItmRpcService> rpcRegistration = getRpcProviderRegistry().addRpcImplementation(ItmRpcService.class, itmRpcService);
  72             itmRpcService.setMdsalManager(mdsalManager);
  73             itmManager.setMdsalManager(mdsalManager);
  74             itmManager.setNotificationPublishService(notificationPublishService);
  75             itmManager.setMdsalManager(mdsalManager);
  76             tzChangeListener.setItmManager(itmManager);
  77             tzChangeListener.registerListener(LogicalDatastoreType.CONFIGURATION, dataBroker);
  78             createIdPool();
  79         } catch (Exception e) {
  80             LOG.error("Error initializing services", e);
  81         }
  82     }
  83
  84     public void setInterfaceManager(IInterfaceManager interfaceManager) {
  85         this.interfaceManager = interfaceManager;
  86     }
  87
  88     public void setNotificationPublishService(NotificationPublishService notificationPublishService) {
  89         this.notificationPublishService = notificationPublishService;
  90     }
  91
  92     public void setMdsalApiManager(IMdsalApiManager mdsalMgr) {
  93         this.mdsalManager = mdsalMgr;
  94     }
  95     public void setNotificationService(NotificationService notificationService) {
  96         this.notificationService = notificationService;
  97     }
  98
  99     @Override
 100     public void close() throws Exception {
 101         if (itmManager != null) {
 102             itmManager.close();
 103         }
 104         if (tzChangeListener != null) {
 105             tzChangeListener.close();
 106         }
 107
 108         LOG.info("ItmProvider Closed");
 109     }
 110
 111     private void createIdPool() {
 112         CreateIdPoolInput createPool = new CreateIdPoolInputBuilder()
 113             .setPoolName(ITMConstants.ITM_IDPOOL_NAME)
 114             .setLow(ITMConstants.ITM_IDPOOL_START)
 115             .setHigh(new BigInteger(ITMConstants.ITM_IDPOOL_SIZE).longValue())
 116             .build();
 117         try {
 118             Future<RpcResult<Void>> result = idManager.createIdPool(createPool);
 119             if ((result != null) && (result.get().isSuccessful())) {
 120                 LOG.debug("Created IdPool for ITM Service");
 121             }
 122         } catch (InterruptedException | ExecutionException e) {
 123             LOG.error("Failed to create idPool for ITM Service",e);
 124         }
 125     }
 126 }