轮询接口和可疑文件限制

jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/OcrMsgReadController.java

@@ -19,9 +19,11 @@ import org.springframework.web.bind.annotation.RequestParam;
 import javax.servlet.http.HttpServletRequest;
 import cn.jyjz.xiaoyao.ocr.service.IOcrMsgReadService;
 
+import java.sql.Array;
 import java.text.SimpleDateFormat;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * ocr控制器
@@ -97,6 +99,22 @@ public class OcrMsgReadController  {
         return ResultVoUtil.success(selectone);
     }
 
+    @ApiOperation(value = "消息轮询", notes = "消息轮询")
+    @RequestMapping(value = "/msgPolling", method = RequestMethod.GET)
+    public ResultVo msgPolling(HttpServletRequest request) {
+
+        UserToken userToken = this.userTokenService.getUserToken(RequestBaseUtil.getToken(request));
+
+        QueryWrapper<OcrMsgRead> wrapperread = new QueryWrapper<>();
+        wrapperread.eq("user_id",userToken.getUserid());
+        List<OcrMsgRead> list = ocrMsgReadService.list(wrapperread);
+
+        QueryWrapper<OcrMsg> wrappermsg = new QueryWrapper<>();
+        wrappermsg.notIn("id", list.stream().map(i->i.getMsgId()).collect(Collectors.toList()));
+        long count = iocrmsgservice.count(wrappermsg);
+
+        return ResultVoUtil.success(count);
+    }
 
     /**
      * 添加

jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/controller/OcrTaskchildDubiousFileController.java

@@ -152,6 +152,11 @@ public class OcrTaskchildDubiousFileController  {
         List<OcrTaskchildDubiousFile> list=new ArrayList<>();
         String[] taskchildid1 = taskchildid.toString().split(",");
         for (String asd:taskchildid1){
+
+            QueryWrapper<OcrTaskchildDubiousFile> filewrapper = new QueryWrapper<>();
+            filewrapper.eq("taskchild_id",asd).eq("userid",userToken.getUserid());
+            OcrTaskchildDubiousFile one1 = ocrTaskchildDubiousFileService.getOne(filewrapper);
+            if(!asd.equals(one1.getTaskchildId())){
                 OcrTaskchildDubiousFile ocrtaskchilddubiousfile=new OcrTaskchildDubiousFile();
                 ocrtaskchilddubiousfile.setTaskchildId(Long.parseLong(asd));
                 ocrtaskchilddubiousfile.setTenantid(Long.parseLong(tenantId));
@@ -170,6 +175,7 @@ public class OcrTaskchildDubiousFileController  {
 
                 list.add(ocrtaskchilddubiousfile);
             }
+        }
 
 
 

jyjz-system/jyjz-system-oa/src/main/java/cn/jyjz/xiaoyao/ocr/service/impl/OcrSearchHistoryServiceImpl.java

@@ -77,6 +77,9 @@ public class OcrSearchHistoryServiceImpl extends ServiceImpl<OcrSearchHistoryMap
     @Override
     public Map<Object, Object> selectaisp(String userid,String id) {
         Map<Object, Object> selectaisp = ocrsearchhistorymapper.selectaisp(id);
+        if(null==selectaisp){
+            return null;
+        }
         String selectaiuser = ocrsearchhistorymapper.selectaiuser(userid);
 
         String[] cps = selectaisp.get("pictureresult").toString().split(",");