From 2db1aa88e8ab53096a936163d686b90d8e056a99 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Wed, 21 Aug 2024 23:18:33 +0800
Subject: [PATCH] 国土对接返回信息加密

---
 src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java |  102 ++++++++++++++++++++++++++++++++-------------------
 1 files changed, 64 insertions(+), 38 deletions(-)

diff --git a/src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java b/src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java
index f1ba312..1207548 100644
--- a/src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java
+++ b/src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java
@@ -17,10 +17,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
-import java.util.ArrayList;
-import java.util.LinkedHashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 @Service
@@ -67,6 +64,12 @@
         if (param.getBsm() != null && !param.getBsm().isEmpty()) {
             queryWrapper.like(LotInfo::getXmc, param.getBsm());
         }
+        if (param.getDkmj()!=null){
+            queryWrapper.eq(LotInfo::getDkmj,param.getDkmj());
+        }
+        if (param.getIsPush() != null) {
+            queryWrapper.eq(LotInfo::getIsPush, param.getIsPush());
+        }
 
         // 执行分页查询
         Page<LotInfo> resultPage = mapper.selectPage(page, queryWrapper);
@@ -95,37 +98,40 @@
 
     @Override
     public PaginationData getPhoto(PatchesParam param, String dkbh) {
-        LotInfo lotInfo = getLotinfoToDb(dkbh);
-        int statue = lotInfo.getIsPush();
-
-        if (statue == 1) {
-            List<MediaFileMarkEntity> allResults = markMapper.selectList(
-                    new LambdaQueryWrapper<MediaFileMarkEntity>().like(MediaFileMarkEntity::getFileName, "%" + dkbh + "~" + "%"));
-
-            // 去重处理
-            Map<String, MediaFileMarkEntity> uniqueFileMap = new LinkedHashMap<>();
-            List<MediaFileMarkEntity> uniqueResults = allResults.stream()
-                    .filter(mediaFile -> uniqueFileMap.putIfAbsent(mediaFile.getFileName(), mediaFile) == null)
-                    .collect(Collectors.toList());
-
-            // 计算分页信息
-            int total = uniqueResults.size();
-            int start = (param.getPage() - 1) * param.getPageSize();
-            int end = Math.min(start + param.getPageSize(), total);
-
-            // 获取当前页的结果
-            List<MediaFileMarkEntity> pageResults = uniqueResults.subList(start, end);
-
-            // 创建临时的 Page 对象
-            Page<MediaFileMarkEntity> resultPage = new Page<>(param.getPage(), param.getPageSize());
-            resultPage.setRecords(pageResults);
-            resultPage.setTotal(total);
-
-            // 返回分页数据
-            return new PaginationData<>(pageResults, new Pagination(resultPage));
-        } else {
+//        LotInfo lotInfo = getLotinfoToDb(dkbh);
+//        int statue = lotInfo.getIsPush();
+//
+//        if (statue == 1) {
+//            List<MediaFileMarkEntity> allResults = markMapper.selectList(
+//                    new LambdaQueryWrapper<MediaFileMarkEntity>()
+//                            .like(MediaFileMarkEntity::getFileName, "%" + dkbh + "~" + "%"));
+//
+//            // 去重处理
+//            Map<String, MediaFileMarkEntity> uniqueFileMap = new LinkedHashMap<>();
+//            List<MediaFileMarkEntity> uniqueResults = allResults.stream()
+//                    .filter(mediaFile -> uniqueFileMap.putIfAbsent(mediaFile.getFileName(), mediaFile) == null)
+//                    .collect(Collectors.toList());
+//
+//            // 计算分页信息
+//            int total = uniqueResults.size();
+//            int start = (param.getPage() - 1) * param.getPageSize();
+//            int end = Math.min(start + param.getPageSize(), total);
+//
+//            // 获取当前页的结果
+//            List<MediaFileMarkEntity> pageResults = uniqueResults.subList(start, end);
+//
+//            // 创建临时的 Page 对象
+//            Page<MediaFileMarkEntity> resultPage = new Page<>(param.getPage(), param.getPageSize());
+//            resultPage.setRecords(pageResults);
+//            resultPage.setTotal(total);
+//
+//            // 返回分页数据
+//            return new PaginationData<>(pageResults, new Pagination(resultPage));
+//        }
+//        else {
             List<MediaFileEntity> allResults = fileMapper.selectList(
-                    new LambdaQueryWrapper<MediaFileEntity>().like(MediaFileEntity::getFileName, "%" + dkbh + "~" + "%"));
+                    new LambdaQueryWrapper<MediaFileEntity>()
+                            .like(MediaFileEntity::getFileName, "%" + dkbh + "~" + "%"));
 
             // 去重处理
             Map<String, MediaFileEntity> uniqueFileMap = new LinkedHashMap<>();
@@ -149,22 +155,38 @@
             // 返回分页数据
             return new PaginationData<>(pageResults, new Pagination(resultPage));
         }
