NetworkSpeed.cpp

#include "NetworkSpeed.h"
#include "ui_NetworkSpeed.h"
#include "Logger.h"
#include "WebsocketServer.h"
#include <QTimer>
#include <QFile>
#include <QtMath>
#include <QRegExp>
#include <QString>
#include <QStringList>
#include <QJsonObject>
#include <QProcess>
NetworkSpeed::NetworkSpeed(QWidget *parent) :
	QWidget(parent),
	ui(new Ui::NetworkSpeed)
{
	ui->setupUi(this);
    //添加webSocket
    WebsocketServer::instance()->registerObject("netSpeed",this);
	//流量统计置零
	uploaded=0;
	downloaded=0;
	timer=new QTimer(this);
	connect(timer,SIGNAL(timeout()),this,SLOT(calcSpeed()));
	timer->start(1000);
}

NetworkSpeed::~NetworkSpeed(){
	delete ui;
}


void NetworkSpeed::calcSpeed(){
#if _WIN32
	ui->download->setText("Windows不支持流量统计");
	ui->upload->setText("Windows不支持流量统计");
	ui->dSpeed->setText("Windows不支持流量统计");
	ui->uSpeed->setText("Windows不支持流量统计");
	timer->stop();
	return;
#endif
	QFile *fp=new QFile("/proc/net/dev");
    if(!fp->open(QIODevice::ReadOnly|QIODevice::Text)){
        qDebug()<<"获取网络流量失败!";
        return;
    }
    QTextStream in(fp);
    for(int i=0;i<2;i++) in.readLine();
    QRegExp virtualDev("/^(virbr|br|vnet|tun|tap)[0-9]+/");
    quint64 newDown=0;
    quint64 newUp=0;
    while(!in.atEnd()){
        QString intf=in.readLine();
        QStringList fields=intf.split(QRegExp("\\W+"),QString::SkipEmptyParts);
        if(fields.length()<2) continue;
        if(fields[0]=="lo"||virtualDev.exactMatch(fields[0])) continue;
        newDown+=fields[1].toULong();
        newUp  +=fields[9].toULong();
    }
	fp->close();
    delete fp;
    quint32 dDown=newDown-downloaded;
    quint32 dUp=newUp-uploaded;
	ui->download->setText(bitCountToString(newDown));
	ui->upload->setText(bitCountToString(newUp));
	ui->dSpeed->setText(bitCountToString(dDown)+"/S");
	ui->uSpeed->setText(bitCountToString(dUp)+"/S");
    dSpeed=dDown;
    uSpeed=dUp;
	downloaded=newDown;
    uploaded=newUp;
}
QString NetworkSpeed::bitCountToString(quint64 sum){
	int rate=sum>0?qFloor(qLn(sum)/qLn(1024)):0;
	const QString units[]={"B","KB","MB","GB","TB","PB","EB"};
	return QString("%1%2").arg(QString::number(sum/qPow(1024,rate),'f',1),units[rate]);
}
QJsonValue NetworkSpeed::getNetworkDev(){
    QJsonObject retn;
    QJsonObject flow;
    QProcess* proc;
    flow.insert("up",(qint64)(this->uploaded));
    flow.insert("down",(qint64)(this->downloaded));
    flow.insert("uSpeed",(qint64)(this->uSpeed));
    flow.insert("dSpeed",(qint64)(this->dSpeed));
    retn.insert("flow",flow);
    proc=new QProcess(this);
    proc->start("netstat -nta --inet");
    proc->waitForFinished();
    int rows=-2;
    while(!proc->atEnd()) {
        proc->readLine();
        rows++;
    }
    proc->close();
    delete proc;
    retn.insert("connections",rows);
    return retn;
}