liuzheng/jimeng-free-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
jonathang4 2025-08-27 19:32:57 +08:00
parent 6753c1f16c
commit d2c78b75d3
4 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TOS_UPLOAD_INTEGRATION.md
View File

@ -115,8 +115,8 @@ TOS_ENDPOINT=tos-cn-beijing.volces.com
## 文件存储路径 ## 文件存储路径
- **图片**: `images/image-{timestamp}-{random}.png` - **图片**: `jimeng_free/images/{taskId}/image-{timestamp}-{random}.png`
- **视频**: `videos/video-{timestamp}-{random}.mp4` - **视频**: `jimeng_free/videos/{taskId}/video-{timestamp}-{random}.mp4`
## 错误处理 ## 错误处理

6
src/api/ImagesTaskCache.ts
View File

@ -77,7 +77,7 @@ export class ImagesTaskCache {
* @param imageUrls 图片URL数组 * @param imageUrls 图片URL数组
* @returns TOS URL数组 * @returns TOS URL数组
*/ */
private async uploadImagesToTOS(imageUrls: string[]): Promise<string[]> { private async uploadImagesToTOS(imageUrls: string[], task_id: string): Promise<string[]> {
const tosUrls: string[] = []; const tosUrls: string[] = [];
for (const imageUrl of imageUrls) { for (const imageUrl of imageUrls) {
@ -85,7 +85,7 @@ export class ImagesTaskCache {
// 从URL获取文件名 // 从URL获取文件名
const fileName = `image-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.png`; const fileName = `image-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.png`;
// 上传到TOS // 上传到TOS
const tosUrl = await TOSService.uploadFromUrl(imageUrl, `jimeng_free/images/${fileName}`); const tosUrl = await TOSService.uploadFromUrl(imageUrl, `jimeng_free/images/${task_id}/${fileName}`);
tosUrls.push(tosUrl); tosUrls.push(tosUrl);
logger.info(`图片上传到TOS成功: ${imageUrl} -> ${tosUrl}`); logger.info(`图片上传到TOS成功: ${imageUrl} -> ${tosUrl}`);
} catch (error) { } catch (error) {
@ -115,7 +115,7 @@ export class ImagesTaskCache {
// 任务成功完成时，自动上传到TOS // 任务成功完成时，自动上传到TOS
cacheLog(`开始上传图片到TOS: ${taskId}`); cacheLog(`开始上传图片到TOS: ${taskId}`);
const imageUrls = url.split(','); const imageUrls = url.split(',');
const tosUrls = await this.uploadImagesToTOS(imageUrls); const tosUrls = await this.uploadImagesToTOS(imageUrls, taskId);
finalUrl = tosUrls.join(','); finalUrl = tosUrls.join(',');
this.tosProcessedTasks.add(taskId); this.tosProcessedTasks.add(taskId);
cacheLog(`Task ${taskId} TOS上传完成，存储TOS地址: ${finalUrl}`); cacheLog(`Task ${taskId} TOS上传完成，存储TOS地址: ${finalUrl}`);

6
src/api/VideoTaskCache.ts
View File

@ -77,7 +77,7 @@ export class VideoTaskCache {
* @param videoUrls 视频URL数组 * @param videoUrls 视频URL数组
* @returns TOS URL数组 * @returns TOS URL数组
*/ */
private async uploadVideosToTOS(videoUrls: string[]): Promise<string[]> { private async uploadVideosToTOS(videoUrls: string[], task_id: string): Promise<string[]> {
const tosUrls: string[] = []; const tosUrls: string[] = [];
for (const videoUrl of videoUrls) { for (const videoUrl of videoUrls) {
@ -85,7 +85,7 @@ export class VideoTaskCache {
// 从URL获取文件名 // 从URL获取文件名
const fileName = `video-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.mp4`; const fileName = `video-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.mp4`;
// 上传到TOS // 上传到TOS
const tosUrl = await TOSService.uploadFromUrl(videoUrl, `jimeng_free/videos/${fileName}`); const tosUrl = await TOSService.uploadFromUrl(videoUrl, `jimeng_free/videos/${task_id}/${fileName}`);
tosUrls.push(tosUrl); tosUrls.push(tosUrl);
logger.info(`视频上传到TOS成功: ${videoUrl} -> ${tosUrl}`); logger.info(`视频上传到TOS成功: ${videoUrl} -> ${tosUrl}`);
} catch (error) { } catch (error) {
@ -115,7 +115,7 @@ export class VideoTaskCache {
// 任务成功完成时，自动上传到TOS // 任务成功完成时，自动上传到TOS
cacheLog(`开始上传视频到TOS: ${taskId}`); cacheLog(`开始上传视频到TOS: ${taskId}`);
const videoUrls = url.split(','); const videoUrls = url.split(',');
const tosUrls = await this.uploadVideosToTOS(videoUrls); const tosUrls = await this.uploadVideosToTOS(videoUrls, taskId);
finalUrl = tosUrls.join(','); finalUrl = tosUrls.join(',');
this.tosProcessedTasks.add(taskId); this.tosProcessedTasks.add(taskId);
cacheLog(`Task ${taskId} TOS上传完成，存储TOS地址: ${finalUrl}`); cacheLog(`Task ${taskId} TOS上传完成，存储TOS地址: ${finalUrl}`);

6
src/lib/services/TaskPollingService.ts
View File

@ -340,7 +340,7 @@ export class TaskPollingService {
taskLog(`Task ${task.task_id} generated ${originalUrls.length} files`); taskLog(`Task ${task.task_id} generated ${originalUrls.length} files`);
// 上传到TOS // 上传到TOS
const tosUrls = await this.uploadToTOS(originalUrls, task.task_type); const tosUrls = await this.uploadToTOS(originalUrls, task.task_type, task.task_id);
// 创建结果记录 // 创建结果记录
const expireTime = currentTime + parseInt(process.env.RESULT_EXPIRE_TIME || '86400'); const expireTime = currentTime + parseInt(process.env.RESULT_EXPIRE_TIME || '86400');
@ -746,7 +746,7 @@ export class TaskPollingService {
/** /**
* 上传文件到TOS * 上传文件到TOS
*/ */
private async uploadToTOS(urls: string[], taskType: 'image' | 'video'): Promise<string[]> { private async uploadToTOS(urls: string[], taskType: 'image' | 'video', task_id: string): Promise<string[]> {
const tosUrls: string[] = []; const tosUrls: string[] = [];
for (const url of urls) { for (const url of urls) {
@ -756,7 +756,7 @@ export class TaskPollingService {
: `video-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.mp4`; : `video-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.mp4`;
const folder = taskType === 'image' ? 'images' : 'videos'; const folder = taskType === 'image' ? 'images' : 'videos';
const tosUrl = await TOSService.uploadFromUrl(url, `jimeng_free/${folder}/${fileName}`); const tosUrl = await TOSService.uploadFromUrl(url, `jimeng_free/${folder}/${task_id}/${fileName}`);
tosUrls.push(tosUrl); tosUrls.push(tosUrl);
taskLog(`${taskType} uploaded to TOS: ${url} -> ${tosUrl}`); taskLog(`${taskType} uploaded to TOS: ${url} -> ${tosUrl}`);