limepotato/jormungandr-bite
1
0
Fork 0
mirror of https://iceshrimp.dev/limepotato/jormungandr-bite.git synced 2024-11-15 06:27:34 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

#160

Browse source
This commit is contained in:
syuilo 2017-02-14 08:24:18 +09:00
parent 0da6e7a161
commit c32954fb28
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,