package net.floodlightcontroller.qos_test;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.HashMap;

 import java.util.Iterator;

 import java.util.Map;

 import java.util.Map.Entry;

 import java.util.concurrent.ScheduledFuture;

 import java.util.concurrent.TimeUnit;

 import org.slf4j.Logger;

 import org.slf4j.LoggerFactory;

 import net.floodlightcontroller.core.IFloodlightProviderService;

 import net.floodlightcontroller.core.internal.IOFSwitchService;

 import net.floodlightcontroller.core.module.FloodlightModuleContext;

 import net.floodlightcontroller.core.module.FloodlightModuleException;

 import net.floodlightcontroller.core.module.IFloodlightModule;

 import net.floodlightcontroller.core.module.IFloodlightService;

 import net.floodlightcontroller.core.types.NodePortTuple;

 import net.floodlightcontroller.statistics.IStatisticsService;

 import net.floodlightcontroller.statistics.StatisticsCollector;

 import net.floodlightcontroller.statistics.SwitchPortBandwidth;

 import net.floodlightcontroller.threadpool.IThreadPoolService;

 /**

  * 带宽获取模块

  * @author cq

  *

  */

 public class MonitorBandwidth implements IFloodlightModule, IMonitorBandwidthService {

     //日志工具

     private static final Logger log = LoggerFactory.getLogger(StatisticsCollector.class);

     //Floodlight最核心的service类，其他service类需要该类提供

     protected static IFloodlightProviderService floodlightProvider;

     //链路数据分析模块，已经由Floodlight实现了，我们只需要调用一下就可以，然后对结果稍做加工，便于我们自己使用

     protected static IStatisticsService statisticsService;

     //Floodllight实现的线程池，当然我们也可以使用Java自带的，但推荐使用这个

     private static IThreadPoolService threadPoolService;

     //Future类，不明白的可以百度 Java现成future,其实C++11也有这个玩意了

     private static ScheduledFuture<?> portBandwidthCollector;

     //交换机相关的service,通过这个服务，我们可以获取所有的交换机，即DataPath

     private static IOFSwitchService switchService;

     //存放每条俩路的带宽使用情况

     private static Map<NodePortTuple,SwitchPortBandwidth> bandwidth;

     //搜集数据的周期

     private static final int portBandwidthInterval = 4;

     /**

      * 获取带宽使用情况，计算带宽：

      * switchPortBand.getBitsPerSecondRx().getValue()/(8*1024) + switchPortBand.getBitsPerSecondTx().getValue()/(8*1024)

      */

     @Override

     public Map<NodePortTuple, SwitchPortBandwidth> getBandwidthMap() {

         bandwidth = statisticsService.getBandwidthConsumption();

         Iterator<Entry<NodePortTuple, SwitchPortBandwidth>> iter = bandwidth.entrySet().iterator();

         while(iter.hasNext()) {

             Entry<NodePortTuple,SwitchPortBandwidth> entry = iter.next();

             NodePortTuple tuple  = entry.getKey();

             SwitchPortBandwidth switchPortBand = entry.getValue();

             System.out.print("节点id以及端口号：" + tuple.getNodeId()+","+tuple.getPortId().getPortNumber()+",");

             System.out.println("端口带宽：" + switchPortBand.getBitsPerSecondRx().getValue()/(8*1024) + switchPortBand.getBitsPerSecondTx().getValue()/(8*1024));

         }

         return bandwidth;

     }

     /**

      * 告诉FL，我们添加了一个模块，提供了IMonitorBandwidthService

      */

     @Override

     public Collection<Class<? extends IFloodlightService>> getModuleServices() {

         Collection<Class<? extends IFloodlightService>> l = new ArrayList<Class<? extends IFloodlightService>>();

         l.add(IMonitorBandwidthService.class);

         return l;

     }

     /**

      * 我们前面声明了几个需要使用的service，在这里说明一下实现类

      */

     @Override

     public Map<Class<? extends IFloodlightService>, IFloodlightService> getServiceImpls() {

         Map<Class<? extends IFloodlightService>, IFloodlightService> m = new HashMap<Class<? extends IFloodlightService>, IFloodlightService>();

         m.put(IMonitorBandwidthService.class, this);

         return m;

     }

     /**

      * 告诉FL我们依赖哪些服务，以便于加载

      */

     @Override

     public Collection<Class<? extends IFloodlightService>> getModuleDependencies() {

         Collection<Class<? extends IFloodlightService>> l = new ArrayList<Class<? extends IFloodlightService>>();

         l.add(IFloodlightProviderService.class);

         l.add(IStatisticsService.class);

         l.add(IOFSwitchService.class);

         l.add(IThreadPoolService.class);

         return l;

     }

     /**

      * 初始化这些service

      */

     @Override

     public void init(FloodlightModuleContext context) throws FloodlightModuleException {

         floodlightProvider = context.getServiceImpl(IFloodlightProviderService.class);

         statisticsService = context.getServiceImpl(IStatisticsService.class);

         switchService = context.getServiceImpl(IOFSwitchService.class);

         threadPoolService = context.getServiceImpl(IThreadPoolService.class);

     }

     /**

      * 启动

      */

     @Override

     public void startUp(FloodlightModuleContext context) throws FloodlightModuleException {

         startCollectBandwidth();

     }

     //自定义的开始收集数据的方法，使用了线程池，定周期的执行

     private synchronized void startCollectBandwidth(){

         portBandwidthCollector = threadPoolService.getScheduledExecutor().scheduleAtFixedRate(new GetBandwidthThread(), portBandwidthInterval, portBandwidthInterval, TimeUnit.SECONDS);

         log.warn("Statistics collection thread(s) started");

     }

     //自定义的线程类，在上面的方法中实例化，并被调用

     private class GetBandwidthThread extends Thread implements Runnable  {

         private Map<NodePortTuple,SwitchPortBandwidth> bandwidth;

         public Map<NodePortTuple, SwitchPortBandwidth> getBandwidth() {

             return bandwidth;

         }

         @Override

         public void run() {

             System.out.println("GetBandwidthThread run()....");

             bandwidth =getBandwidthMap();

             System.out.println("bandwidth.size():"+bandwidth.size());

         }

     }

 }

 package net.floodlightcontroller.qos_test;

 import java.util.Map;

 import net.floodlightcontroller.core.module.IFloodlightService;

 import net.floodlightcontroller.core.types.NodePortTuple;

 import net.floodlightcontroller.statistics.SwitchPortBandwidth;

 public interface IMonitorBandwidthService extends IFloodlightService{

     //带宽使用情况

     public Map<NodePortTuple, SwitchPortBandwidth> getBandwidthMap();

 }