更新版本至 0.0.39,优化 uploadFile 和 uploadChunkedFile 方法,支持自定义 chunkSize 和 maxSize 参数

This commit is contained in:
2026-02-01 16:22:59 +08:00
parent f4afea4560
commit 78f781ef39
2 changed files with 20 additions and 15 deletions

View File

@@ -1,6 +1,6 @@
{ {
"name": "@kevisual/api", "name": "@kevisual/api",
"version": "0.0.38", "version": "0.0.39",
"description": "", "description": "",
"main": "mod.ts", "main": "mod.ts",
"scripts": { "scripts": {

View File

@@ -62,7 +62,7 @@ export class QueryResources {
const url = `${this.prefix}${filepath}`; const url = `${this.prefix}${filepath}`;
return this.get({}, { url, method: 'GET', ...opts, headers: this.header(opts?.headers, false), isText: true }); return this.get({}, { url, method: 'GET', ...opts, headers: this.header(opts?.headers, false), isText: true });
} }
async uploadFile(filepath: string, content: string | Blob, opts?: DataOpts): Promise<Result<any>> { async uploadFile(filepath: string, content: string | Blob, opts?: DataOpts & { chunkSize?: number, maxSize?: number }): Promise<Result<any>> {
const pathname = `${this.prefix}${filepath}`; const pathname = `${this.prefix}${filepath}`;
const filename = path.basename(pathname); const filename = path.basename(pathname);
const type = getContentType(filename); const type = getContentType(filename);
@@ -71,15 +71,15 @@ export class QueryResources {
// Blob 类型时 hashContent 返回 Promise // Blob 类型时 hashContent 返回 Promise
const hash = hashResult instanceof Promise ? await hashResult : hashResult; const hash = hashResult instanceof Promise ? await hashResult : hashResult;
url.searchParams.set('hash', hash); url.searchParams.set('hash', hash);
const { chunkSize, maxSize, ...restOpts } = opts || {};
// 判断是否需要分块上传文件大于20MB // 判断是否需要分块上传文件大于20MB
const isBlob = content instanceof Blob; const isBlob = content instanceof Blob;
const fileSize = isBlob ? content.size : new Blob([content]).size; const fileSize = isBlob ? content.size : new Blob([content]).size;
const CHUNK_THRESHOLD = 20 * 1024 * 1024; // 20MB const CHUNK_THRESHOLD = maxSize ?? 20 * 1024 * 1024; // 20MB
if (fileSize > CHUNK_THRESHOLD && isBlob) { if (fileSize > CHUNK_THRESHOLD && isBlob) {
// 使用分块上传 // 使用分块上传
return this.uploadChunkedFile(filepath, content, hash, opts); return this.uploadChunkedFile(filepath, content, hash, { chunkSize, ...restOpts });
} }
const formData = new FormData(); const formData = new FormData();
@@ -94,15 +94,15 @@ export class QueryResources {
method: 'POST', method: 'POST',
body: formData, body: formData,
timeout: 5 * 60 * 1000, // 5分钟超时 timeout: 5 * 60 * 1000, // 5分钟超时
...opts, ...restOpts,
headers: { ...opts?.headers, ...this.header(opts?.headers, false) }, headers: { ...restOpts?.headers, ...this.header(restOpts?.headers, false) },
params: { params: {
hash: hash, hash: hash,
...opts?.params, ...restOpts?.params,
}, },
}); });
} }
async uploadChunkedFile(filepath: string, file: Blob, hash: string, opts?: DataOpts): Promise<Result<any>> { async uploadChunkedFile(filepath: string, file: Blob, hash: string, opts?: DataOpts & { chunkSize?: number }): Promise<Result<any>> {
const pathname = `${this.prefix}${filepath}`; const pathname = `${this.prefix}${filepath}`;
const filename = path.basename(pathname); const filename = path.basename(pathname);
const url = new URL(pathname, window.location.origin); const url = new URL(pathname, window.location.origin);
@@ -111,8 +111,8 @@ export class QueryResources {
console.log(`url,`, url, hash); console.log(`url,`, url, hash);
// 预留 eventSource 支持(暂不处理) // 预留 eventSource 支持(暂不处理)
// const createEventSource = opts?.createEventSource; // const createEventSource = opts?.createEventSource;
const { chunkSize: _chunkSize, ...restOpts } = opts || {};
const chunkSize = 5 * 1024 * 1024; // 5MB const chunkSize = _chunkSize ?? 5 * 1024 * 1024; // 5MB
const totalChunks = Math.ceil(file.size / chunkSize); const totalChunks = Math.ceil(file.size / chunkSize);
for (let currentChunk = 0; currentChunk < totalChunks; currentChunk++) { for (let currentChunk = 0; currentChunk < totalChunks; currentChunk++) {
@@ -134,18 +134,23 @@ export class QueryResources {
method: 'POST', method: 'POST',
body: formData, body: formData,
timeout: 5 * 60 * 1000, // 5分钟超时 timeout: 5 * 60 * 1000, // 5分钟超时
...opts, ...restOpts,
headers: { ...opts?.headers, ...this.header(opts?.headers, false) }, headers: { ...restOpts?.headers, ...this.header(restOpts?.headers, false) },
params: { params: {
hash: hash, hash: hash,
chunk: '1', chunk: '1',
...opts?.params, chunkIndex: currentChunk,
totalChunks,
...restOpts?.params,
}, },
}); });
if (res.code !== 200) {
throw new Error(`Chunk upload failed with code ${res!.code}, message: ${res!.message}`);
}
console.log(`Chunk ${currentChunk + 1}/${totalChunks} uploaded`, res); console.log(`Chunk ${currentChunk + 1}/${totalChunks} uploaded`, res);
} catch (error) { } catch (error) {
console.error(`Error uploading chunk ${currentChunk + 1}/${totalChunks}`, error); console.error(`Error uploading chunk ${currentChunk + 1}/${totalChunks}`, error);
throw error; return { code: 500, message: `分块上传失败: ${(error as Error).message}` };
} }
} }