-    }
+//    }
 
 
 
     public List<MediaFileEntity> listPohto(String dkbh, String workspaceId) {
         return fileMapper.selectList(new LambdaQueryWrapper<MediaFileEntity>().like(MediaFileEntity::getFileName, "%" + dkbh + "%")
-                .eq(MediaFileEntity::getWorkspaceId, workspaceId));
+                .eq(MediaFileEntity::getWorkspaceId, workspaceId)
+                .eq(MediaFileEntity::getExamine,1));
+    }
+    public List<MediaFileEntity> listPohto(String dkbh) {
+        return fileMapper.selectList(new LambdaQueryWrapper<MediaFileEntity>().like(MediaFileEntity::getFileName, "%" + dkbh + "%")
+                .eq(MediaFileEntity::getExamine,1));
     }
     public LotInfo getLotinfo(String dkbh, String workspaceId) {
         return mapper.selectOne(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getDkbh, dkbh)
                 .eq(LotInfo::getWorkspaceId, workspaceId));
     }
+
+    public LotInfo getLotinfo(String dkbh) {
+        return mapper.selectOne(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getDkbh, dkbh));
+    }
     public void patchesPushed(String taskId, String dkbh, String workspaceId) {
         LambdaUpdateWrapper<LotInfo> updateWrapper = new LambdaUpdateWrapper<>();
         updateWrapper.set(LotInfo::getIsPush, 1)
                 .eq(LotInfo::getWorkspaceId, workspaceId)
+                .eq(LotInfo::getDkbh, dkbh)
+                .eq(LotInfo::getTaskId, taskId);
+        mapper.update(null, updateWrapper);
+    }
+    public void patchesPushed(String taskId, String dkbh) {
+        LambdaUpdateWrapper<LotInfo> updateWrapper = new LambdaUpdateWrapper<>();
+        updateWrapper.set(LotInfo::getIsPush, 1)
                 .eq(LotInfo::getDkbh, dkbh)
                 .eq(LotInfo::getTaskId, taskId);
         mapper.update(null, updateWrapper);
@@ -177,8 +199,8 @@
         queryWrapper.in(LotInfo::getId, ids);
         return mapper.selectList(queryWrapper);
     }
-    public List<LotInfo> listLotinfo() {
-        return mapper.selectList(null);
+    public List<LotInfo> listLotinfo(String workspaceId) {
+        return mapper.selectList(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getWorkspaceId,workspaceId));
     }
     @Override
     public LotInfo getPatchesFromId(String patchesId){
@@ -190,4 +212,8 @@
             mapper.insert(list.get(i));
         }
     }
+    public List<LotInfo> getLotInfosByIDs(String idsStr) {
+        List<String> idList = Arrays.asList(idsStr.split(","));
+        return mapper.selectBatchIds(idList);
+    }
 }

--
Gitblit v1.9.3