From 0c8cd957630d43aa71183d34ecb1faf60689f8a4 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 18 十二月 2024 11:30:15 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/data/DirController.java |  248 +++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 208 insertions(+), 40 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 d66fab8..7444279 100644
--- a/src/main/java/com/lf/server/controller/data/DirController.java
+++ b/src/main/java/com/lf/server/controller/data/DirController.java
@@ -1,10 +1,22 @@
 package com.lf.server.controller.data;
 
+import com.lf.server.annotation.SysLog;
+import com.lf.server.controller.all.BaseController;
+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;
+import io.swagger.annotations.ApiImplicitParam;
+import io.swagger.annotations.ApiImplicitParams;
+import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.util.List;
 
 /**
@@ -12,65 +24,221 @@
  * @author sws
  * @date   2022-09-22
  */
-
+@Api(tags = "鏁版嵁绠$悊\\鐩綍绠$悊")
 @RestController
 @RequestMapping("/dir")
-public class DirController {
+public class DirController extends BaseController {
     @Autowired
     DirService dirService;
 
-    @RequestMapping(value = "/insertDir", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public Integer insertDir(DirEntity dirEntity) {
+    @Autowired
+    TokenService tokenService;
 
-        return dirService.insertDir(dirEntity);
-    }
+    @SysLog()
+    @ApiOperation(value = "鎻掑叆涓�鏉�")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "DirEntity", paramType = "body", example = "")
+    })
+    @PostMapping(value = "/insertDir", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<Integer> insertDir(@RequestBody DirEntity entity, HttpServletRequest req) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setCreateUser(ue.getId());
+            }
 
-    @RequestMapping(value = "/insertDirs", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public Integer insertDirs(@RequestBody List<DirEntity> dirEntity) {
+            dirService.insert(entity);
 
-        return dirService.insertDirs(dirEntity);
-    }
-
-    @ResponseBody
-    @RequestMapping(value = "/deleteDir", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public Integer deleteDir(int id) {
-
-        return dirService.deleteDir(id);
-    }
-
-
-    @RequestMapping(value = "/deleteDirs", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public Integer deleteDirs(@RequestBody List<Integer> ids) {
-
-        if (!ids.isEmpty()) {
-
-            return dirService.deleteDirs(ids);
-
-        } else {
-
-            return -1;
+            return success(entity.getId());
+        } catch (Exception ex) {
+            return fail(ex, -1);
         }
     }
 
-    @ResponseBody
-    @RequestMapping(value = "/updateDir", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public Integer updateDir(DirEntity dirEntity) {
+    @SysLog()
+    @ApiOperation(value = "鎻掑叆澶氭潯")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "DirEntity", paramType = "body", example = "")
+    })
+    @PostMapping(value = "/insertDirs", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<Integer> insertDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            for (DirEntity entity : list) {
+                entity.setCreateUser(ue.getId());
+            }
 
-        return dirService.updateDir(dirEntity);
+            int count = dirService.inserts(list);
+
+            return success(count);
+        } catch (Exception ex) {
+            return fail(ex, -1);
+        }
     }
 
-    @ResponseBody
-    @RequestMapping(value = "/selectDir", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
-    public DirEntity selectDir(int id) {
-        return dirService.selectDir(id);
+    @SysLog()
+    @ApiOperation(value = "鍒犻櫎涓�鏉�")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
+    })
+    @GetMapping(value = "/deleteDir")
+    public ResponseMsg<Integer> deleteDir(int id) {
+        try {
+            int count = dirService.deleteDir(id);
+
+            return success(count);
+        } catch (Exception ex) {
+            return fail(ex, -1);
+        }
     }
 
+    @SysLog()
+    @ApiOperation(value = "鍒犻櫎澶氭潯")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "DirEntity", paramType = "body", example = "")
+    })
+    @GetMapping(value = "/deleteDirs")
+    public ResponseMsg<Integer> deleteDirs(@RequestParam List<Integer> ids) {
+        try {
+            if (ids == null || ids.isEmpty()) {
+                return fail("id鏁扮粍涓嶈兘涓虹┖", -1);
+            }
+
+            int count = dirService.deleteDirs(ids);
+
+            return success(count);
+        } catch (Exception ex) {
+            return fail(ex, -1);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏇存柊涓�鏉�")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "DictEntity", paramType = "body", example = "")
+    })
+    @ResponseBody
+    @PostMapping(value = "/updateDir", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<Integer> updateDir(@RequestBody DirEntity entity, HttpServletRequest req) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setUpdateUser(ue.getId());
+            }
+
+            int count = dirService.update(entity);
+
+            return success(count);
+        } catch (Exception ex) {
+            return fail(ex, -1);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏇存柊澶氭潯")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "DirEntity", paramType = "body", example = "")
+    })
+    @ResponseBody
+    @PostMapping(value = "/updateDirs", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<Integer> updateDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (DirEntity entity : list) {
+                    entity.setUpdateUser(ue.getId());
+                }
+            }
+
+            int count = dirService.updates(list);
+
+            return success(count);
+        } catch (Exception ex) {
+            return fail(ex, -1);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏍规嵁ID鏌ヨ鏁版嵁")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
+    })
+    @GetMapping(value = "/selectDir")
+    public ResponseMsg<DirEntity> selectDir(int id) {
+        try {
+            DirEntity dirEntity = dirService.selectDir(id);
+
+            return success(dirEntity);
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏌ヨ鎵�鏈夋暟鎹�")
     @GetMapping(value = "/selectDirAll")
-    public List<DirEntity> selectDirAll() {
-
-        return dirService.selectDirAll();
+    public ResponseMsg<List<DirEntity>> selectDirAll() {
+        try {
+            List<DirEntity> list = dirService.selectDirAll();
+            return success(list);
+        } catch (Exception ex) {
+            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);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏌ヨ椤圭洰")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = "瑗�")
+    })
+    @GetMapping(value = "/selectProject")
+    public ResponseMsg<List<DirEntity>> selectProject(String name) {
+        try {
+            List<DirEntity> list = dirService.selectProject(name);
+
+            return success(list);
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "閫掑綊鏌ヨ")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "pid", value = "鐖禝D", dataType = "Integer", paramType = "query", example = "1")
+    })
+    @GetMapping(value = "/selectByPid")
+    public ResponseMsg<List<DirEntity>> selectByPid(int pid) {
+        try {
+            List<DirEntity> list = dirService.selectByPid(pid);
+
+            return success(list);
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
 }

--
Gitblit v1.9.3