diff --git a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/ApiTestController.java b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/ApiTestController.java index 3a86692..109276c 100644 --- a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/ApiTestController.java +++ b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/ApiTestController.java @@ -11,6 +11,7 @@ import cn.jyjz.xiaoyao.ocr.api.entity.PictureSourceParameter; import cn.jyjz.xiaoyao.ocr.api.entity.PictureSourceResult; import cn.jyjz.xiaoyao.ocr.api.utils.ApiConfig; import cn.jyjz.xiaoyao.ocr.api.utils.ApiPage; +import cn.jyjz.xiaoyao.ocr.service.OcrPictureService; import cn.jyjz.xiaoyao.ocr.thread.TaskQueue; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.google.common.collect.Maps; @@ -52,6 +53,9 @@ public class ApiTestController { @Resource private DepartmentMybatisDao departmentMybatisDao; + @Resource + private OcrPictureService ocrPictureService; + /** * 拉取无量云任务数据,根据单租户、业务主体拉取 * pull wyl task data @@ -110,7 +114,7 @@ public class ApiTestController { } size++; //将可以处理数据放入处理队列中 - TaskQueue.pictureDisposePushData(pictureSourceResult); + TaskQueue.pictureDisposePushData(pictureSourceResult, ocrPictureService); } return "当前区间内图片总数:"+count+"条,本次获取:"+localCount+"条,可处理数据:"+size+"条"; } catch (Exception e) { @@ -193,7 +197,7 @@ public class ApiTestController { pictureSourceResult.setLivePhoto(livePhoto); } //将可以处理数据放入处理队列中 - TaskQueue.pictureDisposePushData(pictureSourceResult); + TaskQueue.pictureDisposePushData(pictureSourceResult, ocrPictureService); } return "处理成功"; @@ -297,7 +301,7 @@ public class ApiTestController { size++; // 尝试将数据加入队列 while (!addedToQueue && retryCount < 5) { // 最多重试5次 - addedToQueue = TaskQueue.pictureDisposePushData(pictureSourceResult); + addedToQueue = TaskQueue.pictureDisposePushData(pictureSourceResult, ocrPictureService); if (!addedToQueue) { Thread.sleep(60000); retryCount++; diff --git a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/dataDao/OcrPictureMybatisDao.java b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/dataDao/OcrPictureMybatisDao.java index e178988..66ca318 100644 --- a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/dataDao/OcrPictureMybatisDao.java +++ b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/dataDao/OcrPictureMybatisDao.java @@ -5,12 +5,12 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper; import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Param; -import java.util.List; - /** -* 数据持久化 -*/ + * 数据持久化 + */ @Mapper public interface OcrPictureMybatisDao extends BaseMapper { OcrPicture getPackagePictureInfo(@Param("packageId") String packageId, @Param("pictureId") String pictureId); + + OcrPicture selectByRemark(@Param("remark") String remark); } diff --git a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/OcrPictureService.java b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/OcrPictureService.java index 9640b47..6484ec0 100644 --- a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/OcrPictureService.java +++ b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/OcrPictureService.java @@ -1,7 +1,5 @@ package cn.jyjz.xiaoyao.ocr.service; -import java.util.List; -import java.util.Map; import cn.jyjz.xiaoyao.common.base.vo.ResultVo; import cn.jyjz.xiaoyao.common.base.vo.UserToken; import cn.jyjz.xiaoyao.common.mybatisplus.base.BaseService; @@ -11,42 +9,48 @@ import com.baomidou.mybatisplus.core.conditions.Wrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import javax.servlet.http.HttpServletRequest; +import java.util.List; +import java.util.Map; /** -* 业务层 -*/ -public interface OcrPictureService extends BaseService { + * 业务层 + */ +public interface OcrPictureService extends BaseService { List listPage(Wrapper queryWrapper); /** * 保存接口传来的json数据 + * * @param jsonArray * @return */ public List savePicture(JSONObject jsonArray); - public ResultVo createTaskChild(List ocrPictureList, String tenantId , UserToken sysUser, String search_month, HttpServletRequest req, String buessinessno); + public ResultVo createTaskChild(List ocrPictureList, String tenantId, UserToken sysUser, String search_month, HttpServletRequest req, String buessinessno); - public ResultVo createTaskChild(List ocrPictureList, String tenantId,HttpServletRequest request); + public ResultVo createTaskChild(List ocrPictureList, String tenantId, HttpServletRequest request); /** * 查询工单图片信息 + * * @param tenantId * @param createuser * @return */ - public List listByTaskChild(String tenantId,String createuser,String isclose,String tasktype,String packageid); + public List listByTaskChild(String tenantId, String createuser, String isclose, String tasktype, String packageid); /** * 查询工单图片信息 + * * @return */ public List listByTaskChild(List pictureid); /** * 根本预设检索条件id,分页查询图片信息 + * * @param pageNo * @param pageSize * @param userSearchId @@ -56,6 +60,7 @@ public interface OcrPictureService extends BaseService { /** * 批量保存 图片信息 + * * @param ocrPictureList * @return */ @@ -65,7 +70,7 @@ public interface OcrPictureService extends BaseService { ResultVo createPackageTask(List pageList, String tenantId, UserToken userToken, String searchMonth, HttpServletRequest req, String buessinessno, Map paramMap); - Map getOcrPictureClassifyAndHash(String img); + Map getOcrPictureClassifyAndHash(String img); String getServerUrl(); @@ -73,4 +78,12 @@ public interface OcrPictureService extends BaseService { // Long getPhotoDateTime(OcrPicture picture); + /** + * 根据任务id查询图片信息 + * + * @param remark + * @return + */ + OcrPicture selectByRemark(String remark); + } diff --git a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/impl/OcrPictureServiceImpl.java b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/impl/OcrPictureServiceImpl.java index 189d4ea..8f4a04a 100644 --- a/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/impl/OcrPictureServiceImpl.java +++ b/jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/impl/OcrPictureServiceImpl.java @@ -334,6 +334,11 @@ public class OcrPictureServiceImpl extends BaseServiceImpl + +