From c3ca2f26a73dc5fc7dc6f652a159c2697abd53cf Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期一, 05 十二月 2022 22:01:44 +0800 Subject: [PATCH] 1 --- src/main/java/com/lf/server/controller/data/DataLoaderController.java | 223 ++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 217 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/lf/server/controller/data/DataLoaderController.java b/src/main/java/com/lf/server/controller/data/DataLoaderController.java index 6a3f1b1..287a216 100644 --- a/src/main/java/com/lf/server/controller/data/DataLoaderController.java +++ b/src/main/java/com/lf/server/controller/data/DataLoaderController.java @@ -1,11 +1,38 @@ package com.lf.server.controller.data; -import com.lf.server.controller.all.BaseController; -import com.lf.server.service.sys.TokenService; +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.BaseUploadController; +import com.lf.server.entity.all.ResponseMsg; +import com.lf.server.entity.ctrl.TabEntity; +import com.lf.server.entity.ctrl.TabMapperEntity; +import com.lf.server.entity.ctrl.UploadEntity; +import com.lf.server.entity.data.DictEntity; +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.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.DataLoaderService; +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 org.springframework.beans.factory.annotation.Autowired; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletRequest; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Date; +import java.util.List; /** * 鏁版嵁鍏ュ簱 @@ -14,7 +41,191 @@ @Api(tags = "鏁版嵁绠$悊\\鏁版嵁鍏ュ簱") @RestController @RequestMapping("/dataLoader") -public class DataLoaderController extends BaseController { +public class DataLoaderController extends BaseUploadController { @Autowired - TokenService tokenService; + MetaService metaService; + + @Autowired + BaseQueryService baseQueryService; + + @Autowired + DataLoaderService dataLoaderService; + + private final static String POINT = "."; + + private final static String FILE_TYPES = "'xls','shp','gdb','mdb'"; + + public static List<String> extList = new ArrayList<>(Arrays.asList(".xls", ".xlsx", ".mdb", ".zip")); + + @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 = "/selectByPageForUpload") + public ResponseMsg<Object> selectByPageForUpload(String name, Integer pageSize, Integer pageIndex, HttpServletRequest req) { + try { + if (pageSize < 1 || pageIndex < 1) { + return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null); + } + + UserEntity ue = tokenService.getCurrentUser(req); + + int count = metaService.selectCountForUpload(name, ue.getId(), FILE_TYPES); + if (count == 0) { + return success(0, null); + } + + List<MetaEntity> list = metaService.selectByPageForUpload(name, ue.getId(), FILE_TYPES, pageSize, pageSize * (pageIndex - 1)); + + return success(count, list); + } catch (Exception ex) { + return fail(ex.getMessage(), null); + } + } + + @SysLog() + @ApiOperation(value = "鏌ヨ鏂囦欢") + @ApiImplicitParams({ + @ApiImplicitParam(name = "path", value = "璺緞", dataType = "String", paramType = "query") + }) + @GetMapping(value = "/selectFiles") + public ResponseMsg<List<MetaFileEntity>> selectFiles(String path) { + try { + List<MetaFileEntity> list = baseUploadService.selectFiles(path, extList); + + return success(list); + } catch (Exception ex) { + return fail(ex.getMessage(), null); + } + } + + @SysLog() + @ApiOperation(value = "鏌ヨ鎵�鏈夎〃") + @GetMapping(value = "/selectTabs") + public ResponseMsg<List<TabEntity>> selectTabs() { + try { + List<TabEntity> list = baseQueryService.selectTabs(); + + return success(list); + } catch (Exception ex) { + return fail(ex.getMessage(), null); + } + } + + @SysLog() + @ApiOperation(value = "鏌ヨ鏄犲皠") + @ApiImplicitParams({ + @ApiImplicitParam(name = "path", value = "璺緞", dataType = "String", paramType = "query") + }) + @GetMapping(value = "/selectMappers") + public ResponseMsg<List<TabMapperEntity>> selectMappers(String path) { + try { + List<TabMapperEntity> list = dataLoaderService.selectMappers(path); + + return success(list); + } catch (Exception ex) { + return fail(ex.getMessage(), null); + } + } + + @SysLog() + @ApiOperation(value = "鎻掑叆鏂囦欢") + @ApiImplicitParams({ + @ApiImplicitParam(name = "entity", value = "涓婁紶瀹炰綋绫�", dataType = "UploadEntity", paramType = "body") + }) + @ResponseBody + @PostMapping(value = "/insertFiles") + public ResponseMsg<Object> insertFiles(@RequestBody UploadEntity entity, HttpServletRequest req) { + try { + UserEntity ue = tokenService.getCurrentUser(req); + if (ue == null) { + return fail("鐢ㄦ埛鏈櫥褰�", null); + } + if (null == entity || null == entity.getMetaEntity()) { + return fail("鍏冩暟鎹俊鎭负绌�", null); + } + + if (null == entity.getFileEntities() || entity.getFileEntities().isEmpty()) { + return fail("娌℃湁鎵惧埌涓婁紶鏂囦欢", null); + } + + MetaEntity me = entity.getMetaEntity(); + me.setCreateTime(WebHelper.getCurrentTimestamp()); + me.setBatch(StringHelper.YMDHMS_FORMAT.format(new Date(me.getCreateTime().getTime()))); + me.setCreateUser(ue.getId()); + + dataLoaderService.insertFiles(me, entity.getFileEntities(), entity.getTabEntities()); + + return success("鎴愬姛", entity.getTabEntities()); + } 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(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 = "/selectFields") + public ResponseMsg<List<DictEntity>> selectFields(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); + } + } } -- Gitblit v1.9.3