From f6a1323c3c179d304972d00600a9848e773fc4c6 Mon Sep 17 00:00:00 2001
From: huangfeng <huangfeng@xinyingpower.com>
Date: Wed, 15 May 2024 16:51:59 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E6=9F=A5=E8=AF=A2=E5=A2=9E=E5=8A=A0typ?=
 =?UTF-8?q?e=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../shxy/xymanager_admin/controller/UpgradeController.java | 7 ++++---
 .../shxy/xymanager_service/impl/UpgradeServiceImpl.java    | 5 ++++-
 .../com/shxy/xymanager_service/service/UpgradeService.java | 2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/xymanager_admin/src/main/java/com/shxy/xymanager_admin/controller/UpgradeController.java b/xymanager_admin/src/main/java/com/shxy/xymanager_admin/controller/UpgradeController.java
index 9351b1d..7207718 100644
--- a/xymanager_admin/src/main/java/com/shxy/xymanager_admin/controller/UpgradeController.java
+++ b/xymanager_admin/src/main/java/com/shxy/xymanager_admin/controller/UpgradeController.java
@@ -28,8 +28,8 @@ public class UpgradeController extends BaseController {
 
     @GetMapping("listAll")
     @ApiOperation("查询全部列表")
-    public ResponseReult<List<MntnUpgrades>> listAll() {
-        List<MntnUpgrades> result = service.listAll();
+    public ResponseReult<List<MntnUpgrades>> listAll(Integer type) {
+        List<MntnUpgrades> result = service.listAll(type);
         return ResponseReult.success(result);
     }
 
@@ -41,7 +41,8 @@ public class UpgradeController extends BaseController {
         if (file == null) {
             throw new ApiException("缺少上传的文件");
         }
-        if (StringUtils.isBlank(file.getOriginalFilename()) || !file.getOriginalFilename().endsWith(".apk")) {
+        if (StringUtils.isBlank(file.getOriginalFilename()) ||
+                !file.getOriginalFilename().endsWith(".apk") && !file.getOriginalFilename().endsWith(".zip")) {
             throw new ApiException("文件格式不正确");
         }
         MntnUpgrades record = new MntnUpgrades();
diff --git a/xymanager_service/src/main/java/com/shxy/xymanager_service/impl/UpgradeServiceImpl.java b/xymanager_service/src/main/java/com/shxy/xymanager_service/impl/UpgradeServiceImpl.java
index 91b7bce..4d12346 100644
--- a/xymanager_service/src/main/java/com/shxy/xymanager_service/impl/UpgradeServiceImpl.java
+++ b/xymanager_service/src/main/java/com/shxy/xymanager_service/impl/UpgradeServiceImpl.java
@@ -33,9 +33,12 @@ public class UpgradeServiceImpl implements UpgradeService {
     MntnUpgradesMapper mapper;
 
     @Override
-    public List<MntnUpgrades> listAll() {
+    public List<MntnUpgrades> listAll(Integer type) {
         MntnUpgradesExample example = new MntnUpgradesExample();
         MntnUpgradesExample.Criteria criteria = example.createCriteria();
+        if (type != null) {
+            criteria.andTypeEqualTo(type);
+        }
         example.setOrderByClause("id desc");
         List<MntnUpgrades> list = mapper.selectByExample(example);
         for (MntnUpgrades item : list) {
diff --git a/xymanager_service/src/main/java/com/shxy/xymanager_service/service/UpgradeService.java b/xymanager_service/src/main/java/com/shxy/xymanager_service/service/UpgradeService.java
index 30c7f96..3a247d1 100644
--- a/xymanager_service/src/main/java/com/shxy/xymanager_service/service/UpgradeService.java
+++ b/xymanager_service/src/main/java/com/shxy/xymanager_service/service/UpgradeService.java
@@ -7,7 +7,7 @@ import java.util.List;
 
 public interface UpgradeService {
 
-    List<MntnUpgrades> listAll();
+    List<MntnUpgrades> listAll(Integer type);
 
     boolean upload(MntnUpgrades item, MultipartFile file) throws Exception;