From 3e0b7b9a64454e168761442ca58b8b57488afa6d Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 14 十二月 2022 14:22:30 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/data/MetaController.java    |   10 -
 src/main/java/com/lf/server/entity/all/StaticData.java             |    5 +
 src/main/java/com/lf/server/controller/show/DataLibController.java |  258 +++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 265 insertions(+), 8 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 652638a..d4bd77e 100644
--- a/src/main/java/com/lf/server/controller/data/MetaController.java
+++ b/src/main/java/com/lf/server/controller/data/MetaController.java
@@ -55,8 +55,6 @@
     @Autowired
     DownloadService downloadService;
 
-    private final static String EQ = "=";
-
     @SysLog()
     @ApiOperation(value = "鍒嗛〉鏌ヨ骞惰繑鍥炶褰曟暟")
     @ApiImplicitParams({
@@ -163,8 +161,8 @@
             @ApiImplicitParam(name = "ns", value = "鍚嶇О绌洪棿", dataType = "String", paramType = "query", example = "bd"),
             @ApiImplicitParam(name = "tab", value = "琛ㄥ悕", dataType = "String", paramType = "query", example = "dlg25wAanp")
     })
-    @GetMapping(value = "/selectFields")
-    public ResponseMsg<List<DictEntity>> selectFields(String ns, String tab) {
+    @GetMapping(value = "/selectTabFields")
+    public ResponseMsg<List<DictEntity>> selectTabFields(String ns, String tab) {
         try {
             if (StringHelper.isEmpty(ns) || StringHelper.isEmpty(tab)) {
                 return fail("鍚嶇О绌洪棿鍜岃〃鍚嶄笉鑳戒负绌�", null);
@@ -388,7 +386,7 @@
             if (StringHelper.isEmpty(guid) || StringHelper.isEmpty(pwd)) {
                 return fail("鏂囦欢ID鍜屽瘑鐮佷笉鑳戒负绌�", null);
             }
-            if (!pwd.endsWith(EQ)) {
+            if (!pwd.endsWith(StaticData.EQ)) {
                 pwd = URLDecoder.decode(pwd, StandardCharsets.UTF_8.name());
             }
 
@@ -427,7 +425,7 @@
             if (StringHelper.isEmpty(guid) || StringHelper.isEmpty(pwd)) {
                 WebHelper.writeInfo(HttpStatus.BAD_REQUEST, "鏂囦欢ID鍜屽瘑鐮佷笉鑳戒负绌�", res);
             }
-            if (!pwd.endsWith(EQ)) {
+            if (!pwd.endsWith(StaticData.EQ)) {
                 pwd = URLDecoder.decode(pwd, StandardCharsets.UTF_8.name());
             }
 
diff --git a/src/main/java/com/lf/server/controller/show/DataLibController.java b/src/main/java/com/lf/server/controller/show/DataLibController.java
index c62900e..71d3242 100644
--- a/src/main/java/com/lf/server/controller/show/DataLibController.java
+++ b/src/main/java/com/lf/server/controller/show/DataLibController.java
@@ -1,9 +1,42 @@
 package com.lf.server.controller.show;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.metadata.OrderItem;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.lf.server.annotation.SysLog;
 import com.lf.server.controller.all.BaseQueryController;
+import com.lf.server.entity.all.HttpStatus;
+import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.all.StaticData;
+import com.lf.server.entity.ctrl.DownloadReqEntity;
+import com.lf.server.entity.data.DictEntity;
+import com.lf.server.entity.data.DownloadEntity;
+import com.lf.server.entity.data.MetaEntity;
+import com.lf.server.entity.data.MetaFileEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.ClassHelper;
+import com.lf.server.helper.Md5Helper;
+import com.lf.server.helper.StringHelper;
+import com.lf.server.helper.WebHelper;
+import com.lf.server.mapper.all.BasicMapper;
+import com.lf.server.service.all.BaseQueryService;
+import com.lf.server.service.data.DownloadService;
+import com.lf.server.service.data.MetaService;
+import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+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.io.File;
+import java.net.URLDecoder;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
 
 /**
  * 璧勬枡棣�
@@ -13,4 +46,225 @@
 @RestController
 @RequestMapping("/dataLib")
 public class DataLibController extends BaseQueryController {
+    @Autowired
+    MetaService metaService;
+
+    @Autowired
+    TokenService tokenService;
+
+    @Autowired
+    DownloadService downloadService;
+
+    @Autowired
+    BaseQueryService baseQueryService;
+
+    @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 = "/selectByPageForMeta")
+    public ResponseMsg<List<MetaEntity>> selectByPageForMeta(Integer dirid, String name, Integer pageSize, Integer pageIndex) {
+        try {
+            if (pageSize < 1 || pageIndex < 1) {
+                return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null);
+            }
+
+            int count = metaService.selectCount(dirid, name);
+            if (count == 0) {
+                return success(0, null);
+            }
+
+            List<MetaEntity> rs = metaService.selectByPage(dirid, name, pageSize, pageSize * (pageIndex - 1));
+
+            return success(count, rs);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏌ヨ琛ㄤ腑鏁版嵁")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "id", value = "鍏冩暟鎹甀D", dataType = "Integer", paramType = "query", example = "115"),
+            @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1"),
+            @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10")
+    })
+    @GetMapping(value = "/selectDbData")
+    public ResponseMsg<Object> selectDbData(Integer id, Integer pageIndex, Integer pageSize) {
+        // noinspection AlibabaRemoveCommentedCode
+        try {
+            if (null == id || id < 0) {
+                return fail("鍏冩暟鎹甀D涓嶈兘涓虹┖鎴栧皬浜�0", null);
+            }
+
+            MetaEntity meta = metaService.selectById(id);
+            if (null == meta || null == meta.getTab() || !meta.getTab().contains(StaticData.POINT)) {
+                return fail("鎵句笉鍒板厓鏁版嵁淇℃伅", null);
+            }
+
+            String entity = meta.getTab().substring(meta.getTab().indexOf(".") + 1).replace("_", "").toLowerCase();
+            BasicMapper baseMapper = ClassHelper.getBasicMapper(entity);
+            if (null == baseMapper) {
+                return null;
+            }
+
+            QueryWrapper wrapper = new QueryWrapper();
+            //wrapper.eq("dirid", meta.getDirid());
+            //wrapper.eq("depid", meta.getDepid());
+            //wrapper.eq("verid", meta.getVerid());
+            wrapper.eq("createuser", meta.getCreateUser());
+            wrapper.eq("createtime", meta.getCreateTime());
+
+            Page<Object> page = new Page<>(pageIndex, pageSize);
+            page.addOrder(OrderItem.asc("gid"));
+            IPage<Object> paged = baseMapper.selectPage(page, wrapper);
+
+            return success(paged.getTotal(), paged.getRecords());
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏌ヨ瀛楁淇℃伅")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "ns", value = "鍚嶇О绌洪棿", dataType = "String", paramType = "query", example = "bd"),
+            @ApiImplicitParam(name = "tab", value = "琛ㄥ悕", dataType = "String", paramType = "query", example = "dlg25wAanp")
+    })
+    @GetMapping(value = "/selectTabFields")
+    public ResponseMsg<List<DictEntity>> selectTabFields(String ns, String tab) {
+        try {
+            if (StringHelper.isEmpty(ns) || StringHelper.isEmpty(tab)) {
+                return fail("鍚嶇О绌洪棿鍜岃〃鍚嶄笉鑳戒负绌�", null);
+            }
+
+            List<DictEntity> list = baseQueryService.selectFields(ns, tab);
+
+            return success(list);
+        } 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.decryptPwd(reqEntity)) {
+                return fail("瀵嗙爜瑙e瘑澶辫触", null);
+            }
+            if (!StringHelper.checkPwdValid(reqEntity.getPwd())) {
+                return fail("瀵嗙爜涓嶇鍚堣姹�");
+            }
+
+            List<MetaFileEntity> list = metaService.selectMetaFiles(reqEntity.getIds());
+            if (null == list || list.isEmpty()) {
+                return fail("娌℃湁鎵惧埌鍏冩暟鎹�");
+            }
+
+            UserEntity ue = tokenService.getCurrentUser(req);
+            String guid = downloadService.zipFiles(ue, list, reqEntity.getPwd());
+
+            return success(guid);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "鏌ヨ涓嬭浇鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query"),
+            @ApiImplicitParam(name = "pwd", value = "瀵嗙爜", dataType = "String", paramType = "query")
+    })
+    @GetMapping(value = "/selectDownloadFile")
+    public ResponseMsg<Boolean> selectDownloadFile(String guid, String pwd) {
+        try {
+            if (StringHelper.isEmpty(guid) || StringHelper.isEmpty(pwd)) {
+                return fail("鏂囦欢ID鍜屽瘑鐮佷笉鑳戒负绌�", null);
+            }
+            if (!pwd.endsWith(StaticData.EQ)) {
+                pwd = URLDecoder.decode(pwd, StandardCharsets.UTF_8.name());
+            }
+
+            String password = downloadService.decryptPwd(pwd);
+            if (null == password) {
+                return fail("瀵嗙爜瑙e瘑澶辫触", null);
+            }
+
+            DownloadEntity de = downloadService.selectByGuid(guid);
+            if (null == de) {
+                return fail("鏂囦欢涓嶅瓨鍦�", null);
+            }
+            if (!StringHelper.isNull(de.getPwd()) && !Md5Helper.validatePassword(password, de.getPwd())) {
+                return fail("瀵嗙爜涓嶆纭�", null);
+            }
+
+            String filePath = downloadService.getDownloadFilePath(de);
+            File file = new File(filePath);
+
+            return success(file.exists());
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), false);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query"),
+            @ApiImplicitParam(name = "pwd", value = "瀵嗙爜", dataType = "String", paramType = "query")
+    })
+    @ResponseBody
+    @GetMapping(value = "/downloadFile")
+    public void downloadFile(String guid, String pwd, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            if (StringHelper.isEmpty(guid) || StringHelper.isEmpty(pwd)) {
+                WebHelper.writeInfo(HttpStatus.BAD_REQUEST, "鏂囦欢ID鍜屽瘑鐮佷笉鑳戒负绌�", res);
+            }
+            if (!pwd.endsWith(StaticData.EQ)) {
+                pwd = URLDecoder.decode(pwd, StandardCharsets.UTF_8.name());
+            }
+
+            String password = downloadService.decryptPwd(pwd);
+            if (null == password) {
+                WebHelper.writeInfo(HttpStatus.BAD_REQUEST, "瀵嗙爜瑙e瘑澶辫触", res);
+            }
+
+            DownloadEntity de = downloadService.selectByGuid(guid);
+            if (null == de) {
+                WebHelper.writeInfo(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�", res);
+                return;
+            }
+            if (!StringHelper.isNull(de.getPwd()) && !Md5Helper.validatePassword(password, 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);
+        }
+    }
 }
diff --git a/src/main/java/com/lf/server/entity/all/StaticData.java b/src/main/java/com/lf/server/entity/all/StaticData.java
index bcd0451..2ac1e1c 100644
--- a/src/main/java/com/lf/server/entity/all/StaticData.java
+++ b/src/main/java/com/lf/server/entity/all/StaticData.java
@@ -65,6 +65,11 @@
     public final static String POINT = ".";
 
     /**
+     * 绛夊彿
+     */
+    public final static String EQ = "=";
+
+    /**
      * 鎻掑叆鎺掗櫎瀛楁
      */
     public final static List<String> INSERT_EXCLUDE_FIELDS = new ArrayList<>(Arrays.asList("gid", "objectid", "updateuser", "updatetime", "shape_leng", "shape_area", "dirName", "depName", "verName", "createName", "updateName"));

--
Gitblit v1.9.3