62 lines
3.5 KiB
JavaScript
62 lines
3.5 KiB
JavaScript
|
#!/usr/bin/env node
|
||
|
|
/**
|
||
|
|
* storage-report.cjs —— 每用户存储用量上报（档案 28）
|
||
|
|
* 输出：/var/run/dsh-storage-report.json（供门户 GET /api/admin/storage 直接读取，避免每次请求都 du）
|
||
|
|
* cron：每小时一次。
|
||
|
|
*/
|
||
|
|
const { execFileSync } = require('node:child_process')
|
||
|
|
const { existsSync, writeFileSync } = require('node:fs')
|
||
|
|
const { join } = require('node:path')
|
||
|
|
const Database = require('/opt/dshs/node_modules/better-sqlite3')
|
||
|
|
|
||
|
|
const OUT = process.env.DSH_STORAGE_REPORT ?? '/var/run/dsh-storage-report.json'
|
||
|
|
const WS_MB = Number(process.env.DSH_WS_THRESHOLD_MB ?? 2048)
|
||
|
|
const SESS_MB = Number(process.env.DSH_SESSIONS_THRESHOLD_MB ?? 1024)
|
||
|
|
|
||
|
|
const duKB = (p) => { try { return Number(execFileSync('du', ['-sk', p], { encoding: 'utf8' }).split(/\s+/)[0]) * 1024 } catch { return 0 } }
|
||
|
|
const db = new Database('/var/lib/dshs/dshs.db', { readonly: true })
|
||
|
|
const users = db.prepare('SELECT username, home_dir FROM users ORDER BY username').all()
|
||
|
|
|
||
|
|
const rows = users.map((u) => {
|
||
|
|
const ws = join(u.home_dir, '..', 'ws'), sessions = join(u.home_dir, 'sessions'), trash = join(u.home_dir, '..', 'trash')
|
||
|
|
const t1 = [], t2 = []
|
||
|
|
// 仅统计顶层候选（与 ws-cleanup 口径一致），供管理员判断
|
||
|
|
if (existsSync(ws)) {
|
||
|
|
const { readdirSync, statSync } = require('node:fs')
|
||
|
|
const T1_DIRS = ['.local', '.cache', '.poc-backup', 'poc', '__pycache__', '.pytest_cache', 'node_modules', '.ipynb_checkpoints']
|
||
|
|
const T1_SUFFIX = ['.tgz', '.tmp', '.log', '.bak', '.part', '.crdownload']
|
||
|
|
const T2_EXT = new Set(['.py', '.js', '.mjs', '.cjs', '.sh', '.bash', '.ps1', '.psm1', '.bat', '.cmd', '.ts', '.ipynb', '.sql'])
|
||
|
|
const CUTOFF = Date.now() - 90 * 86400_000
|
||
|
|
for (const e of readdirSync(ws)) {
|
||
|
|
const p = join(ws, e)
|
||
|
|
let st; try { st = statSync(p) } catch { continue }
|
||
|
|
if (st.isDirectory()) { if (T1_DIRS.includes(e)) t1.push({ name: e, size: duKB(p) }); continue }
|
||
|
|
const dot = e.lastIndexOf('.')
|
||
|
|
const ext = dot >= 0 ? e.slice(dot).toLowerCase() : ''
|
||
|
|
if (T1_SUFFIX.includes(ext)) t1.push({ name: e, size: st.size })
|
||
|
|
else if (T2_EXT.has(ext) && st.mtimeMs < CUTOFF) t2.push({ name: e, size: st.size })
|
||
|
|
}
|
||
|
|
}
|
||
|
|
const wsB = existsSync(ws) ? duKB(ws) : 0
|
||
|
|
const sessB = existsSync(sessions) ? duKB(sessions) : 0
|
||
|
|
const trashB = existsSync(trash) ? duKB(trash) : 0
|
||
|
|
return {
|
||
|
|
username: u.username,
|
||
|
|
ws: wsB, sessions: sessB, trash: trashB, total: wsB + sessB + trashB,
|
||
|
|
wsOver: wsB >= WS_MB * 1048576, sessionsOver: sessB >= SESS_MB * 1048576,
|
||
|
|
cleanableT1: t1.length, cleanableT2: t2.length,
|
||
|
|
cleanableBytes: [...t1, ...t2].reduce((a, c) => a + c.size, 0),
|
||
|
|
topCleanable: [...t1, ...t2].sort((a, b) => b.size - a.size).slice(0, 5).map((c) => `${c.name} (${(c.size / 1048576).toFixed(1)}MB)`),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
db.close()
|
||
|
|
const report = {
|
||
|
|
generatedAt: new Date().toISOString(),
|
||
|
|
thresholds: { wsMB: WS_MB, sessionsMB: SESS_MB, keepDays: { ws: 90, sessions: 365 }, trashKeepDays: 30 },
|
||
|
|
totals: { ws: rows.reduce((a, r) => a + r.ws, 0), sessions: rows.reduce((a, r) => a + r.sessions, 0), trash: rows.reduce((a, r) => a + r.trash, 0) },
|
||
|
|
users: rows,
|
||
|
|
}
|
||
|
|
writeFileSync(OUT, JSON.stringify(report, null, 2) + '\n')
|
||
|
|
const fmt = (b) => (b >= 1073741824 ? (b / 1073741824).toFixed(2) + 'G' : (b / 1048576).toFixed(1) + 'M')
|
||
|
|
for (const r of rows) console.log(` ${r.username}: ws=${fmt(r.ws)} sessions=${fmt(r.sessions)} trash=${fmt(r.trash)} | 可清 ${r.cleanableT1 + r.cleanableT2} 项/${fmt(r.cleanableBytes)}`)
|
||
|
|
console.log(` → 已写入 ${OUT}`)
|