mirror/MisskeyIO
1
0
Fork 0
mirror of https://github.com/MisskeyIO/misskey synced 2024-12-11 13:18:51 +09:00
Code

#160

Browse Source
This commit is contained in:
syuilo 2017-02-14 08:24:18 +09:00
parent 3e08c670fb
commit 434bced6f4
2 changed files with 28 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/api/common/add-file-to-drive.ts
View File

@ -79,17 +79,20 @@ export default (
}
}
// Fetch all files to calculate drive usage
const files = await DriveFile
.find({ user_id: user._id }, {
fields: {
datasize: true,
_id: false
}
});
// Calculate drive usage (in byte)
const usage = files.map(file => file.datasize).reduce((x, y) => x + y, 0);
// Calculate drive usage
const usage = ((await DriveFile
.aggregate([
{ $match: { user_id: user._id } },
{ $project: {
datasize: true
}},
{ $group: {
_id: null,
usage: { $sum: '$datasize' }
}}
]))[0] || {
usage: 0
}).usage;
log(`drive usage is ${usage}`);

23
src/api/endpoints/drive.js
View File

@ -15,15 +15,20 @@ import DriveFile from '../models/drive-file';
module.exports = (params, user) =>
new Promise(async (res, rej) =>
{
// Fetch all files to calculate drive usage
const files = await DriveFile
.find({ user_id: user._id }, {
datasize: true,
_id: false
});
// Calculate drive usage (in byte)
const usage = files.map(file => file.datasize).reduce((x, y) => x + y, 0);
// Calculate drive usage
const usage = ((await DriveFile
.aggregate([
{ $match: { user_id: user._id } },
{ $project: {
datasize: true
}},
{ $group: {
_id: null,
usage: { $sum: '$datasize' }
}}
]))[0] || {
usage: 0
}).usage;
res({
capacity: user.drive_capacity,