Browse Source

/*

1. 并发任务策略分组个数改成在配置文件中配置;
2. 远端大模型接口请求任务间隔改成在配置文件中配置;
3. 远端大模型接口返回请求处理任务间隔才成在配置文件中配置。
*/
master
htjcAdmin 6 months ago
parent
commit
14b0e1947e
4 changed files with 22 additions and 10 deletions
  1. +10
    -5
      inspect-job/src/main/java/com/inspect/job/task/JobMainTask.java
  2. +6
    -2
      inspect-main/inspect-main-task/src/main/java/com/inspect/analysis/service/impl/AlgorithmRequestProcessConsumer.java
  3. +5
    -2
      inspect-main/inspect-main-task/src/main/java/com/inspect/analysis/service/impl/AlgorithmResponseProcessConsumer.java
  4. +1
    -1
      inspect-main/inspect-main-task/src/main/java/com/inspect/task/controller/PatrolTaskController.java

+ 10
- 5
inspect-job/src/main/java/com/inspect/job/task/JobMainTask.java View File

@ -103,6 +103,9 @@ public class JobMainTask {
@Value("${jobtask.keepAliveTime:100}")
private int keepAliveTime;
@Value("${task.scheduler.batch-size:100}")
private int taskBatchSize;
private ExecutorService threadPool;
final ShaoXinBigModel shaoXinBigModel;
@ -711,7 +714,7 @@ public class JobMainTask {
}
taskExecRecord.setCursorNumber(cursor);
if (bOk) {
//if (bOk) {
success++;
log.info("execute point success, cursor: {}, success: {}, total: {}", cursor, success, total);
taskExecRecord.setFinishNumber(success);
@ -728,9 +731,10 @@ public class JobMainTask {
totalProgress,
TaskStatus.RUNNING.getCode()/*运行中百分比上报*/
);
} else {
log.info("execute point fail, cursor: {}, success: {}, total: {}", cursor, success, total);
}
// }
// else {
// log.info("execute point fail, cursor: {}, success: {}, total: {}", cursor, success, total);
// }
taskExecClient.updatePatrolTaskExecRecord(taskExecRecord);
return cursor;
}
@ -1087,8 +1091,9 @@ public class JobMainTask {
// }
private void prePointExecImmediate(final PatrolTaskExecRecord patrolTaskExecRecord, List<PatrolTaskInfo> patrolTaskInfoList) {
int batchSize = 100;
int batchSize = taskBatchSize;
int total = patrolTaskInfoList.size();
log.info(Color.CYAN + "batchSize: {}, total: {}" + Color.END, batchSize, total);
int index = 1;
for (int i = 0; i < total; i += batchSize) {


+ 6
- 2
inspect-main/inspect-main-task/src/main/java/com/inspect/analysis/service/impl/AlgorithmRequestProcessConsumer.java View File

@ -2,11 +2,13 @@ package com.inspect.analysis.service.impl;
import com.alibaba.nacos.shaded.com.google.gson.Gson;
import com.inspect.analysis.constant.AnalyseConstants;
import com.inspect.base.core.constant.Color;
import com.inspect.base.redis.service.RedisService;
import com.inspect.partrolresult.domain.AnalyseRequest;
import com.inspect.partrolresult.service.AnalyseRemoteService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
@ -23,13 +25,15 @@ public class AlgorithmRequestProcessConsumer {
@Resource
private AnalyseRemoteService analyseRemoteService;
@Scheduled(fixedDelay = 1000)
//@Scheduled(fixedDelay = 1000)
@Scheduled(fixedDelayString = "${task.scheduler.request.delay-ms:1000}")
public void pollAndProcess() {
//logger.info(Color.YELLOW + "AlgorithmRequestProcessConsumerTracer" + Color.END);
String requestDataInRedis;
RedisTemplate<String, String> redisTemplate = redisService.redisTemplate;
while ((requestDataInRedis = redisTemplate.opsForList().leftPop(AnalyseConstants.ALGORITHM_REQUEST_QUEUE)) != null) {
try {
logger.info("AlgorithmProcessConsumer pollAndProcess queueSize: {}, requestData:{}", getQueueSize(), requestDataInRedis);
logger.info(Color.CYAN + "AlgorithmRequestProcessConsumerTracer queueSize: {}, requestData: {}" + Color.END, getQueueSize(), requestDataInRedis);
AnalyseRequest analyseRequest = new Gson().fromJson(requestDataInRedis, AnalyseRequest.class);
analyseRemoteService.sendRequest(analyseRequest, analyseRequest.getTypeList(), analyseRequest.isFilter());
} catch (Exception e) {


+ 5
- 2
inspect-main/inspect-main-task/src/main/java/com/inspect/analysis/service/impl/AlgorithmResponseProcessConsumer.java View File

@ -3,6 +3,7 @@ package com.inspect.analysis.service.impl;
import com.alibaba.nacos.shaded.com.google.gson.Gson;
import com.inspect.analysis.constant.AnalyseConstants;
import com.inspect.analysis.service.IAnalysisService;
import com.inspect.base.core.constant.Color;
import com.inspect.base.redis.service.RedisService;
import com.inspect.partrolresult.domain.AnalyseRequest;
import com.inspect.partrolresult.service.AnalyseRemoteService;
@ -24,13 +25,15 @@ public class AlgorithmResponseProcessConsumer {
@Resource
private IAnalysisService analysisService;
@Scheduled(fixedDelay = 1000)
//@Scheduled(fixedDelay = 1000)
@Scheduled(fixedDelayString = "${task.scheduler.response.delay-ms:1000}")
public void pollAndProcess() {
//logger.info(Color.CYAN + "AlgorithmResponseProcessConsumerTracer" + Color.END);
String responseDataInRedis;
RedisTemplate<String, String> redisTemplate = redisService.redisTemplate;
while ((responseDataInRedis = redisTemplate.opsForList().leftPop(AnalyseConstants.ALGORITHM_RESPONSE_QUEUE)) != null) {
try {
logger.info("AlgorithmResponseProcessConsumer pollAndProcess queueSize: {}, responseData:{}", getQueueSize(), responseDataInRedis);
logger.info(Color.CYAN + "AlgorithmResponseProcessConsumerTracer queueSize: {}, responseData: {}" + Color.END, getQueueSize(), responseDataInRedis);
// AnalyseRequest analyseRequest = new Gson().fromJson(responseDataInRedis, AnalyseRequest.class);
// analyseRemoteService.sendRequest(analyseRequest, analyseRequest.getTypeList(), analyseRequest.isFilter());
analysisService.handleAlgorithmResult(responseDataInRedis);


+ 1
- 1
inspect-main/inspect-main-task/src/main/java/com/inspect/task/controller/PatrolTaskController.java View File

@ -225,7 +225,7 @@ public class PatrolTaskController extends BaseController {
logger.info("---------------[TASK] query resultAnalysis {}", resultAnalysis);
// 初筛数据
List<PatrolData> filterList = patrolResultService.selectPatrolDataResultByTaskCodeV2(resultAnalysis);
logger.info("---------------[TASK] query filterList {}", filterList);
logger.debug("---------------[TASK] query filterList {}", filterList);
List<AlgInfo> algInfos = patrolResultService.selectAlgInfo(new PatrolResult());
resultAnalysis.setFilter("0");
long t2 = System.currentTimeMillis();


Loading…
Cancel
Save