86 lines
3.0 KiB
JavaScript
86 lines
3.0 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const connectDB = require('../config/db');
|
|
const User = require('../models/User');
|
|
const Asset = require('../models/Asset');
|
|
|
|
/**
|
|
* Hàm định dạng dung lượng file
|
|
*/
|
|
function formatBytes(bytes, decimals = 2) {
|
|
if (bytes === 0) return '0 Bytes';
|
|
const k = 1024;
|
|
const dm = decimals < 0 ? 0 : decimals;
|
|
const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB'];
|
|
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
|
return parseFloat((bytes / Math.pow(k, i)).toFixed(dm)) + ' ' + sizes[i];
|
|
}
|
|
|
|
/**
|
|
* Script thống kê dung lượng lưu trữ theo từng người dùng
|
|
*/
|
|
const getStorageStats = async () => {
|
|
try {
|
|
console.log('=== ĐANG TRUY XUẤT THỐNG KÊ DUNG LƯỢNG LƯU TRỮ ===');
|
|
await connectDB();
|
|
|
|
// 1. Lấy tất cả user để map tên
|
|
const users = await User.find().select('username email').lean();
|
|
const userMap = {};
|
|
const stats = {};
|
|
|
|
users.forEach(u => {
|
|
userMap[u._id.toString()] = u.username;
|
|
stats[u._id.toString()] = {
|
|
username: u.username,
|
|
email: u.email,
|
|
fileCount: 0,
|
|
totalBytes: 0
|
|
};
|
|
});
|
|
|
|
// Thêm mục cho các asset không có chủ sở hữu (nếu có)
|
|
stats['unknown'] = { username: 'Không xác định', email: 'N/A', fileCount: 0, totalBytes: 0 };
|
|
|
|
// 2. Lấy tất cả Asset
|
|
const assets = await Asset.find().lean();
|
|
console.log(`- Đang kiểm tra ${assets.length} tệp tin trong hệ thống...`);
|
|
|
|
for (const asset of assets) {
|
|
const userId = asset.uploadedBy ? asset.uploadedBy.toString() : 'unknown';
|
|
|
|
if (asset.filePath && fs.existsSync(asset.filePath)) {
|
|
const fileStat = fs.statSync(asset.filePath);
|
|
if (stats[userId]) {
|
|
stats[userId].totalBytes += fileStat.size;
|
|
stats[userId].fileCount += 1;
|
|
} else {
|
|
stats['unknown'].totalBytes += fileStat.size;
|
|
stats['unknown'].fileCount += 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
// 3. Hiển thị kết quả
|
|
console.log('\nKết quả thống kê:');
|
|
console.log(''.padEnd(60, '-'));
|
|
console.log(`${'Username'.padEnd(20)} | ${'Số file'.padEnd(10)} | ${'Dung lượng'}`);
|
|
console.log(''.padEnd(60, '-'));
|
|
|
|
Object.values(stats).forEach(userStat => {
|
|
if (userStat.fileCount > 0) {
|
|
console.log(`${userStat.username.padEnd(20)} | ${userStat.fileCount.toString().padEnd(10)} | ${formatBytes(userStat.totalBytes)}`);
|
|
}
|
|
});
|
|
console.log(''.padEnd(60, '-'));
|
|
|
|
mongoose.connection.close();
|
|
} catch (err) {
|
|
console.error('Lỗi khi thống kê:', err.message);
|
|
if (mongoose.connection) mongoose.connection.close();
|
|
process.exit(1);
|
|
}
|
|
};
|
|
|
|
getStorageStats(); |