新建module---获取带宽信息

借鉴自http://blog.csdn.net/xjtuse2014/article/details/53968726

1.MoniterBandwidth模块：

 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());
         }
     }
 }

2.IMonitorBandwidthService类：

 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();
 }

在ubuntu命令行，需执行

curl -X POST http://192.168.161.1:8080//wm/statistics/config/enable/json

才能获取到带宽，结果如下所示：

前提是执行了iperf h1 h2命令后才能看到，否则带宽一直显示是0