From 762ba3d007e66a985bc05dde8149949d30a1e3be Mon Sep 17 00:00:00 2001
From: Surpriseplus <845948745@qq.com>
Date: 星期四, 17 十一月 2022 16:32:12 +0800
Subject: [PATCH] 资源表

---
 src/main/java/com/lf/server/controller/show/ExportController.java |  116 ++++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 104 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/show/ExportController.java b/src/main/java/com/lf/server/controller/show/ExportController.java
index 459fcf3..abd0c49 100644
--- a/src/main/java/com/lf/server/controller/show/ExportController.java
+++ b/src/main/java/com/lf/server/controller/show/ExportController.java
@@ -1,17 +1,26 @@
 package com.lf.server.controller.show;
 
-import com.lf.server.aspect.SysLog;
-import com.lf.server.controller.all.BaseQueryController;
+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.data.DictEntity;
+import com.lf.server.entity.data.DownloadEntity;
+import com.lf.server.entity.show.ExportEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.StringHelper;
+import com.lf.server.helper.WebHelper;
+import com.lf.server.service.data.DownloadService;
+import com.lf.server.service.show.ExportService;
+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.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+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;
 
 /**
@@ -21,27 +30,110 @@
 @Api(tags = "缁煎悎灞曠ず\\鍦ㄧ嚎鍒跺浘")
 @RestController
 @RequestMapping("/export")
-public class ExportController  extends BaseQueryController {
+public class ExportController  extends BaseController {
+    @Autowired
+    TokenService tokenService;
+
+    @Autowired
+    ExportService exportService;
+
+    @Autowired
+    DownloadService downloadService;
+
     @SysLog()
     @ApiOperation(value = "鍒嗛〉鏌ヨ")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "ns", value = "琛ㄧ┖闂�", dataType = "String", paramType = "query", example = "bd"),
-            @ApiImplicitParam(name = "tab", value = "琛ㄥ悕", dataType = "String", paramType = "query", example = "dlg_25w_hyda"),
+            @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<DictEntity>> selectByPage(String ns, String tab, Integer pageSize, Integer pageIndex) {
+    public ResponseMsg<List<DownloadEntity>> selectByPage(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);
+            }
 
-            return success(null);
+            int count = downloadService.selectCountForExport(ue.getCreateUser(), name);
+            if (count == 0) {
+                return success(0, null);
+            }
+            List<DownloadEntity> rs = downloadService.selectByPageForExport(ue.getCreateUser(), name, pageSize, pageSize * (pageIndex - 1));
+
+            return success(count, rs);
         } catch (Exception ex) {
             return fail(ex.getMessage(), null);
         }
     }
+
+    @SysLog()
+    @ApiOperation(value = "鏂板缓鍑哄浘")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "entity", value = "鍦ㄧ嚎鍒跺浘瀹炰綋绫�", dataType = "ExportEntity", paramType = "body")
+    })
+    @ResponseBody
+    @PostMapping(value = "/insertMap", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<String> insertMap(@RequestBody ExportEntity entity, HttpServletRequest req) {
+        try {
+            boolean flag = entity.getXmin() == 0 && entity.getYmin() == 0 && entity.getXmax() == 0 && entity.getYmax() == 0;
+            if (entity == null ||
+                    StringHelper.isEmpty(entity.getLayers()) ||
+                    StringHelper.isEmpty(entity.getToken()) ||
+                    flag) {
+                return fail("鍙傛暟涓嶅畬鏁�");
+            }
+
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return fail("鐢ㄦ埛鏈櫥褰�");
+            }
+
+            String str = exportService.post(ue, entity);
+            if (StringHelper.isNull(str)) {
+                return fail("鍑哄浘澶辫触");
+            }
+
+            return success("鍑哄浘鎴愬姛", str);
+        } catch (Exception ex) {
+            return fail(ex.getMessage());
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇鍑哄浘")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query")
+    })
+    @RequestMapping(value = "/downloadFile", method = RequestMethod.GET)
+    public void downloadFile(String guid, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�"));
+            }
+
+            DownloadEntity de = downloadService.selectByGuid(guid);
+            if (de == null) {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�"));
+            }
+
+            de.setDcount(de.getDcount() + 1);
+            de.setDownloadUser(ue.getId());
+            int rows = downloadService.update(de);
+
+            String filePath = exportService.getDownloadFilePath(de);
+            WebHelper.download(filePath, de.getName(), res);
+        } catch (Exception ex) {
+            try {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage()));
+            } catch (Exception e) {
+                log.error(e.getMessage(), e);
+            }
+        }
+    }
 }

--
Gitblit v1.9.3