huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2025-01-18 19:20:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【优化】dall 绘画，改为异步。

Browse Source
This commit is contained in:
cherishsince 2024-05-28 11:39:29 +08:00
parent 63a8cc244d
commit e97408b3ac
1 changed files with 47 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
View File

@ -35,6 +35,9 @@ import org.springframework.transaction.annotation.Transactional;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
/**
* ai 作图
@ -53,6 +56,8 @@ public class AiImageServiceImpl implements AiImageService {
private final OpenAiImageClient openAiImageClient;
private final MidjourneyWebSocketStarter midjourneyWebSocketStarter;
private final MidjourneyInteractionsApi midjourneyInteractionsApi;
private static ThreadPoolExecutor EXECUTOR = new ThreadPoolExecutor(
3, 5, 1, TimeUnit.HOURS, new LinkedBlockingQueue<>(32));
@PostConstruct
public void startMidjourney() {
@ -89,34 +94,48 @@ public class AiImageServiceImpl implements AiImageService {
@Override
public AiImageDallRespVO dallDrawing(AiImageDallReqVO req) {
// 获取 model
OpenAiImageModelEnum openAiImageModelEnum = OpenAiImageModelEnum.valueOfModel(req.getModel());
OpenAiImageStyleEnum openAiImageStyleEnum = OpenAiImageStyleEnum.valueOfStyle(req.getStyle());
try {
// 转换openai 参数
OpenAiImageOptions openAiImageOptions = new OpenAiImageOptions();
openAiImageOptions.setModel(openAiImageModelEnum.getModel());
openAiImageOptions.setStyle(openAiImageStyleEnum.getStyle());
openAiImageOptions.setSize(req.getSize());
ImageResponse imageResponse = openAiImageClient.call(new ImagePrompt(req.getPrompt(), openAiImageOptions));
// 发送
ImageGeneration imageGeneration = imageResponse.getResult();
// 图片保存到服务器
String filePath = fileApi.createFile(HttpUtil.downloadBytes(imageGeneration.getOutput().getUrl()));
// 保存数据库
AiImageDO aiImageDO = doSave(req.getPrompt(), req.getSize(), req.getModel(),
filePath, imageGeneration.getOutput().getUrl(), AiImageStatusEnum.COMPLETE, null,
null, null, null);
// 转换 AiImageDallDrawingRespVO
return AiImageConvert.INSTANCE.convertAiImageDallDrawingRespVO(aiImageDO);
} catch (AiException aiException) {
// 保存数据库
AiImageDO aiImageDO = doSave(req.getPrompt(), req.getSize(), req.getModel(),
null, null, AiImageStatusEnum.FAIL, aiException.getMessage(),
null, null, null);
// 发送错误信息
return AiImageConvert.INSTANCE.convertAiImageDallDrawingRespVO(aiImageDO);
}
// 保存数据库
AiImageDO aiImageDO = doSave(req.getPrompt(), req.getSize(), req.getModel(),
null, null, AiImageStatusEnum.IN_PROGRESS, null,
null, null, null);
// 异步执行
EXECUTOR.execute(() -> {
try {
// 获取 model
OpenAiImageModelEnum openAiImageModelEnum = OpenAiImageModelEnum.valueOfModel(req.getModel());
OpenAiImageStyleEnum openAiImageStyleEnum = OpenAiImageStyleEnum.valueOfStyle(req.getStyle());
// 转换openai 参数
OpenAiImageOptions openAiImageOptions = new OpenAiImageOptions();
openAiImageOptions.setModel(openAiImageModelEnum.getModel());
openAiImageOptions.setStyle(openAiImageStyleEnum.getStyle());
openAiImageOptions.setSize(req.getSize());
ImageResponse imageResponse = openAiImageClient.call(new ImagePrompt(req.getPrompt(), openAiImageOptions));
// 发送
ImageGeneration imageGeneration = imageResponse.getResult();
// 图片保存到服务器
String filePath = fileApi.createFile(HttpUtil.downloadBytes(imageGeneration.getOutput().getUrl()));
// 更新数据库
aiImageMapper.updateById(
new AiImageDO()
.setId(aiImageDO.getId())
.setStatus(AiImageStatusEnum.COMPLETE.getStatus())
.setPicUrl(filePath)
.setOriginalPicUrl(imageGeneration.getOutput().getUrl())
);
} catch (AiException aiException) {
// 更新错误信息
aiImageMapper.updateById(
new AiImageDO()
.setId(aiImageDO.getId())
.setStatus(AiImageStatusEnum.FAIL.getStatus())
.setErrorMessage(aiException.getMessage())
);
}
});
// 转换 AiImageDallDrawingRespVO
return AiImageConvert.INSTANCE.convertAiImageDallDrawingRespVO(aiImageDO);
}
@Override