拉取关联关系优化
qiuyt
3 months ago
parent
9bb53c360f
commit
42d38220e2
@ -0,0 +1,154 @@
|
||||
package com.glxp.api.task;
|
||||
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.glxp.api.common.res.BaseResponse;
|
||||
import com.glxp.api.dao.schedule.ScheduledDao;
|
||||
import com.glxp.api.entity.collect.RelCodeBatch;
|
||||
import com.glxp.api.entity.collect.RelCodeDetail;
|
||||
import com.glxp.api.entity.system.ScheduledEntity;
|
||||
import com.glxp.api.req.alihealth.AlihealthYljgSinglerelationReqeust;
|
||||
import com.glxp.api.req.system.ScheduledRequest;
|
||||
import com.glxp.api.res.alihealth.AlihealthYljgSinglerelationResponse;
|
||||
import com.glxp.api.service.collect.RelCodeBatchService;
|
||||
import com.glxp.api.service.collect.RelCodeDetailService;
|
||||
import com.glxp.api.util.CacheUtils;
|
||||
import com.glxp.api.util.DateUtil;
|
||||
import com.glxp.api.util.RedisUtil;
|
||||
import com.glxp.api.util.alihealth.AlihealthYljgUtils;
|
||||
import com.glxp.api.util.redis.RedisDelayedQueue;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.scheduling.annotation.EnableScheduling;
|
||||
import org.springframework.scheduling.annotation.SchedulingConfigurer;
|
||||
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
|
||||
import org.springframework.scheduling.support.CronTrigger;
|
||||
import org.springframework.stereotype.Component;
|
||||
import redis.clients.jedis.resps.Tuple;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.util.*;
|
||||
|
||||
@Component
|
||||
@EnableScheduling
|
||||
@Slf4j
|
||||
public class AliRelCodeDetailTask implements SchedulingConfigurer {
|
||||
|
||||
final Logger logger = LoggerFactory.getLogger(AsyncDiDlTask.class);
|
||||
@Resource
|
||||
AsyncCompanyDlHelper udiCompanyTask;
|
||||
@Resource
|
||||
RedisUtil redisUtil;
|
||||
|
||||
@Resource
|
||||
private ScheduledDao scheduledDao;
|
||||
@Resource
|
||||
private RedisDelayedQueue redisDelayedQueue;
|
||||
@Resource
|
||||
private AlihealthYljgUtils alihealthYljgUtils;
|
||||
@Resource
|
||||
private RelCodeBatchService relCodeBatchService;
|
||||
@Resource
|
||||
private RelCodeDetailService relCodeDetailService;
|
||||
|
||||
|
||||
@Override
|
||||
public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
|
||||
scheduledTaskRegistrar.addTriggerTask(() -> process(),
|
||||
triggerContext -> {
|
||||
ScheduledRequest scheduledRequest = new ScheduledRequest();
|
||||
scheduledRequest.setCronName("aliRelCodeDetailTask");
|
||||
ScheduledEntity scheduledEntity = scheduledDao.findScheduled(scheduledRequest);
|
||||
if (scheduledEntity == null) {
|
||||
return null;
|
||||
}
|
||||
String cron = scheduledEntity.getCron();
|
||||
if (cron.isEmpty()) {
|
||||
logger.error("cron is null");
|
||||
}
|
||||
return new CronTrigger(cron).nextExecutionTime(triggerContext);
|
||||
});
|
||||
}
|
||||
|
||||
private void process() {
|
||||
|
||||
long currentTimestamp = System.currentTimeMillis() / 1000; // 当前时间戳（秒）
|
||||
|
||||
// 获取所有超时的任务 (score <= 当前时间戳)
|
||||
List<Tuple> tasksToProcess = redisDelayedQueue.getJedis().zrangeByScoreWithScores("rel_code_queue", "-inf", String.valueOf(currentTimestamp));
|
||||
|
||||
Iterator<Tuple> iterator = tasksToProcess.iterator();
|
||||
while (iterator.hasNext()) {
|
||||
|
||||
Tuple task = iterator.next();
|
||||
String taskId = task.getElement();
|
||||
String[] codes = taskId.split("&&&");
|
||||
|
||||
if (codes!=null && codes.length>0) {
|
||||
// List<String> relCodeBatchList =new ArrayList<>();
|
||||
|
||||
String code = codes[0];
|
||||
String erpId = codes[1];
|
||||
log.error("关联关系下载任务ID===" + code);
|
||||
|
||||
// 执行任务逻辑 (你可以在这里调用实际的业务逻辑)
|
||||
AlihealthYljgSinglerelationReqeust alihealthYljgSinglerelationReqeust = new AlihealthYljgSinglerelationReqeust();
|
||||
alihealthYljgSinglerelationReqeust.setCode(code);
|
||||
BaseResponse<AlihealthYljgSinglerelationResponse> baseResponse = alihealthYljgUtils.relCodeInsert(alihealthYljgSinglerelationReqeust);
|
||||
if (baseResponse.getCode() == 20000) {
|
||||
AlihealthYljgSinglerelationResponse alihealthKytSinglerelationResponse = baseResponse.getData();
|
||||
// 返回的码关联关系进行插入
|
||||
List<RelCodeBatch> batchList = alihealthKytSinglerelationResponse.disposeRelCodeBatch(erpId,code);
|
||||
for (RelCodeBatch relCodeBatch : batchList) {
|
||||
RelCodeBatch relCodeBatchOne = relCodeBatchService.getOne(new QueryWrapper<RelCodeBatch>().eq("curCode", relCodeBatch.getCurCode())
|
||||
.last("limit 1")
|
||||
);
|
||||
if (Objects.isNull(relCodeBatchOne)) {
|
||||
relCodeBatch.setCreateUser(erpId);
|
||||
relCodeBatch.setCreateTime(new Date());
|
||||
relCodeBatchService.save(relCodeBatch);
|
||||
|
||||
} else {
|
||||
if(relCodeBatchOne.getOneLevelCount()!=null){
|
||||
relCodeBatch.setOneLevelCount(relCodeBatchOne.getOneLevelCount() +relCodeBatch.getOneLevelCount() );
|
||||
}
|
||||
if(relCodeBatchOne.getTwoLevelCount()!=null){
|
||||
relCodeBatch.setTwoLevelCount(relCodeBatchOne.getTwoLevelCount() +relCodeBatch.getTwoLevelCount() );
|
||||
}
|
||||
if(relCodeBatchOne.getThreeLevelCount()!=null){
|
||||
relCodeBatch.setThreeLevelCount(relCodeBatchOne.getThreeLevelCount() +relCodeBatch.getThreeLevelCount() );
|
||||
}
|
||||
relCodeBatchService.update(relCodeBatch, new QueryWrapper<RelCodeBatch>().eq("curCode", relCodeBatch.getCurCode()));
|
||||
}
|
||||
relCodeBatchOne = relCodeBatchService.getOne(new QueryWrapper<RelCodeBatch>().eq("curCode", relCodeBatch.getCurCode()));
|
||||
|
||||
Integer id = relCodeBatchOne.getId();
|
||||
// relCodeBatchList.add(relCodeBatchOne.getCurCode());
|
||||
|
||||
// 进行查询
|
||||
List<RelCodeDetail> list = alihealthKytSinglerelationResponse.disposeRelRodeDetailList(id,relCodeBatch.getCurCode());
|
||||
if (list != null && list.size() > 0) {
|
||||
List curCodeList = new ArrayList();
|
||||
for (RelCodeDetail relCodeDetail : list) {
|
||||
curCodeList.add(relCodeDetail.getCurCode());
|
||||
}
|
||||
relCodeDetailService.remove(new QueryWrapper<RelCodeDetail>().in("curCode", curCodeList));
|
||||
|
||||
relCodeDetailService.saveBatch(list);
|
||||
}
|
||||
}
|
||||
|
||||
} else {
|
||||
log.error("阿里调用码关联关系失败==="+baseResponse.getMessage());
|
||||
}
|
||||
|
||||
}
|
||||
// 从队列中删除已执行的任务
|
||||
redisDelayedQueue.getJedis().zrem("rel_code_queue", taskId);
|
||||
log.error("关联关系下载删除任务ID " + taskId + " removed from queue.");
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue