From a8e3f406b2c03a122ef81284849f47cb09822eea Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 24 十一月 2022 16:28:42 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/data/MetaController.java |  213 +++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 157 insertions(+), 56 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/data/MetaController.java b/src/main/java/com/lf/server/controller/data/MetaController.java
index 849e20c..da72c71 100644
--- a/src/main/java/com/lf/server/controller/data/MetaController.java
+++ b/src/main/java/com/lf/server/controller/data/MetaController.java
@@ -1,86 +1,66 @@
 package com.lf.server.controller.data;
 
-import com.lf.server.aspect.SysLog;
+import com.lf.server.annotation.SysLog;
 import com.lf.server.controller.all.BaseController;
+import com.lf.server.entity.all.HttpStatus;
 import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.ctrl.DownloadReqEntity;
+import com.lf.server.entity.data.DownloadEntity;
 import com.lf.server.entity.data.MetaEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.Md5Helper;
+import com.lf.server.helper.RsaHelper;
+import com.lf.server.helper.StringHelper;
+import com.lf.server.helper.WebHelper;
+import com.lf.server.service.data.DownloadService;
 import com.lf.server.service.data.MetaService;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiImplicitParam;
-import io.swagger.annotations.ApiImplicitParams;
-import io.swagger.annotations.ApiOperation;
+import com.lf.server.service.sys.TokenService;
+import io.swagger.annotations.*;
 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;
 
 /**
  * 鍏冩暟鎹�
  * @author WWW
  */
-@Api(tags = "鏁版嵁绠$悊\\鍏冩暟鎹鐞�")
+@Api(tags = "鏁版嵁绠$悊\\鍏冩暟鎹�")
 @RestController
-@RequestMapping("/args")
+@RequestMapping("/meta")
 public class MetaController extends BaseController {
     @Autowired
     MetaService metaService;
 
-    @SysLog()
-    @ApiOperation(value = "鏌ヨ璁板綍鏁�")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", required = false, example = "")
-    })
-    @GetMapping({"/selectCount"})
-    public ResponseMsg<Integer> selectCount(String name) {
-        try {
-            int count = metaService.selectCount(name);
+    @Autowired
+    TokenService tokenService;
 
-            return success(count);
-        } catch (Exception ex) {
-            return fail(ex.getMessage(), -1);
-        }
-    }
-
-    @SysLog()
-    @ApiOperation(value = "鍒嗛〉鏌ヨ")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = ""),
-            @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"),
-            @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1")
-    })
-    @GetMapping(value = "/selectByPage")
-    public ResponseMsg<List<MetaEntity>> selectByPage(String name, Integer pageSize, Integer pageIndex) {
-        try {
-            if (pageSize < 1 || pageIndex < 1) {
-                return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null);
-            }
-
-            List<MetaEntity> rs = metaService.selectByPage(name, pageSize, pageSize * (pageIndex - 1));
-
-            return success(rs);
-        } catch (Exception ex) {
-            return fail(ex.getMessage(), null);
-        }
-    }
+    @Autowired
+    DownloadService downloadService;
 
     @SysLog()
     @ApiOperation(value = "鍒嗛〉鏌ヨ骞惰繑鍥炶褰曟暟")
     @ApiImplicitParams({
+            @ApiImplicitParam(name = "dirid", value = "鐩綍ID", dataType = "Integer", paramType = "query", example = "10"),
             @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = ""),
             @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"),
             @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1")
     })
     @GetMapping(value = "/selectByPageAndCount")
-    public ResponseMsg<List<MetaEntity>> selectByPageAndCount(String name, Integer pageSize, Integer pageIndex) {
+    public ResponseMsg<List<MetaEntity>> selectByPageAndCount(Integer dirid, String name, Integer pageSize, Integer pageIndex) {
         try {
             if (pageSize < 1 || pageIndex < 1) {
                 return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null);
             }
-            int count = metaService.selectCount(name);
+
+            int count = metaService.selectCount(dirid, name);
             if (count == 0) {
                 return success(0, null);
             }
-            List<MetaEntity> rs = metaService.selectByPage(name, pageSize, pageSize * (pageIndex - 1));
+
+            List<MetaEntity> rs = metaService.selectByPage(dirid, name, pageSize, pageSize * (pageIndex - 1));
 
             return success(count, rs);
         } catch (Exception ex) {
@@ -118,13 +98,19 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "娣诲姞鏁版嵁")
+    @ApiOperation(value = "鎻掑叆涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "MetaEntity", paramType = "body")
     })
     @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> insert(@RequestBody MetaEntity entity) {
+    public ResponseMsg<Integer> insert(@RequestBody MetaEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setCreateUser(ue.getId());
+            }
+
+            entity.setCreateTime(WebHelper.getCurrentTimestamp());
             int count = metaService.insert(entity);
 
             return success(count);
@@ -134,13 +120,20 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺娣诲姞")
+    @ApiOperation(value = "鎻掑叆澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "MetaEntity", paramType = "body")
     })
     @PostMapping(value = "/inserts", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> inserts(@RequestBody List<MetaEntity> list) {
+    public ResponseMsg<Integer> inserts(@RequestBody List<MetaEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (MetaEntity entity : list) {
+                    entity.setCreateUser(ue.getId());
+                }
+            }
+
             int count = metaService.inserts(list);
 
             return success(count);
@@ -150,7 +143,7 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鍒櫎鏁版嵁")
+    @ApiOperation(value = "鍒犻櫎涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
     })
@@ -166,7 +159,7 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺鍒犻櫎")
+    @ApiOperation(value = "鍒犻櫎澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "Integer", paramType = "query", example = "1,2")
     })
@@ -186,14 +179,19 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "淇敼鏁版嵁")
+    @ApiOperation(value = "鏇存柊涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "MetaEntity", paramType = "body")
     })
     @ResponseBody
     @PostMapping(value = "/update", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> update(@RequestBody MetaEntity entity) {
+    public ResponseMsg<Integer> update(@RequestBody MetaEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setUpdateUser(ue.getId());
+            }
+
             int count = metaService.update(entity);
 
             return success(count);
@@ -203,14 +201,21 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺淇敼")
+    @ApiOperation(value = "鏇存柊澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "MetaEntity", paramType = "body")
     })
     @ResponseBody
     @PostMapping(value = "/updates", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> updates(@RequestBody List<MetaEntity> list) {
+    public ResponseMsg<Integer> updates(@RequestBody List<MetaEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (MetaEntity entity : list) {
+                    entity.setUpdateUser(ue.getId());
+                }
+            }
+
             int count = metaService.updates(list);
 
             return success(count);
@@ -218,4 +223,100 @@
             return fail(ex.getMessage(), -1);
         }
     }
+
+    @SysLog()
+    @ApiOperation(value = "鍒嗛〉鏌ヨ涓嬭浇璁板綍骞惰繑鍥炶褰曟暟")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = ""),
+            @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"),
+            @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1")
+    })
+    @GetMapping(value = "/selectPageCountForDownload")
+    public ResponseMsg<List<DownloadEntity>> selectPageCountForDownload(String name, Integer pageSize, Integer pageIndex, HttpServletRequest req) {
+        try {
+            if (pageSize < 1 || pageIndex < 1) {
+                return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null);
+            }
+
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return fail("鐢ㄦ埛鏈櫥褰�", null);
+            }
+
+            int count = downloadService.selectCountForUser(ue.getCreateUser(), 3, name);
+            if (count == 0) {
+                return success(0, null);
+            }
+            List<DownloadEntity> rs = downloadService.selectByPageForUser(ue.getCreateUser(), 3, name, pageSize, pageSize * (pageIndex - 1));
+
+            return success(count, rs);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "璇锋眰涓嬭浇")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "pwd", value = "瀵嗙爜", dataType = "String", paramType = "body", example = ""),
+            @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "Integer", paramType = "body", example = "1,2")
+    })
+    @ResponseBody
+    @PostMapping(value = "/downloadReq")
+    public ResponseMsg<Object> downloadReq(@RequestBody DownloadReqEntity reqEntity, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            if (null == reqEntity || StringHelper.isEmpty(reqEntity.getPwd())) {
+                return fail("瀵嗙爜涓嶈兘涓虹┖");
+            }
+            if (null == reqEntity.getIds() || reqEntity.getIds().isEmpty()) {
+                return fail("璇烽�夋嫨瑕佷笅杞界殑鏂囦欢");
+            }
+            if (!downloadService.validatePwd(reqEntity)) {
+                return fail("瀵嗙爜涓嶇鍚堣姹�");
+            }
+
+
+            return success(true);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query")
+    })
+    @ResponseBody
+    @RequestMapping(value = "/downloadFile", method = RequestMethod.GET)
+    public void downloadFile(@RequestBody DownloadReqEntity reqEntity, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            if (null == reqEntity || StringHelper.isEmpty(reqEntity.getPwd())) {
+                WebHelper.writeInfo(HttpStatus.BAD_REQUEST, "瀵嗙爜涓嶈兘涓虹┖", res);
+            }
+            if (StringHelper.isEmpty(reqEntity.getGuid())) {
+                WebHelper.writeInfo(HttpStatus.BAD_REQUEST, "鎵句笉鍒版枃浠禝D", res);
+                return;
+            }
+
+            DownloadEntity de = downloadService.selectByGuid(reqEntity.getGuid());
+            if (null == de) {
+                WebHelper.writeInfo(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�", res);
+                return;
+            }
+            if (!StringHelper.isNull(de.getPwd()) && !Md5Helper.validatePassword(reqEntity.getPwd(), de.getPwd())) {
+                WebHelper.writeInfo(HttpStatus.UNAUTHORIZED, "瀵嗙爜涓嶆纭�", res);
+            }
+
+            UserEntity ue = tokenService.getCurrentUser(req);
+            de.setDcount(de.getDcount() + 1);
+            de.setDownloadUser(ue.getId());
+            int rows = downloadService.update(de);
+
+            String filePath = downloadService.getDownloadFilePath(de);
+            WebHelper.download(filePath, de.getName(), res);
+        } catch (Exception ex) {
+            WebHelper.writeInfo(HttpStatus.ERROR, ex.getMessage(), res);
+        }
+    }
 }

--
Gitblit v1.9.3