moris/misskey
1
0
Fork 0
forked from mirror/misskey
Code Pull Requests Activity

#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 // Calculate drive usage
const files = await DriveFile const usage = ((await DriveFile
.find({ user_id: user._id }, { .aggregate([
fields: { { $match: { user_id: user._id } },
datasize: true, { $project: {
_id: false datasize: true
} }},
}); { $group: {
_id: null,
// Calculate drive usage (in byte) usage: { $sum: '$datasize' }
const usage = files.map(file => file.datasize).reduce((x, y) => x + y, 0); }}
]))[0] || {
usage: 0
}).usage;
log(`drive usage is ${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) => module.exports = (params, user) =>
new Promise(async (res, rej) => new Promise(async (res, rej) =>
{ {
// Fetch all files to calculate drive usage // Calculate drive usage
const files = await DriveFile const usage = ((await DriveFile
.find({ user_id: user._id }, { .aggregate([
datasize: true, { $match: { user_id: user._id } },
_id: false { $project: {
}); datasize: true
}},
// Calculate drive usage (in byte) { $group: {
const usage = files.map(file => file.datasize).reduce((x, y) => x + y, 0); _id: null,
usage: { $sum: '$datasize' }
}}
]))[0] || {
usage: 0
}).usage;
res({ res({
capacity: user.drive_capacity, capacity: user.drive_capacity,