Northern_Lights/NapCatQQ
1
0
Fork 0
mirror of https://github.com/NapNeko/NapCatQQ.git synced 2025-12-22 15:20:07 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add file retention option to UploadFileStream

Browse Source 
Introduces a 'file_retention' parameter to control how long uploaded files are retained before automatic deletion. If set, files are deleted after the specified duration; otherwise, they are not automatically removed. This helps manage temporary file storage and cleanup.
This commit is contained in:
手瓜一十雪 2025-09-20 16:13:25 +08:00
parent 74de3d9100
commit 9b88946209
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/onebot/action/stream/UploadFileStream.ts
View File

@ -7,6 +7,7 @@ import fs from 'fs';
import { join as joinPath } from 'node:path'; import { join as joinPath } from 'node:path';
import { randomUUID } from 'crypto'; import { randomUUID } from 'crypto';
import { createHash } from 'crypto'; import { createHash } from 'crypto';
import { unlink } from 'node:fs';
// 简化配置 // 简化配置
const CONFIG = { const CONFIG = {
@ -25,7 +26,8 @@ const SchemaData = Type.Object({
is_complete: Type.Optional(Type.Boolean()), is_complete: Type.Optional(Type.Boolean()),
filename: Type.Optional(Type.String()), filename: Type.Optional(Type.String()),
reset: Type.Optional(Type.Boolean()), reset: Type.Optional(Type.Boolean()),
verify_only: Type.Optional(Type.Boolean()) verify_only: Type.Optional(Type.Boolean()),
file_retention: Type.Number({ default: 5 * 60 * 1000 }) // 默认5分钟 回收 不设置或0为不回收
}); });
type Payload = Static<typeof SchemaData>; type Payload = Static<typeof SchemaData>;
@ -47,6 +49,7 @@ interface StreamState {
memoryChunks?: Map<number, Buffer>; memoryChunks?: Map<number, Buffer>;
tempDir?: string; tempDir?: string;
finalPath?: string; finalPath?: string;
fileRetention?: number;
// 管理 // 管理
createdAt: number; createdAt: number;
@ -131,9 +134,9 @@ export class UploadFileStream extends OneBotAction<Payload, StreamPacket<StreamR
expectedSha256: expected_sha256, expectedSha256: expected_sha256,
useMemory, useMemory,
createdAt: Date.now(), createdAt: Date.now(),
timeoutId: this.setupTimeout(stream_id) timeoutId: this.setupTimeout(stream_id),
fileRetention: payload.file_retention
}; };
try { try {
if (useMemory) { if (useMemory) {
stream.memoryChunks = new Map(); stream.memoryChunks = new Map();
@ -244,7 +247,13 @@ export class UploadFileStream extends OneBotAction<Payload, StreamPacket<StreamR
// 清理资源但保留文件 // 清理资源但保留文件
this.cleanupStream(stream.id, false); this.cleanupStream(stream.id, false);
if (stream.fileRetention && stream.fileRetention > 0) {
setTimeout(() => {
unlink(finalPath, err => {
this.core.context.logger.logError(`Failed to delete retained file ${finalPath}:`, err);
});
}, stream.fileRetention);
}
return { return {
type: StreamStatus.Response, type: StreamStatus.Response,
stream_id: stream.id, stream_id: stream.id,