From 99aa588ae74ed0440bbe5d2bc14ec59f33b2d4f6 Mon Sep 17 00:00:00 2001 From: syuilo Date: Mon, 22 Oct 2018 04:44:10 +0900 Subject: Implement per user drive stats --- src/services/drive/add-file.ts | 3 ++- src/services/drive/delete-file.ts | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) (limited to 'src/services/drive') diff --git a/src/services/drive/add-file.ts b/src/services/drive/add-file.ts index 830f5e0431..77ee70db55 100644 --- a/src/services/drive/add-file.ts +++ b/src/services/drive/add-file.ts @@ -17,7 +17,7 @@ import { isLocalUser, IUser, IRemoteUser } from '../../models/user'; import delFile from './delete-file'; import config from '../../config'; import { getDriveFileThumbnailBucket } from '../../models/drive-file-thumbnail'; -import { driveStats } from '../stats'; +import { driveStats, perUserDriveStats } from '../stats'; const log = debug('misskey:drive:add-file'); @@ -394,6 +394,7 @@ export default async function( // 統計を更新 driveStats.update(driveFile, true); + perUserDriveStats.update(driveFile, true); return driveFile; } diff --git a/src/services/drive/delete-file.ts b/src/services/drive/delete-file.ts index e6aa1a0649..761a5d6d4f 100644 --- a/src/services/drive/delete-file.ts +++ b/src/services/drive/delete-file.ts @@ -2,7 +2,7 @@ import * as Minio from 'minio'; import DriveFile, { DriveFileChunk, IDriveFile } from '../../models/drive-file'; import DriveFileThumbnail, { DriveFileThumbnailChunk } from '../../models/drive-file-thumbnail'; import config from '../../config'; -import { driveStats } from '../stats'; +import { driveStats, perUserDriveStats } from '../stats'; export default async function(file: IDriveFile, isExpired = false) { if (file.metadata.storage == 'minio') { @@ -49,4 +49,5 @@ export default async function(file: IDriveFile, isExpired = false) { // 統計を更新 driveStats.update(file, false); + perUserDriveStats.update(file, false); } -- cgit v1.2.3-freya