parent
132976ba13
commit
b88eafe8ab
@ -0,0 +1,146 @@
|
||||
package org.jeecg.modules.ocr.init;
|
||||
|
||||
import com.alibaba.fastjson.JSONObject;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.collections.CollectionUtils;
|
||||
import org.jeecg.common.constant.OcrConstant;
|
||||
import org.jeecg.common.system.vo.DictModel;
|
||||
import org.jeecg.common.util.AssertUtils;
|
||||
import org.jeecg.common.util.RestUtil;
|
||||
import org.jeecg.modules.ocr.dto.OcrIdentifyDTO;
|
||||
import org.jeecg.modules.ocr.entity.OcrIdentify;
|
||||
import org.jeecg.modules.ocr.service.IOcrIdentifyService;
|
||||
import org.jeecg.modules.ocr.service.impl.TaskService;
|
||||
import org.jeecg.modules.ocr.utils.FileOUtils;
|
||||
import org.jeecg.modules.system.entity.SysLog;
|
||||
import org.jeecg.modules.system.service.ISysDictService;
|
||||
import org.jeecg.modules.system.service.ISysLogService;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Date;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @Description
|
||||
* @Author ZhouWenTao
|
||||
* @Date 2023/9/20 17:53
|
||||
*/
|
||||
@Configuration
|
||||
@Slf4j
|
||||
public class HandleTaskInit implements ApplicationRunner {
|
||||
@Resource
|
||||
IOcrIdentifyService ocrIdentifyService;
|
||||
@Resource
|
||||
TaskService taskService;
|
||||
@Resource
|
||||
private ISysDictService sysDictService;
|
||||
@Resource
|
||||
private ISysLogService sysLogService;
|
||||
@Value("${system.project.enableHandleTask}")
|
||||
private boolean enableHandleTask;
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) throws Exception {
|
||||
if(enableHandleTask){
|
||||
try {
|
||||
Thread.sleep(5000L);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
while (true) {
|
||||
try {
|
||||
Thread.sleep(10000L);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
log.info("---------检测是否有任务...");
|
||||
//获取任务
|
||||
List<OcrIdentify> list= ocrIdentifyService.getSemanticTaskList();
|
||||
if (CollectionUtils.isEmpty(list)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
//获取相对路径
|
||||
List<DictModel> ocr_relative_path = sysDictService.queryDictItemsByCode("ocr_relative_path");
|
||||
String relativePath = null;
|
||||
if (ocr_relative_path != null && ocr_relative_path.size() > 0) {
|
||||
for (DictModel dictModel : ocr_relative_path) {
|
||||
if (dictModel.getValue().equals("0")) {
|
||||
relativePath = dictModel.getText();
|
||||
}
|
||||
}
|
||||
}
|
||||
//获取识别的图片
|
||||
List<String> fileList=null;
|
||||
|
||||
//临时变量
|
||||
List<String> fileUrlList;
|
||||
String image;//图片
|
||||
//执行获取到的任务
|
||||
identifyFor: for (OcrIdentify ocrIdentify : list) {
|
||||
image=null;
|
||||
fileList=new ArrayList<>();
|
||||
//识别的图片路径
|
||||
String imageUrl = ocrIdentify.getIdentifyUrl();
|
||||
//判断是不是网络图片
|
||||
Boolean onlineFile = FileOUtils.isOnlineFile(ocrIdentify.getIdentifyUrl());
|
||||
|
||||
//最终要请求ocr识别的图片对象
|
||||
JSONObject requestBody = new JSONObject();
|
||||
if (onlineFile) {
|
||||
//如果是网络图片，则将图片下载
|
||||
image = FileOUtils.downLoadFromUrl(imageUrl, FileOUtils.getFileName(imageUrl), OcrConstant.FILE_DOWNLOAD_URL_PREFIX);
|
||||
fileList.add(image);
|
||||
}else{
|
||||
//路径下识别到的图片集合
|
||||
fileUrlList = FileOUtils.fileLists(relativePath, ocrIdentify.getIdentifyUrl());
|
||||
AssertUtils.notNull(fileUrlList, "图片地址不存在");
|
||||
for (String fileUrl : fileUrlList) {
|
||||
//判断附件是否是 图片格式
|
||||
if (fileUrl.lastIndexOf(".png") != -1 || fileUrl.lastIndexOf(".jpg") != -1 || fileUrl.lastIndexOf(".jpeg") != -1) {
|
||||
fileList.add(fileUrl);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//最终要识别哪些图片
|
||||
int i=0;
|
||||
for (String img : fileList) {
|
||||
i++;
|
||||
requestBody.put("task_id", ocrIdentify.getId()+"_"+i);
|
||||
requestBody.put("img_path", image);
|
||||
log.info("----------------------请求参数");
|
||||
log.info(requestBody.toJSONString());
|
||||
try {
|
||||
log.info("----------------------请求参数");
|
||||
log.info(requestBody.toJSONString());
|
||||
JSONObject semanticResponseJson = RestUtil.post(OcrConstant.api_test2_identify_url, requestBody);
|
||||
semanticResponseJson.put("identifyId", ocrIdentify.getId());
|
||||
log.info("ocr识别返回数据:");
|
||||
log.info(semanticResponseJson.toJSONString());
|
||||
ocrIdentifyService.getSemanticInfo(semanticResponseJson);
|
||||
}catch (Exception e){
|
||||
log.error("识别图片失败:");
|
||||
SysLog sysLog=new SysLog();
|
||||
sysLog.setLogType(2);
|
||||
sysLog.setLogContent(img+"_识别图片失败:"+e.getMessage());
|
||||
sysLog.setOperateType(2);
|
||||
sysLogService.save(sysLog);
|
||||
log.error(e.getMessage());
|
||||
log.error("正在重试...");
|
||||
continue identifyFor;
|
||||
}
|
||||
|
||||
}
|
||||
//更改任务状态
|
||||
ocrIdentifyService.updateOcrIdentifyStatus(ocrIdentify.getId(), "1");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue