新建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