Implement per user drive stats

2 files changed, 4 insertions, 2 deletions

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);
 }