Files
yenai-plugin/model/State/Monitor.js
2024-04-13 20:21:36 +08:00

107 lines
2.6 KiB
JavaScript

import { Config, Data } from "../../components/index.js"
import _ from "lodash"
import { si, initDependence } from "./DependencyChecker.js"
import { addData } from "./utils.js"
export default new class monitor {
constructor() {
// 网络
this._network = null
// 读写速率
this._fsStats = null
// 记录60条数据一分钟记录一次
this.chartData = {
network: {
// 上行
upload: [],
// 下行
download: []
},
fsStats: {
// 读
readSpeed: [],
// 写
writeSpeed: []
},
// cpu
cpu: [],
// 内存
ram: [],
// 主题
echarts_theme: Data.readJSON("resources/state/theme_westeros.json")
}
this.valueObject = {
networkStats: "rx_sec,tx_sec,iface",
currentLoad: "currentLoad",
mem: "active",
fsStats: "wx_sec,rx_sec"
}
this.init()
}
set network(value) {
if (_.isNumber(value[0]?.tx_sec) && _.isNumber(value[0]?.rx_sec)) {
this._network = value
addData(this.chartData.network.upload, [ Date.now(), value[0].tx_sec ])
addData(this.chartData.network.download, [ Date.now(), value[0].rx_sec ])
}
}
get network() {
return this._network
}
set fsStats(value) {
if (_.isNumber(value?.wx_sec) && _.isNumber(value?.rx_sec)) {
this._fsStats = value
addData(this.chartData.fsStats.writeSpeed, [ Date.now(), value.wx_sec ])
addData(this.chartData.fsStats.readSpeed, [ Date.now(), value.rx_sec ])
}
}
get fsStats() {
return this._fsStats
}
async init() {
if (!await initDependence()) return
// 给有问题的用户关闭定时器
if (!Config.state.statusTask) return
if (Config.state.statusPowerShellStart) si.powerShellStart()
this.getData()
// 网速
const Timer = setInterval(async() => {
let data = await this.getData()
if (_.isEmpty(data)) clearInterval(Timer)
}, 60000)
}
async getData() {
let data = await si.get(this.valueObject)
_.forIn(data, (value, key) => {
if (_.isEmpty(value)) {
logger.debug(`获取${key}数据失败,停止获取对应数据`)
delete this.valueObject[key]
}
})
let {
fsStats,
networkStats,
mem: { active },
currentLoad: { currentLoad }
} = data
this.fsStats = fsStats
this.network = networkStats
if (_.isNumber(active)) {
addData(this.chartData.ram, [ Date.now(), active ])
}
if (_.isNumber(currentLoad)) {
addData(this.chartData.cpu, [ Date.now(), currentLoad ])
}
return data
}
}()