From 861f963ab4750c3efcc19d0d3abd60761ec399b7 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期日, 26 三月 2023 21:06:19 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/data/DirController.java |   26 ++++++++++++++++++++++++++
 1 files changed, 26 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/data/DirController.java b/src/main/java/com/lf/server/controller/data/DirController.java
index e755baf..45f0410 100644
--- a/src/main/java/com/lf/server/controller/data/DirController.java
+++ b/src/main/java/com/lf/server/controller/data/DirController.java
@@ -5,6 +5,7 @@
 import com.lf.server.entity.all.ResponseMsg;
 import com.lf.server.entity.data.DirEntity;
 import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.WebHelper;
 import com.lf.server.service.data.DirService;
 import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
@@ -15,6 +16,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.util.List;
 
 /**
@@ -183,4 +185,28 @@
             return fail(ex, null);
         }
     }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇鐩綍缁撴瀯")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "id", value = "鐩綍ID", dataType = "Integer", paramType = "query", example = "1")
+    })
+    @GetMapping(value = "/downloadDir")
+    public void downloadDir(int id, HttpServletResponse res) {
+        try {
+            DirEntity entity = dirService.selectDir(id);
+            if (null == entity || entity.getPid() > 0) {
+                return;
+            }
+
+            String zipFile = dirService.createDirs(id);
+            if (null == zipFile) {
+                return;
+            }
+
+            WebHelper.download(zipFile, entity.getName() + ".zip", res);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+    }
 }

--
Gitblit v1.9.3