TASK:generate 添加队列消费者,增加并发量

This commit is contained in:
2024-01-24 14:44:31 +08:00
parent d4568ad740
commit 1d2cb17d54
6 changed files with 73 additions and 35 deletions

View File

@@ -2242,7 +2242,7 @@ public class PythonService {
public List<String> generateSketchOrPrint(GenerateToPythonDTO generateToPythonDTO) {
//限流校验
AccessLimitUtils.validate("generateSketchOrPrint", 5);
// AccessLimitUtils.validate("generateSketchOrPrint", 5);
OkHttpClient client = new OkHttpClient().newBuilder()
.connectTimeout(30, TimeUnit.SECONDS)
.pingInterval(5, TimeUnit.SECONDS)//websocket轮训间隔(单位:秒)
@@ -2271,7 +2271,7 @@ public class PythonService {
throw new BusinessException(ioException.getMessage());
}
//去除限流
AccessLimitUtils.validateOut("generateSketchOrPrint");
// AccessLimitUtils.validateOut("generateSketchOrPrint");
// 判断是否生成失败
if (Objects.isNull(response.body())) {