From d4aecf9fb7d3d1dd504693f8d7175235087e4aeb Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Thu, 06 Jun 2024 09:34:06 +0800
Subject: [PATCH] 图斑图片调查状态

---
 src/main/java/com/dji/sample/patches/service/impl/GetPatchesServiceImpl.java |   43 ++++++++++++++++++++++++++++++++++---------
 1 files changed, 34 insertions(+), 9 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 c4bf342..340ebca 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
@@ -1,7 +1,6 @@
 package com.dji.sample.patches.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.dji.sample.common.model.Pagination;
 import com.dji.sample.common.model.PaginationData;
@@ -11,10 +10,11 @@
 import com.dji.sample.patches.model.Param.PatchesParam;
 import com.dji.sample.patches.model.entity.LotInfo;
 import com.dji.sample.patches.service.GetPatchesService;
-import com.dji.sample.wayline.model.entity.WaylineFileEntity;
+import com.dji.sample.patches.utils.DistrictCodeUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -33,13 +33,38 @@
      */
     @Override
     public PaginationData<LotInfo> limitGet(PatchesParam param) {
-        Page<LotInfo> page = mapper.selectPage(new Page<LotInfo>(param.getPage(), param.getPageSize()),
-                new LambdaQueryWrapper<LotInfo>()
-                        .eq(LotInfo::getWorkspaceId, param.getWorkspaceId()));
-        List<LotInfo> records = page.getRecords()
-                .stream()
-                .collect(Collectors.toList());
-        return new PaginationData<LotInfo>(records, new Pagination(page));
+        if (param.getIsPlan() != null) {
+            Page<LotInfo> page = mapper.selectPage(new Page<LotInfo>(param.getPage(), param.getPageSize()),
+                    new LambdaQueryWrapper<LotInfo>()
+                            .eq(LotInfo::getWorkspaceId, param.getWorkspaceId())
+                            .like(LotInfo::getDkbh, param.getDkbh())
+                            .like(LotInfo::getXzqdm, param.getXzqdm())
+                            .like(LotInfo::getBsm,param.getBsm())
+                            .eq(LotInfo::getIsPlan, param.getIsPlan()));
+            List<LotInfo> records = page.getRecords()
+                    .stream()
+                    .peek(lotInfo -> {
+                        // 修改 xzqdm 字段的值
+                        lotInfo.setXzqdm(DistrictCodeUtils.codeToName(lotInfo.getXzqdm()));
+                    })
+                    .collect(Collectors.toList());
+            return new PaginationData<LotInfo>(records, new Pagination(page));
+        } else {
+            Page<LotInfo> page = mapper.selectPage(new Page<LotInfo>(param.getPage(), param.getPageSize()),
+                    new LambdaQueryWrapper<LotInfo>()
+                            .eq(LotInfo::getWorkspaceId, param.getWorkspaceId())
+                            .like(LotInfo::getDkbh, param.getDkbh())
+                            .like(LotInfo::getBsm,param.getBsm())
+                            .like(LotInfo::getXzqdm, param.getXzqdm()));
+            List<LotInfo> records = page.getRecords()
+                    .stream()
+                    .peek(lotInfo -> {
+                        // 修改 xzqdm 字段的值
+                        lotInfo.setXzqdm(DistrictCodeUtils.codeToName(lotInfo.getXzqdm()));
+                    })
+                    .collect(Collectors.toList());
+            return new PaginationData<LotInfo>(records, new Pagination(page));
+        }
     }
 
     @Override

--
Gitblit v1.9.3