From ff968e2cfb132eb86c70e94e99b9937a8a7b15e9 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Mon, 08 Apr 2024 13:58:56 +0800
Subject: [PATCH] 直播

---
 src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java |   19 +++++++++++++++++++
 1 files changed, 19 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java b/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
index 8267a90..a90846f 100644
--- a/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
+++ b/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
@@ -2,6 +2,7 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+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,6 +12,7 @@
 import com.dji.sample.wayline.dao.IWaylineFileMapper;
 import com.dji.sample.wayline.model.dto.KmzFileProperties;
 import com.dji.sample.wayline.model.dto.WaylineFileDTO;
+import com.dji.sample.wayline.model.dto.WaylineListDTO;
 import com.dji.sample.wayline.model.entity.WaylineFileEntity;
 import com.dji.sample.wayline.model.enums.WaylineTemplateTypeEnum;
 import com.dji.sample.wayline.model.param.WaylineQueryParam;
@@ -179,6 +181,18 @@
         }
     }
 
+    @Override
+    public List<WaylineListDTO> waylineList(String workspaceId,String droneName) {
+        LambdaQueryChainWrapper<WaylineFileEntity> wrapper =
+                new LambdaQueryChainWrapper<>(mapper).eq(WaylineFileEntity::getWorkspaceId, workspaceId);
+        if (StringUtils.hasText(droneName)){
+            wrapper.like(WaylineFileEntity::getName,droneName);
+        }
+
+        return  wrapper.list().stream().map(r -> WaylineListDTO.builder().waylineFileId(r.getWaylineId()).name(r.getName()).build())
+                .collect(Collectors.toList());
+    }
+
     private Optional<WaylineFileDTO> validKmzFile(MultipartFile file) {
         String filename = file.getOriginalFilename();
         if (Objects.nonNull(filename) && !filename.endsWith(WAYLINE_FILE_SUFFIX)) {
@@ -285,4 +299,9 @@
 
         return builder.build();
     }
+    @Override
+    public int updateWayline(WaylineFileEntity entity) {
+        return mapper.update(entity,new LambdaQueryWrapper<WaylineFileEntity>()
+                .eq(WaylineFileEntity::getWaylineId,entity.getWaylineId()));
+    }
 }

--
Gitblit v1.9.3