EchoZenith/electricity-monitor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add electricity bill balance alert function

Browse Source 
新增企业微信通知形式的电费余额预警功能，配置阈值环境变量ALERT_THRESHOLD，当余额低于设定值时自动发送提醒，且避免重复告警
This commit is contained in:
EchoZenith
2026-05-23 00:48:06 +08:00
parent 3b43ddad2c
commit e1b4a0ec5d
2 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.env.example
View File

@@ -8,5 +8,8 @@ LOGIN_PASSWORD=你的密码
# 选填 - 企业微信群机器人 Webhook URL # 选填 - 企业微信群机器人 Webhook URL
WECOM_WEBHOOK_URL=https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=你的key WECOM_WEBHOOK_URL=https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=你的key
# 选填 - 电费余额预警阈值（低于此值时发送企业微信通知，设为0或留空关闭）
ALERT_THRESHOLD=20
# 选填 - Cookie 签名密钥 # 选填 - Cookie 签名密钥
COOKIE_SECRET= COOKIE_SECRET=

26
server.js
View File

@@ -54,6 +54,14 @@ if (!WECOM_WEBHOOK_URL) {
console.warn('警告: 环境变量 WECOM_WEBHOOK_URL 未设置，企业微信通知功能将不可用'); console.warn('警告: 环境变量 WECOM_WEBHOOK_URL 未设置，企业微信通知功能将不可用');
} }
const ALERT_THRESHOLD = parseFloat(process.env.ALERT_THRESHOLD) || 0;
if (ALERT_THRESHOLD > 0) {
console.log(`电费预警: 当余额低于 ¥${ALERT_THRESHOLD} 时将发送企业微信通知`);
}
let lastAlertedAmount = null;
function requireAuth(req, res, next) { function requireAuth(req, res, next) {
const auth = req.signedCookies.auth; const auth = req.signedCookies.auth;
if (auth === LOGIN_USERNAME) { if (auth === LOGIN_USERNAME) {
@@ -163,12 +171,30 @@ async function collectData() {
if (sameDay && sameHour) { if (sameDay && sameHour) {
updateStmt.run(data.surplus, data.amount, data.timestamp, latest.id); updateStmt.run(data.surplus, data.amount, data.timestamp, latest.id);
console.log(`[${new Date().toLocaleString()}] 更新本小时记录: ${data.surplus}度, ¥${data.amount}`); console.log(`[${new Date().toLocaleString()}] 更新本小时记录: ${data.surplus}度, ¥${data.amount}`);
await checkThresholdAndAlert(data.amount);
return; return;
} }
} }
insertStmt.run(data.surplus, data.amount, data.timestamp, data.roomName); insertStmt.run(data.surplus, data.amount, data.timestamp, data.roomName);
console.log(`[${new Date().toLocaleString()}] 新增记录: ${data.surplus}度, ¥${data.amount}`); console.log(`[${new Date().toLocaleString()}] 新增记录: ${data.surplus}度, ¥${data.amount}`);
await checkThresholdAndAlert(data.amount);
}
async function checkThresholdAndAlert(amount) {
if (ALERT_THRESHOLD <= 0 || amount > ALERT_THRESHOLD) {
lastAlertedAmount = null;
return;
}
if (lastAlertedAmount !== null && amount >= lastAlertedAmount) return;
lastAlertedAmount = amount;
console.log(`[预警] 余额 ¥${amount} 低于阈值 ¥${ALERT_THRESHOLD}`);
await sendWecomNotification(
'## 电费余额预警\n\n' +
`> 时间：${new Date().toLocaleString('zh-CN')}\n\n` +
`当前余额 **¥${amount.toFixed(2)}**\n\n` +
`已低于预警阈值 **¥${ALERT_THRESHOLD.toFixed(2)}**，请及时充值！`
);
} }
function calculateDailyUsage(records) { function calculateDailyUsage(records) {