refactor: C++ 后端文件移入 src/ 目录,优化项目结构层级
This commit is contained in:
@@ -0,0 +1,339 @@
|
||||
// SPDX-FileCopyrightText: 2024 MyCompany
|
||||
//
|
||||
// SPDX-License-Identifier: LGPL-3.0-or-later
|
||||
|
||||
#include "networkmonitorapplet.h"
|
||||
#include <pluginfactory.h>
|
||||
|
||||
#include <QFile>
|
||||
#include <QTextStream>
|
||||
#include <QDebug>
|
||||
#include <QDir>
|
||||
#include <QRegularExpression>
|
||||
#include <QNetworkInterface>
|
||||
|
||||
DS_BEGIN_NAMESPACE
|
||||
|
||||
NetworkMonitorApplet::NetworkMonitorApplet(QObject *parent)
|
||||
: DApplet(parent)
|
||||
, m_refreshTimer(nullptr)
|
||||
, m_lastRxBytes(0)
|
||||
, m_lastTxBytes(0)
|
||||
, m_downloadSpeed(0)
|
||||
, m_uploadSpeed(0)
|
||||
, m_totalDownload(0)
|
||||
, m_totalUpload(0)
|
||||
, m_ready(false)
|
||||
, m_firstUpdate(true)
|
||||
{
|
||||
}
|
||||
|
||||
NetworkMonitorApplet::~NetworkMonitorApplet()
|
||||
{
|
||||
}
|
||||
|
||||
bool NetworkMonitorApplet::load()
|
||||
{
|
||||
return DApplet::load();
|
||||
}
|
||||
|
||||
bool NetworkMonitorApplet::init()
|
||||
{
|
||||
detectInterfaces();
|
||||
|
||||
// 启动定时刷新
|
||||
m_refreshTimer = new QTimer(this);
|
||||
connect(m_refreshTimer, &QTimer::timeout, this, &NetworkMonitorApplet::refresh);
|
||||
m_refreshTimer->start(1000); // 每秒刷新一次
|
||||
|
||||
// 初始读取
|
||||
readNetworkStats();
|
||||
|
||||
return DApplet::init();
|
||||
}
|
||||
|
||||
double NetworkMonitorApplet::downloadSpeed() const
|
||||
{
|
||||
return static_cast<double>(m_downloadSpeed);
|
||||
}
|
||||
|
||||
double NetworkMonitorApplet::uploadSpeed() const
|
||||
{
|
||||
return static_cast<double>(m_uploadSpeed);
|
||||
}
|
||||
|
||||
double NetworkMonitorApplet::totalDownload() const
|
||||
{
|
||||
return static_cast<double>(m_totalDownload);
|
||||
}
|
||||
|
||||
double NetworkMonitorApplet::totalUpload() const
|
||||
{
|
||||
return static_cast<double>(m_totalUpload);
|
||||
}
|
||||
|
||||
QStringList NetworkMonitorApplet::networkInterfaces() const
|
||||
{
|
||||
return m_interfaceList;
|
||||
}
|
||||
|
||||
QStringList NetworkMonitorApplet::interfaceStats() const
|
||||
{
|
||||
QStringList stats;
|
||||
for (const QString &name : m_interfaceList) {
|
||||
if (m_interfaces.contains(name)) {
|
||||
const NetworkInterface &iface = m_interfaces[name];
|
||||
stats << QString("%1|%2|%3|%4|%5")
|
||||
.arg(iface.name)
|
||||
.arg(iface.rxBytes)
|
||||
.arg(iface.txBytes)
|
||||
.arg(iface.rxPackets)
|
||||
.arg(iface.txPackets);
|
||||
}
|
||||
}
|
||||
return stats;
|
||||
}
|
||||
|
||||
bool NetworkMonitorApplet::ready() const
|
||||
{
|
||||
return m_ready;
|
||||
}
|
||||
|
||||
QString NetworkMonitorApplet::activeInterface() const
|
||||
{
|
||||
return m_activeInterface;
|
||||
}
|
||||
|
||||
// 返回活动接口的 IPv4 地址,供 QML 显示
|
||||
QString NetworkMonitorApplet::ipAddress() const
|
||||
{
|
||||
return m_ipAddress;
|
||||
}
|
||||
|
||||
// 返回活动接口的 IPv6 全球地址,供 QML 显示
|
||||
QString NetworkMonitorApplet::ipv6Address() const
|
||||
{
|
||||
return m_ipv6Address;
|
||||
}
|
||||
|
||||
void NetworkMonitorApplet::refresh()
|
||||
{
|
||||
readNetworkStats();
|
||||
}
|
||||
|
||||
void NetworkMonitorApplet::setActiveInterface(const QString &interface)
|
||||
{
|
||||
if (m_activeInterface != interface) {
|
||||
m_activeInterface = interface;
|
||||
m_firstUpdate = true;
|
||||
emit activeInterfaceChanged();
|
||||
// 接口切换后立即检测新接口的 IP 地址
|
||||
detectIpAddress();
|
||||
}
|
||||
}
|
||||
|
||||
void NetworkMonitorApplet::readNetworkStats()
|
||||
{
|
||||
QFile file("/proc/net/dev");
|
||||
if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||||
qDebug() << "Failed to open /proc/net/dev";
|
||||
return;
|
||||
}
|
||||
|
||||
QTextStream stream(&file);
|
||||
QStringList lines = stream.readAll().split('\n', Qt::SkipEmptyParts);
|
||||
file.close();
|
||||
|
||||
QMap<QString, NetworkInterface> newInterfaces;
|
||||
|
||||
// 跳过前两行标题
|
||||
for (int i = 2; i < lines.size(); ++i) {
|
||||
QString line = lines[i].trimmed();
|
||||
if (line.isEmpty()) continue;
|
||||
|
||||
// 格式: "interface: rx_bytes rx_packets rx_errors rx_dropped ... tx_bytes tx_packets tx_errors tx_dropped ..."
|
||||
QRegularExpression re("^([^:]+):\\s+(\\d+)\\s+(\\d+)\\s+(\\d+)\\s+(\\d+)\\s+\\d+\\s+\\d+\\s+\\d+\\s+\\d+\\s+(\\d+)\\s+(\\d+)\\s+(\\d+)\\s+(\\d+)");
|
||||
QRegularExpressionMatch match = re.match(line);
|
||||
|
||||
if (match.hasMatch()) {
|
||||
NetworkInterface iface;
|
||||
iface.name = match.captured(1);
|
||||
iface.rxBytes = match.captured(2).toLongLong();
|
||||
iface.rxPackets = match.captured(3).toLongLong();
|
||||
iface.rxErrors = match.captured(4).toLongLong();
|
||||
iface.rxDropped = match.captured(5).toLongLong();
|
||||
iface.txBytes = match.captured(6).toLongLong();
|
||||
iface.txPackets = match.captured(7).toLongLong();
|
||||
iface.txErrors = match.captured(8).toLongLong();
|
||||
iface.txDropped = match.captured(9).toLongLong();
|
||||
|
||||
// 过滤掉回环接口和虚拟接口
|
||||
if (iface.name != "lo" && !iface.name.startsWith("veth") &&
|
||||
!iface.name.startsWith("docker") && !iface.name.startsWith("br-")) {
|
||||
newInterfaces[iface.name] = iface;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 检测接口变化
|
||||
if (newInterfaces.keys() != m_interfaces.keys()) {
|
||||
m_interfaceList = newInterfaces.keys();
|
||||
m_interfaces = newInterfaces;
|
||||
|
||||
// 自动选择活动接口
|
||||
if (m_activeInterface.isEmpty() && !m_interfaceList.isEmpty()) {
|
||||
// 优先选择物理网卡(wlp/wlan/enp/eth)中有流量的接口,
|
||||
// 避免选中 Meta/tun0 等虚拟代理接口作为默认显示
|
||||
for (const QString &name : m_interfaceList) {
|
||||
if (!isPhysicalInterface(name)) continue;
|
||||
const NetworkInterface &iface = m_interfaces[name];
|
||||
if (iface.rxBytes > 0 || iface.txBytes > 0) {
|
||||
m_activeInterface = name;
|
||||
break;
|
||||
}
|
||||
}
|
||||
// 其次选择任意有流量的接口(物理网卡都无流量时的兜底)
|
||||
if (m_activeInterface.isEmpty()) {
|
||||
for (const QString &name : m_interfaceList) {
|
||||
const NetworkInterface &iface = m_interfaces[name];
|
||||
if (iface.rxBytes > 0 || iface.txBytes > 0) {
|
||||
m_activeInterface = name;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
// 最后选第一个(无任何流量时)
|
||||
if (m_activeInterface.isEmpty()) {
|
||||
m_activeInterface = m_interfaceList.first();
|
||||
}
|
||||
emit activeInterfaceChanged();
|
||||
}
|
||||
|
||||
m_ready = !m_interfaceList.isEmpty();
|
||||
emit interfacesChanged();
|
||||
emit readyChanged();
|
||||
} else {
|
||||
m_interfaces = newInterfaces;
|
||||
}
|
||||
|
||||
calculateSpeed();
|
||||
emit statsChanged();
|
||||
// 每次刷新都检测 IP,以应对 DHCP 续约等 IP 变更场景
|
||||
detectIpAddress();
|
||||
}
|
||||
|
||||
void NetworkMonitorApplet::calculateSpeed()
|
||||
{
|
||||
qint64 currentRxBytes = getActiveRxBytes();
|
||||
qint64 currentTxBytes = getActiveTxBytes();
|
||||
|
||||
if (m_firstUpdate) {
|
||||
m_firstUpdate = false;
|
||||
m_lastRxBytes = currentRxBytes;
|
||||
m_lastTxBytes = currentTxBytes;
|
||||
return;
|
||||
}
|
||||
|
||||
// 计算速度(字节/秒)
|
||||
m_downloadSpeed = currentRxBytes - m_lastRxBytes;
|
||||
m_uploadSpeed = currentTxBytes - m_lastTxBytes;
|
||||
|
||||
// 更新总量
|
||||
m_totalDownload = currentRxBytes;
|
||||
m_totalUpload = currentTxBytes;
|
||||
|
||||
m_lastRxBytes = currentRxBytes;
|
||||
m_lastTxBytes = currentTxBytes;
|
||||
|
||||
emit speedChanged();
|
||||
emit totalChanged();
|
||||
}
|
||||
|
||||
void NetworkMonitorApplet::detectInterfaces()
|
||||
{
|
||||
QDir sysClass("/sys/class/net");
|
||||
if (!sysClass.exists()) return;
|
||||
|
||||
QStringList entries = sysClass.entryList(QDir::Dirs | QDir::NoDotAndDotDot);
|
||||
m_interfaceList.clear();
|
||||
|
||||
for (const QString &entry : entries) {
|
||||
// 过滤回环接口
|
||||
if (entry != "lo") {
|
||||
m_interfaceList << entry;
|
||||
}
|
||||
}
|
||||
|
||||
m_ready = !m_interfaceList.isEmpty();
|
||||
emit interfacesChanged();
|
||||
emit readyChanged();
|
||||
}
|
||||
|
||||
qint64 NetworkMonitorApplet::getActiveRxBytes() const
|
||||
{
|
||||
if (m_activeInterface.isEmpty() || !m_interfaces.contains(m_activeInterface)) {
|
||||
return 0;
|
||||
}
|
||||
return m_interfaces[m_activeInterface].rxBytes;
|
||||
}
|
||||
|
||||
qint64 NetworkMonitorApplet::getActiveTxBytes() const
|
||||
{
|
||||
if (m_activeInterface.isEmpty() || !m_interfaces.contains(m_activeInterface)) {
|
||||
return 0;
|
||||
}
|
||||
return m_interfaces[m_activeInterface].txBytes;
|
||||
}
|
||||
|
||||
// 检测活动接口的 IPv4 与 IPv6 地址
|
||||
// IPv4:跳过 loopback,取第一个
|
||||
// IPv6:跳过 loopback 和 link-local (fe80::),取第一个全球地址
|
||||
// 任一地址变化时分别 emit 对应信号通知 QML 更新
|
||||
// 设计原因:DHCP 续约、网络切换、IPv6 SLAAC 等场景下地址可能变化,需持续检测
|
||||
void NetworkMonitorApplet::detectIpAddress()
|
||||
{
|
||||
QString newIp;
|
||||
QString newIpv6;
|
||||
if (!m_activeInterface.isEmpty()) {
|
||||
QNetworkInterface iface = QNetworkInterface::interfaceFromName(m_activeInterface);
|
||||
for (const QNetworkAddressEntry &entry : iface.addressEntries()) {
|
||||
// IPv4:跳过 loopback,取第一个
|
||||
if (entry.ip().protocol() == QAbstractSocket::IPv4Protocol
|
||||
&& !entry.ip().isLoopback()
|
||||
&& newIp.isEmpty()) {
|
||||
newIp = entry.ip().toString();
|
||||
}
|
||||
// IPv6:跳过 loopback 和 link-local (fe80::),取第一个全球地址
|
||||
if (entry.ip().protocol() == QAbstractSocket::IPv6Protocol
|
||||
&& !entry.ip().isLoopback()
|
||||
&& !entry.ip().isLinkLocal()
|
||||
&& newIpv6.isEmpty()) {
|
||||
newIpv6 = entry.ip().toString();
|
||||
}
|
||||
}
|
||||
}
|
||||
if (m_ipAddress != newIp) {
|
||||
m_ipAddress = newIp;
|
||||
emit ipAddressChanged();
|
||||
}
|
||||
if (m_ipv6Address != newIpv6) {
|
||||
m_ipv6Address = newIpv6;
|
||||
emit ipv6AddressChanged();
|
||||
}
|
||||
}
|
||||
|
||||
// 判断是否为物理网卡
|
||||
// 物理网卡前缀:无线 wlp/wlan/wifi,有线 enp/eth
|
||||
// 虚拟接口(Meta/tun/tap/docker/veth/br-)返回 false,
|
||||
// 用于自动选择时优先真实网卡而非代理 TUN 接口
|
||||
bool NetworkMonitorApplet::isPhysicalInterface(const QString &name) const
|
||||
{
|
||||
return name.startsWith("wlp") || name.startsWith("wlan")
|
||||
|| name.startsWith("enp") || name.startsWith("eth");
|
||||
}
|
||||
|
||||
D_APPLET_CLASS(NetworkMonitorApplet)
|
||||
|
||||
DS_END_NAMESPACE
|
||||
|
||||
#include "networkmonitorapplet.moc"
|
||||
Reference in New Issue
Block a user