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

Add file_retention parameter to upload test

Browse Source 
Introduces the 'file_retention' field with a value of 30,000 to the upload test payload in OneBotUploadTester. This may be used to specify file retention duration in milliseconds.
This commit is contained in:
手瓜一十雪 2025-09-20 16:19:39 +08:00
parent 9b88946209
commit 937be7678e
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -250,7 +250,7 @@ export class UploadFileStream extends OneBotAction<Payload, StreamPacket<StreamR
if (stream.fileRetention && stream.fileRetention > 0) { if (stream.fileRetention && stream.fileRetention > 0) {
setTimeout(() => { setTimeout(() => {
unlink(finalPath, err => { unlink(finalPath, err => {
this.core.context.logger.logError(`Failed to delete retained file ${finalPath}:`, err); if (err) this.core.context.logger.logError(`Failed to delete retained file ${finalPath}:`, err);
}); });
}, stream.fileRetention); }, stream.fileRetention);
} }

11
src/onebot/action/stream/test_upload_stream.py
View File

@ -29,7 +29,7 @@ class OneBotUploadTester:
headers["Authorization"] = f"Bearer {self.access_token}" headers["Authorization"] = f"Bearer {self.access_token}"
print(f"连接到 {self.ws_url}") print(f"连接到 {self.ws_url}")
self.websocket = await websockets.connect(self.ws_url, extra_headers=headers) self.websocket = await websockets.connect(self.ws_url, additional_headers=headers)
print("WebSocket 连接成功") print("WebSocket 连接成功")
async def disconnect(self): async def disconnect(self):
@ -38,7 +38,7 @@ class OneBotUploadTester:
await self.websocket.close() await self.websocket.close()
print("WebSocket 连接已断开") print("WebSocket 连接已断开")
def calculate_file_chunks(self, file_path: str, chunk_size: int = 64 * 1024) -> tuple[List[bytes], str, int]: def calculate_file_chunks(self, file_path: str, chunk_size: int = 64) -> tuple[List[bytes], str, int]:
""" """
计算文件分片和 SHA256 计算文件分片和 SHA256
@ -97,7 +97,7 @@ class OneBotUploadTester:
print(f"收到其他消息: {data}") print(f"收到其他消息: {data}")
continue continue
async def upload_file_stream_batch(self, file_path: str, chunk_size: int = 64 * 1024) -> str: async def upload_file_stream_batch(self, file_path: str, chunk_size: int = 64 ) -> str:
""" """
一次性批量上传文件流 一次性批量上传文件流
@ -134,7 +134,8 @@ class OneBotUploadTester:
"total_chunks": total_chunks, "total_chunks": total_chunks,
"file_size": total_size, "file_size": total_size,
"expected_sha256": sha256_hash, "expected_sha256": sha256_hash,
"filename": file_path.name "filename": file_path.name,
"file_retention": 30 * 1000
} }
# 发送分片 # 发送分片
@ -171,7 +172,7 @@ class OneBotUploadTester:
else: else:
raise Exception(f"文件状态异常: {result}") raise Exception(f"文件状态异常: {result}")
async def test_upload(self, file_path: str, chunk_size: int = 64 * 1024): async def test_upload(self, file_path: str, chunk_size: int = 64 ):
"""测试文件上传""" """测试文件上传"""
try: try:
await self.connect() await self.connect()