From 94174d2cc22afed6f41c270d970903484bfc5708 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Mon, 05 Aug 2024 16:06:18 +0800
Subject: [PATCH] SM2测试类
---
src/main/java/com/dji/sample/manage/controller/DeviceFirmwareController.java | 90 ++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 80 insertions(+), 10 deletions(-)
diff --git a/src/main/java/com/dji/sample/manage/controller/DeviceFirmwareController.java b/src/main/java/com/dji/sample/manage/controller/DeviceFirmwareController.java
index 217735c..97d46b9 100644
--- a/src/main/java/com/dji/sample/manage/controller/DeviceFirmwareController.java
+++ b/src/main/java/com/dji/sample/manage/controller/DeviceFirmwareController.java
@@ -1,17 +1,28 @@
package com.dji.sample.manage.controller;
+import com.dji.sample.common.model.CustomClaim;
+import com.dji.sample.common.model.PaginationData;
import com.dji.sample.common.model.ResponseResult;
+import com.dji.sample.manage.model.dto.DeviceFirmwareDTO;
import com.dji.sample.manage.model.dto.DeviceFirmwareNoteDTO;
+import com.dji.sample.manage.model.dto.FirmwareFileProperties;
+import com.dji.sample.manage.model.param.DeviceFirmwareQueryParam;
+import com.dji.sample.manage.model.param.DeviceFirmwareUpdateParam;
+import com.dji.sample.manage.model.param.DeviceFirmwareUploadParam;
import com.dji.sample.manage.service.IDeviceFirmwareService;
import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
-import java.util.ArrayList;
+import javax.servlet.http.HttpServletRequest;
+import javax.validation.Valid;
+import javax.validation.constraints.NotNull;
import java.util.List;
import java.util.Optional;
+import java.util.stream.Collectors;
+
+import static com.dji.sample.component.AuthInterceptor.TOKEN_CLAIM;
/**
* @author sean
@@ -20,6 +31,7 @@
*/
@RestController
@RequestMapping("${url.manage.prefix}${url.manage.version}/workspaces")
+@Validated
public class DeviceFirmwareController {
@Autowired
@@ -32,12 +44,70 @@
*/
@GetMapping("/firmware-release-notes/latest")
public ResponseResult<List<DeviceFirmwareNoteDTO>> getLatestFirmwareNote(@RequestParam("device_name") List<String> deviceNames) {
- List<DeviceFirmwareNoteDTO> releaseNotes = new ArrayList<>();
- deviceNames.forEach(deviceName -> {
- Optional<DeviceFirmwareNoteDTO> latestFirmware = service.getLatestFirmwareReleaseNote(deviceName);
- latestFirmware.ifPresent(releaseNotes::add);
- });
+
+ List<DeviceFirmwareNoteDTO> releaseNotes = deviceNames.stream()
+ .map(deviceName -> service.getLatestFirmwareReleaseNote(deviceName))
+ .filter(Optional::isPresent)
+ .map(Optional::get)
+ .collect(Collectors.toList());
+
return ResponseResult.success(releaseNotes);
}
+ /**
+ * Query firmware information based on parameters.
+ * @param workspaceId
+ * @param param
+ * @return
+ */
+ @GetMapping("/{workspace_id}/firmwares")
+ public ResponseResult<PaginationData<DeviceFirmwareDTO>> getAllFirmwarePagination(
+ @PathVariable("workspace_id") String workspaceId, @Valid DeviceFirmwareQueryParam param) {
+
+ PaginationData<DeviceFirmwareDTO> data = service.getAllFirmwarePagination(workspaceId, param);
+ return ResponseResult.success(data);
+ }
+
+ /**
+ * Import firmware file for device upgrades.
+ * @param request
+ * @param workspaceId
+ * @param file
+ * @param param
+ * @return
+ */
+ @PostMapping("/{workspace_id}/firmwares/file/upload")
+ public ResponseResult importFirmwareFile(HttpServletRequest request, @PathVariable("workspace_id") String workspaceId,
+ @NotNull(message = "No file received.") MultipartFile file,
+ @Valid DeviceFirmwareUploadParam param) {
+
+ if (!file.getOriginalFilename().endsWith(FirmwareFileProperties.FIRMWARE_FILE_SUFFIX)) {
+ return ResponseResult.error("文件格式不正确");
+ }
+
+ CustomClaim customClaim = (CustomClaim)request.getAttribute(TOKEN_CLAIM);
+ String creator = customClaim.getUsername();
+
+ service.importFirmwareFile(workspaceId, creator, param, file);
+ return ResponseResult.success();
+ }
+
+ /**
+ * Change the firmware availability status.
+ * @param workspaceId
+ * @param firmwareId
+ * @param param
+ * @return
+ */
+ @PutMapping("/{workspace_id}/firmwares/{firmware_id}")
+ public ResponseResult changeFirmwareStatus(@PathVariable("workspace_id") String workspaceId,
+ @PathVariable("firmware_id") String firmwareId,
+ @Valid @RequestBody DeviceFirmwareUpdateParam param) {
+
+ service.updateFirmwareInfo(DeviceFirmwareDTO.builder()
+ .firmwareId(firmwareId).firmwareStatus(param.getStatus()).build());
+ return ResponseResult.success();
+ }
+
+
}
--
Gitblit v1.9.3