commit
eaaebc05eb
@ -0,0 +1,196 @@
|
|||||||
|
package cn.jyjz.xiaoyao.ocr.controller;
|
||||||
|
|
||||||
|
import cn.jyjz.xiaoyao.common.base.service.UserTokenService;
|
||||||
|
import cn.jyjz.xiaoyao.common.base.util.RequestBaseUtil;
|
||||||
|
import cn.jyjz.xiaoyao.common.base.vo.ResultVo;
|
||||||
|
import cn.jyjz.xiaoyao.common.base.vo.ResultVoUtil;
|
||||||
|
import cn.jyjz.xiaoyao.common.base.vo.UserToken;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.dataobject.*;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.service.OcrPictureService;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.service.OcrTaskchildPictureService;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.springframework.web.bind.annotation.RequestMapping;
|
||||||
|
import org.springframework.web.bind.annotation.PathVariable;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
|
import org.springframework.validation.BindingResult;
|
||||||
|
import org.springframework.web.bind.annotation.RequestParam;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.service.IOcrTaskchildDubiousFileService;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* ocr控制器
|
||||||
|
*
|
||||||
|
* @author 你的肉
|
||||||
|
* @Date 2024-03-21 19:00:57
|
||||||
|
*/
|
||||||
|
@RestController
|
||||||
|
@RequestMapping("/ocr/dubiousfile")
|
||||||
|
public class OcrTaskchildDubiousFileController {
|
||||||
|
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IOcrTaskchildDubiousFileService ocrTaskchildDubiousFileService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private UserTokenService userTokenService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
OcrTaskchildPictureService ocrtaskchildpictureservice;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
OcrPictureService ocrpictureservice;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 分页查询
|
||||||
|
* @param request
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping(value = "/dubiousfilelist", method = RequestMethod.GET)
|
||||||
|
public ResultVo dubiousfilelist(HttpServletRequest request,
|
||||||
|
@RequestParam(value = "pageNo", required = false, defaultValue = "0") int pageNo,
|
||||||
|
@RequestParam(value = "pageSize", required = false, defaultValue = "10") int pageSize) {
|
||||||
|
|
||||||
|
UserToken userToken = this.userTokenService.getUserToken(RequestBaseUtil.getToken(request));
|
||||||
|
|
||||||
|
QueryWrapper<OcrTaskchildDubiousFile> dubiouswrapper = new QueryWrapper<>();
|
||||||
|
dubiouswrapper.eq("userid",userToken.getUserid());
|
||||||
|
List<OcrTaskchildDubiousFile> list = ocrTaskchildDubiousFileService.list(dubiouswrapper);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// for (OcrTaskchildDubiousFile ocrtaskchilddubiousfile:list){
|
||||||
|
// ocrtaskchilddubiousfile.getPictureId()
|
||||||
|
// }
|
||||||
|
|
||||||
|
IPage<OcrPicture> iPage = new Page<>(Integer.valueOf(pageNo),Integer.valueOf(pageSize));
|
||||||
|
|
||||||
|
QueryWrapper<OcrPicture> wrapper = new QueryWrapper<>();
|
||||||
|
wrapper.in("ID",list.stream().map(i->i.getPictureId()).collect(Collectors.toList()));
|
||||||
|
|
||||||
|
|
||||||
|
//执行查询方法
|
||||||
|
IPage<OcrPicture> page = ocrpictureservice.page(iPage, wrapper);
|
||||||
|
|
||||||
|
|
||||||
|
return ResultVoUtil.success(page);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除
|
||||||
|
* @param request
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping(value = "/dubiousfileyc", method = RequestMethod.GET)
|
||||||
|
public ResultVo dubiousfileyc(HttpServletRequest request, @RequestParam(value = "pictureid", required = true) String pictureid) {
|
||||||
|
|
||||||
|
UserToken userToken = this.userTokenService.getUserToken(RequestBaseUtil.getToken(request));
|
||||||
|
|
||||||
|
String[] cps = pictureid.split(",");
|
||||||
|
|
||||||
|
List<OcrPicture> saveList = new ArrayList<>();
|
||||||
|
List<OcrTaskchildDubiousFile> fileList = new ArrayList<>();
|
||||||
|
for (String childpictureid : cps) {
|
||||||
|
OcrPicture ocrPicture = ocrpictureservice.selectDtoById(Long.parseLong(childpictureid));
|
||||||
|
ocrPicture.setSuspiciousfile(0);
|
||||||
|
ocrPicture.setUpdateTime(System.currentTimeMillis());
|
||||||
|
saveList.add(ocrPicture);
|
||||||
|
|
||||||
|
QueryWrapper<OcrTaskchildDubiousFile> dubiouswrapper = new QueryWrapper<>();
|
||||||
|
dubiouswrapper.eq("userid",userToken.getUserid()).eq("picture_id",childpictureid);
|
||||||
|
OcrTaskchildDubiousFile one = ocrTaskchildDubiousFileService.getOne(dubiouswrapper);
|
||||||
|
fileList.add(one);
|
||||||
|
}
|
||||||
|
|
||||||
|
ocrpictureservice.updateBatchById(saveList);
|
||||||
|
|
||||||
|
|
||||||
|
//根据具体情况 判断 是否需要 进行数据返回， 默认不返回数据
|
||||||
|
boolean b = ocrTaskchildDubiousFileService.removeByIds(fileList);
|
||||||
|
return ResultVoUtil.success(b);
|
||||||
|
}
|
||||||
|
|
||||||
|
// /**
|
||||||
|
// * 通过 ID 查询 单条 信息
|
||||||
|
// *
|
||||||
|
// * @param request
|
||||||
|
// * @param id 编号
|
||||||
|
// * @return
|
||||||
|
// */
|
||||||
|
// @RequestMapping(value = "", method = RequestMethod.GET)
|
||||||
|
// public ResponseEntity<?> findById(HttpServletRequest request, @PathVariable("id") Integer id) {
|
||||||
|
// //获取用户信息
|
||||||
|
// User user = userService.getUserInfo(request);
|
||||||
|
// OcrTaskchildDubiousFile ocrTaskchildDubiousFile= ocrTaskchildDubiousFileService.selectById(id);
|
||||||
|
// return new ResponseEntity<OcrTaskchildDubiousFile>(ocrTaskchildDubiousFile, HttpStatus.OK);
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// /** 修改操作
|
||||||
|
// * @param request
|
||||||
|
// * @param ocrTaskchildDubiousFile
|
||||||
|
// * @return
|
||||||
|
// */
|
||||||
|
// @RequestMapping(value = "", method = RequestMethod.PATCH)
|
||||||
|
// public ResponseEntity<?> update(HttpServletRequest request, @RequestBody OcrTaskchildDubiousFile ocrTaskchildDubiousFile) {
|
||||||
|
// //获取用户信息
|
||||||
|
// User user = userService.getUserInfo(request);
|
||||||
|
// ocrTaskchildDubiousFileService.insertOrUpdate(ocrTaskchildDubiousFile);
|
||||||
|
// //此处需要手工处理 返回需要将修改后的数据返回
|
||||||
|
// return new ResponseEntity<OcrTaskchildDubiousFile>(HttpStatus.CREATED);
|
||||||
|
// }
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 添加
|
||||||
|
* @param request
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping(value = "/dubiousfileyd", method = RequestMethod.GET)
|
||||||
|
public ResultVo dubiousfileyd(HttpServletRequest request, @RequestParam(value = "taskchildid", required = true) String taskchildid) {
|
||||||
|
|
||||||
|
String tenantId = request.getHeader("X-Tenant-Id");
|
||||||
|
if(StringUtils.isBlank(tenantId)){
|
||||||
|
return ResultVoUtil.error("租户主键不能为空");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
UserToken userToken = this.userTokenService.getUserToken(RequestBaseUtil.getToken(request));
|
||||||
|
|
||||||
|
List<OcrTaskchildDubiousFile> list=new ArrayList<>();
|
||||||
|
String[] taskchildid1 = taskchildid.toString().split(",");
|
||||||
|
for (String asd:taskchildid1){
|
||||||
|
OcrTaskchildDubiousFile ocrtaskchilddubiousfile=new OcrTaskchildDubiousFile();
|
||||||
|
ocrtaskchilddubiousfile.setTaskchildId(Long.parseLong(asd));
|
||||||
|
ocrtaskchilddubiousfile.setTenantid(Long.parseLong(tenantId));
|
||||||
|
ocrtaskchilddubiousfile.setUserid(Long.parseLong(userToken.getUserid()));
|
||||||
|
ocrtaskchilddubiousfile.setCreatedate(System.currentTimeMillis());
|
||||||
|
|
||||||
|
QueryWrapper<OcrTaskchildPicture> wrapper = new QueryWrapper<>();
|
||||||
|
wrapper.select("PICTUREID as pictureid").eq("ID",asd);
|
||||||
|
OcrTaskchildPicture one = ocrtaskchildpictureservice.getOne(wrapper);
|
||||||
|
ocrtaskchilddubiousfile.setPictureId(one.getPictureid());
|
||||||
|
|
||||||
|
UpdateWrapper<OcrPicture> updatewrapper = new UpdateWrapper<>();
|
||||||
|
updatewrapper.eq("ID", one.getPictureid()).set("suspiciousfile","1");
|
||||||
|
|
||||||
|
ocrpictureservice.update(updatewrapper);
|
||||||
|
|
||||||
|
list.add(ocrtaskchilddubiousfile);
|
||||||
|
}
|
||||||
|
|
||||||
|
boolean b = ocrTaskchildDubiousFileService.saveBatch(list);
|
||||||
|
|
||||||
|
return ResultVoUtil.success(b);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,19 @@
|
|||||||
|
package cn.jyjz.xiaoyao.ocr.dataDao;
|
||||||
|
|
||||||
|
|
||||||
|
import cn.jyjz.xiaoyao.ocr.dataobject.OcrTaskchildDubiousFile;
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import org.apache.ibatis.annotations.Mapper;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p>
|
||||||
|
* 可疑文件夹 Mapper 接口
|
||||||
|
* </p>
|
||||||
|
*
|
||||||
|
* @author 你的肉123
|
||||||
|
* @since 2024-03-21
|
||||||
|
*/
|
||||||
|
@Mapper
|
||||||
|
public interface OcrTaskchildDubiousFileMapper extends BaseMapper<OcrTaskchildDubiousFile> {
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,117 @@
|
|||||||
|
package cn.jyjz.xiaoyao.ocr.dataobject;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.annotation.IdType;
|
||||||
|
import com.baomidou.mybatisplus.annotation.TableField;
|
||||||
|
import com.baomidou.mybatisplus.annotation.TableId;
|
||||||
|
import com.baomidou.mybatisplus.annotation.TableName;
|
||||||
|
import com.baomidou.mybatisplus.extension.activerecord.Model;
|
||||||
|
import java.io.Serializable;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p>
|
||||||
|
* 可疑文件夹
|
||||||
|
* </p>
|
||||||
|
*
|
||||||
|
* @author 你的肉123
|
||||||
|
* @since 2024-03-21
|
||||||
|
*/
|
||||||
|
@TableName("ocr_taskchild_dubious_file")
|
||||||
|
public class OcrTaskchildDubiousFile extends Model<OcrTaskchildDubiousFile> {
|
||||||
|
|
||||||
|
private static final long serialVersionUID = 1L;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 可疑文件夹主键
|
||||||
|
*/
|
||||||
|
@TableId(value = "id", type = IdType.AUTO)
|
||||||
|
private Long id;
|
||||||
|
/**
|
||||||
|
* 任务id
|
||||||
|
*/
|
||||||
|
@TableField("taskchild_id")
|
||||||
|
private Long taskchildId;
|
||||||
|
/**
|
||||||
|
* 图片主键
|
||||||
|
*/
|
||||||
|
@TableField("picture_id")
|
||||||
|
private Long pictureId;
|
||||||
|
/**
|
||||||
|
* 租户id
|
||||||
|
*/
|
||||||
|
private Long tenantid;
|
||||||
|
/**
|
||||||
|
* 用户id
|
||||||
|
*/
|
||||||
|
private Long userid;
|
||||||
|
/**
|
||||||
|
* 创建时间
|
||||||
|
*/
|
||||||
|
private Long createdate;
|
||||||
|
|
||||||
|
|
||||||
|
public Long getId() {
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setId(Long id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getTaskchildId() {
|
||||||
|
return taskchildId;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setTaskchildId(Long taskchildId) {
|
||||||
|
this.taskchildId = taskchildId;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getPictureId() {
|
||||||
|
return pictureId;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setPictureId(Long pictureId) {
|
||||||
|
this.pictureId = pictureId;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getTenantid() {
|
||||||
|
return tenantid;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setTenantid(Long tenantid) {
|
||||||
|
this.tenantid = tenantid;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getUserid() {
|
||||||
|
return userid;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUserid(Long userid) {
|
||||||
|
this.userid = userid;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getCreatedate() {
|
||||||
|
return createdate;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setCreatedate(Long createdate) {
|
||||||
|
this.createdate = createdate;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Serializable pkVal() {
|
||||||
|
return this.id;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString() {
|
||||||
|
return "OcrTaskchildDubiousFile{" +
|
||||||
|
"id=" + id +
|
||||||
|
", taskchildId=" + taskchildId +
|
||||||
|
", pictureId=" + pictureId +
|
||||||
|
", tenantid=" + tenantid +
|
||||||
|
", userid=" + userid +
|
||||||
|
", createdate=" + createdate +
|
||||||
|
"}";
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,17 @@
|
|||||||
|
package cn.jyjz.xiaoyao.ocr.service;
|
||||||
|
|
||||||
|
|
||||||
|
import cn.jyjz.xiaoyao.ocr.dataobject.OcrTaskchildDubiousFile;
|
||||||
|
import com.baomidou.mybatisplus.extension.service.IService;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p>
|
||||||
|
* 可疑文件夹 服务类
|
||||||
|
* </p>
|
||||||
|
*
|
||||||
|
* @author 你的肉123
|
||||||
|
* @since 2024-03-21
|
||||||
|
*/
|
||||||
|
public interface IOcrTaskchildDubiousFileService extends IService<OcrTaskchildDubiousFile> {
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
package cn.jyjz.xiaoyao.ocr.service.impl;
|
||||||
|
|
||||||
|
import cn.jyjz.xiaoyao.ocr.dataDao.OcrTaskchildDubiousFileMapper;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.dataobject.OcrTaskchildDubiousFile;
|
||||||
|
import cn.jyjz.xiaoyao.ocr.service.IOcrTaskchildDubiousFileService;
|
||||||
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p>
|
||||||
|
* 可疑文件夹 服务实现类
|
||||||
|
* </p>
|
||||||
|
*
|
||||||
|
* @author 你的肉123
|
||||||
|
* @since 2024-03-21
|
||||||
|
*/
|
||||||
|
@Service
|
||||||
|
public class OcrTaskchildDubiousFileServiceImpl extends ServiceImpl<OcrTaskchildDubiousFileMapper, OcrTaskchildDubiousFile> implements IOcrTaskchildDubiousFileService {
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
||||||
|
<mapper namespace="cn.jyjz.xiaoyao.ocr.dataDao.OcrTaskchildDubiousFileMapper">
|
||||||
|
|
||||||
|
<!-- 通用查询映射结果 -->
|
||||||
|
<resultMap id="BaseResultMap" type="cn.jyjz.xiaoyao.ocr.dataobject.OcrTaskchildDubiousFile">
|
||||||
|
<id column="id" property="id" />
|
||||||
|
<result column="taskchild_id" property="taskchildId" />
|
||||||
|
<result column="picture_id" property="pictureId" />
|
||||||
|
<result column="tenantid" property="tenantid" />
|
||||||
|
<result column="userid" property="userid" />
|
||||||
|
<result column="createdate" property="createdate" />
|
||||||
|
</resultMap>
|
||||||
|
|
||||||
|
<!-- 通用查询结果列 -->
|
||||||
|
<sql id="Base_Column_List">
|
||||||
|
id, taskchild_id AS taskchildId, picture_id AS pictureId, tenantid, userid, createdate
|
||||||
|
</sql>
|
||||||
|
|
||||||
|
</mapper>
|
||||||
Loading…
Reference in new issue