commit e71dc8a22055e365ec089ea2867f7ef1af7a54b0 Author: LeoAnn <1297441823@qq.com> Date: Mon Apr 14 14:46:06 2025 +0800 first commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ed8368a --- /dev/null +++ b/.gitignore @@ -0,0 +1,47 @@ +###################################################################### +# Build Tools + +.gradle +/build/ +!gradle/wrapper/gradle-wrapper.jar + +target/ +!.mvn/wrapper/maven-wrapper.jar + +###################################################################### +# IDE + +### STS ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans + +### IntelliJ IDEA ### +.idea +*.iws +*.iml +*.ipr + +### JRebel ### +rebel.xml + +### NetBeans ### +nbproject/private/ +build/* +nbbuild/ +dist/ +nbdist/ +.nb-gradle/ + +###################################################################### +# Others +*.log +*.xml.versionsBackup +*.swp + +!*/build/*.java +!*/build/*.html +!*/build/*.xml diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..8564f29 --- /dev/null +++ b/LICENSE @@ -0,0 +1,20 @@ +The MIT License (MIT) + +Copyright (c) 2018 RuoYi + +Permission is hereby granted, free of charge, to any person obtaining a copy of +this software and associated documentation files (the "Software"), to deal in +the Software without restriction, including without limitation the rights to +use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of +the Software, and to permit persons to whom the Software is furnished to do so, +subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS +FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR +COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER +IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN +CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..ba50d3e --- /dev/null +++ b/README.md @@ -0,0 +1,97 @@ +

+ logo +

+

RuoYi v3.8.2

+

基于SpringBoot+Vue前后端分离的Java快速开发框架

+

+ + + +

+ +## 平台简介 + +若依是一套全部开源的快速开发平台,毫无保留给个人及企业免费使用。 + +* 前端采用Vue、Element UI。 +* 后端采用Spring Boot、Spring Security、Redis & Jwt。 +* 权限认证使用Jwt,支持多终端认证系统。 +* 支持加载动态权限菜单,多方式轻松权限控制。 +* 高效率开发,使用代码生成器可以一键生成前后端代码。 +* 提供了技术栈([Vue3](https://v3.cn.vuejs.org) [Element Plus](https://element-plus.org/zh-CN) [Vite](https://cn.vitejs.dev))版本[RuoYi-Vue3](https://github.com/yangzongzhuan/RuoYi-Vue3),保持同步更新。 +* 提供了单应用版本[RuoYi-Vue-fast](https://github.com/yangzongzhuan/RuoYi-Vue-fast),Oracle版本[RuoYi-Vue-Oracle](https://github.com/yangzongzhuan/RuoYi-Vue-Oracle),保持同步更新。 +* 不分离版本,请移步[RuoYi](https://gitee.com/y_project/RuoYi),微服务版本,请移步[RuoYi-Cloud](https://gitee.com/y_project/RuoYi-Cloud) +* 特别鸣谢:[element](https://github.com/ElemeFE/element),[vue-element-admin](https://github.com/PanJiaChen/vue-element-admin),[eladmin-web](https://github.com/elunez/eladmin-web)。 +* 阿里云折扣场:[点我进入](http://aly.ruoyi.vip),腾讯云秒杀场:[点我进入](http://txy.ruoyi.vip)   +* 阿里云优惠券:[点我领取](https://www.aliyun.com/minisite/goods?userCode=brki8iof&share_source=copy_link),腾讯云优惠券:[点我领取](https://cloud.tencent.com/redirect.php?redirect=1025&cps_key=198c8df2ed259157187173bc7f4f32fd&from=console)   + +## 内置功能 + +1. 用户管理:用户是系统操作者,该功能主要完成系统用户配置。 +2. 部门管理:配置系统组织机构(公司、部门、小组),树结构展现支持数据权限。 +3. 岗位管理:配置系统用户所属担任职务。 +4. 菜单管理:配置系统菜单,操作权限,按钮权限标识等。 +5. 角色管理:角色菜单权限分配、设置角色按机构进行数据范围权限划分。 +6. 字典管理:对系统中经常使用的一些较为固定的数据进行维护。 +7. 参数管理:对系统动态配置常用参数。 +8. 通知公告:系统通知公告信息发布维护。 +9. 操作日志:系统正常操作日志记录和查询;系统异常信息日志记录和查询。 +10. 登录日志:系统登录日志记录查询包含登录异常。 +11. 在线用户:当前系统中活跃用户状态监控。 +12. 定时任务:在线(添加、修改、删除)任务调度包含执行结果日志。 +13. 代码生成:前后端代码的生成(java、html、xml、sql)支持CRUD下载 。 +14. 系统接口:根据业务代码自动生成相关的api接口文档。 +15. 服务监控:监视当前系统CPU、内存、磁盘、堆栈等相关信息。 +16. 缓存监控:对系统的缓存信息查询,命令统计等。 +17. 在线构建器:拖动表单元素生成相应的HTML代码。 +18. 连接池监视:监视当前系统数据库连接池状态,可进行分析SQL找出系统性能瓶颈。 + +## 在线体验 + +- admin/admin123 +- 陆陆续续收到一些打赏,为了更好的体验已用于演示服务器升级。谢谢各位小伙伴。 + +演示地址:http://vue.ruoyi.vip +文档地址:http://doc.ruoyi.vip + +## 演示图 + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
+ + +## 若依前后端分离交流群 + +QQ群: [![加入QQ群](https://img.shields.io/badge/已满-937441-blue.svg)](https://jq.qq.com/?_wv=1027&k=5bVB1og) [![加入QQ群](https://img.shields.io/badge/已满-887144332-blue.svg)](https://jq.qq.com/?_wv=1027&k=5eiA4DH) [![加入QQ群](https://img.shields.io/badge/已满-180251782-blue.svg)](https://jq.qq.com/?_wv=1027&k=5AxMKlC) [![加入QQ群](https://img.shields.io/badge/已满-104180207-blue.svg)](https://jq.qq.com/?_wv=1027&k=51G72yr) [![加入QQ群](https://img.shields.io/badge/已满-186866453-blue.svg)](https://jq.qq.com/?_wv=1027&k=VvjN2nvu) [![加入QQ群](https://img.shields.io/badge/已满-201396349-blue.svg)](https://jq.qq.com/?_wv=1027&k=5vYAqA05) [![加入QQ群](https://img.shields.io/badge/已满-101456076-blue.svg)](https://jq.qq.com/?_wv=1027&k=kOIINEb5) [![加入QQ群](https://img.shields.io/badge/已满-101539465-blue.svg)](https://jq.qq.com/?_wv=1027&k=UKtX5jhs) [![加入QQ群](https://img.shields.io/badge/已满-264312783-blue.svg)](https://jq.qq.com/?_wv=1027&k=EI9an8lJ) [![加入QQ群](https://img.shields.io/badge/167385320-blue.svg)](https://jq.qq.com/?_wv=1027&k=SWCtLnMz) 点击按钮入群。 \ No newline at end of file diff --git a/bin/clean.bat b/bin/clean.bat new file mode 100644 index 0000000..24c0974 --- /dev/null +++ b/bin/clean.bat @@ -0,0 +1,12 @@ +@echo off +echo. +echo [Ϣ] target· +echo. + +%~d0 +cd %~dp0 + +cd .. +call mvn clean + +pause \ No newline at end of file diff --git a/bin/package.bat b/bin/package.bat new file mode 100644 index 0000000..c693ec0 --- /dev/null +++ b/bin/package.bat @@ -0,0 +1,12 @@ +@echo off +echo. +echo [Ϣ] Weḅwar/jarļ +echo. + +%~d0 +cd %~dp0 + +cd .. +call mvn clean package -Dmaven.test.skip=true + +pause \ No newline at end of file diff --git a/bin/run.bat b/bin/run.bat new file mode 100644 index 0000000..41efbd0 --- /dev/null +++ b/bin/run.bat @@ -0,0 +1,14 @@ +@echo off +echo. +echo [Ϣ] ʹJarWeb̡ +echo. + +cd %~dp0 +cd ../ruoyi-admin/target + +set JAVA_OPTS=-Xms256m -Xmx1024m -XX:MetaspaceSize=128m -XX:MaxMetaspaceSize=512m + +java -jar %JAVA_OPTS% ruoyi-admin.jar + +cd bin +pause \ No newline at end of file diff --git a/doc/若依环境使用手册.docx b/doc/若依环境使用手册.docx new file mode 100644 index 0000000..9e4daef Binary files /dev/null and b/doc/若依环境使用手册.docx differ diff --git a/pom.xml b/pom.xml new file mode 100644 index 0000000..0758b18 --- /dev/null +++ b/pom.xml @@ -0,0 +1,265 @@ + + + 4.0.0 + + com.ruoyi + ruoyi + 3.8.1 + + ruoyi + http://www.ruoyi.vip + 若依管理系统 + + + 3.8.1 + UTF-8 + UTF-8 + 1.8 + 3.1.1 + 1.2.8 + 1.21 + 3.0.0 + 2.3.2 + 3.4.2 + 1.4.1 + 1.2.80 + 6.1.6 + 2.11.0 + 1.4 + 3.2.2 + 4.1.2 + 2.3 + 0.9.1 + + + + + + + + + org.springframework.boot + spring-boot-dependencies + 2.5.13 + pom + import + + + + + + + + com.alibaba + druid-spring-boot-starter + ${druid.version} + + + + + eu.bitwalker + UserAgentUtils + ${bitwalker.version} + + + + + + + + + + + + com.github.pagehelper + pagehelper-spring-boot-starter + ${pagehelper.boot.version} + + + + + com.github.oshi + oshi-core + ${oshi.version} + + + + + io.springfox + springfox-boot-starter + ${swagger.version} + + + io.swagger + swagger-models + + + + + + + commons-io + commons-io + ${commons.io.version} + + + + + commons-fileupload + commons-fileupload + ${commons.fileupload.version} + + + + + org.apache.poi + poi-ooxml + ${poi.version} + + + + + org.apache.velocity + velocity-engine-core + ${velocity.version} + + + + + commons-collections + commons-collections + ${commons.collections.version} + + + + + com.alibaba + fastjson + ${fastjson.version} + + + + + io.jsonwebtoken + jjwt + ${jwt.version} + + + + + com.github.penggle + kaptcha + ${kaptcha.version} + + + + + com.ruoyi + ruoyi-quartz + ${ruoyi.version} + + + + + com.ruoyi + ruoyi-generator + ${ruoyi.version} + + + + + com.ruoyi + ruoyi-framework + ${ruoyi.version} + + + + + com.ruoyi + ruoyi-system + ${ruoyi.version} + + + + + + com.ruoyi + swlscx + ${ruoyi.version} + + + + + com.ruoyi + ruoyi-common + ${ruoyi.version} + + + + com.ruoyi + ruoyi-api + ${ruoyi.version} + + + + + + ruoyi-admin + ruoyi-framework + ruoyi-system + ruoyi-quartz + ruoyi-generator + ruoyi-common + ruoyi-api + swlscx + + pom + + + + + + + + + + org.apache.maven.plugins + maven-compiler-plugin + 3.1 + + ${java.version} + ${java.version} + ${project.build.sourceEncoding} + + + + + + + + public + aliyun nexus + https://maven.aliyun.com/repository/public + + true + + + + + + + public + aliyun nexus + https://maven.aliyun.com/repository/public + + true + + + false + + + + + diff --git a/ruoyi-admin/pom.xml b/ruoyi-admin/pom.xml new file mode 100644 index 0000000..b6995f6 --- /dev/null +++ b/ruoyi-admin/pom.xml @@ -0,0 +1,135 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + jar + ruoyi-admin + + + web服务入口 + + + + + + + org.springframework.boot + spring-boot-devtools + true + + + + + io.springfox + springfox-boot-starter + + + + + + com.ruoyi + swlscx + + + + + io.swagger + swagger-models + 1.6.2 + + + + + mysql + mysql-connector-java + + + + + com.ruoyi + ruoyi-framework + + + + com.microsoft.sqlserver + mssql-jdbc + 8.4.1.jre8 + + + + + + + + + + + org.springframework.boot + spring-boot-test + test + + + + org.springframework.boot + spring-boot-starter-test + test + + + + + com.ruoyi + ruoyi-quartz + + + + + com.ruoyi + ruoyi-generator + + + + + com.ruoyi + ruoyi-api + + + + + + + + + org.springframework.boot + spring-boot-maven-plugin + 2.1.1.RELEASE + + true + + + + + repackage + + + + + + org.apache.maven.plugins + maven-war-plugin + 3.1.0 + + false + ${project.artifactId} + + + + ${project.artifactId} + + + diff --git a/ruoyi-admin/src/main/java/com/ruoyi/RuoYiApplication.java b/ruoyi-admin/src/main/java/com/ruoyi/RuoYiApplication.java new file mode 100644 index 0000000..f5b7041 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/RuoYiApplication.java @@ -0,0 +1,28 @@ +package com.ruoyi; + +import com.ruoyi.common.utils.BeanUtil; +import org.springframework.boot.SpringApplication; +import org.springframework.boot.autoconfigure.SpringBootApplication; +import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration; +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.context.annotation.Bean; + +/** + * 启动程序 + * + * @author ruoyi + */ +@SpringBootApplication(exclude = { DataSourceAutoConfiguration.class }) +public class RuoYiApplication +{ + public static void main(String[] args) + { + SpringApplication.run(RuoYiApplication.class, args); + System.out.println("(♥◠‿◠)ノ゙ 启动成功 ლ(´ڡ`ლ)゙ "); + } + + @Bean + public BeanUtil beanUtil() { + return new BeanUtil(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/RuoYiServletInitializer.java b/ruoyi-admin/src/main/java/com/ruoyi/RuoYiServletInitializer.java new file mode 100644 index 0000000..6de67dc --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/RuoYiServletInitializer.java @@ -0,0 +1,18 @@ +package com.ruoyi; + +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.boot.web.servlet.support.SpringBootServletInitializer; + +/** + * web容器中进行部署 + * + * @author ruoyi + */ +public class RuoYiServletInitializer extends SpringBootServletInitializer +{ + @Override + protected SpringApplicationBuilder configure(SpringApplicationBuilder application) + { + return application.sources(RuoYiApplication.class); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java new file mode 100644 index 0000000..6bd8ddd --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java @@ -0,0 +1,93 @@ +package com.ruoyi.web.controller.common; + +import java.awt.image.BufferedImage; +import java.io.IOException; +import java.util.concurrent.TimeUnit; +import javax.annotation.Resource; +import javax.imageio.ImageIO; +import javax.servlet.http.HttpServletResponse; +import com.ruoyi.common.config.RuoYiConfig; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.util.FastByteArrayOutputStream; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RestController; +import com.google.code.kaptcha.Producer; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.utils.sign.Base64; +import com.ruoyi.common.utils.uuid.IdUtils; +import com.ruoyi.system.service.ISysConfigService; + +/** + * 验证码操作处理 + * + * @author ruoyi + */ +@RestController +public class CaptchaController +{ + @Resource(name = "captchaProducer") + private Producer captchaProducer; + + @Resource(name = "captchaProducerMath") + private Producer captchaProducerMath; + + @Autowired + private RedisCache redisCache; + + @Autowired + private ISysConfigService configService; + /** + * 生成验证码 + */ + @GetMapping("/captchaImage") + public AjaxResult getCode(HttpServletResponse response) throws IOException + { + AjaxResult ajax = AjaxResult.success(); + boolean captchaOnOff = configService.selectCaptchaOnOff(); + ajax.put("captchaOnOff", captchaOnOff); + if (!captchaOnOff) + { + return ajax; + } + + // 保存验证码信息 + String uuid = IdUtils.simpleUUID(); + String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid; + + String capStr = null, code = null; + BufferedImage image = null; + + // 生成验证码 + String captchaType = RuoYiConfig.getCaptchaType(); + if ("math".equals(captchaType)) + { + String capText = captchaProducerMath.createText(); + capStr = capText.substring(0, capText.lastIndexOf("@")); + code = capText.substring(capText.lastIndexOf("@") + 1); + image = captchaProducerMath.createImage(capStr); + } + else if ("char".equals(captchaType)) + { + capStr = code = captchaProducer.createText(); + image = captchaProducer.createImage(capStr); + } + + redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES); + // 转换流信息写出 + FastByteArrayOutputStream os = new FastByteArrayOutputStream(); + try + { + ImageIO.write(image, "jpg", os); + } + catch (IOException e) + { + return AjaxResult.error(e.getMessage()); + } + + ajax.put("uuid", uuid); + ajax.put("img", Base64.encode(os.toByteArray())); + return ajax; + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java new file mode 100644 index 0000000..cec5006 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java @@ -0,0 +1,163 @@ +package com.ruoyi.web.controller.common; + +import java.util.ArrayList; +import java.util.List; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.http.MediaType; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.file.FileUploadUtils; +import com.ruoyi.common.utils.file.FileUtils; +import com.ruoyi.framework.config.ServerConfig; + +/** + * 通用请求处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/common") +public class CommonController +{ + private static final Logger log = LoggerFactory.getLogger(CommonController.class); + + @Autowired + private ServerConfig serverConfig; + + private static final String FILE_DELIMETER = ","; + + /** + * 通用下载请求 + * + * @param fileName 文件名称 + * @param delete 是否删除 + */ + @GetMapping("/download") + public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request) + { + try + { + if (!FileUtils.checkAllowDownload(fileName)) + { + throw new Exception(StringUtils.format("文件名称({})非法,不允许下载。 ", fileName)); + } + String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1); + String filePath = RuoYiConfig.getDownloadPath() + fileName; + + response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); + FileUtils.setAttachmentResponseHeader(response, realFileName); + FileUtils.writeBytes(filePath, response.getOutputStream()); + if (delete) + { + FileUtils.deleteFile(filePath); + } + } + catch (Exception e) + { + log.error("下载文件失败", e); + } + } + + /** + * 通用上传请求(单个) + */ + @PostMapping("/upload") + public AjaxResult uploadFile(MultipartFile file) throws Exception + { + try + { + // 上传文件路径 + String filePath = RuoYiConfig.getUploadPath(); + // 上传并返回新文件名称 + String fileName = FileUploadUtils.upload(filePath, file); + String url = serverConfig.getUrl() + fileName; + AjaxResult ajax = AjaxResult.success(); + ajax.put("url", url); + ajax.put("fileName", fileName); + ajax.put("newFileName", FileUtils.getName(fileName)); + ajax.put("originalFilename", file.getOriginalFilename()); + return ajax; + } + catch (Exception e) + { + return AjaxResult.error(e.getMessage()); + } + } + + /** + * 通用上传请求(多个) + */ + @PostMapping("/uploads") + public AjaxResult uploadFiles(List files) throws Exception + { + try + { + // 上传文件路径 + String filePath = RuoYiConfig.getUploadPath(); + List urls = new ArrayList(); + List fileNames = new ArrayList(); + List newFileNames = new ArrayList(); + List originalFilenames = new ArrayList(); + for (MultipartFile file : files) + { + // 上传并返回新文件名称 + String fileName = FileUploadUtils.upload(filePath, file); + String url = serverConfig.getUrl() + fileName; + urls.add(url); + fileNames.add(fileName); + newFileNames.add(FileUtils.getName(fileName)); + originalFilenames.add(file.getOriginalFilename()); + } + AjaxResult ajax = AjaxResult.success(); + ajax.put("urls", StringUtils.join(urls, FILE_DELIMETER)); + ajax.put("fileNames", StringUtils.join(fileNames, FILE_DELIMETER)); + ajax.put("newFileNames", StringUtils.join(newFileNames, FILE_DELIMETER)); + ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER)); + return ajax; + } + catch (Exception e) + { + return AjaxResult.error(e.getMessage()); + } + } + + /** + * 本地资源通用下载 + */ + @GetMapping("/download/resource") + public void resourceDownload(String resource, HttpServletRequest request, HttpServletResponse response) + throws Exception + { + try + { + if (!FileUtils.checkAllowDownload(resource)) + { + throw new Exception(StringUtils.format("资源文件({})非法,不允许下载。 ", resource)); + } + // 本地资源路径 + String localPath = RuoYiConfig.getProfile(); + // 数据库资源地址 + String downloadPath = localPath + StringUtils.substringAfter(resource, Constants.RESOURCE_PREFIX); + // 下载名称 + String downloadName = StringUtils.substringAfterLast(downloadPath, "/"); + response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); + FileUtils.setAttachmentResponseHeader(response, downloadName); + FileUtils.writeBytes(downloadPath, response.getOutputStream()); + } + catch (Exception e) + { + log.error("下载文件失败", e); + } + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/CacheController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/CacheController.java new file mode 100644 index 0000000..7b97de2 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/CacheController.java @@ -0,0 +1,53 @@ +package com.ruoyi.web.controller.monitor; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Properties; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.redis.core.RedisCallback; +import org.springframework.data.redis.core.RedisTemplate; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.utils.StringUtils; + +/** + * 缓存监控 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/cache") +public class CacheController +{ + @Autowired + private RedisTemplate redisTemplate; + + @PreAuthorize("@ss.hasPermi('monitor:cache:list')") + @GetMapping() + public AjaxResult getInfo() throws Exception + { + Properties info = (Properties) redisTemplate.execute((RedisCallback) connection -> connection.info()); + Properties commandStats = (Properties) redisTemplate.execute((RedisCallback) connection -> connection.info("commandstats")); + Object dbSize = redisTemplate.execute((RedisCallback) connection -> connection.dbSize()); + + Map result = new HashMap<>(3); + result.put("info", info); + result.put("dbSize", dbSize); + + List> pieList = new ArrayList<>(); + commandStats.stringPropertyNames().forEach(key -> { + Map data = new HashMap<>(2); + String property = commandStats.getProperty(key); + data.put("name", StringUtils.removeStart(key, "cmdstat_")); + data.put("value", StringUtils.substringBetween(property, "calls=", ",usec")); + pieList.add(data); + }); + result.put("commandStats", pieList); + return AjaxResult.success(result); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/ServerController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/ServerController.java new file mode 100644 index 0000000..cc805ad --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/ServerController.java @@ -0,0 +1,27 @@ +package com.ruoyi.web.controller.monitor; + +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.framework.web.domain.Server; + +/** + * 服务器监控 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/server") +public class ServerController +{ + @PreAuthorize("@ss.hasPermi('monitor:server:list')") + @GetMapping() + public AjaxResult getInfo() throws Exception + { + Server server = new Server(); + server.copyTo(); + return AjaxResult.success(server); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysLogininforController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysLogininforController.java new file mode 100644 index 0000000..41dcf98 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysLogininforController.java @@ -0,0 +1,69 @@ +package com.ruoyi.web.controller.monitor; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.domain.SysLogininfor; +import com.ruoyi.system.service.ISysLogininforService; + +/** + * 系统访问记录 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/logininfor") +public class SysLogininforController extends BaseController +{ + @Autowired + private ISysLogininforService logininforService; + + @PreAuthorize("@ss.hasPermi('monitor:logininfor:list')") + @GetMapping("/list") + public TableDataInfo list(SysLogininfor logininfor) + { + startPage(); + List list = logininforService.selectLogininforList(logininfor); + return getDataTable(list); + } + + @Log(title = "登录日志", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('monitor:logininfor:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysLogininfor logininfor) + { + List list = logininforService.selectLogininforList(logininfor); + ExcelUtil util = new ExcelUtil(SysLogininfor.class); + util.exportExcel(response, list, "登录日志"); + } + + @PreAuthorize("@ss.hasPermi('monitor:logininfor:remove')") + @Log(title = "登录日志", businessType = BusinessType.DELETE) + @DeleteMapping("/{infoIds}") + public AjaxResult remove(@PathVariable Long[] infoIds) + { + return toAjax(logininforService.deleteLogininforByIds(infoIds)); + } + + @PreAuthorize("@ss.hasPermi('monitor:logininfor:remove')") + @Log(title = "登录日志", businessType = BusinessType.CLEAN) + @DeleteMapping("/clean") + public AjaxResult clean() + { + logininforService.cleanLogininfor(); + return AjaxResult.success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysOperlogController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysOperlogController.java new file mode 100644 index 0000000..b6ba56b --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysOperlogController.java @@ -0,0 +1,69 @@ +package com.ruoyi.web.controller.monitor; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.domain.SysOperLog; +import com.ruoyi.system.service.ISysOperLogService; + +/** + * 操作日志记录 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/operlog") +public class SysOperlogController extends BaseController +{ + @Autowired + private ISysOperLogService operLogService; + + @PreAuthorize("@ss.hasPermi('monitor:operlog:list')") + @GetMapping("/list") + public TableDataInfo list(SysOperLog operLog) + { + startPage(); + List list = operLogService.selectOperLogList(operLog); + return getDataTable(list); + } + + @Log(title = "操作日志", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('monitor:operlog:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysOperLog operLog) + { + List list = operLogService.selectOperLogList(operLog); + ExcelUtil util = new ExcelUtil(SysOperLog.class); + util.exportExcel(response, list, "操作日志"); + } + + @Log(title = "操作日志", businessType = BusinessType.DELETE) + @PreAuthorize("@ss.hasPermi('monitor:operlog:remove')") + @DeleteMapping("/{operIds}") + public AjaxResult remove(@PathVariable Long[] operIds) + { + return toAjax(operLogService.deleteOperLogByIds(operIds)); + } + + @Log(title = "操作日志", businessType = BusinessType.CLEAN) + @PreAuthorize("@ss.hasPermi('monitor:operlog:remove')") + @DeleteMapping("/clean") + public AjaxResult clean() + { + operLogService.cleanOperLog(); + return AjaxResult.success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysUserOnlineController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysUserOnlineController.java new file mode 100644 index 0000000..9b157ea --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysUserOnlineController.java @@ -0,0 +1,92 @@ +package com.ruoyi.web.controller.monitor; + +import java.util.ArrayList; +import java.util.Collection; +import java.util.Collections; +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.domain.SysUserOnline; +import com.ruoyi.system.service.ISysUserOnlineService; + +/** + * 在线用户监控 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/online") +public class SysUserOnlineController extends BaseController +{ + @Autowired + private ISysUserOnlineService userOnlineService; + + @Autowired + private RedisCache redisCache; + + @PreAuthorize("@ss.hasPermi('monitor:online:list')") + @GetMapping("/list") + public TableDataInfo list(String ipaddr, String userName) + { + Collection keys = redisCache.keys(Constants.LOGIN_TOKEN_KEY + "*"); + List userOnlineList = new ArrayList(); + for (String key : keys) + { + LoginUser user = redisCache.getCacheObject(key); + if (StringUtils.isNotEmpty(ipaddr) && StringUtils.isNotEmpty(userName)) + { + if (StringUtils.equals(ipaddr, user.getIpaddr()) && StringUtils.equals(userName, user.getUsername())) + { + userOnlineList.add(userOnlineService.selectOnlineByInfo(ipaddr, userName, user)); + } + } + else if (StringUtils.isNotEmpty(ipaddr)) + { + if (StringUtils.equals(ipaddr, user.getIpaddr())) + { + userOnlineList.add(userOnlineService.selectOnlineByIpaddr(ipaddr, user)); + } + } + else if (StringUtils.isNotEmpty(userName) && StringUtils.isNotNull(user.getUser())) + { + if (StringUtils.equals(userName, user.getUsername())) + { + userOnlineList.add(userOnlineService.selectOnlineByUserName(userName, user)); + } + } + else + { + userOnlineList.add(userOnlineService.loginUserToUserOnline(user)); + } + } + Collections.reverse(userOnlineList); + userOnlineList.removeAll(Collections.singleton(null)); + return getDataTable(userOnlineList); + } + + /** + * 强退用户 + */ + @PreAuthorize("@ss.hasPermi('monitor:online:forceLogout')") + @Log(title = "在线用户", businessType = BusinessType.FORCE) + @DeleteMapping("/{tokenId}") + public AjaxResult forceLogout(@PathVariable String tokenId) + { + redisCache.deleteObject(Constants.LOGIN_TOKEN_KEY + tokenId); + return AjaxResult.success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/LoginSsoController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/LoginSsoController.java new file mode 100644 index 0000000..08dc709 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/LoginSsoController.java @@ -0,0 +1,86 @@ +package com.ruoyi.web.controller.system; + +import com.alibaba.fastjson.JSON; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.framework.web.service.ThirdLoginService; +import com.ruoyi.framework.web.service.TokenService; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.util.concurrent.TimeUnit; + +/** + * 单点登录Controller + * @author sgc + */ +@Slf4j +@RestController +public class LoginSsoController { + + @Autowired + private ThirdLoginService loginService; + + @Autowired + private TokenService tokenService; + + @Autowired + private RedisCache redisCache; + + @PostMapping("/loginSso") + public AjaxResult loginSso(@RequestParam("code") String code) { + System.out.println("token = " + code); + AjaxResult ajax = AjaxResult.success(); + + String token = redisCache.getCacheObject(code); + LoginUser loginUser = tokenService.getThirdLoginUser(token); + + System.out.println("loginUser = " + loginUser); + if(loginUser != null){ + ajax.put(Constants.TOKEN, token); + ajax.put("msg", "登录成功"); + return ajax; + } + // 生成令牌(免密登录) + String tokenNew = loginService.thirdLogin(code); + if(null==tokenNew){ + ajax = AjaxResult.error("获取认证失败,请重新登入"); + }else { + redisCache.setCacheObject(code,tokenNew, 60, TimeUnit.MINUTES); + ajax.put(Constants.TOKEN, tokenNew); + ajax.put("msg", "登录成功"); + } + return ajax; + + } + + + @PostMapping("/logoutSso") + public void logoutSso(HttpServletRequest request, HttpServletResponse response,@RequestParam("token") String token) + { + String thisToken = redisCache.getCacheObject(token); + LoginUser loginUser = tokenService.getThirdLoginUser(thisToken); + if (StringUtils.isNotNull(loginUser)) + { + String userName = loginUser.getUsername(); + // 删除用户缓存记录 + tokenService.delLoginUser(loginUser.getToken()); + // 记录用户退出日志 + AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGOUT, "退出成功")); + } + ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(HttpStatus.SUCCESS, "退出成功"))); + } + +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysConfigController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysConfigController.java new file mode 100644 index 0000000..b6f0518 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysConfigController.java @@ -0,0 +1,134 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.domain.SysConfig; +import com.ruoyi.system.service.ISysConfigService; + +/** + * 参数配置 信息操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/config") +public class SysConfigController extends BaseController +{ + @Autowired + private ISysConfigService configService; + + /** + * 获取参数配置列表 + */ + @PreAuthorize("@ss.hasPermi('system:config:list')") + @GetMapping("/list") + public TableDataInfo list(SysConfig config) + { + startPage(); + List list = configService.selectConfigList(config); + return getDataTable(list); + } + + @Log(title = "参数管理", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:config:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysConfig config) + { + List list = configService.selectConfigList(config); + ExcelUtil util = new ExcelUtil(SysConfig.class); + util.exportExcel(response, list, "参数数据"); + } + + /** + * 根据参数编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:config:query')") + @GetMapping(value = "/{configId}") + public AjaxResult getInfo(@PathVariable Long configId) + { + return AjaxResult.success(configService.selectConfigById(configId)); + } + + /** + * 根据参数键名查询参数值 + */ + @GetMapping(value = "/configKey/{configKey}") + public AjaxResult getConfigKey(@PathVariable String configKey) + { + return AjaxResult.success(configService.selectConfigByKey(configKey)); + } + + /** + * 新增参数配置 + */ + @PreAuthorize("@ss.hasPermi('system:config:add')") + @Log(title = "参数管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysConfig config) + { + if (UserConstants.NOT_UNIQUE.equals(configService.checkConfigKeyUnique(config))) + { + return AjaxResult.error("新增参数'" + config.getConfigName() + "'失败,参数键名已存在"); + } + config.setCreateBy(getUsername()); + return toAjax(configService.insertConfig(config)); + } + + /** + * 修改参数配置 + */ + @PreAuthorize("@ss.hasPermi('system:config:edit')") + @Log(title = "参数管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysConfig config) + { + if (UserConstants.NOT_UNIQUE.equals(configService.checkConfigKeyUnique(config))) + { + return AjaxResult.error("修改参数'" + config.getConfigName() + "'失败,参数键名已存在"); + } + config.setUpdateBy(getUsername()); + return toAjax(configService.updateConfig(config)); + } + + /** + * 删除参数配置 + */ + @PreAuthorize("@ss.hasPermi('system:config:remove')") + @Log(title = "参数管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{configIds}") + public AjaxResult remove(@PathVariable Long[] configIds) + { + configService.deleteConfigByIds(configIds); + return success(); + } + + /** + * 刷新参数缓存 + */ + @PreAuthorize("@ss.hasPermi('system:config:remove')") + @Log(title = "参数管理", businessType = BusinessType.CLEAN) + @DeleteMapping("/refreshCache") + public AjaxResult refreshCache() + { + configService.resetConfigCache(); + return AjaxResult.success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDeptController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDeptController.java new file mode 100644 index 0000000..867ac71 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDeptController.java @@ -0,0 +1,165 @@ +package com.ruoyi.web.controller.system; + +import java.util.Iterator; +import java.util.List; +import org.apache.commons.lang3.ArrayUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysDept; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.service.ISysDeptService; + +/** + * 部门信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/dept") +public class SysDeptController extends BaseController +{ + @Autowired + private ISysDeptService deptService; + + /** + * 获取部门列表 + */ + @PreAuthorize("@ss.hasPermi('system:dept:list')") + @GetMapping("/list") + public AjaxResult list(SysDept dept) + { + List depts = deptService.selectDeptList(dept); + return AjaxResult.success(depts); + } + + /** + * 查询部门列表(排除节点) + */ + @PreAuthorize("@ss.hasPermi('system:dept:list')") + @GetMapping("/list/exclude/{deptId}") + public AjaxResult excludeChild(@PathVariable(value = "deptId", required = false) Long deptId) + { + List depts = deptService.selectDeptList(new SysDept()); + Iterator it = depts.iterator(); + while (it.hasNext()) + { + SysDept d = (SysDept) it.next(); + if (d.getDeptId().intValue() == deptId + || ArrayUtils.contains(StringUtils.split(d.getAncestors(), ","), deptId + "")) + { + it.remove(); + } + } + return AjaxResult.success(depts); + } + + /** + * 根据部门编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:dept:query')") + @GetMapping(value = "/{deptId}") + public AjaxResult getInfo(@PathVariable Long deptId) + { + deptService.checkDeptDataScope(deptId); + return AjaxResult.success(deptService.selectDeptById(deptId)); + } + + /** + * 获取部门下拉树列表 + */ + @GetMapping("/treeselect") + public AjaxResult treeselect(SysDept dept) + { + List depts = deptService.selectDeptList(dept); + return AjaxResult.success(deptService.buildDeptTreeSelect(depts)); + } + + /** + * 加载对应角色部门列表树 + */ + @GetMapping(value = "/roleDeptTreeselect/{roleId}") + public AjaxResult roleDeptTreeselect(@PathVariable("roleId") Long roleId) + { + List depts = deptService.selectDeptList(new SysDept()); + AjaxResult ajax = AjaxResult.success(); + ajax.put("checkedKeys", deptService.selectDeptListByRoleId(roleId)); + ajax.put("depts", deptService.buildDeptTreeSelect(depts)); + return ajax; + } + + /** + * 新增部门 + */ + @PreAuthorize("@ss.hasPermi('system:dept:add')") + @Log(title = "部门管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysDept dept) + { + if (UserConstants.NOT_UNIQUE.equals(deptService.checkDeptNameUnique(dept))) + { + return AjaxResult.error("新增部门'" + dept.getDeptName() + "'失败,部门名称已存在"); + } + dept.setCreateBy(getUsername()); + return toAjax(deptService.insertDept(dept)); + } + + /** + * 修改部门 + */ + @PreAuthorize("@ss.hasPermi('system:dept:edit')") + @Log(title = "部门管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysDept dept) + { + Long deptId = dept.getDeptId(); + deptService.checkDeptDataScope(deptId); + if (UserConstants.NOT_UNIQUE.equals(deptService.checkDeptNameUnique(dept))) + { + return AjaxResult.error("修改部门'" + dept.getDeptName() + "'失败,部门名称已存在"); + } + else if (dept.getParentId().equals(deptId)) + { + return AjaxResult.error("修改部门'" + dept.getDeptName() + "'失败,上级部门不能是自己"); + } + else if (StringUtils.equals(UserConstants.DEPT_DISABLE, dept.getStatus()) && deptService.selectNormalChildrenDeptById(deptId) > 0) + { + return AjaxResult.error("该部门包含未停用的子部门!"); + } + dept.setUpdateBy(getUsername()); + return toAjax(deptService.updateDept(dept)); + } + + /** + * 删除部门 + */ + @PreAuthorize("@ss.hasPermi('system:dept:remove')") + @Log(title = "部门管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{deptId}") + public AjaxResult remove(@PathVariable Long deptId) + { + if (deptService.hasChildByDeptId(deptId)) + { + return AjaxResult.error("存在下级部门,不允许删除"); + } + if (deptService.checkDeptExistUser(deptId)) + { + return AjaxResult.error("部门存在用户,不允许删除"); + } + deptService.checkDeptDataScope(deptId); + return toAjax(deptService.deleteDeptById(deptId)); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictDataController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictDataController.java new file mode 100644 index 0000000..b53fe47 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictDataController.java @@ -0,0 +1,121 @@ +package com.ruoyi.web.controller.system; + +import java.util.ArrayList; +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysDictData; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.service.ISysDictDataService; +import com.ruoyi.system.service.ISysDictTypeService; + +/** + * 数据字典信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/dict/data") +public class SysDictDataController extends BaseController +{ + @Autowired + private ISysDictDataService dictDataService; + + @Autowired + private ISysDictTypeService dictTypeService; + + @PreAuthorize("@ss.hasPermi('system:dict:list')") + @GetMapping("/list") + public TableDataInfo list(SysDictData dictData) + { + startPage(); + List list = dictDataService.selectDictDataList(dictData); + return getDataTable(list); + } + + @Log(title = "字典数据", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:dict:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysDictData dictData) + { + List list = dictDataService.selectDictDataList(dictData); + ExcelUtil util = new ExcelUtil(SysDictData.class); + util.exportExcel(response, list, "字典数据"); + } + + /** + * 查询字典数据详细 + */ + @PreAuthorize("@ss.hasPermi('system:dict:query')") + @GetMapping(value = "/{dictCode}") + public AjaxResult getInfo(@PathVariable Long dictCode) + { + return AjaxResult.success(dictDataService.selectDictDataById(dictCode)); + } + + /** + * 根据字典类型查询字典数据信息 + */ + @GetMapping(value = "/type/{dictType}") + public AjaxResult dictType(@PathVariable String dictType) + { + List data = dictTypeService.selectDictDataByType(dictType); + if (StringUtils.isNull(data)) + { + data = new ArrayList(); + } + return AjaxResult.success(data); + } + + /** + * 新增字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:add')") + @Log(title = "字典数据", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysDictData dict) + { + dict.setCreateBy(getUsername()); + return toAjax(dictDataService.insertDictData(dict)); + } + + /** + * 修改保存字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:edit')") + @Log(title = "字典数据", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysDictData dict) + { + dict.setUpdateBy(getUsername()); + return toAjax(dictDataService.updateDictData(dict)); + } + + /** + * 删除字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:remove')") + @Log(title = "字典类型", businessType = BusinessType.DELETE) + @DeleteMapping("/{dictCodes}") + public AjaxResult remove(@PathVariable Long[] dictCodes) + { + dictDataService.deleteDictDataByIds(dictCodes); + return success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictTypeController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictTypeController.java new file mode 100644 index 0000000..f8b7dbc --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictTypeController.java @@ -0,0 +1,132 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysDictType; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.service.ISysDictTypeService; + +/** + * 数据字典信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/dict/type") +public class SysDictTypeController extends BaseController +{ + @Autowired + private ISysDictTypeService dictTypeService; + + @PreAuthorize("@ss.hasPermi('system:dict:list')") + @GetMapping("/list") + public TableDataInfo list(SysDictType dictType) + { + startPage(); + List list = dictTypeService.selectDictTypeList(dictType); + return getDataTable(list); + } + + @Log(title = "字典类型", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:dict:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysDictType dictType) + { + List list = dictTypeService.selectDictTypeList(dictType); + ExcelUtil util = new ExcelUtil(SysDictType.class); + util.exportExcel(response, list, "字典类型"); + } + + /** + * 查询字典类型详细 + */ + @PreAuthorize("@ss.hasPermi('system:dict:query')") + @GetMapping(value = "/{dictId}") + public AjaxResult getInfo(@PathVariable Long dictId) + { + return AjaxResult.success(dictTypeService.selectDictTypeById(dictId)); + } + + /** + * 新增字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:add')") + @Log(title = "字典类型", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysDictType dict) + { + if (UserConstants.NOT_UNIQUE.equals(dictTypeService.checkDictTypeUnique(dict))) + { + return AjaxResult.error("新增字典'" + dict.getDictName() + "'失败,字典类型已存在"); + } + dict.setCreateBy(getUsername()); + return toAjax(dictTypeService.insertDictType(dict)); + } + + /** + * 修改字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:edit')") + @Log(title = "字典类型", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysDictType dict) + { + if (UserConstants.NOT_UNIQUE.equals(dictTypeService.checkDictTypeUnique(dict))) + { + return AjaxResult.error("修改字典'" + dict.getDictName() + "'失败,字典类型已存在"); + } + dict.setUpdateBy(getUsername()); + return toAjax(dictTypeService.updateDictType(dict)); + } + + /** + * 删除字典类型 + */ + @PreAuthorize("@ss.hasPermi('system:dict:remove')") + @Log(title = "字典类型", businessType = BusinessType.DELETE) + @DeleteMapping("/{dictIds}") + public AjaxResult remove(@PathVariable Long[] dictIds) + { + dictTypeService.deleteDictTypeByIds(dictIds); + return success(); + } + + /** + * 刷新字典缓存 + */ + @PreAuthorize("@ss.hasPermi('system:dict:remove')") + @Log(title = "字典类型", businessType = BusinessType.CLEAN) + @DeleteMapping("/refreshCache") + public AjaxResult refreshCache() + { + dictTypeService.resetDictCache(); + return AjaxResult.success(); + } + + /** + * 获取字典选择框列表 + */ + @GetMapping("/optionselect") + public AjaxResult optionselect() + { + List dictTypes = dictTypeService.selectDictTypeAll(); + return AjaxResult.success(dictTypes); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysIndexController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysIndexController.java new file mode 100644 index 0000000..13007eb --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysIndexController.java @@ -0,0 +1,29 @@ +package com.ruoyi.web.controller.system; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.utils.StringUtils; + +/** + * 首页 + * + * @author ruoyi + */ +@RestController +public class SysIndexController +{ + /** 系统基础配置 */ + @Autowired + private RuoYiConfig ruoyiConfig; + + /** + * 访问首页,提示语 + */ + @RequestMapping("/") + public String index() + { + return StringUtils.format("欢迎使用{}后台管理框架,当前版本:v{},请通过前端地址访问。", ruoyiConfig.getName(), ruoyiConfig.getVersion()); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java new file mode 100644 index 0000000..1a80755 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java @@ -0,0 +1,89 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import java.util.Set; + +import com.ruoyi.common.utils.sign.RsaUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysMenu; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginBody; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.framework.web.service.SysLoginService; +import com.ruoyi.framework.web.service.SysPermissionService; +import com.ruoyi.system.service.ISysMenuService; + +/** + * 登录验证 + * + * @author ruoyi + */ +@RestController +public class SysLoginController +{ + @Autowired + private SysLoginService loginService; + + @Autowired + private ISysMenuService menuService; + + @Autowired + private SysPermissionService permissionService; + + /** + * 登录方法 + * + * @param loginBody 登录信息 + * @return 结果 + */ + @PostMapping("/login") + public AjaxResult login(@RequestBody LoginBody loginBody) throws Exception { + AjaxResult ajax = AjaxResult.success(); + // 生成令牌 + String token = loginService.login(loginBody.getUsername(), + RsaUtils.decryptByPrivateKey(loginBody.getPassword()), + loginBody.getCode(), + loginBody.getUuid()); + ajax.put(Constants.TOKEN, token); + return ajax; + } + + /** + * 获取用户信息 + * + * @return 用户信息 + */ + @GetMapping("getInfo") + public AjaxResult getInfo() + { + SysUser user = SecurityUtils.getLoginUser().getUser(); + // 角色集合 + Set roles = permissionService.getRolePermission(user); + // 权限集合 + Set permissions = permissionService.getMenuPermission(user); + AjaxResult ajax = AjaxResult.success(); + ajax.put("user", user); + ajax.put("roles", roles); + ajax.put("permissions", permissions); + return ajax; + } + + /** + * 获取路由信息 + * + * @return 路由信息 + */ + @GetMapping("getRouters") + public AjaxResult getRouters() + { + Long userId = SecurityUtils.getUserId(); + List menus = menuService.selectMenuTreeByUserId(userId); + return AjaxResult.success(menuService.buildMenus(menus)); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysMenuController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysMenuController.java new file mode 100644 index 0000000..4a9f73f --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysMenuController.java @@ -0,0 +1,142 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysMenu; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.service.ISysMenuService; + +/** + * 菜单信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/menu") +public class SysMenuController extends BaseController +{ + @Autowired + private ISysMenuService menuService; + + /** + * 获取菜单列表 + */ + @PreAuthorize("@ss.hasPermi('system:menu:list')") + @GetMapping("/list") + public AjaxResult list(SysMenu menu) + { + List menus = menuService.selectMenuList(menu, getUserId()); + return AjaxResult.success(menus); + } + + /** + * 根据菜单编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:menu:query')") + @GetMapping(value = "/{menuId}") + public AjaxResult getInfo(@PathVariable Long menuId) + { + return AjaxResult.success(menuService.selectMenuById(menuId)); + } + + /** + * 获取菜单下拉树列表 + */ + @GetMapping("/treeselect") + public AjaxResult treeselect(SysMenu menu) + { + List menus = menuService.selectMenuList(menu, getUserId()); + return AjaxResult.success(menuService.buildMenuTreeSelect(menus)); + } + + /** + * 加载对应角色菜单列表树 + */ + @GetMapping(value = "/roleMenuTreeselect/{roleId}") + public AjaxResult roleMenuTreeselect(@PathVariable("roleId") Long roleId) + { + List menus = menuService.selectMenuList(getUserId()); + AjaxResult ajax = AjaxResult.success(); + ajax.put("checkedKeys", menuService.selectMenuListByRoleId(roleId)); + ajax.put("menus", menuService.buildMenuTreeSelect(menus)); + return ajax; + } + + /** + * 新增菜单 + */ + @PreAuthorize("@ss.hasPermi('system:menu:add')") + @Log(title = "菜单管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysMenu menu) + { + if (UserConstants.NOT_UNIQUE.equals(menuService.checkMenuNameUnique(menu))) + { + return AjaxResult.error("新增菜单'" + menu.getMenuName() + "'失败,菜单名称已存在"); + } + else if (UserConstants.YES_FRAME.equals(menu.getIsFrame()) && !StringUtils.ishttp(menu.getPath())) + { + return AjaxResult.error("新增菜单'" + menu.getMenuName() + "'失败,地址必须以http(s)://开头"); + } + menu.setCreateBy(getUsername()); + return toAjax(menuService.insertMenu(menu)); + } + + /** + * 修改菜单 + */ + @PreAuthorize("@ss.hasPermi('system:menu:edit')") + @Log(title = "菜单管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysMenu menu) + { + if (UserConstants.NOT_UNIQUE.equals(menuService.checkMenuNameUnique(menu))) + { + return AjaxResult.error("修改菜单'" + menu.getMenuName() + "'失败,菜单名称已存在"); + } + else if (UserConstants.YES_FRAME.equals(menu.getIsFrame()) && !StringUtils.ishttp(menu.getPath())) + { + return AjaxResult.error("修改菜单'" + menu.getMenuName() + "'失败,地址必须以http(s)://开头"); + } + else if (menu.getMenuId().equals(menu.getParentId())) + { + return AjaxResult.error("修改菜单'" + menu.getMenuName() + "'失败,上级菜单不能选择自己"); + } + menu.setUpdateBy(getUsername()); + return toAjax(menuService.updateMenu(menu)); + } + + /** + * 删除菜单 + */ + @PreAuthorize("@ss.hasPermi('system:menu:remove')") + @Log(title = "菜单管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{menuId}") + public AjaxResult remove(@PathVariable("menuId") Long menuId) + { + if (menuService.hasChildByMenuId(menuId)) + { + return AjaxResult.error("存在子菜单,不允许删除"); + } + if (menuService.checkMenuExistRole(menuId)) + { + return AjaxResult.error("菜单已分配,不允许删除"); + } + return toAjax(menuService.deleteMenuById(menuId)); + } +} \ No newline at end of file diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysNoticeController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysNoticeController.java new file mode 100644 index 0000000..4da9f04 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysNoticeController.java @@ -0,0 +1,91 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.system.domain.SysNotice; +import com.ruoyi.system.service.ISysNoticeService; + +/** + * 公告 信息操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/notice") +public class SysNoticeController extends BaseController +{ + @Autowired + private ISysNoticeService noticeService; + + /** + * 获取通知公告列表 + */ + @PreAuthorize("@ss.hasPermi('system:notice:list')") + @GetMapping("/list") + public TableDataInfo list(SysNotice notice) + { + startPage(); + List list = noticeService.selectNoticeList(notice); + return getDataTable(list); + } + + /** + * 根据通知公告编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:notice:query')") + @GetMapping(value = "/{noticeId}") + public AjaxResult getInfo(@PathVariable Long noticeId) + { + return AjaxResult.success(noticeService.selectNoticeById(noticeId)); + } + + /** + * 新增通知公告 + */ + @PreAuthorize("@ss.hasPermi('system:notice:add')") + @Log(title = "通知公告", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysNotice notice) + { + notice.setCreateBy(getUsername()); + return toAjax(noticeService.insertNotice(notice)); + } + + /** + * 修改通知公告 + */ + @PreAuthorize("@ss.hasPermi('system:notice:edit')") + @Log(title = "通知公告", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysNotice notice) + { + notice.setUpdateBy(getUsername()); + return toAjax(noticeService.updateNotice(notice)); + } + + /** + * 删除通知公告 + */ + @PreAuthorize("@ss.hasPermi('system:notice:remove')") + @Log(title = "通知公告", businessType = BusinessType.DELETE) + @DeleteMapping("/{noticeIds}") + public AjaxResult remove(@PathVariable Long[] noticeIds) + { + return toAjax(noticeService.deleteNoticeByIds(noticeIds)); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysPostController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysPostController.java new file mode 100644 index 0000000..f73cea4 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysPostController.java @@ -0,0 +1,130 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.domain.SysPost; +import com.ruoyi.system.service.ISysPostService; + +/** + * 岗位信息操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/post") +public class SysPostController extends BaseController +{ + @Autowired + private ISysPostService postService; + + /** + * 获取岗位列表 + */ + @PreAuthorize("@ss.hasPermi('system:post:list')") + @GetMapping("/list") + public TableDataInfo list(SysPost post) + { + startPage(); + List list = postService.selectPostList(post); + return getDataTable(list); + } + + @Log(title = "岗位管理", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:post:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysPost post) + { + List list = postService.selectPostList(post); + ExcelUtil util = new ExcelUtil(SysPost.class); + util.exportExcel(response, list, "岗位数据"); + } + + /** + * 根据岗位编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:post:query')") + @GetMapping(value = "/{postId}") + public AjaxResult getInfo(@PathVariable Long postId) + { + return AjaxResult.success(postService.selectPostById(postId)); + } + + /** + * 新增岗位 + */ + @PreAuthorize("@ss.hasPermi('system:post:add')") + @Log(title = "岗位管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysPost post) + { + if (UserConstants.NOT_UNIQUE.equals(postService.checkPostNameUnique(post))) + { + return AjaxResult.error("新增岗位'" + post.getPostName() + "'失败,岗位名称已存在"); + } + else if (UserConstants.NOT_UNIQUE.equals(postService.checkPostCodeUnique(post))) + { + return AjaxResult.error("新增岗位'" + post.getPostName() + "'失败,岗位编码已存在"); + } + post.setCreateBy(getUsername()); + return toAjax(postService.insertPost(post)); + } + + /** + * 修改岗位 + */ + @PreAuthorize("@ss.hasPermi('system:post:edit')") + @Log(title = "岗位管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysPost post) + { + if (UserConstants.NOT_UNIQUE.equals(postService.checkPostNameUnique(post))) + { + return AjaxResult.error("修改岗位'" + post.getPostName() + "'失败,岗位名称已存在"); + } + else if (UserConstants.NOT_UNIQUE.equals(postService.checkPostCodeUnique(post))) + { + return AjaxResult.error("修改岗位'" + post.getPostName() + "'失败,岗位编码已存在"); + } + post.setUpdateBy(getUsername()); + return toAjax(postService.updatePost(post)); + } + + /** + * 删除岗位 + */ + @PreAuthorize("@ss.hasPermi('system:post:remove')") + @Log(title = "岗位管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{postIds}") + public AjaxResult remove(@PathVariable Long[] postIds) + { + return toAjax(postService.deletePostByIds(postIds)); + } + + /** + * 获取岗位选择框列表 + */ + @GetMapping("/optionselect") + public AjaxResult optionselect() + { + List posts = postService.selectPostAll(); + return AjaxResult.success(posts); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java new file mode 100644 index 0000000..ba38bd0 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java @@ -0,0 +1,142 @@ +package com.ruoyi.web.controller.system; + +import java.io.IOException; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.file.FileUploadUtils; +import com.ruoyi.framework.web.service.TokenService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 个人信息 业务处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/user/profile") +public class SysProfileController extends BaseController +{ + @Autowired + private ISysUserService userService; + + @Autowired + private TokenService tokenService; + + /** + * 个人信息 + */ + @GetMapping + public AjaxResult profile() + { + LoginUser loginUser = getLoginUser(); + SysUser user = loginUser.getUser(); + AjaxResult ajax = AjaxResult.success(user); + ajax.put("roleGroup", userService.selectUserRoleGroup(loginUser.getUsername())); + ajax.put("postGroup", userService.selectUserPostGroup(loginUser.getUsername())); + return ajax; + } + + /** + * 修改用户 + */ + @Log(title = "个人信息", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult updateProfile(@RequestBody SysUser user) + { + LoginUser loginUser = getLoginUser(); + SysUser sysUser = loginUser.getUser(); + user.setUserName(sysUser.getUserName()); + if (StringUtils.isNotEmpty(user.getPhonenumber()) + && UserConstants.NOT_UNIQUE.equals(userService.checkPhoneUnique(user))) + { + return AjaxResult.error("修改用户'" + user.getUserName() + "'失败,手机号码已存在"); + } + if (StringUtils.isNotEmpty(user.getEmail()) + && UserConstants.NOT_UNIQUE.equals(userService.checkEmailUnique(user))) + { + return AjaxResult.error("修改用户'" + user.getUserName() + "'失败,邮箱账号已存在"); + } + user.setUserId(sysUser.getUserId()); + user.setPassword(null); + if (userService.updateUserProfile(user) > 0) + { + // 更新缓存用户信息 + sysUser.setNickName(user.getNickName()); + sysUser.setPhonenumber(user.getPhonenumber()); + sysUser.setEmail(user.getEmail()); + sysUser.setSex(user.getSex()); + tokenService.setLoginUser(loginUser); + return AjaxResult.success(); + } + return AjaxResult.error("修改个人信息异常,请联系管理员"); + } + + /** + * 重置密码 + */ + @Log(title = "个人信息", businessType = BusinessType.UPDATE) + @PutMapping("/updatePwd") + public AjaxResult updatePwd(String oldPassword, String newPassword) + { + LoginUser loginUser = getLoginUser(); + String userName = loginUser.getUsername(); + String password = loginUser.getPassword(); + if (!SecurityUtils.matchesPassword(oldPassword, password)) + { + return AjaxResult.error("修改密码失败,旧密码错误"); + } + if (SecurityUtils.matchesPassword(newPassword, password)) + { + return AjaxResult.error("新密码不能与旧密码相同"); + } + if (userService.resetUserPwd(userName, SecurityUtils.encryptPassword(newPassword)) > 0) + { + // 更新缓存用户密码 + loginUser.getUser().setPassword(SecurityUtils.encryptPassword(newPassword)); + tokenService.setLoginUser(loginUser); + return AjaxResult.success(); + } + return AjaxResult.error("修改密码异常,请联系管理员"); + } + + /** + * 头像上传 + */ + @Log(title = "用户头像", businessType = BusinessType.UPDATE) + @PostMapping("/avatar") + public AjaxResult avatar(@RequestParam("avatarfile") MultipartFile file) throws IOException + { + if (!file.isEmpty()) + { + LoginUser loginUser = getLoginUser(); + String avatar = FileUploadUtils.upload(RuoYiConfig.getAvatarPath(), file); + if (userService.updateUserAvatar(loginUser.getUsername(), avatar)) + { + AjaxResult ajax = AjaxResult.success(); + ajax.put("imgUrl", avatar); + // 更新缓存用户头像 + loginUser.getUser().setAvatar(avatar); + tokenService.setLoginUser(loginUser); + return ajax; + } + } + return AjaxResult.error("上传图片异常,请联系管理员"); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRegisterController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRegisterController.java new file mode 100644 index 0000000..fe19249 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRegisterController.java @@ -0,0 +1,38 @@ +package com.ruoyi.web.controller.system; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.model.RegisterBody; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.web.service.SysRegisterService; +import com.ruoyi.system.service.ISysConfigService; + +/** + * 注册验证 + * + * @author ruoyi + */ +@RestController +public class SysRegisterController extends BaseController +{ + @Autowired + private SysRegisterService registerService; + + @Autowired + private ISysConfigService configService; + + @PostMapping("/register") + public AjaxResult register(@RequestBody RegisterBody user) + { + if (!("true".equals(configService.selectConfigByKey("sys.account.registerUser")))) + { + return error("当前系统没有开启注册功能!"); + } + String msg = registerService.register(user); + return StringUtils.isEmpty(msg) ? success() : error(msg); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRoleController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRoleController.java new file mode 100644 index 0000000..2abec1b --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRoleController.java @@ -0,0 +1,245 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.framework.web.service.SysPermissionService; +import com.ruoyi.framework.web.service.TokenService; +import com.ruoyi.system.domain.SysUserRole; +import com.ruoyi.system.service.ISysRoleService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 角色信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/role") +public class SysRoleController extends BaseController +{ + @Autowired + private ISysRoleService roleService; + + @Autowired + private TokenService tokenService; + + @Autowired + private SysPermissionService permissionService; + + @Autowired + private ISysUserService userService; + + @PreAuthorize("@ss.hasPermi('system:role:list')") + @GetMapping("/list") + public TableDataInfo list(SysRole role) + { + startPage(); + List list = roleService.selectRoleList(role); + return getDataTable(list); + } + + @Log(title = "角色管理", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:role:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysRole role) + { + List list = roleService.selectRoleList(role); + ExcelUtil util = new ExcelUtil(SysRole.class); + util.exportExcel(response, list, "角色数据"); + } + + /** + * 根据角色编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:role:query')") + @GetMapping(value = "/{roleId}") + public AjaxResult getInfo(@PathVariable Long roleId) + { + roleService.checkRoleDataScope(roleId); + return AjaxResult.success(roleService.selectRoleById(roleId)); + } + + /** + * 新增角色 + */ + @PreAuthorize("@ss.hasPermi('system:role:add')") + @Log(title = "角色管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysRole role) + { + if (UserConstants.NOT_UNIQUE.equals(roleService.checkRoleNameUnique(role))) + { + return AjaxResult.error("新增角色'" + role.getRoleName() + "'失败,角色名称已存在"); + } + else if (UserConstants.NOT_UNIQUE.equals(roleService.checkRoleKeyUnique(role))) + { + return AjaxResult.error("新增角色'" + role.getRoleName() + "'失败,角色权限已存在"); + } + role.setCreateBy(getUsername()); + return toAjax(roleService.insertRole(role)); + + } + + /** + * 修改保存角色 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysRole role) + { + roleService.checkRoleAllowed(role); + roleService.checkRoleDataScope(role.getRoleId()); + if (UserConstants.NOT_UNIQUE.equals(roleService.checkRoleNameUnique(role))) + { + return AjaxResult.error("修改角色'" + role.getRoleName() + "'失败,角色名称已存在"); + } + else if (UserConstants.NOT_UNIQUE.equals(roleService.checkRoleKeyUnique(role))) + { + return AjaxResult.error("修改角色'" + role.getRoleName() + "'失败,角色权限已存在"); + } + role.setUpdateBy(getUsername()); + + if (roleService.updateRole(role) > 0) + { + // 更新缓存用户权限 + LoginUser loginUser = getLoginUser(); + if (StringUtils.isNotNull(loginUser.getUser()) && !loginUser.getUser().isAdmin()) + { + loginUser.setPermissions(permissionService.getMenuPermission(loginUser.getUser())); + loginUser.setUser(userService.selectUserByUserName(loginUser.getUser().getUserName())); + tokenService.setLoginUser(loginUser); + } + return AjaxResult.success(); + } + return AjaxResult.error("修改角色'" + role.getRoleName() + "'失败,请联系管理员"); + } + + /** + * 修改保存数据权限 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.UPDATE) + @PutMapping("/dataScope") + public AjaxResult dataScope(@RequestBody SysRole role) + { + roleService.checkRoleAllowed(role); + roleService.checkRoleDataScope(role.getRoleId()); + return toAjax(roleService.authDataScope(role)); + } + + /** + * 状态修改 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.UPDATE) + @PutMapping("/changeStatus") + public AjaxResult changeStatus(@RequestBody SysRole role) + { + roleService.checkRoleAllowed(role); + roleService.checkRoleDataScope(role.getRoleId()); + role.setUpdateBy(getUsername()); + return toAjax(roleService.updateRoleStatus(role)); + } + + /** + * 删除角色 + */ + @PreAuthorize("@ss.hasPermi('system:role:remove')") + @Log(title = "角色管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{roleIds}") + public AjaxResult remove(@PathVariable Long[] roleIds) + { + return toAjax(roleService.deleteRoleByIds(roleIds)); + } + + /** + * 获取角色选择框列表 + */ + @PreAuthorize("@ss.hasPermi('system:role:query')") + @GetMapping("/optionselect") + public AjaxResult optionselect() + { + return AjaxResult.success(roleService.selectRoleAll()); + } + + /** + * 查询已分配用户角色列表 + */ + @PreAuthorize("@ss.hasPermi('system:role:list')") + @GetMapping("/authUser/allocatedList") + public TableDataInfo allocatedList(SysUser user) + { + startPage(); + List list = userService.selectAllocatedList(user); + return getDataTable(list); + } + + /** + * 查询未分配用户角色列表 + */ + @PreAuthorize("@ss.hasPermi('system:role:list')") + @GetMapping("/authUser/unallocatedList") + public TableDataInfo unallocatedList(SysUser user) + { + startPage(); + List list = userService.selectUnallocatedList(user); + return getDataTable(list); + } + + /** + * 取消授权用户 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.GRANT) + @PutMapping("/authUser/cancel") + public AjaxResult cancelAuthUser(@RequestBody SysUserRole userRole) + { + return toAjax(roleService.deleteAuthUser(userRole)); + } + + /** + * 批量取消授权用户 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.GRANT) + @PutMapping("/authUser/cancelAll") + public AjaxResult cancelAuthUserAll(Long roleId, Long[] userIds) + { + return toAjax(roleService.deleteAuthUsers(roleId, userIds)); + } + + /** + * 批量选择用户授权 + */ + @PreAuthorize("@ss.hasPermi('system:role:edit')") + @Log(title = "角色管理", businessType = BusinessType.GRANT) + @PutMapping("/authUser/selectAll") + public AjaxResult selectAuthUserAll(Long roleId, Long[] userIds) + { + roleService.checkRoleDataScope(roleId); + return toAjax(roleService.insertAuthUsers(roleId, userIds)); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysUserController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysUserController.java new file mode 100644 index 0000000..089f711 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysUserController.java @@ -0,0 +1,237 @@ +package com.ruoyi.web.controller.system; + +import java.util.List; +import java.util.stream.Collectors; +import javax.servlet.http.HttpServletResponse; +import org.apache.commons.lang3.ArrayUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.system.service.ISysPostService; +import com.ruoyi.system.service.ISysRoleService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 用户信息 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/system/user") +public class SysUserController extends BaseController +{ + @Autowired + private ISysUserService userService; + + @Autowired + private ISysRoleService roleService; + + @Autowired + private ISysPostService postService; + + /** + * 获取用户列表 + */ + @PreAuthorize("@ss.hasPermi('system:user:list')") + @GetMapping("/list") + public TableDataInfo list(SysUser user) + { + startPage(); + List list = userService.selectUserList(user); + return getDataTable(list); + } + + @Log(title = "用户管理", businessType = BusinessType.EXPORT) + @PreAuthorize("@ss.hasPermi('system:user:export')") + @PostMapping("/export") + public void export(HttpServletResponse response, SysUser user) + { + List list = userService.selectUserList(user); + ExcelUtil util = new ExcelUtil(SysUser.class); + util.exportExcel(response, list, "用户数据"); + } + + @Log(title = "用户管理", businessType = BusinessType.IMPORT) + @PreAuthorize("@ss.hasPermi('system:user:import')") + @PostMapping("/importData") + public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception + { + ExcelUtil util = new ExcelUtil(SysUser.class); + List userList = util.importExcel(file.getInputStream()); + String operName = getUsername(); + String message = userService.importUser(userList, updateSupport, operName); + return AjaxResult.success(message); + } + + @PostMapping("/importTemplate") + public void importTemplate(HttpServletResponse response) + { + ExcelUtil util = new ExcelUtil(SysUser.class); + util.importTemplateExcel(response, "用户数据"); + } + + /** + * 根据用户编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('system:user:query')") + @GetMapping(value = { "/", "/{userId}" }) + public AjaxResult getInfo(@PathVariable(value = "userId", required = false) Long userId) + { + userService.checkUserDataScope(userId); + AjaxResult ajax = AjaxResult.success(); + List roles = roleService.selectRoleAll(); + ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); + ajax.put("posts", postService.selectPostAll()); + if (StringUtils.isNotNull(userId)) + { + SysUser sysUser = userService.selectUserById(userId); + ajax.put(AjaxResult.DATA_TAG, sysUser); + ajax.put("postIds", postService.selectPostListByUserId(userId)); + ajax.put("roleIds", sysUser.getRoles().stream().map(SysRole::getRoleId).collect(Collectors.toList())); + } + return ajax; + } + + /** + * 新增用户 + */ + @PreAuthorize("@ss.hasPermi('system:user:add')") + @Log(title = "用户管理", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@Validated @RequestBody SysUser user) + { + if (UserConstants.NOT_UNIQUE.equals(userService.checkUserNameUnique(user.getUserName()))) + { + return AjaxResult.error("新增用户'" + user.getUserName() + "'失败,登录账号已存在"); + } + else if (StringUtils.isNotEmpty(user.getPhonenumber()) + && UserConstants.NOT_UNIQUE.equals(userService.checkPhoneUnique(user))) + { + return AjaxResult.error("新增用户'" + user.getUserName() + "'失败,手机号码已存在"); + } + else if (StringUtils.isNotEmpty(user.getEmail()) + && UserConstants.NOT_UNIQUE.equals(userService.checkEmailUnique(user))) + { + return AjaxResult.error("新增用户'" + user.getUserName() + "'失败,邮箱账号已存在"); + } + user.setCreateBy(getUsername()); + user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); + return toAjax(userService.insertUser(user)); + } + + /** + * 修改用户 + */ + @PreAuthorize("@ss.hasPermi('system:user:edit')") + @Log(title = "用户管理", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@Validated @RequestBody SysUser user) + { + userService.checkUserAllowed(user); + userService.checkUserDataScope(user.getUserId()); + if (StringUtils.isNotEmpty(user.getPhonenumber()) + && UserConstants.NOT_UNIQUE.equals(userService.checkPhoneUnique(user))) + { + return AjaxResult.error("修改用户'" + user.getUserName() + "'失败,手机号码已存在"); + } + else if (StringUtils.isNotEmpty(user.getEmail()) + && UserConstants.NOT_UNIQUE.equals(userService.checkEmailUnique(user))) + { + return AjaxResult.error("修改用户'" + user.getUserName() + "'失败,邮箱账号已存在"); + } + user.setUpdateBy(getUsername()); + return toAjax(userService.updateUser(user)); + } + + /** + * 删除用户 + */ + @PreAuthorize("@ss.hasPermi('system:user:remove')") + @Log(title = "用户管理", businessType = BusinessType.DELETE) + @DeleteMapping("/{userIds}") + public AjaxResult remove(@PathVariable Long[] userIds) + { + if (ArrayUtils.contains(userIds, getUserId())) + { + return error("当前用户不能删除"); + } + return toAjax(userService.deleteUserByIds(userIds)); + } + + /** + * 重置密码 + */ + @PreAuthorize("@ss.hasPermi('system:user:resetPwd')") + @Log(title = "用户管理", businessType = BusinessType.UPDATE) + @PutMapping("/resetPwd") + public AjaxResult resetPwd(@RequestBody SysUser user) + { + userService.checkUserAllowed(user); + userService.checkUserDataScope(user.getUserId()); + user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); + user.setUpdateBy(getUsername()); + return toAjax(userService.resetPwd(user)); + } + + /** + * 状态修改 + */ + @PreAuthorize("@ss.hasPermi('system:user:edit')") + @Log(title = "用户管理", businessType = BusinessType.UPDATE) + @PutMapping("/changeStatus") + public AjaxResult changeStatus(@RequestBody SysUser user) + { + userService.checkUserAllowed(user); + userService.checkUserDataScope(user.getUserId()); + user.setUpdateBy(getUsername()); + return toAjax(userService.updateUserStatus(user)); + } + + /** + * 根据用户编号获取授权角色 + */ + @PreAuthorize("@ss.hasPermi('system:user:query')") + @GetMapping("/authRole/{userId}") + public AjaxResult authRole(@PathVariable("userId") Long userId) + { + AjaxResult ajax = AjaxResult.success(); + SysUser user = userService.selectUserById(userId); + List roles = roleService.selectRolesByUserId(userId); + ajax.put("user", user); + ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); + return ajax; + } + + /** + * 用户授权角色 + */ + @PreAuthorize("@ss.hasPermi('system:user:edit')") + @Log(title = "用户管理", businessType = BusinessType.GRANT) + @PutMapping("/authRole") + public AjaxResult insertAuthRole(Long userId, Long[] roleIds) + { + userService.checkUserDataScope(userId); + userService.insertUserAuth(userId, roleIds); + return success(); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/SwaggerController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/SwaggerController.java new file mode 100644 index 0000000..f66ca24 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/SwaggerController.java @@ -0,0 +1,24 @@ +package com.ruoyi.web.controller.tool; + +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.stereotype.Controller; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import com.ruoyi.common.core.controller.BaseController; + +/** + * swagger 接口 + * + * @author ruoyi + */ +@Controller +@RequestMapping("/tool/swagger") +public class SwaggerController extends BaseController +{ + @PreAuthorize("@ss.hasPermi('tool:swagger:view')") + @GetMapping() + public String index() + { + return redirect("/swagger-ui.html"); + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TestController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TestController.java new file mode 100644 index 0000000..3361ef9 --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TestController.java @@ -0,0 +1,181 @@ +package com.ruoyi.web.controller.tool; + +import java.util.ArrayList; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.utils.StringUtils; +import io.swagger.annotations.Api; +import io.swagger.annotations.ApiImplicitParam; +import io.swagger.annotations.ApiImplicitParams; +import io.swagger.annotations.ApiModel; +import io.swagger.annotations.ApiModelProperty; +import io.swagger.annotations.ApiOperation; + +/** + * swagger 用户测试方法 + * + * @author ruoyi + */ +@Api("用户信息管理") +@RestController +@RequestMapping("/test/user") +public class TestController extends BaseController +{ + private final static Map users = new LinkedHashMap(); + { + users.put(1, new UserEntity(1, "admin", "admin123", "15888888888")); + users.put(2, new UserEntity(2, "ry", "admin123", "15666666666")); + } + + @ApiOperation("获取用户列表") + @GetMapping("/list") + public AjaxResult userList() + { + List userList = new ArrayList(users.values()); + return AjaxResult.success(userList); + } + + @ApiOperation("获取用户详细") + @ApiImplicitParam(name = "userId", value = "用户ID", required = true, dataType = "int", paramType = "path", dataTypeClass = Integer.class) + @GetMapping("/{userId}") + public AjaxResult getUser(@PathVariable Integer userId) + { + if (!users.isEmpty() && users.containsKey(userId)) + { + return AjaxResult.success(users.get(userId)); + } + else + { + return error("用户不存在"); + } + } + + @ApiOperation("新增用户") + @ApiImplicitParams({ + @ApiImplicitParam(name = "userId", value = "用户id", dataType = "Integer", dataTypeClass = Integer.class), + @ApiImplicitParam(name = "username", value = "用户名称", dataType = "String", dataTypeClass = String.class), + @ApiImplicitParam(name = "password", value = "用户密码", dataType = "String", dataTypeClass = String.class), + @ApiImplicitParam(name = "mobile", value = "用户手机", dataType = "String", dataTypeClass = String.class) + }) + @PostMapping("/save") + public AjaxResult save(UserEntity user) + { + if (StringUtils.isNull(user) || StringUtils.isNull(user.getUserId())) + { + return error("用户ID不能为空"); + } + return AjaxResult.success(users.put(user.getUserId(), user)); + } + + @ApiOperation("更新用户") + @PutMapping("/update") + public AjaxResult update(@RequestBody UserEntity user) + { + if (StringUtils.isNull(user) || StringUtils.isNull(user.getUserId())) + { + return error("用户ID不能为空"); + } + if (users.isEmpty() || !users.containsKey(user.getUserId())) + { + return error("用户不存在"); + } + users.remove(user.getUserId()); + return AjaxResult.success(users.put(user.getUserId(), user)); + } + + @ApiOperation("删除用户信息") + @ApiImplicitParam(name = "userId", value = "用户ID", required = true, dataType = "int", paramType = "path", dataTypeClass = Integer.class) + @DeleteMapping("/{userId}") + public AjaxResult delete(@PathVariable Integer userId) + { + if (!users.isEmpty() && users.containsKey(userId)) + { + users.remove(userId); + return success(); + } + else + { + return error("用户不存在"); + } + } +} + +@ApiModel(value = "UserEntity", description = "用户实体") +class UserEntity +{ + @ApiModelProperty("用户ID") + private Integer userId; + + @ApiModelProperty("用户名称") + private String username; + + @ApiModelProperty("用户密码") + private String password; + + @ApiModelProperty("用户手机") + private String mobile; + + public UserEntity() + { + + } + + public UserEntity(Integer userId, String username, String password, String mobile) + { + this.userId = userId; + this.username = username; + this.password = password; + this.mobile = mobile; + } + + public Integer getUserId() + { + return userId; + } + + public void setUserId(Integer userId) + { + this.userId = userId; + } + + public String getUsername() + { + return username; + } + + public void setUsername(String username) + { + this.username = username; + } + + public String getPassword() + { + return password; + } + + public void setPassword(String password) + { + this.password = password; + } + + public String getMobile() + { + return mobile; + } + + public void setMobile(String mobile) + { + this.mobile = mobile; + } +} diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/core/config/SwaggerConfig.java b/ruoyi-admin/src/main/java/com/ruoyi/web/core/config/SwaggerConfig.java new file mode 100644 index 0000000..ae1c3ec --- /dev/null +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/core/config/SwaggerConfig.java @@ -0,0 +1,125 @@ +package com.ruoyi.web.core.config; + +import java.util.ArrayList; +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import com.ruoyi.common.config.RuoYiConfig; +import io.swagger.annotations.ApiOperation; +import io.swagger.models.auth.In; +import springfox.documentation.builders.ApiInfoBuilder; +import springfox.documentation.builders.PathSelectors; +import springfox.documentation.builders.RequestHandlerSelectors; +import springfox.documentation.service.ApiInfo; +import springfox.documentation.service.ApiKey; +import springfox.documentation.service.AuthorizationScope; +import springfox.documentation.service.Contact; +import springfox.documentation.service.SecurityReference; +import springfox.documentation.service.SecurityScheme; +import springfox.documentation.spi.DocumentationType; +import springfox.documentation.spi.service.contexts.SecurityContext; +import springfox.documentation.spring.web.plugins.Docket; + +/** + * Swagger2的接口配置 + * + * @author ruoyi + */ +@Configuration +public class SwaggerConfig +{ + /** 系统基础配置 */ + @Autowired + private RuoYiConfig ruoyiConfig; + + /** 是否开启swagger */ + @Value("${swagger.enabled}") + private boolean enabled; + + /** 设置请求的统一前缀 */ + @Value("${swagger.pathMapping}") + private String pathMapping; + + /** + * 创建API + */ + @Bean + public Docket createRestApi() + { + return new Docket(DocumentationType.OAS_30) + // 是否启用Swagger + .enable(enabled) + // 用来创建该API的基本信息,展示在文档的页面中(自定义展示的信息) + .apiInfo(apiInfo()) + // 设置哪些接口暴露给Swagger展示 + .select() + // 扫描所有有注解的api,用这种方式更灵活 + .apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class)) + // 扫描指定包中的swagger注解 + // .apis(RequestHandlerSelectors.basePackage("com.ruoyi.project.tool.swagger")) + // 扫描所有 .apis(RequestHandlerSelectors.any()) + .paths(PathSelectors.any()) + .build() + /* 设置安全模式,swagger可以设置访问token */ + .securitySchemes(securitySchemes()) + .securityContexts(securityContexts()) + .pathMapping(pathMapping); + } + + /** + * 安全模式,这里指定token通过Authorization头请求头传递 + */ + private List securitySchemes() + { + List apiKeyList = new ArrayList(); + apiKeyList.add(new ApiKey("Authorization", "Authorization", In.HEADER.toValue())); + return apiKeyList; + } + + /** + * 安全上下文 + */ + private List securityContexts() + { + List securityContexts = new ArrayList<>(); + securityContexts.add( + SecurityContext.builder() + .securityReferences(defaultAuth()) + .operationSelector(o -> o.requestMappingPattern().matches("/.*")) + .build()); + return securityContexts; + } + + /** + * 默认的安全上引用 + */ + private List defaultAuth() + { + AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything"); + AuthorizationScope[] authorizationScopes = new AuthorizationScope[1]; + authorizationScopes[0] = authorizationScope; + List securityReferences = new ArrayList<>(); + securityReferences.add(new SecurityReference("Authorization", authorizationScopes)); + return securityReferences; + } + + /** + * 添加摘要信息 + */ + private ApiInfo apiInfo() + { + // 用ApiInfoBuilder进行定制 + return new ApiInfoBuilder() + // 设置标题 + .title("标题:若依管理系统_接口文档") + // 描述 + .description("描述:用于管理集团旗下公司的人员信息,具体包括XXX,XXX模块...") + // 作者信息 + .contact(new Contact(ruoyiConfig.getName(), null, null)) + // 版本 + .version("版本号:" + ruoyiConfig.getVersion()) + .build(); + } +} diff --git a/ruoyi-admin/src/main/resources/META-INF/spring-devtools.properties b/ruoyi-admin/src/main/resources/META-INF/spring-devtools.properties new file mode 100644 index 0000000..2b23f85 --- /dev/null +++ b/ruoyi-admin/src/main/resources/META-INF/spring-devtools.properties @@ -0,0 +1 @@ +restart.include.json=/com.alibaba.fastjson.*.jar \ No newline at end of file diff --git a/ruoyi-admin/src/main/resources/application-druid.yml b/ruoyi-admin/src/main/resources/application-druid.yml new file mode 100644 index 0000000..290afe7 --- /dev/null +++ b/ruoyi-admin/src/main/resources/application-druid.yml @@ -0,0 +1,57 @@ +# 数据源配置 +spring: + datasource: + type: com.alibaba.druid.pool.DruidDataSource + driverClassName: com.mysql.cj.jdbc.Driver + druid: + # 主库数据源 + master: + url: jdbc:mysql://192.168.16.219:3306/ywpt?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&serverTimezone=Asia/Shanghai&allowMultiQueries=true&allowPublicKeyRetrieval=true + username: ywpt + password: ywPT_2022! + # 从库数据源 + slave: + # 从数据源开关/默认关闭 + enabled: false + url: + username: + password: + # 初始连接数 + initialSize: 5 + # 最小连接池数量 + minIdle: 10 + # 最大连接池数量 + maxActive: 20 + # 配置获取连接等待超时的时间 + maxWait: 60000 + # 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 + timeBetweenEvictionRunsMillis: 60000 + # 配置一个连接在池中最小生存的时间,单位是毫秒 + minEvictableIdleTimeMillis: 300000 + # 配置一个连接在池中最大生存的时间,单位是毫秒 + maxEvictableIdleTimeMillis: 900000 + # 配置检测连接是否有效 + validationQuery: SELECT 1 FROM DUAL + testWhileIdle: true + testOnBorrow: false + testOnReturn: false + webStatFilter: + enabled: true + statViewServlet: + enabled: true + # 设置白名单,不填则允许所有访问 + allow: + url-pattern: /druid/* + # 控制台管理用户名和密码 + login-username: ruoyi + login-password: 123456 + filter: + stat: + enabled: true + # 慢SQL记录 + log-slow-sql: true + slow-sql-millis: 1000 + merge-sql: true + wall: + config: + multi-statement-allow: true diff --git a/ruoyi-admin/src/main/resources/application-test.yml b/ruoyi-admin/src/main/resources/application-test.yml new file mode 100644 index 0000000..b684930 --- /dev/null +++ b/ruoyi-admin/src/main/resources/application-test.yml @@ -0,0 +1,60 @@ +# 数据源配置 +spring: + datasource: + type: com.alibaba.druid.pool.DruidDataSource + druid: + # 主库数据源 + master: + driverClassName: com.mysql.cj.jdbc.Driver + url: jdbc:mysql://192.168.16.219:3306/lscx?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&serverTimezone=Asia/Shanghai&allowMultiQueries=true&allowPublicKeyRetrieval=true&rewriteBatchedStatements=true + username: lscx + password: lscx_2024! + # 从库数据源 + shardsource: + # 从数据源开关/默认关闭 + enabled: true + driverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriver +# url: jdbc:sqlserver://localhost:1433;DatabaseName=softdb;encrypt=false; + url: jdbc:sqlserver://192.168.16.232:1433;DatabaseName=hy_shdp;ncrypt=false; + username: sa +# password: 1q2w#E$R + password: admin + # 初始连接数 + initialSize: 5 + # 最小连接池数量 + minIdle: 10 + # 最大连接池数量 + maxActive: 20 + # 配置获取连接等待超时的时间 + maxWait: 60000 + # 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 + timeBetweenEvictionRunsMillis: 60000 + # 配置一个连接在池中最小生存的时间,单位是毫秒 + minEvictableIdleTimeMillis: 300000 + # 配置一个连接在池中最大生存的时间,单位是毫秒 + maxEvictableIdleTimeMillis: 900000 + # 配置检测连接是否有效 + validationQuery: SELECT 1 + testWhileIdle: true + testOnBorrow: false + testOnReturn: false + webStatFilter: + enabled: true + statViewServlet: + enabled: true + # 设置白名单,不填则允许所有访问 + allow: + url-pattern: /druid/* + # 控制台管理用户名和密码 + login-username: ruoyi + login-password: 123456 + filter: + stat: + enabled: true + # 慢SQL记录 + log-slow-sql: true + slow-sql-millis: 1000 + merge-sql: true + wall: + config: + multi-statement-allow: true diff --git a/ruoyi-admin/src/main/resources/application.yml b/ruoyi-admin/src/main/resources/application.yml new file mode 100644 index 0000000..7dceacb --- /dev/null +++ b/ruoyi-admin/src/main/resources/application.yml @@ -0,0 +1,147 @@ +# 项目相关配置 +ruoyi: + # 名称 + name: RuoYi + # 版本 + version: 3.8.2 + # 版权年份 + copyrightYear: 2022 + # 实例演示开关 + demoEnabled: true + # 文件路径 示例( Windows配置D:/ruoyi/uploadPath,Linux配置 /home/ruoyi/uploadPath) + profile: profile + # 获取ip地址开关 + addressEnabled: false + # 验证码类型 math 数组计算 char 字符验证 + captchaType: math + +# 开发环境配置 +server: + # 服务器的HTTP端口,默认为8080 + port: 8111 + servlet: + # 应用的访问路径 + context-path: /swls + tomcat: + # tomcat的URI编码 + uri-encoding: UTF-8 + # 连接数满后的排队数,默认为100 + accept-count: 1000 + threads: + # tomcat最大线程数,默认为200 + max: 800 + # Tomcat启动初始化的线程数,默认值10 + min-spare: 100 + +# 日志配置 +logging: + level: + com.ruoyi: debug + org.springframework: warn + +# Spring配置 +spring: + # 资源信息 + messages: + # 国际化资源文件路径 + basename: i18n/messages + profiles: +# active: druid + active: test + # 文件上传 + servlet: + multipart: + # 单个文件大小 + max-file-size: 10MB + # 设置总上传的文件大小 + max-request-size: 20MB + jackson: + date-format: yyyy-MM-dd HH:mm:ss + time-zone: GMT+8 + # 服务模块 +# devtools: +# restart: +# # 热部署开关 +# enabled: true + # redis 配置 + redis: + # 地址 + host: 127.0.0.1 + # 端口,默认为6379 + port: 6379 + # 数据库索引 + database: 8 + # 密码 + password: + # 连接超时时间 + timeout: 10s + lettuce: + pool: + # 连接池中的最小空闲连接 + min-idle: 0 + # 连接池中的最大空闲连接 + max-idle: 8 + # 连接池的最大数据库连接数 + max-active: 8 + # #连接池最大阻塞等待时间(使用负值表示没有限制) + max-wait: -1ms + +# token配置 +token: + # 令牌自定义标识 + header: Authorization + # 令牌密钥 + secret: abcdefghijklmnopqrstuvwxyz + # 令牌有效期(默认30分钟) + expireTime: 30 + +# MyBatis-plus配置 +mybatis-plus: + # 搜索指定包别名 + typeAliasesPackage: com.ruoyi.**.domain + # 配置mapper的扫描,找到所有的mapper.xml映射文件 + mapperLocations: classpath*:mapper/**/*Mapper.xml + # 加载全局的配置文件 + configLocation: classpath:mybatis/mybatis-config.xml +# global-config: +# db-config: +# logic-delete-field: flag # 全局逻辑删除的实体字段名(since 3.3.0,配置后可以忽略不配置步骤2) +# logic-delete-value: 1 # 逻辑已删除值(默认为 1) +# logic-not-delete-value: 0 # 逻辑未删除值(默认为 0) + + +# PageHelper分页插件 +pagehelper: + helperDialect: mysql + supportMethodsArguments: true + params: count=countSql + +# Swagger配置 +swagger: + # 是否开启swagger + enabled: true + # 请求前缀 + pathMapping: /dev-api + +# 防止XSS攻击 +xss: + # 过滤开关 + enabled: true + # 排除链接(多个用逗号分隔) + excludes: /system/notice + # 匹配链接 + urlPatterns: /system/*,/monitor/*,/tool/* + +#camera: +# register: +# ip: 192.168.1.11 +## ip: 172.17.0.2 +# port: 8020 +# username: admin +# password: admin123 +localip: 172.16.46.58 +serverip: 121.40.203.197 + + +ssoip: localhost +ssoport: 9000 diff --git a/ruoyi-admin/src/main/resources/banner.txt b/ruoyi-admin/src/main/resources/banner.txt new file mode 100644 index 0000000..0931cb8 --- /dev/null +++ b/ruoyi-admin/src/main/resources/banner.txt @@ -0,0 +1,24 @@ +Application Version: ${ruoyi.version} +Spring Boot Version: ${spring-boot.version} +//////////////////////////////////////////////////////////////////// +// _ooOoo_ // +// o8888888o // +// 88" . "88 // +// (| ^_^ |) // +// O\ = /O // +// ____/`---'\____ // +// .' \\| |// `. // +// / \\||| : |||// \ // +// / _||||| -:- |||||- \ // +// | | \\\ - /// | | // +// | \_| ''\---/'' | | // +// \ .-\__ `-` ___/-. / // +// ___`. .' /--.--\ `. . ___ // +// ."" '< `.___\_<|>_/___.' >'"". // +// | | : `- \`.;`\ _ /`;.`/ - ` : | | // +// \ \ `-. \_ __\ /__ _/ .-` / / // +// ========`-.____`-.___\_____/___.-`____.-'======== // +// `=---=' // +// ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ // +// 佛祖保佑 永不宕机 永无BUG // +//////////////////////////////////////////////////////////////////// \ No newline at end of file diff --git a/ruoyi-admin/src/main/resources/excel/blank.xls b/ruoyi-admin/src/main/resources/excel/blank.xls new file mode 100644 index 0000000..857a6dc Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/blank.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayEvaporationWater.xls b/ruoyi-admin/src/main/resources/excel/dayEvaporationWater.xls new file mode 100644 index 0000000..5073a45 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayEvaporationWater.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayFlow.xls b/ruoyi-admin/src/main/resources/excel/dayFlow.xls new file mode 100644 index 0000000..927ce85 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayFlow.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayMaxRain.xls b/ruoyi-admin/src/main/resources/excel/dayMaxRain.xls new file mode 100644 index 0000000..b91245c Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayMaxRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayRain.xls b/ruoyi-admin/src/main/resources/excel/dayRain.xls new file mode 100644 index 0000000..07e01ac Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/daySedimentConcentration.xls b/ruoyi-admin/src/main/resources/excel/daySedimentConcentration.xls new file mode 100644 index 0000000..f4d5a82 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/daySedimentConcentration.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayWaterLever.xls b/ruoyi-admin/src/main/resources/excel/dayWaterLever.xls new file mode 100644 index 0000000..f7b8f4d Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayWaterLever.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/dayWaterTemperature.xls b/ruoyi-admin/src/main/resources/excel/dayWaterTemperature.xls new file mode 100644 index 0000000..789ab2f Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/dayWaterTemperature.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/floodExcerpt.xls b/ruoyi-admin/src/main/resources/excel/floodExcerpt.xls new file mode 100644 index 0000000..18ce9e9 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/floodExcerpt.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/followingTheTideExcerpt.xls b/ruoyi-admin/src/main/resources/excel/followingTheTideExcerpt.xls new file mode 100644 index 0000000..e7ca1e0 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/followingTheTideExcerpt.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/hourMaxRain.xls b/ruoyi-admin/src/main/resources/excel/hourMaxRain.xls new file mode 100644 index 0000000..4431994 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/hourMaxRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/maxFloodScale.xls b/ruoyi-admin/src/main/resources/excel/maxFloodScale.xls new file mode 100644 index 0000000..f4891e6 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/maxFloodScale.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/minMaxRain.xls b/ruoyi-admin/src/main/resources/excel/minMaxRain.xls new file mode 100644 index 0000000..dfae87d Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/minMaxRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/momthSedimentTransportRate.xls b/ruoyi-admin/src/main/resources/excel/momthSedimentTransportRate.xls new file mode 100644 index 0000000..83b0b5c Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/momthSedimentTransportRate.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthEvaporationWater.xls b/ruoyi-admin/src/main/resources/excel/monthEvaporationWater.xls new file mode 100644 index 0000000..ca0b48f Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthEvaporationWater.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthFlow.xls b/ruoyi-admin/src/main/resources/excel/monthFlow.xls new file mode 100644 index 0000000..a07cdfb Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthFlow.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthRain.xls b/ruoyi-admin/src/main/resources/excel/monthRain.xls new file mode 100644 index 0000000..2fd53f9 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthSedimentConcentration.xls b/ruoyi-admin/src/main/resources/excel/monthSedimentConcentration.xls new file mode 100644 index 0000000..42d20eb Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthSedimentConcentration.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthTideLever.xls b/ruoyi-admin/src/main/resources/excel/monthTideLever.xls new file mode 100644 index 0000000..cb594bb Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthTideLever.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthWaterLever.xls b/ruoyi-admin/src/main/resources/excel/monthWaterLever.xls new file mode 100644 index 0000000..294a433 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthWaterLever.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/monthWaterTemperature.xls b/ruoyi-admin/src/main/resources/excel/monthWaterTemperature.xls new file mode 100644 index 0000000..b28392e Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/monthWaterTemperature.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/rainExcerpt.xls b/ruoyi-admin/src/main/resources/excel/rainExcerpt.xls new file mode 100644 index 0000000..8041d8f Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/rainExcerpt.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/rsverFloodExcerpt.xls b/ruoyi-admin/src/main/resources/excel/rsverFloodExcerpt.xls new file mode 100644 index 0000000..4917c9d Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/rsverFloodExcerpt.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/sedimentConcentration.xls b/ruoyi-admin/src/main/resources/excel/sedimentConcentration.xls new file mode 100644 index 0000000..2dd1e17 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/sedimentConcentration.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/sedimentTransportRate.xls b/ruoyi-admin/src/main/resources/excel/sedimentTransportRate.xls new file mode 100644 index 0000000..4258f05 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/sedimentTransportRate.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/station.xls b/ruoyi-admin/src/main/resources/excel/station.xls new file mode 100644 index 0000000..aebe2d7 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/station.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/szFloodExcerpt.xls b/ruoyi-admin/src/main/resources/excel/szFloodExcerpt.xls new file mode 100644 index 0000000..387609e Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/szFloodExcerpt.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearEvaporationWater.xls b/ruoyi-admin/src/main/resources/excel/yearEvaporationWater.xls new file mode 100644 index 0000000..ef03cf5 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearEvaporationWater.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearFlow.xls b/ruoyi-admin/src/main/resources/excel/yearFlow.xls new file mode 100644 index 0000000..14b8678 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearFlow.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearRain.xls b/ruoyi-admin/src/main/resources/excel/yearRain.xls new file mode 100644 index 0000000..c5758b5 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearRain.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearTideLever.xls b/ruoyi-admin/src/main/resources/excel/yearTideLever.xls new file mode 100644 index 0000000..02928c9 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearTideLever.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearWaterLever.xls b/ruoyi-admin/src/main/resources/excel/yearWaterLever.xls new file mode 100644 index 0000000..d9d43e8 Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearWaterLever.xls differ diff --git a/ruoyi-admin/src/main/resources/excel/yearWaterTemperature.xls b/ruoyi-admin/src/main/resources/excel/yearWaterTemperature.xls new file mode 100644 index 0000000..30222fb Binary files /dev/null and b/ruoyi-admin/src/main/resources/excel/yearWaterTemperature.xls differ diff --git a/ruoyi-admin/src/main/resources/i18n/messages.properties b/ruoyi-admin/src/main/resources/i18n/messages.properties new file mode 100644 index 0000000..71cf52d --- /dev/null +++ b/ruoyi-admin/src/main/resources/i18n/messages.properties @@ -0,0 +1,37 @@ +#错误消息 +not.null=* 必须填写 +user.jcaptcha.error=验证码错误 +user.jcaptcha.expire=验证码已失效 +user.not.exists=用户不存在/密码错误 +user.password.not.match=用户不存在/密码错误 +user.password.retry.limit.count=密码输入错误{0}次 +user.password.retry.limit.exceed=密码输入错误{0}次,帐户锁定10分钟 +user.password.delete=对不起,您的账号已被删除 +user.blocked=用户已封禁,请联系管理员 +role.blocked=角色已封禁,请联系管理员 +user.logout.success=退出成功 + +length.not.valid=长度必须在{min}到{max}个字符之间 + +user.username.not.valid=* 2到20个汉字、字母、数字或下划线组成,且必须以非数字开头 +user.password.not.valid=* 5-50个字符 + +user.email.not.valid=邮箱格式错误 +user.mobile.phone.number.not.valid=手机号格式错误 +user.login.success=登录成功 +user.register.success=注册成功 +user.notfound=请重新登录 +user.forcelogout=管理员强制退出,请重新登录 +user.unknown.error=未知错误,请重新登录 + +##文件上传消息 +upload.exceed.maxSize=上传的文件大小超出限制的文件大小!
允许的文件最大大小是:{0}MB! +upload.filename.exceed.length=上传的文件名最长{0}个字符 + +##权限 +no.permission=您没有数据的权限,请联系管理员添加权限 [{0}] +no.create.permission=您没有创建数据的权限,请联系管理员添加权限 [{0}] +no.update.permission=您没有修改数据的权限,请联系管理员添加权限 [{0}] +no.delete.permission=您没有删除数据的权限,请联系管理员添加权限 [{0}] +no.export.permission=您没有导出数据的权限,请联系管理员添加权限 [{0}] +no.view.permission=您没有查看数据的权限,请联系管理员添加权限 [{0}] diff --git a/ruoyi-admin/src/main/resources/logback.xml b/ruoyi-admin/src/main/resources/logback.xml new file mode 100644 index 0000000..05ad0da --- /dev/null +++ b/ruoyi-admin/src/main/resources/logback.xml @@ -0,0 +1,93 @@ + + + + + + + + + + + ${log.pattern} + + + + + + ${log.path}/sys-info.log + + + + ${log.path}/sys-info.%d{yyyy-MM-dd}.log + + 60 + + + ${log.pattern} + + + + INFO + + ACCEPT + + DENY + + + + + ${log.path}/sys-error.log + + + + ${log.path}/sys-error.%d{yyyy-MM-dd}.log + + 60 + + + ${log.pattern} + + + + ERROR + + ACCEPT + + DENY + + + + + + ${log.path}/sys-user.log + + + ${log.path}/sys-user.%d{yyyy-MM-dd}.log + + 60 + + + ${log.pattern} + + + + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/ruoyi-admin/src/main/resources/mybatis/mybatis-config.xml b/ruoyi-admin/src/main/resources/mybatis/mybatis-config.xml new file mode 100644 index 0000000..ac47c03 --- /dev/null +++ b/ruoyi-admin/src/main/resources/mybatis/mybatis-config.xml @@ -0,0 +1,20 @@ + + + + + + + + + + + + + + + + + + diff --git a/ruoyi-admin/src/test/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImplTest.java b/ruoyi-admin/src/test/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImplTest.java new file mode 100644 index 0000000..0d34a0f --- /dev/null +++ b/ruoyi-admin/src/test/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImplTest.java @@ -0,0 +1,77 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import com.ruoyi.swlscx.basic.service.HyStscAService; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.service.HyDcsFService; +import com.ruoyi.swlscx.day.service.HyDwtCService; +import com.ruoyi.swlscx.excerpt.service.HyHltdzBService; +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; +import org.springframework.mock.web.MockMultipartFile; +import org.springframework.web.multipart.MultipartFile; + +import java.io.File; +import java.io.FileInputStream; +import java.util.Arrays; +import java.util.List; + + +/** + * @Author al + * @Date 2024/10/16 9:48 + * @Description: TODO + * @Version + */ +@SpringBootTest(classes = com.ruoyi.RuoYiApplication.class) +class HyDpCServiceImplTest { + + + @Autowired + private HyDcsFService hyDcsFService; + @Autowired + private YcExportTaskService ycExportTaskService; + + @Autowired + private HyStscAService hyStscAService; + + @Test + public void importData() throws Exception { + +// List stringList = Arrays.asList("逐潮高低潮位表200501-200801" +// , "逐潮高低潮位表200801-201001", +// "逐潮高低潮位表201001-201301" +// , "逐潮高低潮位表201301-201501", +// "逐潮高低潮位表201501-201701", +// "逐潮高低潮位表201701-202001" +// ,"逐潮高低潮位表202001-"); + List stringList = Arrays.asList("日平均含沙量表195001-199001","日平均含沙量表199001-"); + long newTime = System.currentTimeMillis(); + System.out.println("当前时间:" + newTime); + for (String s : stringList) { + File file = new File("C:\\Users\\12974\\Desktop\\历史查询子系统\\历史数据\\日表类\\日平均含沙量表-已完成\\" + s + ".xls"); + + // 创建 FileInputStream 对象 + FileInputStream fileInputStream = new FileInputStream(file); + + // 创建 MockMultipartFile 对象 + MultipartFile multipartFile = new MockMultipartFile( + "file", + file.getName(), + "application/vnd.ms-excel", + fileInputStream + ); + hyDcsFService.importHyDcsFData(multipartFile); + } + System.out.println("总耗时:" + (System.currentTimeMillis() - newTime)); + } + + @Test + public void test() { + System.out.println(hyStscAService.testSqlServer()); + } +} \ No newline at end of file diff --git a/ruoyi-api/pom.xml b/ruoyi-api/pom.xml new file mode 100644 index 0000000..7cd5133 --- /dev/null +++ b/ruoyi-api/pom.xml @@ -0,0 +1,28 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-api + + + ruoyi-api + + + + + + + com.ruoyi + ruoyi-common + + + + + diff --git a/ruoyi-common/pom.xml b/ruoyi-common/pom.xml new file mode 100644 index 0000000..0612064 --- /dev/null +++ b/ruoyi-common/pom.xml @@ -0,0 +1,160 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-common + + + common通用工具 + + + + + + + org.springframework + spring-context-support + + + + + org.springframework + spring-web + + + + + org.springframework.boot + spring-boot-starter-security + + + + + com.github.pagehelper + pagehelper-spring-boot-starter + + + + + org.springframework.boot + spring-boot-starter-validation + + + + + org.apache.commons + commons-lang3 + + + + + com.fasterxml.jackson.core + jackson-databind + + + + + com.alibaba + fastjson + + + + + commons-io + commons-io + + + + + commons-fileupload + commons-fileupload + + + + + org.apache.poi + poi-ooxml + + + + + org.yaml + snakeyaml + + + + + io.jsonwebtoken + jjwt + + + + + javax.xml.bind + jaxb-api + + + + + org.springframework.boot + spring-boot-starter-data-redis + + + + + org.apache.commons + commons-pool2 + + + + + eu.bitwalker + UserAgentUtils + + + + + javax.servlet + javax.servlet-api + + + + + com.baomidou + mybatis-plus-boot-starter + 3.4.2 + + + + org.projectlombok + lombok + 1.18.4 + + + + org.bouncycastle + bcprov-jdk16 + 1.46 + + + + com.alibaba + easyexcel + 4.0.1 + + + + org.apache.poi + poi-ooxml + + + + + diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataScope.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataScope.java new file mode 100644 index 0000000..176878e --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataScope.java @@ -0,0 +1,28 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 数据权限过滤注解 + * + * @author ruoyi + */ +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +@Documented +public @interface DataScope +{ + /** + * 部门表的别名 + */ + public String deptAlias() default ""; + + /** + * 用户表的别名 + */ + public String userAlias() default ""; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataSource.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataSource.java new file mode 100644 index 0000000..79cd191 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/DataSource.java @@ -0,0 +1,28 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Inherited; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import com.ruoyi.common.enums.DataSourceType; + +/** + * 自定义多数据源切换注解 + * + * 优先级:先方法,后类,如果方法覆盖了类上的数据源类型,以方法的为准,否则以类上的为准 + * + * @author ruoyi + */ +@Target({ ElementType.METHOD, ElementType.TYPE }) +@Retention(RetentionPolicy.RUNTIME) +@Documented +@Inherited +public @interface DataSource +{ + /** + * 切换数据源名称 + */ + public DataSourceType value() default DataSourceType.MASTER; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excel.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excel.java new file mode 100644 index 0000000..6626986 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excel.java @@ -0,0 +1,183 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import java.math.BigDecimal; +import org.apache.poi.ss.usermodel.HorizontalAlignment; +import org.apache.poi.ss.usermodel.IndexedColors; +import com.ruoyi.common.utils.poi.ExcelHandlerAdapter; + +/** + * 自定义导出Excel数据注解 + * + * @author ruoyi + */ +@Retention(RetentionPolicy.RUNTIME) +@Target(ElementType.FIELD) +public @interface Excel +{ + /** + * 导出时在excel中排序 + */ + public int sort() default Integer.MAX_VALUE; + + /** + * 导出到Excel中的名字. + */ + public String name() default ""; + + /** + * 日期格式, 如: yyyy-MM-dd + */ + public String dateFormat() default ""; + + /** + * 如果是字典类型,请设置字典的type值 (如: sys_user_sex) + */ + public String dictType() default ""; + + /** + * 读取内容转表达式 (如: 0=男,1=女,2=未知) + */ + public String readConverterExp() default ""; + + /** + * 分隔符,读取字符串组内容 + */ + public String separator() default ","; + + /** + * BigDecimal 精度 默认:-1(默认不开启BigDecimal格式化) + */ + public int scale() default -1; + + /** + * BigDecimal 舍入规则 默认:BigDecimal.ROUND_HALF_EVEN + */ + public int roundingMode() default BigDecimal.ROUND_HALF_EVEN; + + /** + * 导出时在excel中每个列的高度 单位为字符 + */ + public double height() default 14; + + /** + * 导出时在excel中每个列的宽 单位为字符 + */ + public double width() default 16; + + /** + * 文字后缀,如% 90 变成90% + */ + public String suffix() default ""; + + /** + * 当值为空时,字段的默认值 + */ + public String defaultValue() default ""; + + /** + * 提示信息 + */ + public String prompt() default ""; + + /** + * 设置只能选择不能输入的列内容. + */ + public String[] combo() default {}; + + /** + * 是否导出数据,应对需求:有时我们需要导出一份模板,这是标题需要但内容需要用户手工填写. + */ + public boolean isExport() default true; + + /** + * 另一个类中的属性名称,支持多级获取,以小数点隔开 + */ + public String targetAttr() default ""; + + /** + * 是否自动统计数据,在最后追加一行统计数据总和 + */ + public boolean isStatistics() default false; + + /** + * 导出类型(0数字 1字符串) + */ + public ColumnType cellType() default ColumnType.STRING; + + /** + * 导出字体颜色 + */ + public IndexedColors color() default IndexedColors.BLACK; + + /** + * 导出字段对齐方式 + */ + public HorizontalAlignment align() default HorizontalAlignment.CENTER; + + /** + * 自定义数据处理器 + */ + public Class handler() default ExcelHandlerAdapter.class; + + /** + * 自定义数据处理器参数 + */ + public String[] args() default {}; + + public enum Align + { + AUTO(0), LEFT(1), CENTER(2), RIGHT(3); + private final int value; + + Align(int value) + { + this.value = value; + } + + public int value() + { + return this.value; + } + } + + /** + * 字段类型(0:导出导入;1:仅导出;2:仅导入) + */ + Type type() default Type.ALL; + + public enum Type + { + ALL(0), EXPORT(1), IMPORT(2); + private final int value; + + Type(int value) + { + this.value = value; + } + + public int value() + { + return this.value; + } + } + + public enum ColumnType + { + NUMERIC(0), STRING(1), IMAGE(2); + private final int value; + + ColumnType(int value) + { + this.value = value; + } + + public int value() + { + return this.value; + } + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excels.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excels.java new file mode 100644 index 0000000..1f1cc81 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Excels.java @@ -0,0 +1,18 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * Excel注解集 + * + * @author ruoyi + */ +@Target(ElementType.FIELD) +@Retention(RetentionPolicy.RUNTIME) +public @interface Excels +{ + public Excel[] value(); +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Log.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Log.java new file mode 100644 index 0000000..ca02c6c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/Log.java @@ -0,0 +1,46 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.enums.OperatorType; + +/** + * 自定义操作日志记录注解 + * + * @author ruoyi + * + */ +@Target({ ElementType.PARAMETER, ElementType.METHOD }) +@Retention(RetentionPolicy.RUNTIME) +@Documented +public @interface Log +{ + /** + * 模块 + */ + public String title() default ""; + + /** + * 功能 + */ + public BusinessType businessType() default BusinessType.OTHER; + + /** + * 操作人类别 + */ + public OperatorType operatorType() default OperatorType.MANAGE; + + /** + * 是否保存请求的参数 + */ + public boolean isSaveRequestData() default true; + + /** + * 是否保存响应的参数 + */ + public boolean isSaveResponseData() default true; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RateLimiter.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RateLimiter.java new file mode 100644 index 0000000..69461ea --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RateLimiter.java @@ -0,0 +1,40 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.enums.LimitType; + +/** + * 限流注解 + * + * @author ruoyi + */ +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +@Documented +public @interface RateLimiter +{ + /** + * 限流key + */ + public String key() default Constants.RATE_LIMIT_KEY; + + /** + * 限流时间,单位秒 + */ + public int time() default 60; + + /** + * 限流次数 + */ + public int count() default 100; + + /** + * 限流类型 + */ + public LimitType limitType() default LimitType.DEFAULT; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RepeatSubmit.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RepeatSubmit.java new file mode 100644 index 0000000..b769748 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RepeatSubmit.java @@ -0,0 +1,31 @@ +package com.ruoyi.common.annotation; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Inherited; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 自定义注解防止表单重复提交 + * + * @author ruoyi + * + */ +@Inherited +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +@Documented +public @interface RepeatSubmit +{ + /** + * 间隔时间(ms),小于此时间视为重复提交 + */ + public int interval() default 5000; + + /** + * 提示消息 + */ + public String message() default "不允许重复提交,请稍候再试"; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/config/RuoYiConfig.java b/ruoyi-common/src/main/java/com/ruoyi/common/config/RuoYiConfig.java new file mode 100644 index 0000000..751fe3b --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/config/RuoYiConfig.java @@ -0,0 +1,144 @@ +package com.ruoyi.common.config; + +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.stereotype.Component; + +/** + * 读取项目相关配置 + * + * @author ruoyi + */ +@Component +@ConfigurationProperties(prefix = "ruoyi") +public class RuoYiConfig +{ + /** 项目名称 */ + private String name; + + /** 版本 */ + private String version; + + /** 版权年份 */ + private String copyrightYear; + + /** 实例演示开关 */ + private boolean demoEnabled; + + /** 上传路径 */ + private static String profile; + + /** 获取地址开关 */ + private static boolean addressEnabled; + + /** 验证码类型 */ + private static String captchaType; + + public String getName() + { + return name; + } + + public void setName(String name) + { + this.name = name; + } + + public String getVersion() + { + return version; + } + + public void setVersion(String version) + { + this.version = version; + } + + public String getCopyrightYear() + { + return copyrightYear; + } + + public void setCopyrightYear(String copyrightYear) + { + this.copyrightYear = copyrightYear; + } + + public boolean isDemoEnabled() + { + return demoEnabled; + } + + public void setDemoEnabled(boolean demoEnabled) + { + this.demoEnabled = demoEnabled; + } + + public static String getProfile() + { + return profile; + } + + public void setProfile(String profile) + { + RuoYiConfig.profile = profile; + } + + public static boolean isAddressEnabled() + { + return addressEnabled; + } + + public void setAddressEnabled(boolean addressEnabled) + { + RuoYiConfig.addressEnabled = addressEnabled; + } + + public static String getCaptchaType() { + return captchaType; + } + + public void setCaptchaType(String captchaType) { + RuoYiConfig.captchaType = captchaType; + } + + /** + * 获取导入上传路径 + */ + public static String getImportPath() + { + return getProfile() + "/import"; + } + + /** + * 获取头像上传路径 + */ + public static String getAvatarPath() + { + return getProfile() + "/avatar"; + } + + /** + * 获取下载路径 + */ + public static String getDownloadPath() + { + return getProfile() + "/download/"; + } + + /** + * 获取上传路径 + */ + public static String getUploadPath() + { + return getProfile() + "/upload"; + } + + + /** + * 获取上传路径 + */ + public static String getExportPath() + { + return getProfile() + "/report"; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/constant/Constants.java b/ruoyi-common/src/main/java/com/ruoyi/common/constant/Constants.java new file mode 100644 index 0000000..3d05ccf --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/constant/Constants.java @@ -0,0 +1,189 @@ +package com.ruoyi.common.constant; + +import io.jsonwebtoken.Claims; + +/** + * 通用常量信息 + * + * @author ruoyi + */ +public class Constants +{ + /** + * UTF-8 字符集 + */ + public static final String UTF8 = "UTF-8"; + + /** + * GBK 字符集 + */ + public static final String GBK = "GBK"; + + /** + * http请求 + */ + public static final String HTTP = "http://"; + + /** + * https请求 + */ + public static final String HTTPS = "https://"; + + /** + * 通用成功标识 + */ + public static final String SUCCESS = "0"; + + /** + * 通用失败标识 + */ + public static final String FAIL = "1"; + + /** + * 登录成功 + */ + public static final String LOGIN_SUCCESS = "Success"; + + /** + * 注销 + */ + public static final String LOGOUT = "Logout"; + + /** + * 注册 + */ + public static final String REGISTER = "Register"; + + /** + * 登录失败 + */ + public static final String LOGIN_FAIL = "Error"; + + /** + * 验证码 redis key + */ + public static final String CAPTCHA_CODE_KEY = "captcha_codes:"; + + /** + * 登录用户 redis key + */ + public static final String LOGIN_TOKEN_KEY = "login_tokens:"; + + /** + * 防重提交 redis key + */ + public static final String REPEAT_SUBMIT_KEY = "repeat_submit:"; + + /** + * 限流 redis key + */ + public static final String RATE_LIMIT_KEY = "rate_limit:"; + + /** + * 验证码有效期(分钟) + */ + public static final Integer CAPTCHA_EXPIRATION = 2; + + /** + * 令牌 + */ + public static final String TOKEN = "token"; + + /** + * 令牌前缀 + */ + public static final String TOKEN_PREFIX = "Bearer "; + + /** + * 令牌前缀 + */ + public static final String LOGIN_USER_KEY = "login_user_key"; + + /** + * 用户ID + */ + public static final String JWT_USERID = "userid"; + + /** + * 用户名称 + */ + public static final String JWT_USERNAME = Claims.SUBJECT; + + /** + * 用户头像 + */ + public static final String JWT_AVATAR = "avatar"; + + /** + * 创建时间 + */ + public static final String JWT_CREATED = "created"; + + /** + * 用户权限 + */ + public static final String JWT_AUTHORITIES = "authorities"; + + /** + * 参数管理 cache key + */ + public static final String SYS_CONFIG_KEY = "sys_config:"; + + /** + * 字典管理 cache key + */ + public static final String SYS_DICT_KEY = "sys_dict:"; + + /** + * 资源映射路径 前缀 + */ + public static final String RESOURCE_PREFIX = "/profile"; + + /** + * RMI 远程方法调用 + */ + public static final String LOOKUP_RMI = "rmi:"; + + /** + * LDAP 远程方法调用 + */ + public static final String LOOKUP_LDAP = "ldap:"; + + /** + * LDAPS 远程方法调用 + */ + public static final String LOOKUP_LDAPS = "ldaps:"; + + /** + * 定时任务白名单配置(仅允许访问的包名,如其他需要可以自行添加) + */ + public static final String[] JOB_WHITELIST_STR = { "com.ruoyi" }; + + /** + * 定时任务违规的字符 + */ + public static final String[] JOB_ERROR_STR = { "java.net.URL", "javax.naming.InitialContext", "org.yaml.snakeyaml", + "org.springframework", "org.apache", "com.ruoyi.common.utils.file" }; + + + /** + * 当前页码 + */ + public static final String PAGE = "page"; + /** + * 每页显示记录数 + */ + public static final String LIMIT = "limit"; + /** + * 排序字段 + */ + public static final String ORDER_FIELD = "sidx"; + /** + * 排序方式 + */ + public static final String ORDER = "order"; + /** + * 升序 + */ + public static final String ASC = "asc"; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/constant/GenConstants.java b/ruoyi-common/src/main/java/com/ruoyi/common/constant/GenConstants.java new file mode 100644 index 0000000..7d899d4 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/constant/GenConstants.java @@ -0,0 +1,117 @@ +package com.ruoyi.common.constant; + +/** + * 代码生成通用常量 + * + * @author ruoyi + */ +public class GenConstants +{ + /** 单表(增删改查) */ + public static final String TPL_CRUD = "crud"; + + /** 树表(增删改查) */ + public static final String TPL_TREE = "tree"; + + /** 主子表(增删改查) */ + public static final String TPL_SUB = "sub"; + + /** 树编码字段 */ + public static final String TREE_CODE = "treeCode"; + + /** 树父编码字段 */ + public static final String TREE_PARENT_CODE = "treeParentCode"; + + /** 树名称字段 */ + public static final String TREE_NAME = "treeName"; + + /** 上级菜单ID字段 */ + public static final String PARENT_MENU_ID = "parentMenuId"; + + /** 上级菜单名称字段 */ + public static final String PARENT_MENU_NAME = "parentMenuName"; + + /** 数据库字符串类型 */ + public static final String[] COLUMNTYPE_STR = { "char", "varchar", "nvarchar", "varchar2" }; + + /** 数据库文本类型 */ + public static final String[] COLUMNTYPE_TEXT = { "tinytext", "text", "mediumtext", "longtext" }; + + /** 数据库时间类型 */ + public static final String[] COLUMNTYPE_TIME = { "datetime", "time", "date", "timestamp" }; + + /** 数据库数字类型 */ + public static final String[] COLUMNTYPE_NUMBER = { "tinyint", "smallint", "mediumint", "int", "number", "integer", + "bit", "bigint", "float", "double", "decimal" }; + + /** 页面不需要编辑字段 */ + public static final String[] COLUMNNAME_NOT_EDIT = { "id", "create_by", "create_time", "del_flag" }; + + /** 页面不需要显示的列表字段 */ + public static final String[] COLUMNNAME_NOT_LIST = { "id", "create_by", "create_time", "del_flag", "update_by", + "update_time" }; + + /** 页面不需要查询字段 */ + public static final String[] COLUMNNAME_NOT_QUERY = { "id", "create_by", "create_time", "del_flag", "update_by", + "update_time", "remark" }; + + /** Entity基类字段 */ + public static final String[] BASE_ENTITY = { "createBy", "createTime", "updateBy", "updateTime", "remark" }; + + /** Tree基类字段 */ + public static final String[] TREE_ENTITY = { "parentName", "parentId", "orderNum", "ancestors", "children" }; + + /** 文本框 */ + public static final String HTML_INPUT = "input"; + + /** 文本域 */ + public static final String HTML_TEXTAREA = "textarea"; + + /** 下拉框 */ + public static final String HTML_SELECT = "select"; + + /** 单选框 */ + public static final String HTML_RADIO = "radio"; + + /** 复选框 */ + public static final String HTML_CHECKBOX = "checkbox"; + + /** 日期控件 */ + public static final String HTML_DATETIME = "datetime"; + + /** 图片上传控件 */ + public static final String HTML_IMAGE_UPLOAD = "imageUpload"; + + /** 文件上传控件 */ + public static final String HTML_FILE_UPLOAD = "fileUpload"; + + /** 富文本控件 */ + public static final String HTML_EDITOR = "editor"; + + /** 字符串类型 */ + public static final String TYPE_STRING = "String"; + + /** 整型 */ + public static final String TYPE_INTEGER = "Integer"; + + /** 长整型 */ + public static final String TYPE_LONG = "Long"; + + /** 浮点型 */ + public static final String TYPE_DOUBLE = "Double"; + + /** 高精度计算类型 */ + public static final String TYPE_BIGDECIMAL = "BigDecimal"; + + /** 时间类型 */ + public static final String TYPE_DATE = "Date"; + + /** 模糊查询 */ + public static final String QUERY_LIKE = "LIKE"; + + /** 相等查询 */ + public static final String QUERY_EQ = "EQ"; + + /** 需要 */ + public static final String REQUIRE = "1"; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/constant/HttpStatus.java b/ruoyi-common/src/main/java/com/ruoyi/common/constant/HttpStatus.java new file mode 100644 index 0000000..d60afee --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/constant/HttpStatus.java @@ -0,0 +1,89 @@ +package com.ruoyi.common.constant; + +/** + * 返回状态码 + * + * @author ruoyi + */ +public class HttpStatus +{ + /** + * 操作成功 + */ + public static final int SUCCESS = 200; + + /** + * 对象创建成功 + */ + public static final int CREATED = 201; + + /** + * 请求已经被接受 + */ + public static final int ACCEPTED = 202; + + /** + * 操作已经执行成功,但是没有返回数据 + */ + public static final int NO_CONTENT = 204; + + /** + * 资源已被移除 + */ + public static final int MOVED_PERM = 301; + + /** + * 重定向 + */ + public static final int SEE_OTHER = 303; + + /** + * 资源没有被修改 + */ + public static final int NOT_MODIFIED = 304; + + /** + * 参数列表错误(缺少,格式不匹配) + */ + public static final int BAD_REQUEST = 400; + + /** + * 未授权 + */ + public static final int UNAUTHORIZED = 401; + + /** + * 访问受限,授权过期 + */ + public static final int FORBIDDEN = 403; + + /** + * 资源,服务未找到 + */ + public static final int NOT_FOUND = 404; + + /** + * 不允许的http方法 + */ + public static final int BAD_METHOD = 405; + + /** + * 资源冲突,或者资源被锁 + */ + public static final int CONFLICT = 409; + + /** + * 不支持的数据,媒体类型 + */ + public static final int UNSUPPORTED_TYPE = 415; + + /** + * 系统内部错误 + */ + public static final int ERROR = 500; + + /** + * 接口未实现 + */ + public static final int NOT_IMPLEMENTED = 501; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/constant/ScheduleConstants.java b/ruoyi-common/src/main/java/com/ruoyi/common/constant/ScheduleConstants.java new file mode 100644 index 0000000..62ad815 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/constant/ScheduleConstants.java @@ -0,0 +1,50 @@ +package com.ruoyi.common.constant; + +/** + * 任务调度通用常量 + * + * @author ruoyi + */ +public class ScheduleConstants +{ + public static final String TASK_CLASS_NAME = "TASK_CLASS_NAME"; + + /** 执行目标key */ + public static final String TASK_PROPERTIES = "TASK_PROPERTIES"; + + /** 默认 */ + public static final String MISFIRE_DEFAULT = "0"; + + /** 立即触发执行 */ + public static final String MISFIRE_IGNORE_MISFIRES = "1"; + + /** 触发一次执行 */ + public static final String MISFIRE_FIRE_AND_PROCEED = "2"; + + /** 不触发立即执行 */ + public static final String MISFIRE_DO_NOTHING = "3"; + + public enum Status + { + /** + * 正常 + */ + NORMAL("0"), + /** + * 暂停 + */ + PAUSE("1"); + + private String value; + + private Status(String value) + { + this.value = value; + } + + public String getValue() + { + return value; + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/constant/UserConstants.java b/ruoyi-common/src/main/java/com/ruoyi/common/constant/UserConstants.java new file mode 100644 index 0000000..4ed6009 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/constant/UserConstants.java @@ -0,0 +1,78 @@ +package com.ruoyi.common.constant; + +/** + * 用户常量信息 + * + * @author ruoyi + */ +public class UserConstants +{ + /** + * 平台内系统用户的唯一标志 + */ + public static final String SYS_USER = "SYS_USER"; + + /** 正常状态 */ + public static final String NORMAL = "0"; + + /** 异常状态 */ + public static final String EXCEPTION = "1"; + + /** 用户封禁状态 */ + public static final String USER_DISABLE = "1"; + + /** 角色封禁状态 */ + public static final String ROLE_DISABLE = "1"; + + /** 部门正常状态 */ + public static final String DEPT_NORMAL = "0"; + + /** 部门停用状态 */ + public static final String DEPT_DISABLE = "1"; + + /** 字典正常状态 */ + public static final String DICT_NORMAL = "0"; + + /** 是否为系统默认(是) */ + public static final String YES = "Y"; + + /** 是否菜单外链(是) */ + public static final String YES_FRAME = "0"; + + /** 是否菜单外链(否) */ + public static final String NO_FRAME = "1"; + + /** 菜单类型(目录) */ + public static final String TYPE_DIR = "M"; + + /** 菜单类型(菜单) */ + public static final String TYPE_MENU = "C"; + + /** 菜单类型(按钮) */ + public static final String TYPE_BUTTON = "F"; + + /** Layout组件标识 */ + public final static String LAYOUT = "Layout"; + + /** ParentView组件标识 */ + public final static String PARENT_VIEW = "ParentView"; + + /** InnerLink组件标识 */ + public final static String INNER_LINK = "InnerLink"; + + /** 校验返回结果码 */ + public final static String UNIQUE = "0"; + public final static String NOT_UNIQUE = "1"; + + /** + * 用户名长度限制 + */ + public static final int USERNAME_MIN_LENGTH = 2; + public static final int USERNAME_MAX_LENGTH = 20; + + /** + * 密码长度限制 + */ + public static final int PASSWORD_MIN_LENGTH = 5; + public static final int PASSWORD_MAX_LENGTH = 20; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/controller/BaseController.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/controller/BaseController.java new file mode 100644 index 0000000..80e31cb --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/controller/BaseController.java @@ -0,0 +1,186 @@ +package com.ruoyi.common.core.controller; + +import java.beans.PropertyEditorSupport; +import java.util.Date; +import java.util.List; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.web.bind.WebDataBinder; +import org.springframework.web.bind.annotation.InitBinder; +import com.github.pagehelper.PageHelper; +import com.github.pagehelper.PageInfo; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.page.PageDomain; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.core.page.TableSupport; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.PageUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.sql.SqlUtil; + +/** + * web层通用数据处理 + * + * @author ruoyi + */ +public class BaseController +{ + protected final Logger logger = LoggerFactory.getLogger(this.getClass()); + + /** + * 将前台传递过来的日期格式的字符串,自动转化为Date类型 + */ + @InitBinder + public void initBinder(WebDataBinder binder) + { + // Date 类型转换 + binder.registerCustomEditor(Date.class, new PropertyEditorSupport() + { + @Override + public void setAsText(String text) + { + setValue(DateUtils.parseDate(text)); + } + }); + } + + /** + * 设置请求分页数据 + */ + protected void startPage() + { + PageUtils.startPage(); + } + + /** + * 设置请求排序数据 + */ + protected void startOrderBy() + { + PageDomain pageDomain = TableSupport.buildPageRequest(); + if (StringUtils.isNotEmpty(pageDomain.getOrderBy())) + { + String orderBy = SqlUtil.escapeOrderBySql(pageDomain.getOrderBy()); + PageHelper.orderBy(orderBy); + } + } + + /** + * 清理分页的线程变量 + */ + protected void clearPage() + { + PageUtils.clearPage(); + } + + /** + * 响应请求分页数据 + */ + @SuppressWarnings({ "rawtypes", "unchecked" }) + protected TableDataInfo getDataTable(List list) + { + TableDataInfo rspData = new TableDataInfo(); + rspData.setCode(HttpStatus.SUCCESS); + rspData.setMsg("查询成功"); + rspData.setRows(list); + rspData.setTotal(new PageInfo(list).getTotal()); + return rspData; + } + + /** + * 返回成功 + */ + public AjaxResult success() + { + return AjaxResult.success(); + } + + /** + * 返回失败消息 + */ + public AjaxResult error() + { + return AjaxResult.error(); + } + + /** + * 返回成功消息 + */ + public AjaxResult success(String message) + { + return AjaxResult.success(message); + } + + /** + * 返回失败消息 + */ + public AjaxResult error(String message) + { + return AjaxResult.error(message); + } + + /** + * 响应返回结果 + * + * @param rows 影响行数 + * @return 操作结果 + */ + protected AjaxResult toAjax(int rows) + { + return rows > 0 ? AjaxResult.success() : AjaxResult.error(); + } + + /** + * 响应返回结果 + * + * @param result 结果 + * @return 操作结果 + */ + protected AjaxResult toAjax(boolean result) + { + return result ? success() : error(); + } + + /** + * 页面跳转 + */ + public String redirect(String url) + { + return StringUtils.format("redirect:{}", url); + } + + /** + * 获取用户缓存信息 + */ + public LoginUser getLoginUser() + { + return SecurityUtils.getLoginUser(); + } + + /** + * 获取登录用户id + */ + public Long getUserId() + { + return getLoginUser().getUserId(); + } + + /** + * 获取登录部门id + */ + public Long getDeptId() + { + return getLoginUser().getDeptId(); + } + + /** + * 获取登录用户名 + */ + public String getUsername() + { + return getLoginUser().getUsername(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/AjaxResult.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/AjaxResult.java new file mode 100644 index 0000000..4950bd0 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/AjaxResult.java @@ -0,0 +1,162 @@ +package com.ruoyi.common.core.domain; + +import java.util.HashMap; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.utils.StringUtils; + +/** + * 操作消息提醒 + * + * @author ruoyi + */ +public class AjaxResult extends HashMap +{ + private static final long serialVersionUID = 1L; + + /** 状态码 */ + public static final String CODE_TAG = "code"; + + /** 返回内容 */ + public static final String MSG_TAG = "msg"; + + /** 数据对象 */ + public static final String DATA_TAG = "data"; + + /** + * 初始化一个新创建的 AjaxResult 对象,使其表示一个空消息。 + */ + public AjaxResult() + { + } + + /** + * 初始化一个新创建的 AjaxResult 对象 + * + * @param code 状态码 + * @param msg 返回内容 + */ + public AjaxResult(int code, String msg) + { + super.put(CODE_TAG, code); + super.put(MSG_TAG, msg); + } + + /** + * 初始化一个新创建的 AjaxResult 对象 + * + * @param code 状态码 + * @param msg 返回内容 + * @param data 数据对象 + */ + public AjaxResult(int code, String msg, Object data) + { + super.put(CODE_TAG, code); + super.put(MSG_TAG, msg); + if (StringUtils.isNotNull(data)) + { + super.put(DATA_TAG, data); + } + } + + /** + * 返回成功消息 + * + * @return 成功消息 + */ + public static AjaxResult success() + { + return AjaxResult.success("操作成功"); + } + + /** + * 返回成功数据 + * + * @return 成功消息 + */ + public static AjaxResult success(Object data) + { + return AjaxResult.success("操作成功", data); + } + + /** + * 返回成功消息 + * + * @param msg 返回内容 + * @return 成功消息 + */ + public static AjaxResult success(String msg) + { + return AjaxResult.success(msg, null); + } + + /** + * 返回成功消息 + * + * @param msg 返回内容 + * @param data 数据对象 + * @return 成功消息 + */ + public static AjaxResult success(String msg, Object data) + { + return new AjaxResult(HttpStatus.SUCCESS, msg, data); + } + + /** + * 返回错误消息 + * + * @return + */ + public static AjaxResult error() + { + return AjaxResult.error("操作失败"); + } + + /** + * 返回错误消息 + * + * @param msg 返回内容 + * @return 警告消息 + */ + public static AjaxResult error(String msg) + { + return AjaxResult.error(msg, null); + } + + /** + * 返回错误消息 + * + * @param msg 返回内容 + * @param data 数据对象 + * @return 警告消息 + */ + public static AjaxResult error(String msg, Object data) + { + return new AjaxResult(HttpStatus.ERROR, msg, data); + } + + /** + * 返回错误消息 + * + * @param code 状态码 + * @param msg 返回内容 + * @return 警告消息 + */ + public static AjaxResult error(int code, String msg) + { + return new AjaxResult(code, msg, null); + } + + /** + * 方便链式调用 + * + * @param key 键 + * @param value 值 + * @return 数据对象 + */ + @Override + public AjaxResult put(String key, Object value) + { + super.put(key, value); + return this; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java new file mode 100644 index 0000000..f7d5bf4 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java @@ -0,0 +1,114 @@ +package com.ruoyi.common.core.domain; + +import java.io.Serializable; +import java.util.Date; +import java.util.HashMap; +import java.util.Map; +import com.fasterxml.jackson.annotation.JsonFormat; + +/** + * Entity基类 + * + * @author ruoyi + */ +public class BaseEntity implements Serializable +{ + private static final long serialVersionUID = 1L; + + /** 搜索值 */ + private String searchValue; + + /** 创建者 */ + private String createBy; + + /** 创建时间 */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + private Date createTime; + + /** 更新者 */ + private String updateBy; + + /** 更新时间 */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + private Date updateTime; + + /** 备注 */ + private String remark; + + /** 请求参数 */ + private Map params; + + public String getSearchValue() + { + return searchValue; + } + + public void setSearchValue(String searchValue) + { + this.searchValue = searchValue; + } + + public String getCreateBy() + { + return createBy; + } + + public void setCreateBy(String createBy) + { + this.createBy = createBy; + } + + public Date getCreateTime() + { + return createTime; + } + + public void setCreateTime(Date createTime) + { + this.createTime = createTime; + } + + public String getUpdateBy() + { + return updateBy; + } + + public void setUpdateBy(String updateBy) + { + this.updateBy = updateBy; + } + + public Date getUpdateTime() + { + return updateTime; + } + + public void setUpdateTime(Date updateTime) + { + this.updateTime = updateTime; + } + + public String getRemark() + { + return remark; + } + + public void setRemark(String remark) + { + this.remark = remark; + } + + public Map getParams() + { + if (params == null) + { + params = new HashMap<>(); + } + return params; + } + + public void setParams(Map params) + { + this.params = params; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeEntity.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeEntity.java new file mode 100644 index 0000000..a180a18 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeEntity.java @@ -0,0 +1,79 @@ +package com.ruoyi.common.core.domain; + +import java.util.ArrayList; +import java.util.List; + +/** + * Tree基类 + * + * @author ruoyi + */ +public class TreeEntity extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 父菜单名称 */ + private String parentName; + + /** 父菜单ID */ + private Long parentId; + + /** 显示顺序 */ + private Integer orderNum; + + /** 祖级列表 */ + private String ancestors; + + /** 子部门 */ + private List children = new ArrayList<>(); + + public String getParentName() + { + return parentName; + } + + public void setParentName(String parentName) + { + this.parentName = parentName; + } + + public Long getParentId() + { + return parentId; + } + + public void setParentId(Long parentId) + { + this.parentId = parentId; + } + + public Integer getOrderNum() + { + return orderNum; + } + + public void setOrderNum(Integer orderNum) + { + this.orderNum = orderNum; + } + + public String getAncestors() + { + return ancestors; + } + + public void setAncestors(String ancestors) + { + this.ancestors = ancestors; + } + + public List getChildren() + { + return children; + } + + public void setChildren(List children) + { + this.children = children; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeSelect.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeSelect.java new file mode 100644 index 0000000..bd835db --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/TreeSelect.java @@ -0,0 +1,77 @@ +package com.ruoyi.common.core.domain; + +import java.io.Serializable; +import java.util.List; +import java.util.stream.Collectors; +import com.fasterxml.jackson.annotation.JsonInclude; +import com.ruoyi.common.core.domain.entity.SysDept; +import com.ruoyi.common.core.domain.entity.SysMenu; + +/** + * Treeselect树结构实体类 + * + * @author ruoyi + */ +public class TreeSelect implements Serializable +{ + private static final long serialVersionUID = 1L; + + /** 节点ID */ + private Long id; + + /** 节点名称 */ + private String label; + + /** 子节点 */ + @JsonInclude(JsonInclude.Include.NON_EMPTY) + private List children; + + public TreeSelect() + { + + } + + public TreeSelect(SysDept dept) + { + this.id = dept.getDeptId(); + this.label = dept.getDeptName(); + this.children = dept.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList()); + } + + public TreeSelect(SysMenu menu) + { + this.id = menu.getMenuId(); + this.label = menu.getMenuName(); + this.children = menu.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList()); + } + + public Long getId() + { + return id; + } + + public void setId(Long id) + { + this.id = id; + } + + public String getLabel() + { + return label; + } + + public void setLabel(String label) + { + this.label = label; + } + + public List getChildren() + { + return children; + } + + public void setChildren(List children) + { + this.children = children; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDept.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDept.java new file mode 100644 index 0000000..fb18c5c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDept.java @@ -0,0 +1,203 @@ +package com.ruoyi.common.core.domain.entity; + +import java.util.ArrayList; +import java.util.List; +import javax.validation.constraints.Email; +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.NotNull; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 部门表 sys_dept + * + * @author ruoyi + */ +public class SysDept extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 部门ID */ + private Long deptId; + + /** 父部门ID */ + private Long parentId; + + /** 祖级列表 */ + private String ancestors; + + /** 部门名称 */ + private String deptName; + + /** 显示顺序 */ + private Integer orderNum; + + /** 负责人 */ + private String leader; + + /** 联系电话 */ + private String phone; + + /** 邮箱 */ + private String email; + + /** 部门状态:0正常,1停用 */ + private String status; + + /** 删除标志(0代表存在 2代表删除) */ + private String delFlag; + + /** 父部门名称 */ + private String parentName; + + /** 子部门 */ + private List children = new ArrayList(); + + public Long getDeptId() + { + return deptId; + } + + public void setDeptId(Long deptId) + { + this.deptId = deptId; + } + + public Long getParentId() + { + return parentId; + } + + public void setParentId(Long parentId) + { + this.parentId = parentId; + } + + public String getAncestors() + { + return ancestors; + } + + public void setAncestors(String ancestors) + { + this.ancestors = ancestors; + } + + @NotBlank(message = "部门名称不能为空") + @Size(min = 0, max = 30, message = "部门名称长度不能超过30个字符") + public String getDeptName() + { + return deptName; + } + + public void setDeptName(String deptName) + { + this.deptName = deptName; + } + + @NotNull(message = "显示顺序不能为空") + public Integer getOrderNum() + { + return orderNum; + } + + public void setOrderNum(Integer orderNum) + { + this.orderNum = orderNum; + } + + public String getLeader() + { + return leader; + } + + public void setLeader(String leader) + { + this.leader = leader; + } + + @Size(min = 0, max = 11, message = "联系电话长度不能超过11个字符") + public String getPhone() + { + return phone; + } + + public void setPhone(String phone) + { + this.phone = phone; + } + + @Email(message = "邮箱格式不正确") + @Size(min = 0, max = 50, message = "邮箱长度不能超过50个字符") + public String getEmail() + { + return email; + } + + public void setEmail(String email) + { + this.email = email; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getDelFlag() + { + return delFlag; + } + + public void setDelFlag(String delFlag) + { + this.delFlag = delFlag; + } + + public String getParentName() + { + return parentName; + } + + public void setParentName(String parentName) + { + this.parentName = parentName; + } + + public List getChildren() + { + return children; + } + + public void setChildren(List children) + { + this.children = children; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("deptId", getDeptId()) + .append("parentId", getParentId()) + .append("ancestors", getAncestors()) + .append("deptName", getDeptName()) + .append("orderNum", getOrderNum()) + .append("leader", getLeader()) + .append("phone", getPhone()) + .append("email", getEmail()) + .append("status", getStatus()) + .append("delFlag", getDelFlag()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictData.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictData.java new file mode 100644 index 0000000..3f152b3 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictData.java @@ -0,0 +1,176 @@ +package com.ruoyi.common.core.domain.entity; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 字典数据表 sys_dict_data + * + * @author ruoyi + */ +public class SysDictData extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 字典编码 */ + @Excel(name = "字典编码", cellType = ColumnType.NUMERIC) + private Long dictCode; + + /** 字典排序 */ + @Excel(name = "字典排序", cellType = ColumnType.NUMERIC) + private Long dictSort; + + /** 字典标签 */ + @Excel(name = "字典标签") + private String dictLabel; + + /** 字典键值 */ + @Excel(name = "字典键值") + private String dictValue; + + /** 字典类型 */ + @Excel(name = "字典类型") + private String dictType; + + /** 样式属性(其他样式扩展) */ + private String cssClass; + + /** 表格字典样式 */ + private String listClass; + + /** 是否默认(Y是 N否) */ + @Excel(name = "是否默认", readConverterExp = "Y=是,N=否") + private String isDefault; + + /** 状态(0正常 1停用) */ + @Excel(name = "状态", readConverterExp = "0=正常,1=停用") + private String status; + + public Long getDictCode() + { + return dictCode; + } + + public void setDictCode(Long dictCode) + { + this.dictCode = dictCode; + } + + public Long getDictSort() + { + return dictSort; + } + + public void setDictSort(Long dictSort) + { + this.dictSort = dictSort; + } + + @NotBlank(message = "字典标签不能为空") + @Size(min = 0, max = 100, message = "字典标签长度不能超过100个字符") + public String getDictLabel() + { + return dictLabel; + } + + public void setDictLabel(String dictLabel) + { + this.dictLabel = dictLabel; + } + + @NotBlank(message = "字典键值不能为空") + @Size(min = 0, max = 100, message = "字典键值长度不能超过100个字符") + public String getDictValue() + { + return dictValue; + } + + public void setDictValue(String dictValue) + { + this.dictValue = dictValue; + } + + @NotBlank(message = "字典类型不能为空") + @Size(min = 0, max = 100, message = "字典类型长度不能超过100个字符") + public String getDictType() + { + return dictType; + } + + public void setDictType(String dictType) + { + this.dictType = dictType; + } + + @Size(min = 0, max = 100, message = "样式属性长度不能超过100个字符") + public String getCssClass() + { + return cssClass; + } + + public void setCssClass(String cssClass) + { + this.cssClass = cssClass; + } + + public String getListClass() + { + return listClass; + } + + public void setListClass(String listClass) + { + this.listClass = listClass; + } + + public boolean getDefault() + { + return UserConstants.YES.equals(this.isDefault) ? true : false; + } + + public String getIsDefault() + { + return isDefault; + } + + public void setIsDefault(String isDefault) + { + this.isDefault = isDefault; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("dictCode", getDictCode()) + .append("dictSort", getDictSort()) + .append("dictLabel", getDictLabel()) + .append("dictValue", getDictValue()) + .append("dictType", getDictType()) + .append("cssClass", getCssClass()) + .append("listClass", getListClass()) + .append("isDefault", getIsDefault()) + .append("status", getStatus()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictType.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictType.java new file mode 100644 index 0000000..e324fcf --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysDictType.java @@ -0,0 +1,96 @@ +package com.ruoyi.common.core.domain.entity; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Pattern; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 字典类型表 sys_dict_type + * + * @author ruoyi + */ +public class SysDictType extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 字典主键 */ + @Excel(name = "字典主键", cellType = ColumnType.NUMERIC) + private Long dictId; + + /** 字典名称 */ + @Excel(name = "字典名称") + private String dictName; + + /** 字典类型 */ + @Excel(name = "字典类型") + private String dictType; + + /** 状态(0正常 1停用) */ + @Excel(name = "状态", readConverterExp = "0=正常,1=停用") + private String status; + + public Long getDictId() + { + return dictId; + } + + public void setDictId(Long dictId) + { + this.dictId = dictId; + } + + @NotBlank(message = "字典名称不能为空") + @Size(min = 0, max = 100, message = "字典类型名称长度不能超过100个字符") + public String getDictName() + { + return dictName; + } + + public void setDictName(String dictName) + { + this.dictName = dictName; + } + + @NotBlank(message = "字典类型不能为空") + @Size(min = 0, max = 100, message = "字典类型类型长度不能超过100个字符") + @Pattern(regexp = "^[a-z][a-z0-9_]*$", message = "字典类型必须以字母开头,且只能为(小写字母,数字,下滑线)") + public String getDictType() + { + return dictType; + } + + public void setDictType(String dictType) + { + this.dictType = dictType; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("dictId", getDictId()) + .append("dictName", getDictName()) + .append("dictType", getDictType()) + .append("status", getStatus()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysMenu.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysMenu.java new file mode 100644 index 0000000..042bc76 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysMenu.java @@ -0,0 +1,259 @@ +package com.ruoyi.common.core.domain.entity; + +import java.util.ArrayList; +import java.util.List; +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.NotNull; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 菜单权限表 sys_menu + * + * @author ruoyi + */ +public class SysMenu extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 菜单ID */ + private Long menuId; + + /** 菜单名称 */ + private String menuName; + + /** 父菜单名称 */ + private String parentName; + + /** 父菜单ID */ + private Long parentId; + + /** 显示顺序 */ + private Integer orderNum; + + /** 路由地址 */ + private String path; + + /** 组件路径 */ + private String component; + + /** 路由参数 */ + private String query; + + /** 是否为外链(0是 1否) */ + private String isFrame; + + /** 是否缓存(0缓存 1不缓存) */ + private String isCache; + + /** 类型(M目录 C菜单 F按钮) */ + private String menuType; + + /** 显示状态(0显示 1隐藏) */ + private String visible; + + /** 菜单状态(0显示 1隐藏) */ + private String status; + + /** 权限字符串 */ + private String perms; + + /** 菜单图标 */ + private String icon; + + /** 子菜单 */ + private List children = new ArrayList(); + + public Long getMenuId() + { + return menuId; + } + + public void setMenuId(Long menuId) + { + this.menuId = menuId; + } + + @NotBlank(message = "菜单名称不能为空") + @Size(min = 0, max = 50, message = "菜单名称长度不能超过50个字符") + public String getMenuName() + { + return menuName; + } + + public void setMenuName(String menuName) + { + this.menuName = menuName; + } + + public String getParentName() + { + return parentName; + } + + public void setParentName(String parentName) + { + this.parentName = parentName; + } + + public Long getParentId() + { + return parentId; + } + + public void setParentId(Long parentId) + { + this.parentId = parentId; + } + + @NotNull(message = "显示顺序不能为空") + public Integer getOrderNum() + { + return orderNum; + } + + public void setOrderNum(Integer orderNum) + { + this.orderNum = orderNum; + } + + @Size(min = 0, max = 200, message = "路由地址不能超过200个字符") + public String getPath() + { + return path; + } + + public void setPath(String path) + { + this.path = path; + } + + @Size(min = 0, max = 200, message = "组件路径不能超过255个字符") + public String getComponent() + { + return component; + } + + public void setComponent(String component) + { + this.component = component; + } + + public String getQuery() + { + return query; + } + + public void setQuery(String query) + { + this.query = query; + } + + public String getIsFrame() + { + return isFrame; + } + + public void setIsFrame(String isFrame) + { + this.isFrame = isFrame; + } + + public String getIsCache() + { + return isCache; + } + + public void setIsCache(String isCache) + { + this.isCache = isCache; + } + + @NotBlank(message = "菜单类型不能为空") + public String getMenuType() + { + return menuType; + } + + public void setMenuType(String menuType) + { + this.menuType = menuType; + } + + public String getVisible() + { + return visible; + } + + public void setVisible(String visible) + { + this.visible = visible; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + @Size(min = 0, max = 100, message = "权限标识长度不能超过100个字符") + public String getPerms() + { + return perms; + } + + public void setPerms(String perms) + { + this.perms = perms; + } + + public String getIcon() + { + return icon; + } + + public void setIcon(String icon) + { + this.icon = icon; + } + + public List getChildren() + { + return children; + } + + public void setChildren(List children) + { + this.children = children; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("menuId", getMenuId()) + .append("menuName", getMenuName()) + .append("parentId", getParentId()) + .append("orderNum", getOrderNum()) + .append("path", getPath()) + .append("component", getComponent()) + .append("isFrame", getIsFrame()) + .append("IsCache", getIsCache()) + .append("menuType", getMenuType()) + .append("visible", getVisible()) + .append("status ", getStatus()) + .append("perms", getPerms()) + .append("icon", getIcon()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysRole.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysRole.java new file mode 100644 index 0000000..36629eb --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysRole.java @@ -0,0 +1,226 @@ +package com.ruoyi.common.core.domain.entity; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 角色表 sys_role + * + * @author ruoyi + */ +public class SysRole extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 角色ID */ + @Excel(name = "角色序号", cellType = ColumnType.NUMERIC) + private Long roleId; + + /** 角色名称 */ + @Excel(name = "角色名称") + private String roleName; + + /** 角色权限 */ + @Excel(name = "角色权限") + private String roleKey; + + /** 角色排序 */ + @Excel(name = "角色排序") + private String roleSort; + + /** 数据范围(1:所有数据权限;2:自定义数据权限;3:本部门数据权限;4:本部门及以下数据权限;5:仅本人数据权限) */ + @Excel(name = "数据范围", readConverterExp = "1=所有数据权限,2=自定义数据权限,3=本部门数据权限,4=本部门及以下数据权限,5=仅本人数据权限") + private String dataScope; + + /** 菜单树选择项是否关联显示( 0:父子不互相关联显示 1:父子互相关联显示) */ + private boolean menuCheckStrictly; + + /** 部门树选择项是否关联显示(0:父子不互相关联显示 1:父子互相关联显示 ) */ + private boolean deptCheckStrictly; + + /** 角色状态(0正常 1停用) */ + @Excel(name = "角色状态", readConverterExp = "0=正常,1=停用") + private String status; + + /** 删除标志(0代表存在 2代表删除) */ + private String delFlag; + + /** 用户是否存在此角色标识 默认不存在 */ + private boolean flag = false; + + /** 菜单组 */ + private Long[] menuIds; + + /** 部门组(数据权限) */ + private Long[] deptIds; + + public SysRole() + { + + } + + public SysRole(Long roleId) + { + this.roleId = roleId; + } + + public Long getRoleId() + { + return roleId; + } + + public void setRoleId(Long roleId) + { + this.roleId = roleId; + } + + public boolean isAdmin() + { + return isAdmin(this.roleId); + } + + public static boolean isAdmin(Long roleId) + { + return roleId != null && 1L == roleId; + } + + @NotBlank(message = "角色名称不能为空") + @Size(min = 0, max = 30, message = "角色名称长度不能超过30个字符") + public String getRoleName() + { + return roleName; + } + + public void setRoleName(String roleName) + { + this.roleName = roleName; + } + + @NotBlank(message = "权限字符不能为空") + @Size(min = 0, max = 100, message = "权限字符长度不能超过100个字符") + public String getRoleKey() + { + return roleKey; + } + + public void setRoleKey(String roleKey) + { + this.roleKey = roleKey; + } + + @NotBlank(message = "显示顺序不能为空") + public String getRoleSort() + { + return roleSort; + } + + public void setRoleSort(String roleSort) + { + this.roleSort = roleSort; + } + + public String getDataScope() + { + return dataScope; + } + + public void setDataScope(String dataScope) + { + this.dataScope = dataScope; + } + + public boolean isMenuCheckStrictly() + { + return menuCheckStrictly; + } + + public void setMenuCheckStrictly(boolean menuCheckStrictly) + { + this.menuCheckStrictly = menuCheckStrictly; + } + + public boolean isDeptCheckStrictly() + { + return deptCheckStrictly; + } + + public void setDeptCheckStrictly(boolean deptCheckStrictly) + { + this.deptCheckStrictly = deptCheckStrictly; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getDelFlag() + { + return delFlag; + } + + public void setDelFlag(String delFlag) + { + this.delFlag = delFlag; + } + + public boolean isFlag() + { + return flag; + } + + public void setFlag(boolean flag) + { + this.flag = flag; + } + + public Long[] getMenuIds() + { + return menuIds; + } + + public void setMenuIds(Long[] menuIds) + { + this.menuIds = menuIds; + } + + public Long[] getDeptIds() + { + return deptIds; + } + + public void setDeptIds(Long[] deptIds) + { + this.deptIds = deptIds; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("roleId", getRoleId()) + .append("roleName", getRoleName()) + .append("roleKey", getRoleKey()) + .append("roleSort", getRoleSort()) + .append("dataScope", getDataScope()) + .append("menuCheckStrictly", isMenuCheckStrictly()) + .append("deptCheckStrictly", isDeptCheckStrictly()) + .append("status", getStatus()) + .append("delFlag", getDelFlag()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysUser.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysUser.java new file mode 100644 index 0000000..432ea56 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/entity/SysUser.java @@ -0,0 +1,342 @@ +package com.ruoyi.common.core.domain.entity; + +import java.util.Date; +import java.util.List; +import javax.validation.constraints.*; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.fasterxml.jackson.annotation.JsonIgnore; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.annotation.Excel.Type; +import com.ruoyi.common.annotation.Excels; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.xss.Xss; + +/** + * 用户对象 sys_user + * + * @author ruoyi + */ +public class SysUser extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 用户ID */ + @Excel(name = "用户序号", cellType = ColumnType.NUMERIC, prompt = "用户编号") + private Long userId; + + /** 部门ID */ + @Excel(name = "部门编号", type = Type.IMPORT) + private Long deptId; + + /** 用户账号 */ + @Excel(name = "登录名称") + private String userName; + + /** 用户昵称 */ + @Excel(name = "用户名称") + private String nickName; + + /** 用户邮箱 */ + @Excel(name = "用户邮箱") + private String email; + + /** 手机号码 */ + @Excel(name = "手机号码") + private String phonenumber; + + /** 用户性别 */ + @Excel(name = "用户性别", readConverterExp = "0=男,1=女,2=未知") + private String sex; + + /** 用户头像 */ + private String avatar; + + /** 密码 */ + private String password; + + /** 盐加密 */ + private String salt; + + /** 帐号状态(0正常 1停用) */ + @Excel(name = "帐号状态", readConverterExp = "0=正常,1=停用") + private String status; + + /** 删除标志(0代表存在 2代表删除) */ + private String delFlag; + + /** 最后登录IP */ + @Excel(name = "最后登录IP", type = Type.EXPORT) + private String loginIp; + + /** 最后登录时间 */ + @Excel(name = "最后登录时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss", type = Type.EXPORT) + private Date loginDate; + + /** 部门对象 */ + @Excels({ + @Excel(name = "部门名称", targetAttr = "deptName", type = Type.EXPORT), + @Excel(name = "部门负责人", targetAttr = "leader", type = Type.EXPORT) + }) + private SysDept dept; + + /** 角色对象 */ + private List roles; + + /** 角色组 */ + private Long[] roleIds; + + /** 岗位组 */ + private Long[] postIds; + + /** 角色ID */ + private Long roleId; + + public SysUser() + { + + } + + public SysUser(Long userId) + { + this.userId = userId; + } + + public Long getUserId() + { + return userId; + } + + public void setUserId(Long userId) + { + this.userId = userId; + } + + public boolean isAdmin() + { + return isAdmin(this.userId); + } + + public static boolean isAdmin(Long userId) + { + return userId != null && 1L == userId; + } + + public Long getDeptId() + { + return deptId; + } + + public void setDeptId(Long deptId) + { + this.deptId = deptId; + } + + @Xss(message = "用户昵称不能包含脚本字符") + @Size(min = 0, max = 30, message = "用户昵称长度不能超过30个字符") + public String getNickName() + { + return nickName; + } + + public void setNickName(String nickName) + { + this.nickName = nickName; + } + + @Xss(message = "用户账号不能包含脚本字符") + @NotBlank(message = "用户账号不能为空") + @Size(min = 0, max = 30, message = "用户账号长度不能超过30个字符") + public String getUserName() + { + return userName; + } + + public void setUserName(String userName) + { + this.userName = userName; + } + + @Email(message = "邮箱格式不正确") + @Size(min = 0, max = 50, message = "邮箱长度不能超过50个字符") + public String getEmail() + { + return email; + } + + public void setEmail(String email) + { + this.email = email; + } + + @Size(min = 0, max = 11, message = "手机号码长度不能超过11个字符") + public String getPhonenumber() + { + return phonenumber; + } + + public void setPhonenumber(String phonenumber) + { + this.phonenumber = phonenumber; + } + + public String getSex() + { + return sex; + } + + public void setSex(String sex) + { + this.sex = sex; + } + + public String getAvatar() + { + return avatar; + } + + public void setAvatar(String avatar) + { + this.avatar = avatar; + } + + @JsonIgnore + @JsonProperty + public String getPassword() + { + return password; + } + + public void setPassword(String password) + { + this.password = password; + } + + public String getSalt() + { + return salt; + } + + public void setSalt(String salt) + { + this.salt = salt; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getDelFlag() + { + return delFlag; + } + + public void setDelFlag(String delFlag) + { + this.delFlag = delFlag; + } + + public String getLoginIp() + { + return loginIp; + } + + public void setLoginIp(String loginIp) + { + this.loginIp = loginIp; + } + + public Date getLoginDate() + { + return loginDate; + } + + public void setLoginDate(Date loginDate) + { + this.loginDate = loginDate; + } + + public SysDept getDept() + { + return dept; + } + + public void setDept(SysDept dept) + { + this.dept = dept; + } + + public List getRoles() + { + return roles; + } + + public void setRoles(List roles) + { + this.roles = roles; + } + + public Long[] getRoleIds() + { + return roleIds; + } + + public void setRoleIds(Long[] roleIds) + { + this.roleIds = roleIds; + } + + public Long[] getPostIds() + { + return postIds; + } + + public void setPostIds(Long[] postIds) + { + this.postIds = postIds; + } + + public Long getRoleId() + { + return roleId; + } + + public void setRoleId(Long roleId) + { + this.roleId = roleId; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("userId", getUserId()) + .append("deptId", getDeptId()) + .append("userName", getUserName()) + .append("nickName", getNickName()) + .append("email", getEmail()) + .append("phonenumber", getPhonenumber()) + .append("sex", getSex()) + .append("avatar", getAvatar()) + .append("password", getPassword()) + .append("salt", getSalt()) + .append("status", getStatus()) + .append("delFlag", getDelFlag()) + .append("loginIp", getLoginIp()) + .append("loginDate", getLoginDate()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .append("dept", getDept()) + .toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginBody.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginBody.java new file mode 100644 index 0000000..b5bc8c8 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginBody.java @@ -0,0 +1,69 @@ +package com.ruoyi.common.core.domain.model; + +/** + * 用户登录对象 + * + * @author ruoyi + */ +public class LoginBody +{ + /** + * 用户名 + */ + private String username; + + /** + * 用户密码 + */ + private String password; + + /** + * 验证码 + */ + private String code; + + /** + * 唯一标识 + */ + private String uuid; + + public String getUsername() + { + return username; + } + + public void setUsername(String username) + { + this.username = username; + } + + public String getPassword() + { + return password; + } + + public void setPassword(String password) + { + this.password = password; + } + + public String getCode() + { + return code; + } + + public void setCode(String code) + { + this.code = code; + } + + public String getUuid() + { + return uuid; + } + + public void setUuid(String uuid) + { + this.uuid = uuid; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginUser.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginUser.java new file mode 100644 index 0000000..db4d2a5 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/LoginUser.java @@ -0,0 +1,266 @@ +package com.ruoyi.common.core.domain.model; + +import java.util.Collection; +import java.util.Set; +import org.springframework.security.core.GrantedAuthority; +import org.springframework.security.core.userdetails.UserDetails; +import com.alibaba.fastjson.annotation.JSONField; +import com.ruoyi.common.core.domain.entity.SysUser; + +/** + * 登录用户身份权限 + * + * @author ruoyi + */ +public class LoginUser implements UserDetails +{ + private static final long serialVersionUID = 1L; + + /** + * 用户ID + */ + private Long userId; + + /** + * 部门ID + */ + private Long deptId; + + /** + * 用户唯一标识 + */ + private String token; + + /** + * 登录时间 + */ + private Long loginTime; + + /** + * 过期时间 + */ + private Long expireTime; + + /** + * 登录IP地址 + */ + private String ipaddr; + + /** + * 登录地点 + */ + private String loginLocation; + + /** + * 浏览器类型 + */ + private String browser; + + /** + * 操作系统 + */ + private String os; + + /** + * 权限列表 + */ + private Set permissions; + + /** + * 用户信息 + */ + private SysUser user; + + public Long getUserId() + { + return userId; + } + + public void setUserId(Long userId) + { + this.userId = userId; + } + + public Long getDeptId() + { + return deptId; + } + + public void setDeptId(Long deptId) + { + this.deptId = deptId; + } + + public String getToken() + { + return token; + } + + public void setToken(String token) + { + this.token = token; + } + + public LoginUser() + { + } + + public LoginUser(SysUser user, Set permissions) + { + this.user = user; + this.permissions = permissions; + } + + public LoginUser(Long userId, Long deptId, SysUser user, Set permissions) + { + this.userId = userId; + this.deptId = deptId; + this.user = user; + this.permissions = permissions; + } + + @JSONField(serialize = false) + @Override + public String getPassword() + { + return user.getPassword(); + } + + @Override + public String getUsername() + { + return user.getUserName(); + } + + /** + * 账户是否未过期,过期无法验证 + */ + @JSONField(serialize = false) + @Override + public boolean isAccountNonExpired() + { + return true; + } + + /** + * 指定用户是否解锁,锁定的用户无法进行身份验证 + * + * @return + */ + @JSONField(serialize = false) + @Override + public boolean isAccountNonLocked() + { + return true; + } + + /** + * 指示是否已过期的用户的凭据(密码),过期的凭据防止认证 + * + * @return + */ + @JSONField(serialize = false) + @Override + public boolean isCredentialsNonExpired() + { + return true; + } + + /** + * 是否可用 ,禁用的用户不能身份验证 + * + * @return + */ + @JSONField(serialize = false) + @Override + public boolean isEnabled() + { + return true; + } + + public Long getLoginTime() + { + return loginTime; + } + + public void setLoginTime(Long loginTime) + { + this.loginTime = loginTime; + } + + public String getIpaddr() + { + return ipaddr; + } + + public void setIpaddr(String ipaddr) + { + this.ipaddr = ipaddr; + } + + public String getLoginLocation() + { + return loginLocation; + } + + public void setLoginLocation(String loginLocation) + { + this.loginLocation = loginLocation; + } + + public String getBrowser() + { + return browser; + } + + public void setBrowser(String browser) + { + this.browser = browser; + } + + public String getOs() + { + return os; + } + + public void setOs(String os) + { + this.os = os; + } + + public Long getExpireTime() + { + return expireTime; + } + + public void setExpireTime(Long expireTime) + { + this.expireTime = expireTime; + } + + public Set getPermissions() + { + return permissions; + } + + public void setPermissions(Set permissions) + { + this.permissions = permissions; + } + + public SysUser getUser() + { + return user; + } + + public void setUser(SysUser user) + { + this.user = user; + } + + @Override + public Collection getAuthorities() + { + return null; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/RegisterBody.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/RegisterBody.java new file mode 100644 index 0000000..868a1fc --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/model/RegisterBody.java @@ -0,0 +1,11 @@ +package com.ruoyi.common.core.domain.model; + +/** + * 用户注册对象 + * + * @author ruoyi + */ +public class RegisterBody extends LoginBody +{ + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/page/PageDomain.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/PageDomain.java new file mode 100644 index 0000000..8966cb4 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/PageDomain.java @@ -0,0 +1,101 @@ +package com.ruoyi.common.core.page; + +import com.ruoyi.common.utils.StringUtils; + +/** + * 分页数据 + * + * @author ruoyi + */ +public class PageDomain +{ + /** 当前记录起始索引 */ + private Integer pageNum; + + /** 每页显示记录数 */ + private Integer pageSize; + + /** 排序列 */ + private String orderByColumn; + + /** 排序的方向desc或者asc */ + private String isAsc = "asc"; + + /** 分页参数合理化 */ + private Boolean reasonable = true; + + public String getOrderBy() + { + if (StringUtils.isEmpty(orderByColumn)) + { + return ""; + } + return StringUtils.toUnderScoreCase(orderByColumn) + " " + isAsc; + } + + public Integer getPageNum() + { + return pageNum; + } + + public void setPageNum(Integer pageNum) + { + this.pageNum = pageNum; + } + + public Integer getPageSize() + { + return pageSize; + } + + public void setPageSize(Integer pageSize) + { + this.pageSize = pageSize; + } + + public String getOrderByColumn() + { + return orderByColumn; + } + + public void setOrderByColumn(String orderByColumn) + { + this.orderByColumn = orderByColumn; + } + + public String getIsAsc() + { + return isAsc; + } + + public void setIsAsc(String isAsc) + { + if (StringUtils.isNotEmpty(isAsc)) + { + // 兼容前端排序类型 + if ("ascending".equals(isAsc)) + { + isAsc = "asc"; + } + else if ("descending".equals(isAsc)) + { + isAsc = "desc"; + } + this.isAsc = isAsc; + } + } + + public Boolean getReasonable() + { + if (StringUtils.isNull(reasonable)) + { + return Boolean.TRUE; + } + return reasonable; + } + + public void setReasonable(Boolean reasonable) + { + this.reasonable = reasonable; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/page/R.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/R.java new file mode 100644 index 0000000..7329aed --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/R.java @@ -0,0 +1,66 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.common.core.page; + +import java.util.HashMap; +import java.util.Map; + +/** + * 返回数据 + * + * @author Mark sunlightcs@gmail.com + */ +public class R extends HashMap { + private static final long serialVersionUID = 1L; + + public R() { + put("code", 0); + put("msg", "操作成功"); + } + + public R(Object o) {} + + public static R error() { + return error(500, "发生错误,请联系管理员"); + } + + public static R error(String msg) { + return error(500, msg); + } + + public static R error(int code, String msg) { + R r = new R(); + r.put("code", code); + r.put("msg", msg); + return r; + } + + public static R ok(String msg) { + R r = new R(); + r.put("msg", msg); + return r; + } + + public static R ok(Map map) { + R r = new R(); + r.putAll(map); + return r; + } + + public static R ok() { + return new R(); + } + + @Override + public R put(String key, Object value) { + super.put(key, value); + return this; + } + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableDataInfo.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableDataInfo.java new file mode 100644 index 0000000..847685b --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableDataInfo.java @@ -0,0 +1,85 @@ +package com.ruoyi.common.core.page; + +import java.io.Serializable; +import java.util.List; + +/** + * 表格分页数据对象 + * + * @author ruoyi + */ +public class TableDataInfo implements Serializable +{ + private static final long serialVersionUID = 1L; + + /** 总记录数 */ + private long total; + + /** 列表数据 */ + private List rows; + + /** 消息状态码 */ + private int code; + + /** 消息内容 */ + private String msg; + + /** + * 表格数据对象 + */ + public TableDataInfo() + { + } + + /** + * 分页 + * + * @param list 列表数据 + * @param total 总记录数 + */ + public TableDataInfo(List list, int total) + { + this.rows = list; + this.total = total; + } + + public long getTotal() + { + return total; + } + + public void setTotal(long total) + { + this.total = total; + } + + public List getRows() + { + return rows; + } + + public void setRows(List rows) + { + this.rows = rows; + } + + public int getCode() + { + return code; + } + + public void setCode(int code) + { + this.code = code; + } + + public String getMsg() + { + return msg; + } + + public void setMsg(String msg) + { + this.msg = msg; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableSupport.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableSupport.java new file mode 100644 index 0000000..a120c30 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableSupport.java @@ -0,0 +1,56 @@ +package com.ruoyi.common.core.page; + +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.utils.ServletUtils; + +/** + * 表格数据处理 + * + * @author ruoyi + */ +public class TableSupport +{ + /** + * 当前记录起始索引 + */ + public static final String PAGE_NUM = "pageNum"; + + /** + * 每页显示记录数 + */ + public static final String PAGE_SIZE = "pageSize"; + + /** + * 排序列 + */ + public static final String ORDER_BY_COLUMN = "orderByColumn"; + + /** + * 排序的方向 "desc" 或者 "asc". + */ + public static final String IS_ASC = "isAsc"; + + /** + * 分页参数合理化 + */ + public static final String REASONABLE = "reasonable"; + + /** + * 封装分页对象 + */ + public static PageDomain getPageDomain() + { + PageDomain pageDomain = new PageDomain(); + pageDomain.setPageNum(Convert.toInt(ServletUtils.getParameter(PAGE_NUM), 1)); + pageDomain.setPageSize(Convert.toInt(ServletUtils.getParameter(PAGE_SIZE), 10)); + pageDomain.setOrderByColumn(ServletUtils.getParameter(ORDER_BY_COLUMN)); + pageDomain.setIsAsc(ServletUtils.getParameter(IS_ASC)); + pageDomain.setReasonable(ServletUtils.getParameterToBool(REASONABLE)); + return pageDomain; + } + + public static PageDomain buildPageRequest() + { + return getPageDomain(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisCache.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisCache.java new file mode 100644 index 0000000..de4a9d4 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisCache.java @@ -0,0 +1,246 @@ +package com.ruoyi.common.core.redis; + +import java.util.Collection; +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.concurrent.TimeUnit; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.redis.core.BoundSetOperations; +import org.springframework.data.redis.core.HashOperations; +import org.springframework.data.redis.core.RedisTemplate; +import org.springframework.data.redis.core.ValueOperations; +import org.springframework.stereotype.Component; + +/** + * spring redis 工具类 + * + * @author ruoyi + **/ +@SuppressWarnings(value = { "unchecked", "rawtypes" }) +@Component +public class RedisCache +{ + @Autowired + public RedisTemplate redisTemplate; + + /** + * 缓存基本的对象,Integer、String、实体类等 + * + * @param key 缓存的键值 + * @param value 缓存的值 + */ + public void setCacheObject(final String key, final T value) + { + redisTemplate.opsForValue().set(key, value); + } + + /** + * 缓存基本的对象,Integer、String、实体类等 + * + * @param key 缓存的键值 + * @param value 缓存的值 + * @param timeout 时间 + * @param timeUnit 时间颗粒度 + */ + public void setCacheObject(final String key, final T value, final Integer timeout, final TimeUnit timeUnit) + { + redisTemplate.opsForValue().set(key, value, timeout, timeUnit); + } + + /** + * 设置有效时间 + * + * @param key Redis键 + * @param timeout 超时时间 + * @return true=设置成功;false=设置失败 + */ + public boolean expire(final String key, final long timeout) + { + return expire(key, timeout, TimeUnit.SECONDS); + } + + /** + * 设置有效时间 + * + * @param key Redis键 + * @param timeout 超时时间 + * @param unit 时间单位 + * @return true=设置成功;false=设置失败 + */ + public boolean expire(final String key, final long timeout, final TimeUnit unit) + { + return redisTemplate.expire(key, timeout, unit); + } + + /** + * 获得缓存的基本对象。 + * + * @param key 缓存键值 + * @return 缓存键值对应的数据 + */ + public T getCacheObject(final String key) + { + ValueOperations operation = redisTemplate.opsForValue(); + return operation.get(key); + } + + /** + * 删除单个对象 + * + * @param key + */ + public boolean deleteObject(final String key) + { + return redisTemplate.delete(key); + } + + /** + * 删除集合对象 + * + * @param collection 多个对象 + * @return + */ + public long deleteObject(final Collection collection) + { + return redisTemplate.delete(collection); + } + + /** + * 缓存List数据 + * + * @param key 缓存的键值 + * @param dataList 待缓存的List数据 + * @return 缓存的对象 + */ + public long setCacheList(final String key, final List dataList) + { + Long count = redisTemplate.opsForList().rightPushAll(key, dataList); + return count == null ? 0 : count; + } + + /** + * 获得缓存的list对象 + * + * @param key 缓存的键值 + * @return 缓存键值对应的数据 + */ + public List getCacheList(final String key) + { + return redisTemplate.opsForList().range(key, 0, -1); + } + + /** + * 缓存Set + * + * @param key 缓存键值 + * @param dataSet 缓存的数据 + * @return 缓存数据的对象 + */ + public BoundSetOperations setCacheSet(final String key, final Set dataSet) + { + BoundSetOperations setOperation = redisTemplate.boundSetOps(key); + Iterator it = dataSet.iterator(); + while (it.hasNext()) + { + setOperation.add(it.next()); + } + return setOperation; + } + + /** + * 获得缓存的set + * + * @param key + * @return + */ + public Set getCacheSet(final String key) + { + return redisTemplate.opsForSet().members(key); + } + + /** + * 缓存Map + * + * @param key + * @param dataMap + */ + public void setCacheMap(final String key, final Map dataMap) + { + if (dataMap != null) { + redisTemplate.opsForHash().putAll(key, dataMap); + } + } + + /** + * 获得缓存的Map + * + * @param key + * @return + */ + public Map getCacheMap(final String key) + { + return redisTemplate.opsForHash().entries(key); + } + + /** + * 往Hash中存入数据 + * + * @param key Redis键 + * @param hKey Hash键 + * @param value 值 + */ + public void setCacheMapValue(final String key, final String hKey, final T value) + { + redisTemplate.opsForHash().put(key, hKey, value); + } + + /** + * 获取Hash中的数据 + * + * @param key Redis键 + * @param hKey Hash键 + * @return Hash中的对象 + */ + public T getCacheMapValue(final String key, final String hKey) + { + HashOperations opsForHash = redisTemplate.opsForHash(); + return opsForHash.get(key, hKey); + } + + /** + * 删除Hash中的数据 + * + * @param key + * @param hKey + */ + public void delCacheMapValue(final String key, final String hKey) + { + HashOperations hashOperations = redisTemplate.opsForHash(); + hashOperations.delete(key, hKey); + } + + /** + * 获取多个Hash中的数据 + * + * @param key Redis键 + * @param hKeys Hash键集合 + * @return Hash对象集合 + */ + public List getMultiCacheMapValue(final String key, final Collection hKeys) + { + return redisTemplate.opsForHash().multiGet(key, hKeys); + } + + /** + * 获得缓存的基本对象列表 + * + * @param pattern 字符串前缀 + * @return 对象列表 + */ + public Collection keys(final String pattern) + { + return redisTemplate.keys(pattern); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/text/CharsetKit.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/CharsetKit.java new file mode 100644 index 0000000..84124aa --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/CharsetKit.java @@ -0,0 +1,86 @@ +package com.ruoyi.common.core.text; + +import java.nio.charset.Charset; +import java.nio.charset.StandardCharsets; +import com.ruoyi.common.utils.StringUtils; + +/** + * 字符集工具类 + * + * @author ruoyi + */ +public class CharsetKit +{ + /** ISO-8859-1 */ + public static final String ISO_8859_1 = "ISO-8859-1"; + /** UTF-8 */ + public static final String UTF_8 = "UTF-8"; + /** GBK */ + public static final String GBK = "GBK"; + + /** ISO-8859-1 */ + public static final Charset CHARSET_ISO_8859_1 = Charset.forName(ISO_8859_1); + /** UTF-8 */ + public static final Charset CHARSET_UTF_8 = Charset.forName(UTF_8); + /** GBK */ + public static final Charset CHARSET_GBK = Charset.forName(GBK); + + /** + * 转换为Charset对象 + * + * @param charset 字符集,为空则返回默认字符集 + * @return Charset + */ + public static Charset charset(String charset) + { + return StringUtils.isEmpty(charset) ? Charset.defaultCharset() : Charset.forName(charset); + } + + /** + * 转换字符串的字符集编码 + * + * @param source 字符串 + * @param srcCharset 源字符集,默认ISO-8859-1 + * @param destCharset 目标字符集,默认UTF-8 + * @return 转换后的字符集 + */ + public static String convert(String source, String srcCharset, String destCharset) + { + return convert(source, Charset.forName(srcCharset), Charset.forName(destCharset)); + } + + /** + * 转换字符串的字符集编码 + * + * @param source 字符串 + * @param srcCharset 源字符集,默认ISO-8859-1 + * @param destCharset 目标字符集,默认UTF-8 + * @return 转换后的字符集 + */ + public static String convert(String source, Charset srcCharset, Charset destCharset) + { + if (null == srcCharset) + { + srcCharset = StandardCharsets.ISO_8859_1; + } + + if (null == destCharset) + { + destCharset = StandardCharsets.UTF_8; + } + + if (StringUtils.isEmpty(source) || srcCharset.equals(destCharset)) + { + return source; + } + return new String(source.getBytes(srcCharset), destCharset); + } + + /** + * @return 系统字符集编码 + */ + public static String systemCharset() + { + return Charset.defaultCharset().name(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/text/Convert.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/Convert.java new file mode 100644 index 0000000..1fb7461 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/Convert.java @@ -0,0 +1,1005 @@ +package com.ruoyi.common.core.text; + +import java.math.BigDecimal; +import java.math.BigInteger; +import java.nio.ByteBuffer; +import java.nio.charset.Charset; +import java.text.NumberFormat; +import java.util.Set; +import com.ruoyi.common.utils.StringUtils; +import org.apache.commons.lang3.ArrayUtils; + +/** + * 类型转换器 + * + * @author ruoyi + */ +public class Convert +{ + /** + * 转换为字符串
+ * 如果给定的值为null,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static String toStr(Object value, String defaultValue) + { + if (null == value) + { + return defaultValue; + } + if (value instanceof String) + { + return (String) value; + } + return value.toString(); + } + + /** + * 转换为字符串
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static String toStr(Object value) + { + return toStr(value, null); + } + + /** + * 转换为字符
+ * 如果给定的值为null,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Character toChar(Object value, Character defaultValue) + { + if (null == value) + { + return defaultValue; + } + if (value instanceof Character) + { + return (Character) value; + } + + final String valueStr = toStr(value, null); + return StringUtils.isEmpty(valueStr) ? defaultValue : valueStr.charAt(0); + } + + /** + * 转换为字符
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Character toChar(Object value) + { + return toChar(value, null); + } + + /** + * 转换为byte
+ * 如果给定的值为null,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Byte toByte(Object value, Byte defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Byte) + { + return (Byte) value; + } + if (value instanceof Number) + { + return ((Number) value).byteValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return Byte.parseByte(valueStr); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为byte
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Byte toByte(Object value) + { + return toByte(value, null); + } + + /** + * 转换为Short
+ * 如果给定的值为null,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Short toShort(Object value, Short defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Short) + { + return (Short) value; + } + if (value instanceof Number) + { + return ((Number) value).shortValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return Short.parseShort(valueStr.trim()); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为Short
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Short toShort(Object value) + { + return toShort(value, null); + } + + /** + * 转换为Number
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Number toNumber(Object value, Number defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Number) + { + return (Number) value; + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return NumberFormat.getInstance().parse(valueStr); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为Number
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Number toNumber(Object value) + { + return toNumber(value, null); + } + + /** + * 转换为int
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Integer toInt(Object value, Integer defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Integer) + { + return (Integer) value; + } + if (value instanceof Number) + { + return ((Number) value).intValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return Integer.parseInt(valueStr.trim()); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为int
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Integer toInt(Object value) + { + return toInt(value, null); + } + + /** + * 转换为Integer数组
+ * + * @param str 被转换的值 + * @return 结果 + */ + public static Integer[] toIntArray(String str) + { + return toIntArray(",", str); + } + + /** + * 转换为Long数组
+ * + * @param str 被转换的值 + * @return 结果 + */ + public static Long[] toLongArray(String str) + { + return toLongArray(",", str); + } + + /** + * 转换为Integer数组
+ * + * @param split 分隔符 + * @param split 被转换的值 + * @return 结果 + */ + public static Integer[] toIntArray(String split, String str) + { + if (StringUtils.isEmpty(str)) + { + return new Integer[] {}; + } + String[] arr = str.split(split); + final Integer[] ints = new Integer[arr.length]; + for (int i = 0; i < arr.length; i++) + { + final Integer v = toInt(arr[i], 0); + ints[i] = v; + } + return ints; + } + + /** + * 转换为Long数组
+ * + * @param split 分隔符 + * @param str 被转换的值 + * @return 结果 + */ + public static Long[] toLongArray(String split, String str) + { + if (StringUtils.isEmpty(str)) + { + return new Long[] {}; + } + String[] arr = str.split(split); + final Long[] longs = new Long[arr.length]; + for (int i = 0; i < arr.length; i++) + { + final Long v = toLong(arr[i], null); + longs[i] = v; + } + return longs; + } + + /** + * 转换为String数组
+ * + * @param str 被转换的值 + * @return 结果 + */ + public static String[] toStrArray(String str) + { + return toStrArray(",", str); + } + + /** + * 转换为String数组
+ * + * @param split 分隔符 + * @param split 被转换的值 + * @return 结果 + */ + public static String[] toStrArray(String split, String str) + { + return str.split(split); + } + + /** + * 转换为long
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Long toLong(Object value, Long defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Long) + { + return (Long) value; + } + if (value instanceof Number) + { + return ((Number) value).longValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + // 支持科学计数法 + return new BigDecimal(valueStr.trim()).longValue(); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为long
+ * 如果给定的值为null,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Long toLong(Object value) + { + return toLong(value, null); + } + + /** + * 转换为double
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Double toDouble(Object value, Double defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Double) + { + return (Double) value; + } + if (value instanceof Number) + { + return ((Number) value).doubleValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + // 支持科学计数法 + return new BigDecimal(valueStr.trim()).doubleValue(); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为double
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Double toDouble(Object value) + { + return toDouble(value, null); + } + + /** + * 转换为Float
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Float toFloat(Object value, Float defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Float) + { + return (Float) value; + } + if (value instanceof Number) + { + return ((Number) value).floatValue(); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return Float.parseFloat(valueStr.trim()); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为Float
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Float toFloat(Object value) + { + return toFloat(value, null); + } + + /** + * 转换为boolean
+ * String支持的值为:true、false、yes、ok、no,1,0 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static Boolean toBool(Object value, Boolean defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof Boolean) + { + return (Boolean) value; + } + String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + valueStr = valueStr.trim().toLowerCase(); + switch (valueStr) + { + case "true": + return true; + case "false": + return false; + case "yes": + return true; + case "ok": + return true; + case "no": + return false; + case "1": + return true; + case "0": + return false; + default: + return defaultValue; + } + } + + /** + * 转换为boolean
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static Boolean toBool(Object value) + { + return toBool(value, null); + } + + /** + * 转换为Enum对象
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * + * @param clazz Enum的Class + * @param value 值 + * @param defaultValue 默认值 + * @return Enum + */ + public static > E toEnum(Class clazz, Object value, E defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (clazz.isAssignableFrom(value.getClass())) + { + @SuppressWarnings("unchecked") + E myE = (E) value; + return myE; + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return Enum.valueOf(clazz, valueStr); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为Enum对象
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * + * @param clazz Enum的Class + * @param value 值 + * @return Enum + */ + public static > E toEnum(Class clazz, Object value) + { + return toEnum(clazz, value, null); + } + + /** + * 转换为BigInteger
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static BigInteger toBigInteger(Object value, BigInteger defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof BigInteger) + { + return (BigInteger) value; + } + if (value instanceof Long) + { + return BigInteger.valueOf((Long) value); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return new BigInteger(valueStr); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为BigInteger
+ * 如果给定的值为空,或者转换失败,返回默认值null
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static BigInteger toBigInteger(Object value) + { + return toBigInteger(value, null); + } + + /** + * 转换为BigDecimal
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @param defaultValue 转换错误时的默认值 + * @return 结果 + */ + public static BigDecimal toBigDecimal(Object value, BigDecimal defaultValue) + { + if (value == null) + { + return defaultValue; + } + if (value instanceof BigDecimal) + { + return (BigDecimal) value; + } + if (value instanceof Long) + { + return new BigDecimal((Long) value); + } + if (value instanceof Double) + { + return new BigDecimal((Double) value); + } + if (value instanceof Integer) + { + return new BigDecimal((Integer) value); + } + final String valueStr = toStr(value, null); + if (StringUtils.isEmpty(valueStr)) + { + return defaultValue; + } + try + { + return new BigDecimal(valueStr); + } + catch (Exception e) + { + return defaultValue; + } + } + + /** + * 转换为BigDecimal
+ * 如果给定的值为空,或者转换失败,返回默认值
+ * 转换失败不会报错 + * + * @param value 被转换的值 + * @return 结果 + */ + public static BigDecimal toBigDecimal(Object value) + { + return toBigDecimal(value, null); + } + + /** + * 将对象转为字符串
+ * 1、Byte数组和ByteBuffer会被转换为对应字符串的数组 2、对象数组会调用Arrays.toString方法 + * + * @param obj 对象 + * @return 字符串 + */ + public static String utf8Str(Object obj) + { + return str(obj, CharsetKit.CHARSET_UTF_8); + } + + /** + * 将对象转为字符串
+ * 1、Byte数组和ByteBuffer会被转换为对应字符串的数组 2、对象数组会调用Arrays.toString方法 + * + * @param obj 对象 + * @param charsetName 字符集 + * @return 字符串 + */ + public static String str(Object obj, String charsetName) + { + return str(obj, Charset.forName(charsetName)); + } + + /** + * 将对象转为字符串
+ * 1、Byte数组和ByteBuffer会被转换为对应字符串的数组 2、对象数组会调用Arrays.toString方法 + * + * @param obj 对象 + * @param charset 字符集 + * @return 字符串 + */ + public static String str(Object obj, Charset charset) + { + if (null == obj) + { + return null; + } + + if (obj instanceof String) + { + return (String) obj; + } + else if (obj instanceof byte[]) + { + return str((byte[]) obj, charset); + } + else if (obj instanceof Byte[]) + { + byte[] bytes = ArrayUtils.toPrimitive((Byte[]) obj); + return str(bytes, charset); + } + else if (obj instanceof ByteBuffer) + { + return str((ByteBuffer) obj, charset); + } + return obj.toString(); + } + + /** + * 将byte数组转为字符串 + * + * @param bytes byte数组 + * @param charset 字符集 + * @return 字符串 + */ + public static String str(byte[] bytes, String charset) + { + return str(bytes, StringUtils.isEmpty(charset) ? Charset.defaultCharset() : Charset.forName(charset)); + } + + /** + * 解码字节码 + * + * @param data 字符串 + * @param charset 字符集,如果此字段为空,则解码的结果取决于平台 + * @return 解码后的字符串 + */ + public static String str(byte[] data, Charset charset) + { + if (data == null) + { + return null; + } + + if (null == charset) + { + return new String(data); + } + return new String(data, charset); + } + + /** + * 将编码的byteBuffer数据转换为字符串 + * + * @param data 数据 + * @param charset 字符集,如果为空使用当前系统字符集 + * @return 字符串 + */ + public static String str(ByteBuffer data, String charset) + { + if (data == null) + { + return null; + } + + return str(data, Charset.forName(charset)); + } + + /** + * 将编码的byteBuffer数据转换为字符串 + * + * @param data 数据 + * @param charset 字符集,如果为空使用当前系统字符集 + * @return 字符串 + */ + public static String str(ByteBuffer data, Charset charset) + { + if (null == charset) + { + charset = Charset.defaultCharset(); + } + return charset.decode(data).toString(); + } + + // ----------------------------------------------------------------------- 全角半角转换 + /** + * 半角转全角 + * + * @param input String. + * @return 全角字符串. + */ + public static String toSBC(String input) + { + return toSBC(input, null); + } + + /** + * 半角转全角 + * + * @param input String + * @param notConvertSet 不替换的字符集合 + * @return 全角字符串. + */ + public static String toSBC(String input, Set notConvertSet) + { + char c[] = input.toCharArray(); + for (int i = 0; i < c.length; i++) + { + if (null != notConvertSet && notConvertSet.contains(c[i])) + { + // 跳过不替换的字符 + continue; + } + + if (c[i] == ' ') + { + c[i] = '\u3000'; + } + else if (c[i] < '\177') + { + c[i] = (char) (c[i] + 65248); + + } + } + return new String(c); + } + + /** + * 全角转半角 + * + * @param input String. + * @return 半角字符串 + */ + public static String toDBC(String input) + { + return toDBC(input, null); + } + + /** + * 替换全角为半角 + * + * @param text 文本 + * @param notConvertSet 不替换的字符集合 + * @return 替换后的字符 + */ + public static String toDBC(String text, Set notConvertSet) + { + char c[] = text.toCharArray(); + for (int i = 0; i < c.length; i++) + { + if (null != notConvertSet && notConvertSet.contains(c[i])) + { + // 跳过不替换的字符 + continue; + } + + if (c[i] == '\u3000') + { + c[i] = ' '; + } + else if (c[i] > '\uFF00' && c[i] < '\uFF5F') + { + c[i] = (char) (c[i] - 65248); + } + } + String returnString = new String(c); + + return returnString; + } + + /** + * 数字金额大写转换 先写个完整的然后将如零拾替换成零 + * + * @param n 数字 + * @return 中文大写数字 + */ + public static String digitUppercase(double n) + { + String[] fraction = { "角", "分" }; + String[] digit = { "零", "壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖" }; + String[][] unit = { { "元", "万", "亿" }, { "", "拾", "佰", "仟" } }; + + String head = n < 0 ? "负" : ""; + n = Math.abs(n); + + String s = ""; + for (int i = 0; i < fraction.length; i++) + { + s += (digit[(int) (Math.floor(n * 10 * Math.pow(10, i)) % 10)] + fraction[i]).replaceAll("(零.)+", ""); + } + if (s.length() < 1) + { + s = "整"; + } + int integerPart = (int) Math.floor(n); + + for (int i = 0; i < unit[0].length && integerPart > 0; i++) + { + String p = ""; + for (int j = 0; j < unit[1].length && n > 0; j++) + { + p = digit[integerPart % 10] + unit[1][j] + p; + integerPart = integerPart / 10; + } + s = p.replaceAll("(零.)*零$", "").replaceAll("^$", "零") + unit[0][i] + s; + } + return head + s.replaceAll("(零.)*零元", "元").replaceFirst("(零.)+", "").replaceAll("(零.)+", "零").replaceAll("^整$", "零元整"); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/text/StrFormatter.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/StrFormatter.java new file mode 100644 index 0000000..c78ac77 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/text/StrFormatter.java @@ -0,0 +1,92 @@ +package com.ruoyi.common.core.text; + +import com.ruoyi.common.utils.StringUtils; + +/** + * 字符串格式化 + * + * @author ruoyi + */ +public class StrFormatter +{ + public static final String EMPTY_JSON = "{}"; + public static final char C_BACKSLASH = '\\'; + public static final char C_DELIM_START = '{'; + public static final char C_DELIM_END = '}'; + + /** + * 格式化字符串
+ * 此方法只是简单将占位符 {} 按照顺序替换为参数
+ * 如果想输出 {} 使用 \\转义 { 即可,如果想输出 {} 之前的 \ 使用双转义符 \\\\ 即可
+ * 例:
+ * 通常使用:format("this is {} for {}", "a", "b") -> this is a for b
+ * 转义{}: format("this is \\{} for {}", "a", "b") -> this is \{} for a
+ * 转义\: format("this is \\\\{} for {}", "a", "b") -> this is \a for b
+ * + * @param strPattern 字符串模板 + * @param argArray 参数列表 + * @return 结果 + */ + public static String format(final String strPattern, final Object... argArray) + { + if (StringUtils.isEmpty(strPattern) || StringUtils.isEmpty(argArray)) + { + return strPattern; + } + final int strPatternLength = strPattern.length(); + + // 初始化定义好的长度以获得更好的性能 + StringBuilder sbuf = new StringBuilder(strPatternLength + 50); + + int handledPosition = 0; + int delimIndex;// 占位符所在位置 + for (int argIndex = 0; argIndex < argArray.length; argIndex++) + { + delimIndex = strPattern.indexOf(EMPTY_JSON, handledPosition); + if (delimIndex == -1) + { + if (handledPosition == 0) + { + return strPattern; + } + else + { // 字符串模板剩余部分不再包含占位符,加入剩余部分后返回结果 + sbuf.append(strPattern, handledPosition, strPatternLength); + return sbuf.toString(); + } + } + else + { + if (delimIndex > 0 && strPattern.charAt(delimIndex - 1) == C_BACKSLASH) + { + if (delimIndex > 1 && strPattern.charAt(delimIndex - 2) == C_BACKSLASH) + { + // 转义符之前还有一个转义符,占位符依旧有效 + sbuf.append(strPattern, handledPosition, delimIndex - 1); + sbuf.append(Convert.utf8Str(argArray[argIndex])); + handledPosition = delimIndex + 2; + } + else + { + // 占位符被转义 + argIndex--; + sbuf.append(strPattern, handledPosition, delimIndex - 1); + sbuf.append(C_DELIM_START); + handledPosition = delimIndex + 1; + } + } + else + { + // 正常占位符 + sbuf.append(strPattern, handledPosition, delimIndex); + sbuf.append(Convert.utf8Str(argArray[argIndex])); + handledPosition = delimIndex + 2; + } + } + } + // 加入最后一个占位符后所有的字符 + sbuf.append(strPattern, handledPosition, strPattern.length()); + + return sbuf.toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessStatus.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessStatus.java new file mode 100644 index 0000000..10b7306 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessStatus.java @@ -0,0 +1,20 @@ +package com.ruoyi.common.enums; + +/** + * 操作状态 + * + * @author ruoyi + * + */ +public enum BusinessStatus +{ + /** + * 成功 + */ + SUCCESS, + + /** + * 失败 + */ + FAIL, +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessType.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessType.java new file mode 100644 index 0000000..2e17c4a --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/BusinessType.java @@ -0,0 +1,59 @@ +package com.ruoyi.common.enums; + +/** + * 业务操作类型 + * + * @author ruoyi + */ +public enum BusinessType +{ + /** + * 其它 + */ + OTHER, + + /** + * 新增 + */ + INSERT, + + /** + * 修改 + */ + UPDATE, + + /** + * 删除 + */ + DELETE, + + /** + * 授权 + */ + GRANT, + + /** + * 导出 + */ + EXPORT, + + /** + * 导入 + */ + IMPORT, + + /** + * 强退 + */ + FORCE, + + /** + * 生成代码 + */ + GENCODE, + + /** + * 清空数据 + */ + CLEAN, +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/DataSourceType.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/DataSourceType.java new file mode 100644 index 0000000..e7cc6bb --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/DataSourceType.java @@ -0,0 +1,23 @@ +package com.ruoyi.common.enums; + +/** + * 数据源 + * + * @author ruoyi + */ +public enum DataSourceType +{ + /** + * 主库 + */ + MASTER, + + /** + * 从库 + */ + SLAVE, + /** + * 分库分表 + */ + SHARDING +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/HttpMethod.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/HttpMethod.java new file mode 100644 index 0000000..be6f739 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/HttpMethod.java @@ -0,0 +1,36 @@ +package com.ruoyi.common.enums; + +import java.util.HashMap; +import java.util.Map; +import org.springframework.lang.Nullable; + +/** + * 请求方式 + * + * @author ruoyi + */ +public enum HttpMethod +{ + GET, HEAD, POST, PUT, PATCH, DELETE, OPTIONS, TRACE; + + private static final Map mappings = new HashMap<>(16); + + static + { + for (HttpMethod httpMethod : values()) + { + mappings.put(httpMethod.name(), httpMethod); + } + } + + @Nullable + public static HttpMethod resolve(@Nullable String method) + { + return (method != null ? mappings.get(method) : null); + } + + public boolean matches(String method) + { + return (this == resolve(method)); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/LimitType.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/LimitType.java new file mode 100644 index 0000000..c609fd8 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/LimitType.java @@ -0,0 +1,20 @@ +package com.ruoyi.common.enums; + +/** + * 限流类型 + * + * @author ruoyi + */ + +public enum LimitType +{ + /** + * 默认策略全局限流 + */ + DEFAULT, + + /** + * 根据请求者IP进行限流 + */ + IP +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/OperatorType.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/OperatorType.java new file mode 100644 index 0000000..bdd143c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/OperatorType.java @@ -0,0 +1,24 @@ +package com.ruoyi.common.enums; + +/** + * 操作人类别 + * + * @author ruoyi + */ +public enum OperatorType +{ + /** + * 其它 + */ + OTHER, + + /** + * 后台用户 + */ + MANAGE, + + /** + * 手机端用户 + */ + MOBILE +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/enums/UserStatus.java b/ruoyi-common/src/main/java/com/ruoyi/common/enums/UserStatus.java new file mode 100644 index 0000000..d7ff44a --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/enums/UserStatus.java @@ -0,0 +1,30 @@ +package com.ruoyi.common.enums; + +/** + * 用户状态 + * + * @author ruoyi + */ +public enum UserStatus +{ + OK("0", "正常"), DISABLE("1", "停用"), DELETED("2", "删除"); + + private final String code; + private final String info; + + UserStatus(String code, String info) + { + this.code = code; + this.info = info; + } + + public String getCode() + { + return code; + } + + public String getInfo() + { + return info; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/DemoModeException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/DemoModeException.java new file mode 100644 index 0000000..f6ad2ab --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/DemoModeException.java @@ -0,0 +1,15 @@ +package com.ruoyi.common.exception; + +/** + * 演示模式异常 + * + * @author ruoyi + */ +public class DemoModeException extends RuntimeException +{ + private static final long serialVersionUID = 1L; + + public DemoModeException() + { + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/GlobalException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/GlobalException.java new file mode 100644 index 0000000..211441b --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/GlobalException.java @@ -0,0 +1,58 @@ +package com.ruoyi.common.exception; + +/** + * 全局异常 + * + * @author ruoyi + */ +public class GlobalException extends RuntimeException +{ + + private static final long serialVersionUID = 1L; + + /** + * 错误提示 + */ + private String message; + + /** + * 错误明细,内部调试错误 + * + * 和 {@link CommonResult#getDetailMessage()} 一致的设计 + */ + private String detailMessage; + + /** + * 空构造方法,避免反序列化问题 + */ + public GlobalException() + { + } + + public GlobalException(String message) + { + this.message = message; + } + + public String getDetailMessage() + { + return detailMessage; + } + + public GlobalException setDetailMessage(String detailMessage) + { + this.detailMessage = detailMessage; + return this; + } + + public String getMessage() + { + return message; + } + + public GlobalException setMessage(String message) + { + this.message = message; + return this; + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/RYException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/RYException.java new file mode 100644 index 0000000..7bb2fb5 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/RYException.java @@ -0,0 +1,61 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.common.exception; + +/** + * 自定义异常 + * + * @author Mark sunlightcs@gmail.com + */ +public class RYException extends RuntimeException { + private static final long serialVersionUID = 1L; + + private String msg; + private int code = 500; + + public RYException(String msg) { + super(msg); + this.msg = msg; + } + + public RYException(String msg, Throwable e) { + super(msg, e); + this.msg = msg; + } + + public RYException(String msg, int code) { + super(msg); + this.msg = msg; + this.code = code; + } + + public RYException(String msg, int code, Throwable e) { + super(msg, e); + this.msg = msg; + this.code = code; + } + + public String getMsg() { + return msg; + } + + public void setMsg(String msg) { + this.msg = msg; + } + + public int getCode() { + return code; + } + + public void setCode(int code) { + this.code = code; + } + + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/ServiceException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/ServiceException.java new file mode 100644 index 0000000..6297f87 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/ServiceException.java @@ -0,0 +1,73 @@ +package com.ruoyi.common.exception; + +/** + * 业务异常 + * + * @author ruoyi + */ +public final class ServiceException extends RuntimeException +{ + private static final long serialVersionUID = 1L; + + /** + * 错误码 + */ + private Integer code; + + /** + * 错误提示 + */ + private String message; + + /** + * 错误明细,内部调试错误 + * + * 和 {@link CommonResult#getDetailMessage()} 一致的设计 + */ + private String detailMessage; + + /** + * 空构造方法,避免反序列化问题 + */ + public ServiceException() + { + } + + public ServiceException(String message) + { + this.message = message; + } + + public ServiceException(String message, Integer code) + { + this.message = message; + this.code = code; + } + + public String getDetailMessage() + { + return detailMessage; + } + + public String getMessage() + { + return message; + } + + public Integer getCode() + { + return code; + } + + public ServiceException setMessage(String message) + { + this.message = message; + return this; + } + + public ServiceException setDetailMessage(String detailMessage) + { + this.detailMessage = detailMessage; + return this; + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/UtilException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/UtilException.java new file mode 100644 index 0000000..980fa46 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/UtilException.java @@ -0,0 +1,26 @@ +package com.ruoyi.common.exception; + +/** + * 工具类异常 + * + * @author ruoyi + */ +public class UtilException extends RuntimeException +{ + private static final long serialVersionUID = 8247610319171014183L; + + public UtilException(Throwable e) + { + super(e.getMessage(), e); + } + + public UtilException(String message) + { + super(message); + } + + public UtilException(String message, Throwable throwable) + { + super(message, throwable); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/base/BaseException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/base/BaseException.java new file mode 100644 index 0000000..b55d72e --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/base/BaseException.java @@ -0,0 +1,97 @@ +package com.ruoyi.common.exception.base; + +import com.ruoyi.common.utils.MessageUtils; +import com.ruoyi.common.utils.StringUtils; + +/** + * 基础异常 + * + * @author ruoyi + */ +public class BaseException extends RuntimeException +{ + private static final long serialVersionUID = 1L; + + /** + * 所属模块 + */ + private String module; + + /** + * 错误码 + */ + private String code; + + /** + * 错误码对应的参数 + */ + private Object[] args; + + /** + * 错误消息 + */ + private String defaultMessage; + + public BaseException(String module, String code, Object[] args, String defaultMessage) + { + this.module = module; + this.code = code; + this.args = args; + this.defaultMessage = defaultMessage; + } + + public BaseException(String module, String code, Object[] args) + { + this(module, code, args, null); + } + + public BaseException(String module, String defaultMessage) + { + this(module, null, null, defaultMessage); + } + + public BaseException(String code, Object[] args) + { + this(null, code, args, null); + } + + public BaseException(String defaultMessage) + { + this(null, null, null, defaultMessage); + } + + @Override + public String getMessage() + { + String message = null; + if (!StringUtils.isEmpty(code)) + { + message = MessageUtils.message(code, args); + } + if (message == null) + { + message = defaultMessage; + } + return message; + } + + public String getModule() + { + return module; + } + + public String getCode() + { + return code; + } + + public Object[] getArgs() + { + return args; + } + + public String getDefaultMessage() + { + return defaultMessage; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileException.java new file mode 100644 index 0000000..871f09b --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileException.java @@ -0,0 +1,19 @@ +package com.ruoyi.common.exception.file; + +import com.ruoyi.common.exception.base.BaseException; + +/** + * 文件信息异常类 + * + * @author ruoyi + */ +public class FileException extends BaseException +{ + private static final long serialVersionUID = 1L; + + public FileException(String code, Object[] args) + { + super("file", code, args, null); + } + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileNameLengthLimitExceededException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileNameLengthLimitExceededException.java new file mode 100644 index 0000000..70e0ec9 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileNameLengthLimitExceededException.java @@ -0,0 +1,16 @@ +package com.ruoyi.common.exception.file; + +/** + * 文件名称超长限制异常类 + * + * @author ruoyi + */ +public class FileNameLengthLimitExceededException extends FileException +{ + private static final long serialVersionUID = 1L; + + public FileNameLengthLimitExceededException(int defaultFileNameLength) + { + super("upload.filename.exceed.length", new Object[] { defaultFileNameLength }); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileSizeLimitExceededException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileSizeLimitExceededException.java new file mode 100644 index 0000000..ec6ab05 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/FileSizeLimitExceededException.java @@ -0,0 +1,16 @@ +package com.ruoyi.common.exception.file; + +/** + * 文件名大小限制异常类 + * + * @author ruoyi + */ +public class FileSizeLimitExceededException extends FileException +{ + private static final long serialVersionUID = 1L; + + public FileSizeLimitExceededException(long defaultMaxSize) + { + super("upload.exceed.maxSize", new Object[] { defaultMaxSize }); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/InvalidExtensionException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/InvalidExtensionException.java new file mode 100644 index 0000000..f1c8e83 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/file/InvalidExtensionException.java @@ -0,0 +1,81 @@ +package com.ruoyi.common.exception.file; + +import java.util.Arrays; +import org.apache.commons.fileupload.FileUploadException; + +/** + * 文件上传 误异常类 + * + * @author ruoyi + */ +public class InvalidExtensionException extends FileUploadException +{ + private static final long serialVersionUID = 1L; + + private String[] allowedExtension; + private String extension; + private String filename; + + public InvalidExtensionException(String[] allowedExtension, String extension, String filename) + { + super("filename : [" + filename + "], extension : [" + extension + "], allowed extension : [" + Arrays.toString(allowedExtension) + "]"); + this.allowedExtension = allowedExtension; + this.extension = extension; + this.filename = filename; + } + + public String[] getAllowedExtension() + { + return allowedExtension; + } + + public String getExtension() + { + return extension; + } + + public String getFilename() + { + return filename; + } + + public static class InvalidImageExtensionException extends InvalidExtensionException + { + private static final long serialVersionUID = 1L; + + public InvalidImageExtensionException(String[] allowedExtension, String extension, String filename) + { + super(allowedExtension, extension, filename); + } + } + + public static class InvalidFlashExtensionException extends InvalidExtensionException + { + private static final long serialVersionUID = 1L; + + public InvalidFlashExtensionException(String[] allowedExtension, String extension, String filename) + { + super(allowedExtension, extension, filename); + } + } + + public static class InvalidMediaExtensionException extends InvalidExtensionException + { + private static final long serialVersionUID = 1L; + + public InvalidMediaExtensionException(String[] allowedExtension, String extension, String filename) + { + super(allowedExtension, extension, filename); + } + } + + public static class InvalidVideoExtensionException extends InvalidExtensionException + { + private static final long serialVersionUID = 1L; + + public InvalidVideoExtensionException(String[] allowedExtension, String extension, String filename) + { + super(allowedExtension, extension, filename); + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/job/TaskException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/job/TaskException.java new file mode 100644 index 0000000..a567b40 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/job/TaskException.java @@ -0,0 +1,34 @@ +package com.ruoyi.common.exception.job; + +/** + * 计划策略异常 + * + * @author ruoyi + */ +public class TaskException extends Exception +{ + private static final long serialVersionUID = 1L; + + private Code code; + + public TaskException(String msg, Code code) + { + this(msg, code, null); + } + + public TaskException(String msg, Code code, Exception nestedEx) + { + super(msg, nestedEx); + this.code = code; + } + + public Code getCode() + { + return code; + } + + public enum Code + { + TASK_EXISTS, NO_TASK_EXISTS, TASK_ALREADY_STARTED, UNKNOWN, CONFIG_ERROR, TASK_NODE_NOT_AVAILABLE + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaException.java new file mode 100644 index 0000000..389dbc7 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaException.java @@ -0,0 +1,16 @@ +package com.ruoyi.common.exception.user; + +/** + * 验证码错误异常类 + * + * @author ruoyi + */ +public class CaptchaException extends UserException +{ + private static final long serialVersionUID = 1L; + + public CaptchaException() + { + super("user.jcaptcha.error", null); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaExpireException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaExpireException.java new file mode 100644 index 0000000..85f9486 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/CaptchaExpireException.java @@ -0,0 +1,16 @@ +package com.ruoyi.common.exception.user; + +/** + * 验证码失效异常类 + * + * @author ruoyi + */ +public class CaptchaExpireException extends UserException +{ + private static final long serialVersionUID = 1L; + + public CaptchaExpireException() + { + super("user.jcaptcha.expire", null); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserException.java new file mode 100644 index 0000000..c292d70 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserException.java @@ -0,0 +1,18 @@ +package com.ruoyi.common.exception.user; + +import com.ruoyi.common.exception.base.BaseException; + +/** + * 用户信息异常类 + * + * @author ruoyi + */ +public class UserException extends BaseException +{ + private static final long serialVersionUID = 1L; + + public UserException(String code, Object[] args) + { + super("user", code, args, null); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserPasswordNotMatchException.java b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserPasswordNotMatchException.java new file mode 100644 index 0000000..a7f3e5f --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/exception/user/UserPasswordNotMatchException.java @@ -0,0 +1,16 @@ +package com.ruoyi.common.exception.user; + +/** + * 用户密码不正确或不符合规范异常类 + * + * @author ruoyi + */ +public class UserPasswordNotMatchException extends UserException +{ + private static final long serialVersionUID = 1L; + + public UserPasswordNotMatchException() + { + super("user.password.not.match", null); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatableFilter.java b/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatableFilter.java new file mode 100644 index 0000000..a1bcfe2 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatableFilter.java @@ -0,0 +1,52 @@ +package com.ruoyi.common.filter; + +import java.io.IOException; +import javax.servlet.Filter; +import javax.servlet.FilterChain; +import javax.servlet.FilterConfig; +import javax.servlet.ServletException; +import javax.servlet.ServletRequest; +import javax.servlet.ServletResponse; +import javax.servlet.http.HttpServletRequest; +import org.springframework.http.MediaType; +import com.ruoyi.common.utils.StringUtils; + +/** + * Repeatable 过滤器 + * + * @author ruoyi + */ +public class RepeatableFilter implements Filter +{ + @Override + public void init(FilterConfig filterConfig) throws ServletException + { + + } + + @Override + public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) + throws IOException, ServletException + { + ServletRequest requestWrapper = null; + if (request instanceof HttpServletRequest + && StringUtils.startsWithIgnoreCase(request.getContentType(), MediaType.APPLICATION_JSON_VALUE)) + { + requestWrapper = new RepeatedlyRequestWrapper((HttpServletRequest) request, response); + } + if (null == requestWrapper) + { + chain.doFilter(request, response); + } + else + { + chain.doFilter(requestWrapper, response); + } + } + + @Override + public void destroy() + { + + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatedlyRequestWrapper.java b/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatedlyRequestWrapper.java new file mode 100644 index 0000000..614c24c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/filter/RepeatedlyRequestWrapper.java @@ -0,0 +1,75 @@ +package com.ruoyi.common.filter; + +import java.io.BufferedReader; +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.io.InputStreamReader; +import javax.servlet.ReadListener; +import javax.servlet.ServletInputStream; +import javax.servlet.ServletResponse; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletRequestWrapper; +import com.ruoyi.common.utils.http.HttpHelper; + +/** + * 构建可重复读取inputStream的request + * + * @author ruoyi + */ +public class RepeatedlyRequestWrapper extends HttpServletRequestWrapper +{ + private final byte[] body; + + public RepeatedlyRequestWrapper(HttpServletRequest request, ServletResponse response) throws IOException + { + super(request); + request.setCharacterEncoding("UTF-8"); + response.setCharacterEncoding("UTF-8"); + + body = HttpHelper.getBodyString(request).getBytes("UTF-8"); + } + + @Override + public BufferedReader getReader() throws IOException + { + return new BufferedReader(new InputStreamReader(getInputStream())); + } + + @Override + public ServletInputStream getInputStream() throws IOException + { + final ByteArrayInputStream bais = new ByteArrayInputStream(body); + return new ServletInputStream() + { + @Override + public int read() throws IOException + { + return bais.read(); + } + + @Override + public int available() throws IOException + { + return body.length; + } + + @Override + public boolean isFinished() + { + return false; + } + + @Override + public boolean isReady() + { + return false; + } + + @Override + public void setReadListener(ReadListener readListener) + { + + } + }; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssFilter.java b/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssFilter.java new file mode 100644 index 0000000..99323ed --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssFilter.java @@ -0,0 +1,74 @@ +package com.ruoyi.common.filter; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.List; +import javax.servlet.Filter; +import javax.servlet.FilterChain; +import javax.servlet.FilterConfig; +import javax.servlet.ServletException; +import javax.servlet.ServletRequest; +import javax.servlet.ServletResponse; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import com.ruoyi.common.utils.StringUtils; + +/** + * 防止XSS攻击的过滤器 + * + * @author ruoyi + */ +public class XssFilter implements Filter +{ + /** + * 排除链接 + */ + public List excludes = new ArrayList<>(); + + @Override + public void init(FilterConfig filterConfig) throws ServletException + { + String tempExcludes = filterConfig.getInitParameter("excludes"); + if (StringUtils.isNotEmpty(tempExcludes)) + { + String[] url = tempExcludes.split(","); + for (int i = 0; url != null && i < url.length; i++) + { + excludes.add(url[i]); + } + } + } + + @Override + public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) + throws IOException, ServletException + { + HttpServletRequest req = (HttpServletRequest) request; + HttpServletResponse resp = (HttpServletResponse) response; + if (handleExcludeURL(req, resp)) + { + chain.doFilter(request, response); + return; + } + XssHttpServletRequestWrapper xssRequest = new XssHttpServletRequestWrapper((HttpServletRequest) request); + chain.doFilter(xssRequest, response); + } + + private boolean handleExcludeURL(HttpServletRequest request, HttpServletResponse response) + { + String url = request.getServletPath(); + String method = request.getMethod(); + // GET DELETE 不过滤 + if (method == null || method.matches("GET") || method.matches("DELETE")) + { + return true; + } + return StringUtils.matches(url, excludes); + } + + @Override + public void destroy() + { + + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssHttpServletRequestWrapper.java b/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssHttpServletRequestWrapper.java new file mode 100644 index 0000000..b1eeb65 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/filter/XssHttpServletRequestWrapper.java @@ -0,0 +1,111 @@ +package com.ruoyi.common.filter; + +import java.io.ByteArrayInputStream; +import java.io.IOException; +import javax.servlet.ReadListener; +import javax.servlet.ServletInputStream; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletRequestWrapper; +import org.apache.commons.io.IOUtils; +import org.springframework.http.HttpHeaders; +import org.springframework.http.MediaType; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.html.EscapeUtil; + +/** + * XSS过滤处理 + * + * @author ruoyi + */ +public class XssHttpServletRequestWrapper extends HttpServletRequestWrapper +{ + /** + * @param request + */ + public XssHttpServletRequestWrapper(HttpServletRequest request) + { + super(request); + } + + @Override + public String[] getParameterValues(String name) + { + String[] values = super.getParameterValues(name); + if (values != null) + { + int length = values.length; + String[] escapseValues = new String[length]; + for (int i = 0; i < length; i++) + { + // 防xss攻击和过滤前后空格 + escapseValues[i] = EscapeUtil.clean(values[i]).trim(); + } + return escapseValues; + } + return super.getParameterValues(name); + } + + @Override + public ServletInputStream getInputStream() throws IOException + { + // 非json类型,直接返回 + if (!isJsonRequest()) + { + return super.getInputStream(); + } + + // 为空,直接返回 + String json = IOUtils.toString(super.getInputStream(), "utf-8"); + if (StringUtils.isEmpty(json)) + { + return super.getInputStream(); + } + + // xss过滤 + json = EscapeUtil.clean(json).trim(); + byte[] jsonBytes = json.getBytes("utf-8"); + final ByteArrayInputStream bis = new ByteArrayInputStream(jsonBytes); + return new ServletInputStream() + { + @Override + public boolean isFinished() + { + return true; + } + + @Override + public boolean isReady() + { + return true; + } + + @Override + public int available() throws IOException + { + return jsonBytes.length; + } + + @Override + public void setReadListener(ReadListener readListener) + { + } + + @Override + public int read() throws IOException + { + return bis.read(); + } + }; + } + + /** + * 是否是Json请求 + * + * @param request + */ + public boolean isJsonRequest() + { + String header = super.getHeader(HttpHeaders.CONTENT_TYPE); + return StringUtils.startsWithIgnoreCase(header, MediaType.APPLICATION_JSON_VALUE); + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/Arith.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Arith.java new file mode 100644 index 0000000..b6326c2 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Arith.java @@ -0,0 +1,114 @@ +package com.ruoyi.common.utils; + +import java.math.BigDecimal; +import java.math.RoundingMode; + +/** + * 精确的浮点数运算 + * + * @author ruoyi + */ +public class Arith +{ + + /** 默认除法运算精度 */ + private static final int DEF_DIV_SCALE = 10; + + /** 这个类不能实例化 */ + private Arith() + { + } + + /** + * 提供精确的加法运算。 + * @param v1 被加数 + * @param v2 加数 + * @return 两个参数的和 + */ + public static double add(double v1, double v2) + { + BigDecimal b1 = new BigDecimal(Double.toString(v1)); + BigDecimal b2 = new BigDecimal(Double.toString(v2)); + return b1.add(b2).doubleValue(); + } + + /** + * 提供精确的减法运算。 + * @param v1 被减数 + * @param v2 减数 + * @return 两个参数的差 + */ + public static double sub(double v1, double v2) + { + BigDecimal b1 = new BigDecimal(Double.toString(v1)); + BigDecimal b2 = new BigDecimal(Double.toString(v2)); + return b1.subtract(b2).doubleValue(); + } + + /** + * 提供精确的乘法运算。 + * @param v1 被乘数 + * @param v2 乘数 + * @return 两个参数的积 + */ + public static double mul(double v1, double v2) + { + BigDecimal b1 = new BigDecimal(Double.toString(v1)); + BigDecimal b2 = new BigDecimal(Double.toString(v2)); + return b1.multiply(b2).doubleValue(); + } + + /** + * 提供(相对)精确的除法运算,当发生除不尽的情况时,精确到 + * 小数点以后10位,以后的数字四舍五入。 + * @param v1 被除数 + * @param v2 除数 + * @return 两个参数的商 + */ + public static double div(double v1, double v2) + { + return div(v1, v2, DEF_DIV_SCALE); + } + + /** + * 提供(相对)精确的除法运算。当发生除不尽的情况时,由scale参数指 + * 定精度,以后的数字四舍五入。 + * @param v1 被除数 + * @param v2 除数 + * @param scale 表示表示需要精确到小数点以后几位。 + * @return 两个参数的商 + */ + public static double div(double v1, double v2, int scale) + { + if (scale < 0) + { + throw new IllegalArgumentException( + "The scale must be a positive integer or zero"); + } + BigDecimal b1 = new BigDecimal(Double.toString(v1)); + BigDecimal b2 = new BigDecimal(Double.toString(v2)); + if (b1.compareTo(BigDecimal.ZERO) == 0) + { + return BigDecimal.ZERO.doubleValue(); + } + return b1.divide(b2, scale, RoundingMode.HALF_UP).doubleValue(); + } + + /** + * 提供精确的小数位四舍五入处理。 + * @param v 需要四舍五入的数字 + * @param scale 小数点后保留几位 + * @return 四舍五入后的结果 + */ + public static double round(double v, int scale) + { + if (scale < 0) + { + throw new IllegalArgumentException( + "The scale must be a positive integer or zero"); + } + BigDecimal b = new BigDecimal(Double.toString(v)); + BigDecimal one = BigDecimal.ONE; + return b.divide(one, scale, RoundingMode.HALF_UP).doubleValue(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/BeanUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/BeanUtil.java new file mode 100644 index 0000000..4f399f2 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/BeanUtil.java @@ -0,0 +1,36 @@ +package com.ruoyi.common.utils; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.io.output.CloseShieldOutputStream; +import org.springframework.beans.BeansException; +import org.springframework.beans.factory.DisposableBean; +import org.springframework.context.ApplicationContext; +import org.springframework.context.ApplicationContextAware; +import java.util.ArrayList; +import java.util.List; + +public class BeanUtil implements ApplicationContextAware, DisposableBean { + private static ApplicationContext applicationContext = null; + + /** + * 从静态变量applicationContext中取得Bean, 自动转型为所赋值对象的类型. + */ + public static T getBean(Class requiredType) { + if(applicationContext==null){ + throw new IllegalStateException("applicaitonContext属性未注入, 请在SpringBoot启动类中注册BeanUtil."); + } + return applicationContext.getBean(requiredType); + } + + @Override + public void destroy() { + applicationContext = null; + } + + @Override + public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { + if (BeanUtil.applicationContext != null) { + System.out.println("BeanUtil中的ApplicationContext被覆盖, 原有ApplicationContext为:" + BeanUtil.applicationContext); + } + BeanUtil.applicationContext = applicationContext; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/ClassReflection.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ClassReflection.java new file mode 100644 index 0000000..15acf9d --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ClassReflection.java @@ -0,0 +1,36 @@ +package com.ruoyi.common.utils; + +import java.lang.reflect.Field; + +public class ClassReflection { + + /** + * @param class1 from 用于赋值的实体类 + * @param class2 to 需要待赋值的实体类 + * 描述:反射实体类赋值 + */ + public static void reflectionAttr(Object class1, Object class2) throws Exception { + Class clazz1 = class1.getClass(); + Class clazz2 = class2.getClass(); + // 获取两个实体类的所有属性 + Field[] fields1 = clazz1.getDeclaredFields(); + Field[] fields2 = clazz2.getDeclaredFields(); + // 遍历class1Bean,获取逐个属性值,然后遍历class2Bean查找是否有相同的属性,如有相同则赋值 + for (Field f1 : fields1) { + if (f1.getName().equals("serialVersionUID")) + continue; + //设置访问权限 + f1.setAccessible(true); + Object value = f1.get(class1); + for (Field f2 : fields2) { + //字段名称相同、字段类型相同,进行赋值 + if (f1.getName().equals(f2.getName()) + && f1.getType().getName().equals(f2.getType().getName())) { + //设置访问权限 + f2.setAccessible(true); + f2.set(class2, value); + } + } + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/DateUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/DateUtils.java new file mode 100644 index 0000000..d0f8c69 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/DateUtils.java @@ -0,0 +1,187 @@ +package com.ruoyi.common.utils; + +import java.lang.management.ManagementFactory; +import java.text.ParseException; +import java.text.SimpleDateFormat; +import java.time.LocalDate; +import java.time.LocalDateTime; +import java.time.LocalTime; +import java.time.ZoneId; +import java.time.ZonedDateTime; +import java.util.Date; +import org.apache.commons.lang3.time.DateFormatUtils; + +/** + * 时间工具类 + * + * @author ruoyi + */ +public class DateUtils extends org.apache.commons.lang3.time.DateUtils +{ + public static String YYYY = "yyyy"; + + public static String YYYY_MM = "yyyy-MM"; + + public static String YYYY_MM_DD = "yyyy-MM-dd"; + + public static String YYYYMMDDHHMMSS = "yyyyMMddHHmmss"; + + public static String YYYY_MM_DD_HH_MM_SS = "yyyy-MM-dd HH:mm:ss"; + + private static String[] parsePatterns = { + "yyyy-MM-dd", "yyyy-MM-dd HH:mm:ss", "yyyy-MM-dd HH:mm", "yyyy-MM", + "yyyy/MM/dd", "yyyy/MM/dd HH:mm:ss", "yyyy/MM/dd HH:mm", "yyyy/MM", + "yyyy.MM.dd", "yyyy.MM.dd HH:mm:ss", "yyyy.MM.dd HH:mm", "yyyy.MM"}; + + /** + * 获取当前Date型日期 + * + * @return Date() 当前日期 + */ + public static Date getNowDate() + { + return new Date(); + } + + /** + * 获取当前日期, 默认格式为yyyy-MM-dd + * + * @return String + */ + public static String getDate() + { + return dateTimeNow(YYYY_MM_DD); + } + + public static final String getTime() + { + return dateTimeNow(YYYY_MM_DD_HH_MM_SS); + } + + public static final String dateTimeNow() + { + return dateTimeNow(YYYYMMDDHHMMSS); + } + + public static final String dateTimeNow(final String format) + { + return parseDateToStr(format, new Date()); + } + + public static final String dateTime(final Date date) + { + return parseDateToStr(YYYY_MM_DD, date); + } + + public static final String parseDateToStr(final String format, final Date date) + { + return new SimpleDateFormat(format).format(date); + } + + public static final Date dateTime(final String format, final String ts) + { + try + { + return new SimpleDateFormat(format).parse(ts); + } + catch (ParseException e) + { + throw new RuntimeException(e); + } + } + + /** + * 日期路径 即年/月/日 如2018/08/08 + */ + public static final String datePath() + { + Date now = new Date(); + return DateFormatUtils.format(now, "yyyy/MM/dd"); + } + + /** + * 日期路径 即年/月/日 如20180808 + */ + public static final String dateTime() + { + Date now = new Date(); + return DateFormatUtils.format(now, "yyyyMMdd"); + } + + /** + * 日期型字符串转化为日期 格式 + */ + public static Date parseDate(Object str) + { + if (str == null) + { + return null; + } + try + { + return parseDate(str.toString(), parsePatterns); + } + catch (ParseException e) + { + return null; + } + } + + /** + * 获取服务器启动时间 + */ + public static Date getServerStartDate() + { + long time = ManagementFactory.getRuntimeMXBean().getStartTime(); + return new Date(time); + } + + /** + * 计算相差天数 + */ + public static int differentDaysByMillisecond(Date date1, Date date2) + { + return Math.abs((int) ((date2.getTime() - date1.getTime()) / (1000 * 3600 * 24))); + } + + /** + * 计算两个时间差 + */ + public static String getDatePoor(Date endDate, Date nowDate) + { + long nd = 1000 * 24 * 60 * 60; + long nh = 1000 * 60 * 60; + long nm = 1000 * 60; + // long ns = 1000; + // 获得两个时间的毫秒时间差异 + long diff = endDate.getTime() - nowDate.getTime(); + // 计算差多少天 + long day = diff / nd; + // 计算差多少小时 + long hour = diff % nd / nh; + // 计算差多少分钟 + long min = diff % nd % nh / nm; + // 计算差多少秒//输出结果 + // long sec = diff % nd % nh % nm / ns; + return day + "天" + hour + "小时" + min + "分钟"; + } + + /** + * 增加 LocalDateTime ==> Date + */ + public static Date toDate(LocalDateTime temporalAccessor) + { + ZonedDateTime zdt = temporalAccessor.atZone(ZoneId.systemDefault()); + return Date.from(zdt.toInstant()); + } + + /** + * 增加 LocalDate ==> Date + */ + public static Date toDate(LocalDate temporalAccessor) + { + LocalDateTime localDateTime = LocalDateTime.of(temporalAccessor, LocalTime.of(0, 0, 0)); + ZonedDateTime zdt = localDateTime.atZone(ZoneId.systemDefault()); + return Date.from(zdt.toInstant()); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/DictUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/DictUtils.java new file mode 100644 index 0000000..34909ff --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/DictUtils.java @@ -0,0 +1,182 @@ +package com.ruoyi.common.utils; + +import java.util.Collection; +import java.util.List; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.entity.SysDictData; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.utils.spring.SpringUtils; + +/** + * 字典工具类 + * + * @author ruoyi + */ +public class DictUtils +{ + /** + * 分隔符 + */ + public static final String SEPARATOR = ","; + + /** + * 设置字典缓存 + * + * @param key 参数键 + * @param dictDatas 字典数据列表 + */ + public static void setDictCache(String key, List dictDatas) + { + SpringUtils.getBean(RedisCache.class).setCacheObject(getCacheKey(key), dictDatas); + } + + /** + * 获取字典缓存 + * + * @param key 参数键 + * @return dictDatas 字典数据列表 + */ + public static List getDictCache(String key) + { + Object cacheObj = SpringUtils.getBean(RedisCache.class).getCacheObject(getCacheKey(key)); + if (StringUtils.isNotNull(cacheObj)) + { + return StringUtils.cast(cacheObj); + } + return null; + } + + /** + * 根据字典类型和字典值获取字典标签 + * + * @param dictType 字典类型 + * @param dictValue 字典值 + * @return 字典标签 + */ + public static String getDictLabel(String dictType, String dictValue) + { + return getDictLabel(dictType, dictValue, SEPARATOR); + } + + /** + * 根据字典类型和字典标签获取字典值 + * + * @param dictType 字典类型 + * @param dictLabel 字典标签 + * @return 字典值 + */ + public static String getDictValue(String dictType, String dictLabel) + { + return getDictValue(dictType, dictLabel, SEPARATOR); + } + + /** + * 根据字典类型和字典值获取字典标签 + * + * @param dictType 字典类型 + * @param dictValue 字典值 + * @param separator 分隔符 + * @return 字典标签 + */ + public static String getDictLabel(String dictType, String dictValue, String separator) + { + StringBuilder propertyString = new StringBuilder(); + List datas = getDictCache(dictType); + + if (StringUtils.containsAny(separator, dictValue) && StringUtils.isNotEmpty(datas)) + { + for (SysDictData dict : datas) + { + for (String value : dictValue.split(separator)) + { + if (value.equals(dict.getDictValue())) + { + propertyString.append(dict.getDictLabel()).append(separator); + break; + } + } + } + } + else + { + for (SysDictData dict : datas) + { + if (dictValue.equals(dict.getDictValue())) + { + return dict.getDictLabel(); + } + } + } + return StringUtils.stripEnd(propertyString.toString(), separator); + } + + /** + * 根据字典类型和字典标签获取字典值 + * + * @param dictType 字典类型 + * @param dictLabel 字典标签 + * @param separator 分隔符 + * @return 字典值 + */ + public static String getDictValue(String dictType, String dictLabel, String separator) + { + StringBuilder propertyString = new StringBuilder(); + List datas = getDictCache(dictType); + + if (StringUtils.containsAny(separator, dictLabel) && StringUtils.isNotEmpty(datas)) + { + for (SysDictData dict : datas) + { + for (String label : dictLabel.split(separator)) + { + if (label.equals(dict.getDictLabel())) + { + propertyString.append(dict.getDictValue()).append(separator); + break; + } + } + } + } + else + { + for (SysDictData dict : datas) + { + if (dictLabel.equals(dict.getDictLabel())) + { + return dict.getDictValue(); + } + } + } + return StringUtils.stripEnd(propertyString.toString(), separator); + } + + /** + * 删除指定字典缓存 + * + * @param key 字典键 + */ + public static void removeDictCache(String key) + { + SpringUtils.getBean(RedisCache.class).deleteObject(getCacheKey(key)); + } + + /** + * 清空字典缓存 + */ + public static void clearDictCache() + { + Collection keys = SpringUtils.getBean(RedisCache.class).keys(Constants.SYS_DICT_KEY + "*"); + SpringUtils.getBean(RedisCache.class).deleteObject(keys); + } + + /** + * 设置cache key + * + * @param configKey 参数键 + * @return 缓存键key + */ + public static String getCacheKey(String configKey) + { + return Constants.SYS_DICT_KEY + configKey; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/ExceptionUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ExceptionUtil.java new file mode 100644 index 0000000..214e4a0 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ExceptionUtil.java @@ -0,0 +1,39 @@ +package com.ruoyi.common.utils; + +import java.io.PrintWriter; +import java.io.StringWriter; +import org.apache.commons.lang3.exception.ExceptionUtils; + +/** + * 错误信息处理类。 + * + * @author ruoyi + */ +public class ExceptionUtil +{ + /** + * 获取exception的详细错误信息。 + */ + public static String getExceptionMessage(Throwable e) + { + StringWriter sw = new StringWriter(); + e.printStackTrace(new PrintWriter(sw, true)); + return sw.toString(); + } + + public static String getRootErrorMessage(Exception e) + { + Throwable root = ExceptionUtils.getRootCause(e); + root = (root == null ? e : root); + if (root == null) + { + return ""; + } + String msg = root.getMessage(); + if (msg == null) + { + return "null"; + } + return StringUtils.defaultString(msg); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/IPUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/IPUtils.java new file mode 100644 index 0000000..2e3bef6 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/IPUtils.java @@ -0,0 +1,91 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.common.utils; + +import lombok.extern.slf4j.Slf4j; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import javax.servlet.http.HttpServletRequest; +import java.net.Inet4Address; +import java.net.InetAddress; +import java.net.NetworkInterface; +import java.util.Enumeration; + +/** + * IP地址 + * + * @author Mark sunlightcs@gmail.com + */ +@Slf4j +public class IPUtils { + private static Logger logger = LoggerFactory.getLogger(IPUtils.class); + + /** + * 获取IP地址 + * + * 使用Nginx等反向代理软件, 则不能通过request.getRemoteAddr()获取IP地址 + * 如果使用了多级反向代理的话,X-Forwarded-For的值并不止一个,而是一串IP地址,X-Forwarded-For中第一个非unknown的有效IP字符串,则为真实IP地址 + */ + public static String getIpAddr(HttpServletRequest request) { + String ip = null; + try { + ip = request.getHeader("x-forwarded-for"); + if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { + ip = request.getHeader("Proxy-Client-IP"); + } + if (StringUtils.isEmpty(ip) || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) { + ip = request.getHeader("WL-Proxy-Client-IP"); + } + if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { + ip = request.getHeader("HTTP_CLIENT_IP"); + } + if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { + ip = request.getHeader("HTTP_X_FORWARDED_FOR"); + } + if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) { + ip = request.getRemoteAddr(); + } + } catch (Exception e) { + logger.error("IPUtils ERROR ", e); + } + +// //使用代理,则获取第一个IP地址 +// if(StringUtils.isEmpty(ip) && ip.length() > 15) { +// if(ip.indexOf(",") > 0) { +// ip = ip.substring(0, ip.indexOf(",")); +// } +// } + + return ip; + } + + + public static String getLocalIpAddress() { + try { + Enumeration allNetInterfaces = NetworkInterface.getNetworkInterfaces(); + InetAddress ip; + while (allNetInterfaces.hasMoreElements()) { + NetworkInterface netInterface = allNetInterfaces.nextElement(); + if (!netInterface.isLoopback() && !netInterface.isVirtual() && netInterface.isUp()) { + Enumeration addresses = netInterface.getInetAddresses(); + while (addresses.hasMoreElements()) { + ip = addresses.nextElement(); + if (ip instanceof Inet4Address) { + return ip.getHostAddress(); + } + } + } + } + } catch (Exception e) { + logger.error("本地IP地址获取失败", e); + } + return ""; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/LogUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/LogUtils.java new file mode 100644 index 0000000..0de30c6 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/LogUtils.java @@ -0,0 +1,18 @@ +package com.ruoyi.common.utils; + +/** + * 处理并记录日志文件 + * + * @author ruoyi + */ +public class LogUtils +{ + public static String getBlock(Object msg) + { + if (msg == null) + { + msg = ""; + } + return "[" + msg.toString() + "]"; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/MessageUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/MessageUtils.java new file mode 100644 index 0000000..7dac75a --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/MessageUtils.java @@ -0,0 +1,26 @@ +package com.ruoyi.common.utils; + +import org.springframework.context.MessageSource; +import org.springframework.context.i18n.LocaleContextHolder; +import com.ruoyi.common.utils.spring.SpringUtils; + +/** + * 获取i18n资源文件 + * + * @author ruoyi + */ +public class MessageUtils +{ + /** + * 根据消息键和参数 获取消息 委托给spring messageSource + * + * @param code 消息键 + * @param args 参数 + * @return 获取国际化翻译值 + */ + public static String message(String code, Object... args) + { + MessageSource messageSource = SpringUtils.getBean(MessageSource.class); + return messageSource.getMessage(code, args, LocaleContextHolder.getLocale()); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/PageUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/PageUtils.java new file mode 100644 index 0000000..70e9b08 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/PageUtils.java @@ -0,0 +1,35 @@ +package com.ruoyi.common.utils; + +import com.github.pagehelper.PageHelper; +import com.ruoyi.common.core.page.PageDomain; +import com.ruoyi.common.core.page.TableSupport; +import com.ruoyi.common.utils.sql.SqlUtil; + +/** + * 分页工具类 + * + * @author ruoyi + */ +public class PageUtils extends PageHelper +{ + /** + * 设置请求分页数据 + */ + public static void startPage() + { + PageDomain pageDomain = TableSupport.buildPageRequest(); + Integer pageNum = pageDomain.getPageNum(); + Integer pageSize = pageDomain.getPageSize(); + String orderBy = SqlUtil.escapeOrderBySql(pageDomain.getOrderBy()); + Boolean reasonable = pageDomain.getReasonable(); + PageHelper.startPage(pageNum, pageSize, orderBy).setReasonable(reasonable); + } + + /** + * 清理分页的线程变量 + */ + public static void clearPage() + { + PageHelper.clearPage(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/Query.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Query.java new file mode 100644 index 0000000..98c9788 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Query.java @@ -0,0 +1,75 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.common.utils; + +import com.baomidou.mybatisplus.core.metadata.OrderItem; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.xss.SQLFilter; + +import java.util.Map; + +/** + * 查询参数 + * + * @author Mark sunlightcs@gmail.com + */ +public class Query { + + public Page getPage(Map params) { + return this.getPage(params, null, false); + } + + public Page getPage(Map params, String defaultOrderField, boolean isAsc) { + //分页参数 + long curPage = 1; + long limit = 10; + + if(params.get(Constants.PAGE) != null){ + curPage = Long.parseLong((String)params.get(Constants.PAGE)); + } + if(params.get(Constants.LIMIT) != null){ + limit = Long.parseLong((String)params.get(Constants.LIMIT)); + } + + //分页对象 + Page page = new Page<>(curPage, limit); + + //分页参数 + params.put(Constants.PAGE, page); + + //排序字段 + //防止SQL注入(因为sidx、order是通过拼接SQL实现排序的,会有SQL注入风险) + String orderField = SQLFilter.sqlInject((String)params.get(Constants.ORDER_FIELD)); + String order = (String)params.get(Constants.ORDER); + + //前端字段排序 + if(StringUtils.isNotEmpty(orderField) && StringUtils.isNotEmpty(order)){ + if(Constants.ASC.equalsIgnoreCase(order)) { + return page.addOrder(OrderItem.asc(orderField)); + }else { + return page.addOrder(OrderItem.desc(orderField)); + } + } + + //没有排序字段,则不排序 + if(StringUtils.isBlank(defaultOrderField)){ + return page; + } + + //默认排序 + if(isAsc) { + page.addOrder(OrderItem.asc(defaultOrderField)); + }else { + page.addOrder(OrderItem.desc(defaultOrderField)); + } + + return page; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/SecurityUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/SecurityUtils.java new file mode 100644 index 0000000..a6f3d53 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/SecurityUtils.java @@ -0,0 +1,120 @@ +package com.ruoyi.common.utils; + +import org.springframework.security.core.Authentication; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.exception.ServiceException; + +/** + * 安全服务工具类 + * + * @author ruoyi + */ +public class SecurityUtils +{ + /** + * 用户ID + **/ + public static Long getUserId() + { + try + { + return getLoginUser().getUserId(); + } + catch (Exception e) + { + throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED); + } + } + + /** + * 获取部门ID + **/ + public static Long getDeptId() + { + try + { + return getLoginUser().getDeptId(); + } + catch (Exception e) + { + throw new ServiceException("获取部门ID异常", HttpStatus.UNAUTHORIZED); + } + } + + /** + * 获取用户账户 + **/ + public static String getUsername() + { + try + { + return getLoginUser().getUsername(); + } + catch (Exception e) + { + throw new ServiceException("获取用户账户异常", HttpStatus.UNAUTHORIZED); + } + } + + /** + * 获取用户 + **/ + public static LoginUser getLoginUser() + { + try + { + return (LoginUser) getAuthentication().getPrincipal(); + } + catch (Exception e) + { + throw new ServiceException("获取用户信息异常", HttpStatus.UNAUTHORIZED); + } + } + + /** + * 获取Authentication + */ + public static Authentication getAuthentication() + { + return SecurityContextHolder.getContext().getAuthentication(); + } + + /** + * 生成BCryptPasswordEncoder密码 + * + * @param password 密码 + * @return 加密字符串 + */ + public static String encryptPassword(String password) + { + BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder(); + return passwordEncoder.encode(password); + } + + /** + * 判断密码是否相同 + * + * @param rawPassword 真实密码 + * @param encodedPassword 加密后字符 + * @return 结果 + */ + public static boolean matchesPassword(String rawPassword, String encodedPassword) + { + BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder(); + return passwordEncoder.matches(rawPassword, encodedPassword); + } + + /** + * 是否为管理员 + * + * @param userId 用户ID + * @return 结果 + */ + public static boolean isAdmin(Long userId) + { + return userId != null && 1L == userId; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/ServletUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ServletUtils.java new file mode 100644 index 0000000..85af068 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ServletUtils.java @@ -0,0 +1,146 @@ +package com.ruoyi.common.utils; + +import java.io.IOException; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import javax.servlet.http.HttpSession; +import org.springframework.web.context.request.RequestAttributes; +import org.springframework.web.context.request.RequestContextHolder; +import org.springframework.web.context.request.ServletRequestAttributes; +import com.ruoyi.common.core.text.Convert; + +/** + * 客户端工具类 + * + * @author ruoyi + */ +public class ServletUtils +{ + /** + * 获取String参数 + */ + public static String getParameter(String name) + { + return getRequest().getParameter(name); + } + + /** + * 获取String参数 + */ + public static String getParameter(String name, String defaultValue) + { + return Convert.toStr(getRequest().getParameter(name), defaultValue); + } + + /** + * 获取Integer参数 + */ + public static Integer getParameterToInt(String name) + { + return Convert.toInt(getRequest().getParameter(name)); + } + + /** + * 获取Integer参数 + */ + public static Integer getParameterToInt(String name, Integer defaultValue) + { + return Convert.toInt(getRequest().getParameter(name), defaultValue); + } + + /** + * 获取Boolean参数 + */ + public static Boolean getParameterToBool(String name) + { + return Convert.toBool(getRequest().getParameter(name)); + } + + /** + * 获取Boolean参数 + */ + public static Boolean getParameterToBool(String name, Boolean defaultValue) + { + return Convert.toBool(getRequest().getParameter(name), defaultValue); + } + + /** + * 获取request + */ + public static HttpServletRequest getRequest() + { + return getRequestAttributes().getRequest(); + } + + /** + * 获取response + */ + public static HttpServletResponse getResponse() + { + return getRequestAttributes().getResponse(); + } + + /** + * 获取session + */ + public static HttpSession getSession() + { + return getRequest().getSession(); + } + + public static ServletRequestAttributes getRequestAttributes() + { + RequestAttributes attributes = RequestContextHolder.getRequestAttributes(); + return (ServletRequestAttributes) attributes; + } + + /** + * 将字符串渲染到客户端 + * + * @param response 渲染对象 + * @param string 待渲染的字符串 + */ + public static void renderString(HttpServletResponse response, String string) + { + try + { + response.setStatus(200); + response.setContentType("application/json"); + response.setCharacterEncoding("utf-8"); + response.getWriter().print(string); + } + catch (IOException e) + { + e.printStackTrace(); + } + } + + /** + * 是否是Ajax异步请求 + * + * @param request + */ + public static boolean isAjaxRequest(HttpServletRequest request) + { + String accept = request.getHeader("accept"); + if (accept != null && accept.contains("application/json")) + { + return true; + } + + String xRequestedWith = request.getHeader("X-Requested-With"); + if (xRequestedWith != null && xRequestedWith.contains("XMLHttpRequest")) + { + return true; + } + + String uri = request.getRequestURI(); + if (StringUtils.inStringIgnoreCase(uri, ".json", ".xml")) + { + return true; + } + + String ajax = request.getParameter("__ajax"); + return StringUtils.inStringIgnoreCase(ajax, "json", "xml"); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/StringUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/StringUtils.java new file mode 100644 index 0000000..6b70889 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/StringUtils.java @@ -0,0 +1,583 @@ +package com.ruoyi.common.utils; + +import java.util.ArrayList; +import java.util.Collection; +import java.util.HashSet; +import java.util.List; +import java.util.Map; +import java.util.Set; +import org.springframework.util.AntPathMatcher; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.text.StrFormatter; + +/** + * 字符串工具类 + * + * @author ruoyi + */ +public class StringUtils extends org.apache.commons.lang3.StringUtils +{ + /** 空字符串 */ + private static final String NULLSTR = ""; + + /** 下划线 */ + private static final char SEPARATOR = '_'; + + /** + * 获取参数不为空值 + * + * @param value defaultValue 要判断的value + * @return value 返回值 + */ + public static T nvl(T value, T defaultValue) + { + return value != null ? value : defaultValue; + } + + /** + * * 判断一个Collection是否为空, 包含List,Set,Queue + * + * @param coll 要判断的Collection + * @return true:为空 false:非空 + */ + public static boolean isEmpty(Collection coll) + { + return isNull(coll) || coll.isEmpty(); + } + + /** + * * 判断一个Collection是否非空,包含List,Set,Queue + * + * @param coll 要判断的Collection + * @return true:非空 false:空 + */ + public static boolean isNotEmpty(Collection coll) + { + return !isEmpty(coll); + } + + /** + * * 判断一个对象数组是否为空 + * + * @param objects 要判断的对象数组 + ** @return true:为空 false:非空 + */ + public static boolean isEmpty(Object[] objects) + { + return isNull(objects) || (objects.length == 0); + } + + /** + * * 判断一个对象数组是否非空 + * + * @param objects 要判断的对象数组 + * @return true:非空 false:空 + */ + public static boolean isNotEmpty(Object[] objects) + { + return !isEmpty(objects); + } + + /** + * * 判断一个Map是否为空 + * + * @param map 要判断的Map + * @return true:为空 false:非空 + */ + public static boolean isEmpty(Map map) + { + return isNull(map) || map.isEmpty(); + } + + /** + * * 判断一个Map是否为空 + * + * @param map 要判断的Map + * @return true:非空 false:空 + */ + public static boolean isNotEmpty(Map map) + { + return !isEmpty(map); + } + + /** + * * 判断一个字符串是否为空串 + * + * @param str String + * @return true:为空 false:非空 + */ + public static boolean isEmpty(String str) + { + return isNull(str) || NULLSTR.equals(str.trim()); + } + + /** + * * 判断一个字符串是否为非空串 + * + * @param str String + * @return true:非空串 false:空串 + */ + public static boolean isNotEmpty(String str) + { + return !isEmpty(str); + } + + /** + * * 判断一个对象是否为空 + * + * @param object Object + * @return true:为空 false:非空 + */ + public static boolean isNull(Object object) + { + return object == null; + } + + /** + * * 判断一个对象是否非空 + * + * @param object Object + * @return true:非空 false:空 + */ + public static boolean isNotNull(Object object) + { + return !isNull(object); + } + + /** + * * 判断一个对象是否是数组类型(Java基本型别的数组) + * + * @param object 对象 + * @return true:是数组 false:不是数组 + */ + public static boolean isArray(Object object) + { + return isNotNull(object) && object.getClass().isArray(); + } + + /** + * 去空格 + */ + public static String trim(String str) + { + return (str == null ? "" : str.trim()); + } + + /** + * 截取字符串 + * + * @param str 字符串 + * @param start 开始 + * @return 结果 + */ + public static String substring(final String str, int start) + { + if (str == null) + { + return NULLSTR; + } + + if (start < 0) + { + start = str.length() + start; + } + + if (start < 0) + { + start = 0; + } + if (start > str.length()) + { + return NULLSTR; + } + + return str.substring(start); + } + + /** + * 截取字符串 + * + * @param str 字符串 + * @param start 开始 + * @param end 结束 + * @return 结果 + */ + public static String substring(final String str, int start, int end) + { + if (str == null) + { + return NULLSTR; + } + + if (end < 0) + { + end = str.length() + end; + } + if (start < 0) + { + start = str.length() + start; + } + + if (end > str.length()) + { + end = str.length(); + } + + if (start > end) + { + return NULLSTR; + } + + if (start < 0) + { + start = 0; + } + if (end < 0) + { + end = 0; + } + + return str.substring(start, end); + } + + /** + * 格式化文本, {} 表示占位符
+ * 此方法只是简单将占位符 {} 按照顺序替换为参数
+ * 如果想输出 {} 使用 \\转义 { 即可,如果想输出 {} 之前的 \ 使用双转义符 \\\\ 即可
+ * 例:
+ * 通常使用:format("this is {} for {}", "a", "b") -> this is a for b
+ * 转义{}: format("this is \\{} for {}", "a", "b") -> this is \{} for a
+ * 转义\: format("this is \\\\{} for {}", "a", "b") -> this is \a for b
+ * + * @param template 文本模板,被替换的部分用 {} 表示 + * @param params 参数值 + * @return 格式化后的文本 + */ + public static String format(String template, Object... params) + { + if (isEmpty(params) || isEmpty(template)) + { + return template; + } + return StrFormatter.format(template, params); + } + + /** + * 是否为http(s)://开头 + * + * @param link 链接 + * @return 结果 + */ + public static boolean ishttp(String link) + { + return StringUtils.startsWithAny(link, Constants.HTTP, Constants.HTTPS); + } + + /** + * 字符串转set + * + * @param str 字符串 + * @param sep 分隔符 + * @return set集合 + */ + public static final Set str2Set(String str, String sep) + { + return new HashSet(str2List(str, sep, true, false)); + } + + /** + * 字符串转list + * + * @param str 字符串 + * @param sep 分隔符 + * @param filterBlank 过滤纯空白 + * @param trim 去掉首尾空白 + * @return list集合 + */ + public static final List str2List(String str, String sep, boolean filterBlank, boolean trim) + { + List list = new ArrayList(); + if (StringUtils.isEmpty(str)) + { + return list; + } + + // 过滤空白字符串 + if (filterBlank && StringUtils.isBlank(str)) + { + return list; + } + String[] split = str.split(sep); + for (String string : split) + { + if (filterBlank && StringUtils.isBlank(string)) + { + continue; + } + if (trim) + { + string = string.trim(); + } + list.add(string); + } + + return list; + } + + /** + * 查找指定字符串是否包含指定字符串列表中的任意一个字符串同时串忽略大小写 + * + * @param cs 指定字符串 + * @param searchCharSequences 需要检查的字符串数组 + * @return 是否包含任意一个字符串 + */ + public static boolean containsAnyIgnoreCase(CharSequence cs, CharSequence... searchCharSequences) + { + if (isEmpty(cs) || isEmpty(searchCharSequences)) + { + return false; + } + for (CharSequence testStr : searchCharSequences) + { + if (containsIgnoreCase(cs, testStr)) + { + return true; + } + } + return false; + } + + /** + * 驼峰转下划线命名 + */ + public static String toUnderScoreCase(String str) + { + if (str == null) + { + return null; + } + StringBuilder sb = new StringBuilder(); + // 前置字符是否大写 + boolean preCharIsUpperCase = true; + // 当前字符是否大写 + boolean curreCharIsUpperCase = true; + // 下一字符是否大写 + boolean nexteCharIsUpperCase = true; + for (int i = 0; i < str.length(); i++) + { + char c = str.charAt(i); + if (i > 0) + { + preCharIsUpperCase = Character.isUpperCase(str.charAt(i - 1)); + } + else + { + preCharIsUpperCase = false; + } + + curreCharIsUpperCase = Character.isUpperCase(c); + + if (i < (str.length() - 1)) + { + nexteCharIsUpperCase = Character.isUpperCase(str.charAt(i + 1)); + } + + if (preCharIsUpperCase && curreCharIsUpperCase && !nexteCharIsUpperCase) + { + sb.append(SEPARATOR); + } + else if ((i != 0 && !preCharIsUpperCase) && curreCharIsUpperCase) + { + sb.append(SEPARATOR); + } + sb.append(Character.toLowerCase(c)); + } + + return sb.toString(); + } + + /** + * 是否包含字符串 + * + * @param str 验证字符串 + * @param strs 字符串组 + * @return 包含返回true + */ + public static boolean inStringIgnoreCase(String str, String... strs) + { + if (str != null && strs != null) + { + for (String s : strs) + { + if (str.equalsIgnoreCase(trim(s))) + { + return true; + } + } + } + return false; + } + + /** + * 将下划线大写方式命名的字符串转换为驼峰式。如果转换前的下划线大写方式命名的字符串为空,则返回空字符串。 例如:HELLO_WORLD->HelloWorld + * + * @param name 转换前的下划线大写方式命名的字符串 + * @return 转换后的驼峰式命名的字符串 + */ + public static String convertToCamelCase(String name) + { + StringBuilder result = new StringBuilder(); + // 快速检查 + if (name == null || name.isEmpty()) + { + // 没必要转换 + return ""; + } + else if (!name.contains("_")) + { + // 不含下划线,仅将首字母大写 + return name.substring(0, 1).toUpperCase() + name.substring(1); + } + // 用下划线将原始字符串分割 + String[] camels = name.split("_"); + for (String camel : camels) + { + // 跳过原始字符串中开头、结尾的下换线或双重下划线 + if (camel.isEmpty()) + { + continue; + } + // 首字母大写 + result.append(camel.substring(0, 1).toUpperCase()); + result.append(camel.substring(1).toLowerCase()); + } + return result.toString(); + } + + /** + * 驼峰式命名法 例如:user_name->userName + */ + public static String toCamelCase(String s) + { + if (s == null) + { + return null; + } + s = s.toLowerCase(); + StringBuilder sb = new StringBuilder(s.length()); + boolean upperCase = false; + for (int i = 0; i < s.length(); i++) + { + char c = s.charAt(i); + + if (c == SEPARATOR) + { + upperCase = true; + } + else if (upperCase) + { + sb.append(Character.toUpperCase(c)); + upperCase = false; + } + else + { + sb.append(c); + } + } + return sb.toString(); + } + + /** + * 查找指定字符串是否匹配指定字符串列表中的任意一个字符串 + * + * @param str 指定字符串 + * @param strs 需要检查的字符串数组 + * @return 是否匹配 + */ + public static boolean matches(String str, List strs) + { + if (isEmpty(str) || isEmpty(strs)) + { + return false; + } + for (String pattern : strs) + { + if (isMatch(pattern, str)) + { + return true; + } + } + return false; + } + + /** + * 判断url是否与规则配置: + * ? 表示单个字符; + * * 表示一层路径内的任意字符串,不可跨层级; + * ** 表示任意层路径; + * + * @param pattern 匹配规则 + * @param url 需要匹配的url + * @return + */ + public static boolean isMatch(String pattern, String url) + { + AntPathMatcher matcher = new AntPathMatcher(); + return matcher.match(pattern, url); + } + + @SuppressWarnings("unchecked") + public static T cast(Object obj) + { + return (T) obj; + } + + /** + * 数字左边补齐0,使之达到指定长度。注意,如果数字转换为字符串后,长度大于size,则只保留 最后size个字符。 + * + * @param num 数字对象 + * @param size 字符串指定长度 + * @return 返回数字的字符串格式,该字符串为指定长度。 + */ + public static final String padl(final Number num, final int size) + { + return padl(num.toString(), size, '0'); + } + + /** + * 字符串左补齐。如果原始字符串s长度大于size,则只保留最后size个字符。 + * + * @param s 原始字符串 + * @param size 字符串指定长度 + * @param c 用于补齐的字符 + * @return 返回指定长度的字符串,由原字符串左补齐或截取得到。 + */ + public static final String padl(final String s, final int size, final char c) + { + final StringBuilder sb = new StringBuilder(size); + if (s != null) + { + final int len = s.length(); + if (s.length() <= size) + { + for (int i = size - len; i > 0; i--) + { + sb.append(c); + } + sb.append(s); + } + else + { + return s.substring(len - size, len); + } + } + else + { + for (int i = size; i > 0; i--) + { + sb.append(c); + } + } + return sb.toString(); + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/Threads.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Threads.java new file mode 100644 index 0000000..71fe6d5 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/Threads.java @@ -0,0 +1,99 @@ +package com.ruoyi.common.utils; + +import java.util.concurrent.CancellationException; +import java.util.concurrent.ExecutionException; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Future; +import java.util.concurrent.TimeUnit; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** + * 线程相关工具类. + * + * @author ruoyi + */ +public class Threads +{ + private static final Logger logger = LoggerFactory.getLogger(Threads.class); + + /** + * sleep等待,单位为毫秒 + */ + public static void sleep(long milliseconds) + { + try + { + Thread.sleep(milliseconds); + } + catch (InterruptedException e) + { + return; + } + } + + /** + * 停止线程池 + * 先使用shutdown, 停止接收新任务并尝试完成所有已存在任务. + * 如果超时, 则调用shutdownNow, 取消在workQueue中Pending的任务,并中断所有阻塞函数. + * 如果仍然超時,則強制退出. + * 另对在shutdown时线程本身被调用中断做了处理. + */ + public static void shutdownAndAwaitTermination(ExecutorService pool) + { + if (pool != null && !pool.isShutdown()) + { + pool.shutdown(); + try + { + if (!pool.awaitTermination(120, TimeUnit.SECONDS)) + { + pool.shutdownNow(); + if (!pool.awaitTermination(120, TimeUnit.SECONDS)) + { + logger.info("Pool did not terminate"); + } + } + } + catch (InterruptedException ie) + { + pool.shutdownNow(); + Thread.currentThread().interrupt(); + } + } + } + + /** + * 打印线程异常信息 + */ + public static void printException(Runnable r, Throwable t) + { + if (t == null && r instanceof Future) + { + try + { + Future future = (Future) r; + if (future.isDone()) + { + future.get(); + } + } + catch (CancellationException ce) + { + t = ce; + } + catch (ExecutionException ee) + { + t = ee.getCause(); + } + catch (InterruptedException ie) + { + Thread.currentThread().interrupt(); + } + } + if (t != null) + { + logger.error(t.getMessage(), t); + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanUtils.java new file mode 100644 index 0000000..4463662 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanUtils.java @@ -0,0 +1,110 @@ +package com.ruoyi.common.utils.bean; + +import java.lang.reflect.Method; +import java.util.ArrayList; +import java.util.List; +import java.util.regex.Matcher; +import java.util.regex.Pattern; + +/** + * Bean 工具类 + * + * @author ruoyi + */ +public class BeanUtils extends org.springframework.beans.BeanUtils +{ + /** Bean方法名中属性名开始的下标 */ + private static final int BEAN_METHOD_PROP_INDEX = 3; + + /** * 匹配getter方法的正则表达式 */ + private static final Pattern GET_PATTERN = Pattern.compile("get(\\p{javaUpperCase}\\w*)"); + + /** * 匹配setter方法的正则表达式 */ + private static final Pattern SET_PATTERN = Pattern.compile("set(\\p{javaUpperCase}\\w*)"); + + /** + * Bean属性复制工具方法。 + * + * @param dest 目标对象 + * @param src 源对象 + */ + public static void copyBeanProp(Object dest, Object src) + { + try + { + copyProperties(src, dest); + } + catch (Exception e) + { + e.printStackTrace(); + } + } + + /** + * 获取对象的setter方法。 + * + * @param obj 对象 + * @return 对象的setter方法列表 + */ + public static List getSetterMethods(Object obj) + { + // setter方法列表 + List setterMethods = new ArrayList(); + + // 获取所有方法 + Method[] methods = obj.getClass().getMethods(); + + // 查找setter方法 + + for (Method method : methods) + { + Matcher m = SET_PATTERN.matcher(method.getName()); + if (m.matches() && (method.getParameterTypes().length == 1)) + { + setterMethods.add(method); + } + } + // 返回setter方法列表 + return setterMethods; + } + + /** + * 获取对象的getter方法。 + * + * @param obj 对象 + * @return 对象的getter方法列表 + */ + + public static List getGetterMethods(Object obj) + { + // getter方法列表 + List getterMethods = new ArrayList(); + // 获取所有方法 + Method[] methods = obj.getClass().getMethods(); + // 查找getter方法 + for (Method method : methods) + { + Matcher m = GET_PATTERN.matcher(method.getName()); + if (m.matches() && (method.getParameterTypes().length == 0)) + { + getterMethods.add(method); + } + } + // 返回getter方法列表 + return getterMethods; + } + + /** + * 检查Bean方法名中的属性名是否相等。
+ * 如getName()和setName()属性名一样,getName()和setAge()属性名不一样。 + * + * @param m1 方法名1 + * @param m2 方法名2 + * @return 属性名一样返回true,否则返回false + */ + + public static boolean isMethodPropEquals(String m1, String m2) + { + return m1.substring(BEAN_METHOD_PROP_INDEX).equals(m2.substring(BEAN_METHOD_PROP_INDEX)); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanValidators.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanValidators.java new file mode 100644 index 0000000..80bfed7 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/bean/BeanValidators.java @@ -0,0 +1,24 @@ +package com.ruoyi.common.utils.bean; + +import java.util.Set; +import javax.validation.ConstraintViolation; +import javax.validation.ConstraintViolationException; +import javax.validation.Validator; + +/** + * bean对象属性验证 + * + * @author ruoyi + */ +public class BeanValidators +{ + public static void validateWithException(Validator validator, Object object, Class... groups) + throws ConstraintViolationException + { + Set> constraintViolations = validator.validate(object, groups); + if (!constraintViolations.isEmpty()) + { + throw new ConstraintViolationException(constraintViolations); + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileTypeUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileTypeUtils.java new file mode 100644 index 0000000..68130b9 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileTypeUtils.java @@ -0,0 +1,76 @@ +package com.ruoyi.common.utils.file; + +import java.io.File; +import org.apache.commons.lang3.StringUtils; + +/** + * 文件类型工具类 + * + * @author ruoyi + */ +public class FileTypeUtils +{ + /** + * 获取文件类型 + *

+ * 例如: ruoyi.txt, 返回: txt + * + * @param file 文件名 + * @return 后缀(不含".") + */ + public static String getFileType(File file) + { + if (null == file) + { + return StringUtils.EMPTY; + } + return getFileType(file.getName()); + } + + /** + * 获取文件类型 + *

+ * 例如: ruoyi.txt, 返回: txt + * + * @param fileName 文件名 + * @return 后缀(不含".") + */ + public static String getFileType(String fileName) + { + int separatorIndex = fileName.lastIndexOf("."); + if (separatorIndex < 0) + { + return ""; + } + return fileName.substring(separatorIndex + 1).toLowerCase(); + } + + /** + * 获取文件类型 + * + * @param photoByte 文件字节码 + * @return 后缀(不含".") + */ + public static String getFileExtendName(byte[] photoByte) + { + String strFileExtendName = "JPG"; + if ((photoByte[0] == 71) && (photoByte[1] == 73) && (photoByte[2] == 70) && (photoByte[3] == 56) + && ((photoByte[4] == 55) || (photoByte[4] == 57)) && (photoByte[5] == 97)) + { + strFileExtendName = "GIF"; + } + else if ((photoByte[6] == 74) && (photoByte[7] == 70) && (photoByte[8] == 73) && (photoByte[9] == 70)) + { + strFileExtendName = "JPG"; + } + else if ((photoByte[0] == 66) && (photoByte[1] == 77)) + { + strFileExtendName = "BMP"; + } + else if ((photoByte[1] == 80) && (photoByte[2] == 78) && (photoByte[3] == 71)) + { + strFileExtendName = "PNG"; + } + return strFileExtendName; + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java new file mode 100644 index 0000000..23b4cf1 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java @@ -0,0 +1,233 @@ +package com.ruoyi.common.utils.file; + +import java.io.File; +import java.io.IOException; +import java.nio.file.Paths; +import java.util.Objects; +import org.apache.commons.io.FilenameUtils; +import org.springframework.web.multipart.MultipartFile; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.exception.file.FileNameLengthLimitExceededException; +import com.ruoyi.common.exception.file.FileSizeLimitExceededException; +import com.ruoyi.common.exception.file.InvalidExtensionException; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.uuid.Seq; + +/** + * 文件上传工具类 + * + * @author ruoyi + */ +public class FileUploadUtils +{ + /** + * 默认大小 50M + */ + public static final long DEFAULT_MAX_SIZE = 50 * 1024 * 1024; + + /** + * 默认的文件名最大长度 100 + */ + public static final int DEFAULT_FILE_NAME_LENGTH = 100; + + /** + * 默认上传的地址 + */ + private static String defaultBaseDir = RuoYiConfig.getProfile(); + + public static void setDefaultBaseDir(String defaultBaseDir) + { + FileUploadUtils.defaultBaseDir = defaultBaseDir; + } + + public static String getDefaultBaseDir() + { + return defaultBaseDir; + } + + /** + * 以默认配置进行文件上传 + * + * @param file 上传的文件 + * @return 文件名称 + * @throws Exception + */ + public static final String upload(MultipartFile file) throws IOException + { + try + { + return upload(getDefaultBaseDir(), file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); + } + catch (Exception e) + { + throw new IOException(e.getMessage(), e); + } + } + + /** + * 根据文件路径上传 + * + * @param baseDir 相对应用的基目录 + * @param file 上传的文件 + * @return 文件名称 + * @throws IOException + */ + public static final String upload(String baseDir, MultipartFile file) throws IOException + { + try + { + return upload(baseDir, file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); + } + catch (Exception e) + { + throw new IOException(e.getMessage(), e); + } + } + + /** + * 文件上传 + * + * @param baseDir 相对应用的基目录 + * @param file 上传的文件 + * @param allowedExtension 上传文件类型 + * @return 返回上传成功的文件名 + * @throws FileSizeLimitExceededException 如果超出最大大小 + * @throws FileNameLengthLimitExceededException 文件名太长 + * @throws IOException 比如读写文件出错时 + * @throws InvalidExtensionException 文件校验异常 + */ + public static final String upload(String baseDir, MultipartFile file, String[] allowedExtension) + throws FileSizeLimitExceededException, IOException, FileNameLengthLimitExceededException, + InvalidExtensionException + { + int fileNamelength = Objects.requireNonNull(file.getOriginalFilename()).length(); + if (fileNamelength > FileUploadUtils.DEFAULT_FILE_NAME_LENGTH) + { + throw new FileNameLengthLimitExceededException(FileUploadUtils.DEFAULT_FILE_NAME_LENGTH); + } + + assertAllowed(file, allowedExtension); + + String fileName = extractFilename(file); + + String absPath = getAbsoluteFile(baseDir, fileName).getAbsolutePath(); + file.transferTo(Paths.get(absPath)); + return getPathFileName(baseDir, fileName); + } + + /** + * 编码文件名 + */ + public static final String extractFilename(MultipartFile file) + { + return StringUtils.format("{}/{}_{}.{}", DateUtils.datePath(), + FilenameUtils.getBaseName(file.getOriginalFilename()), Seq.getId(Seq.uploadSeqType), getExtension(file)); + } + + public static final File getAbsoluteFile(String uploadDir, String fileName) throws IOException + { + File desc = new File(uploadDir + File.separator + fileName); + + if (!desc.exists()) + { + if (!desc.getParentFile().exists()) + { + desc.getParentFile().mkdirs(); + } + } + return desc; + } + + public static final String getPathFileName(String uploadDir, String fileName) throws IOException + { + int dirLastIndex = RuoYiConfig.getProfile().length() + 1; + String currentDir = StringUtils.substring(uploadDir, dirLastIndex); + return Constants.RESOURCE_PREFIX + "/" + currentDir + "/" + fileName; + } + + /** + * 文件大小校验 + * + * @param file 上传的文件 + * @return + * @throws FileSizeLimitExceededException 如果超出最大大小 + * @throws InvalidExtensionException + */ + public static final void assertAllowed(MultipartFile file, String[] allowedExtension) + throws FileSizeLimitExceededException, InvalidExtensionException + { + long size = file.getSize(); + if (size > DEFAULT_MAX_SIZE) + { + throw new FileSizeLimitExceededException(DEFAULT_MAX_SIZE / 1024 / 1024); + } + + String fileName = file.getOriginalFilename(); + String extension = getExtension(file); + if (allowedExtension != null && !isAllowedExtension(extension, allowedExtension)) + { + if (allowedExtension == MimeTypeUtils.IMAGE_EXTENSION) + { + throw new InvalidExtensionException.InvalidImageExtensionException(allowedExtension, extension, + fileName); + } + else if (allowedExtension == MimeTypeUtils.FLASH_EXTENSION) + { + throw new InvalidExtensionException.InvalidFlashExtensionException(allowedExtension, extension, + fileName); + } + else if (allowedExtension == MimeTypeUtils.MEDIA_EXTENSION) + { + throw new InvalidExtensionException.InvalidMediaExtensionException(allowedExtension, extension, + fileName); + } + else if (allowedExtension == MimeTypeUtils.VIDEO_EXTENSION) + { + throw new InvalidExtensionException.InvalidVideoExtensionException(allowedExtension, extension, + fileName); + } + else + { + throw new InvalidExtensionException(allowedExtension, extension, fileName); + } + } + + } + + /** + * 判断MIME类型是否是允许的MIME类型 + * + * @param extension + * @param allowedExtension + * @return + */ + public static final boolean isAllowedExtension(String extension, String[] allowedExtension) + { + for (String str : allowedExtension) + { + if (str.equalsIgnoreCase(extension)) + { + return true; + } + } + return false; + } + + /** + * 获取文件名的后缀 + * + * @param file 表单文件 + * @return 后缀名 + */ + public static final String getExtension(MultipartFile file) + { + String extension = FilenameUtils.getExtension(file.getOriginalFilename()); + if (StringUtils.isEmpty(extension)) + { + extension = MimeTypeUtils.getExtension(Objects.requireNonNull(file.getContentType())); + } + return extension; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUtils.java new file mode 100644 index 0000000..f844270 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUtils.java @@ -0,0 +1,293 @@ +package com.ruoyi.common.utils.file; + +import java.io.File; +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.OutputStream; +import java.io.UnsupportedEncodingException; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.apache.commons.io.IOUtils; +import org.apache.commons.lang3.ArrayUtils; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.uuid.IdUtils; +import org.apache.commons.io.FilenameUtils; + +/** + * 文件处理工具类 + * + * @author ruoyi + */ +public class FileUtils +{ + public static String FILENAME_PATTERN = "[a-zA-Z0-9_\\-\\|\\.\\u4e00-\\u9fa5]+"; + + /** + * 输出指定文件的byte数组 + * + * @param filePath 文件路径 + * @param os 输出流 + * @return + */ + public static void writeBytes(String filePath, OutputStream os) throws IOException + { + FileInputStream fis = null; + try + { + File file = new File(filePath); + if (!file.exists()) + { + throw new FileNotFoundException(filePath); + } + fis = new FileInputStream(file); + byte[] b = new byte[1024]; + int length; + while ((length = fis.read(b)) > 0) + { + os.write(b, 0, length); + } + } + catch (IOException e) + { + throw e; + } + finally + { + IOUtils.close(os); + IOUtils.close(fis); + } + } + + /** + * 写数据到文件中 + * + * @param data 数据 + * @return 目标文件 + * @throws IOException IO异常 + */ + public static String writeImportBytes(byte[] data) throws IOException + { + return writeBytes(data, RuoYiConfig.getImportPath()); + } + + /** + * 写数据到文件中 + * + * @param data 数据 + * @param uploadDir 目标文件 + * @return 目标文件 + * @throws IOException IO异常 + */ + public static String writeBytes(byte[] data, String uploadDir) throws IOException + { + FileOutputStream fos = null; + String pathName = ""; + try + { + String extension = getFileExtendName(data); + pathName = DateUtils.datePath() + "/" + IdUtils.fastUUID() + "." + extension; + File file = FileUploadUtils.getAbsoluteFile(uploadDir, pathName); + fos = new FileOutputStream(file); + fos.write(data); + } + finally + { + IOUtils.close(fos); + } + return FileUploadUtils.getPathFileName(uploadDir, pathName); + } + + /** + * 删除文件 + * + * @param filePath 文件 + * @return + */ + public static boolean deleteFile(String filePath) + { + boolean flag = false; + File file = new File(filePath); + // 路径为文件且不为空则进行删除 + if (file.isFile() && file.exists()) + { + file.delete(); + flag = true; + } + return flag; + } + + /** + * 文件名称验证 + * + * @param filename 文件名称 + * @return true 正常 false 非法 + */ + public static boolean isValidFilename(String filename) + { + return filename.matches(FILENAME_PATTERN); + } + + /** + * 检查文件是否可下载 + * + * @param resource 需要下载的文件 + * @return true 正常 false 非法 + */ + public static boolean checkAllowDownload(String resource) + { + // 禁止目录上跳级别 + if (StringUtils.contains(resource, "..")) + { + return false; + } + + // 检查允许下载的文件规则 + if (ArrayUtils.contains(MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION, FileTypeUtils.getFileType(resource))) + { + return true; + } + + // 不在允许下载的文件规则 + return false; + } + + /** + * 下载文件名重新编码 + * + * @param request 请求对象 + * @param fileName 文件名 + * @return 编码后的文件名 + */ + public static String setFileDownloadHeader(HttpServletRequest request, String fileName) throws UnsupportedEncodingException + { + final String agent = request.getHeader("USER-AGENT"); + String filename = fileName; + if (agent.contains("MSIE")) + { + // IE浏览器 + filename = URLEncoder.encode(filename, "utf-8"); + filename = filename.replace("+", " "); + } + else if (agent.contains("Firefox")) + { + // 火狐浏览器 + filename = new String(fileName.getBytes(), "ISO8859-1"); + } + else if (agent.contains("Chrome")) + { + // google浏览器 + filename = URLEncoder.encode(filename, "utf-8"); + } + else + { + // 其它浏览器 + filename = URLEncoder.encode(filename, "utf-8"); + } + return filename; + } + + /** + * 下载文件名重新编码 + * + * @param response 响应对象 + * @param realFileName 真实文件名 + */ + public static void setAttachmentResponseHeader(HttpServletResponse response, String realFileName) throws UnsupportedEncodingException + { + String percentEncodedFileName = percentEncode(realFileName); + + StringBuilder contentDispositionValue = new StringBuilder(); + contentDispositionValue.append("attachment; filename=") + .append(percentEncodedFileName) + .append(";") + .append("filename*=") + .append("utf-8''") + .append(percentEncodedFileName); + + response.addHeader("Access-Control-Expose-Headers", "Content-Disposition,download-filename"); + response.setHeader("Content-disposition", contentDispositionValue.toString()); + response.setHeader("download-filename", percentEncodedFileName); + } + + /** + * 百分号编码工具方法 + * + * @param s 需要百分号编码的字符串 + * @return 百分号编码后的字符串 + */ + public static String percentEncode(String s) throws UnsupportedEncodingException + { + String encode = URLEncoder.encode(s, StandardCharsets.UTF_8.toString()); + return encode.replaceAll("\\+", "%20"); + } + + /** + * 获取图像后缀 + * + * @param photoByte 图像数据 + * @return 后缀名 + */ + public static String getFileExtendName(byte[] photoByte) + { + String strFileExtendName = "jpg"; + if ((photoByte[0] == 71) && (photoByte[1] == 73) && (photoByte[2] == 70) && (photoByte[3] == 56) + && ((photoByte[4] == 55) || (photoByte[4] == 57)) && (photoByte[5] == 97)) + { + strFileExtendName = "gif"; + } + else if ((photoByte[6] == 74) && (photoByte[7] == 70) && (photoByte[8] == 73) && (photoByte[9] == 70)) + { + strFileExtendName = "jpg"; + } + else if ((photoByte[0] == 66) && (photoByte[1] == 77)) + { + strFileExtendName = "bmp"; + } + else if ((photoByte[1] == 80) && (photoByte[2] == 78) && (photoByte[3] == 71)) + { + strFileExtendName = "png"; + } + return strFileExtendName; + } + + /** + * 获取文件名称 /profile/upload/2022/04/16/ruoyi.png -- ruoyi.png + * + * @param fileName 路径名称 + * @return 没有文件路径的名称 + */ + public static String getName(String fileName) + { + if (fileName == null) + { + return null; + } + int lastUnixPos = fileName.lastIndexOf('/'); + int lastWindowsPos = fileName.lastIndexOf('\\'); + int index = Math.max(lastUnixPos, lastWindowsPos); + return fileName.substring(index + 1); + } + + /** + * 获取不带后缀文件名称 /profile/upload/2022/04/16/ruoyi.png -- ruoyi + * + * @param fileName 路径名称 + * @return 没有文件路径和后缀的名称 + */ + public static String getNameNotSuffix(String fileName) + { + if (fileName == null) + { + return null; + } + String baseName = FilenameUtils.getBaseName(fileName); + return baseName; + } + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/ImageUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/ImageUtils.java new file mode 100644 index 0000000..432dfda --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/ImageUtils.java @@ -0,0 +1,98 @@ +package com.ruoyi.common.utils.file; + +import java.io.ByteArrayInputStream; +import java.io.FileInputStream; +import java.io.InputStream; +import java.net.URL; +import java.net.URLConnection; +import java.util.Arrays; +import org.apache.poi.util.IOUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.utils.StringUtils; + +/** + * 图片处理工具类 + * + * @author ruoyi + */ +public class ImageUtils +{ + private static final Logger log = LoggerFactory.getLogger(ImageUtils.class); + + public static byte[] getImage(String imagePath) + { + InputStream is = getFile(imagePath); + try + { + return IOUtils.toByteArray(is); + } + catch (Exception e) + { + log.error("图片加载异常 {}", e); + return null; + } + finally + { + IOUtils.closeQuietly(is); + } + } + + public static InputStream getFile(String imagePath) + { + try + { + byte[] result = readFile(imagePath); + result = Arrays.copyOf(result, result.length); + return new ByteArrayInputStream(result); + } + catch (Exception e) + { + log.error("获取图片异常 {}", e); + } + return null; + } + + /** + * 读取文件为字节数据 + * + * @param url 地址 + * @return 字节数据 + */ + public static byte[] readFile(String url) + { + InputStream in = null; + try + { + if (url.startsWith("http")) + { + // 网络地址 + URL urlObj = new URL(url); + URLConnection urlConnection = urlObj.openConnection(); + urlConnection.setConnectTimeout(30 * 1000); + urlConnection.setReadTimeout(60 * 1000); + urlConnection.setDoInput(true); + in = urlConnection.getInputStream(); + } + else + { + // 本机地址 + String localPath = RuoYiConfig.getProfile(); + String downloadPath = localPath + StringUtils.substringAfter(url, Constants.RESOURCE_PREFIX); + in = new FileInputStream(downloadPath); + } + return IOUtils.toByteArray(in); + } + catch (Exception e) + { + log.error("获取文件路径异常 {}", e); + return null; + } + finally + { + IOUtils.closeQuietly(in); + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java new file mode 100644 index 0000000..f968f1a --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java @@ -0,0 +1,59 @@ +package com.ruoyi.common.utils.file; + +/** + * 媒体类型工具类 + * + * @author ruoyi + */ +public class MimeTypeUtils +{ + public static final String IMAGE_PNG = "image/png"; + + public static final String IMAGE_JPG = "image/jpg"; + + public static final String IMAGE_JPEG = "image/jpeg"; + + public static final String IMAGE_BMP = "image/bmp"; + + public static final String IMAGE_GIF = "image/gif"; + + public static final String[] IMAGE_EXTENSION = { "bmp", "gif", "jpg", "jpeg", "png" }; + + public static final String[] FLASH_EXTENSION = { "swf", "flv" }; + + public static final String[] MEDIA_EXTENSION = { "swf", "flv", "mp3", "wav", "wma", "wmv", "mid", "avi", "mpg", + "asf", "rm", "rmvb" }; + + public static final String[] VIDEO_EXTENSION = { "mp4", "avi", "rmvb" }; + + public static final String[] DEFAULT_ALLOWED_EXTENSION = { + // 图片 + "bmp", "gif", "jpg", "jpeg", "png", + // word excel powerpoint + "doc", "docx", "xls", "xlsx", "ppt", "pptx", "html", "htm", "txt", + // 压缩文件 + "rar", "zip", "gz", "bz2", + // 视频格式 + "mp4", "avi", "rmvb", + // pdf + "pdf" }; + + public static String getExtension(String prefix) + { + switch (prefix) + { + case IMAGE_PNG: + return "png"; + case IMAGE_JPG: + return "jpg"; + case IMAGE_JPEG: + return "jpeg"; + case IMAGE_BMP: + return "bmp"; + case IMAGE_GIF: + return "gif"; + default: + return ""; + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/EscapeUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/EscapeUtil.java new file mode 100644 index 0000000..f52e83e --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/EscapeUtil.java @@ -0,0 +1,167 @@ +package com.ruoyi.common.utils.html; + +import com.ruoyi.common.utils.StringUtils; + +/** + * 转义和反转义工具类 + * + * @author ruoyi + */ +public class EscapeUtil +{ + public static final String RE_HTML_MARK = "(<[^<]*?>)|(<[\\s]*?/[^<]*?>)|(<[^<]*?/[\\s]*?>)"; + + private static final char[][] TEXT = new char[64][]; + + static + { + for (int i = 0; i < 64; i++) + { + TEXT[i] = new char[] { (char) i }; + } + + // special HTML characters + TEXT['\''] = "'".toCharArray(); // 单引号 + TEXT['"'] = """.toCharArray(); // 双引号 + TEXT['&'] = "&".toCharArray(); // &符 + TEXT['<'] = "<".toCharArray(); // 小于号 + TEXT['>'] = ">".toCharArray(); // 大于号 + } + + /** + * 转义文本中的HTML字符为安全的字符 + * + * @param text 被转义的文本 + * @return 转义后的文本 + */ + public static String escape(String text) + { + return encode(text); + } + + /** + * 还原被转义的HTML特殊字符 + * + * @param content 包含转义符的HTML内容 + * @return 转换后的字符串 + */ + public static String unescape(String content) + { + return decode(content); + } + + /** + * 清除所有HTML标签,但是不删除标签内的内容 + * + * @param content 文本 + * @return 清除标签后的文本 + */ + public static String clean(String content) + { + return new HTMLFilter().filter(content); + } + + /** + * Escape编码 + * + * @param text 被编码的文本 + * @return 编码后的字符 + */ + private static String encode(String text) + { + if (StringUtils.isEmpty(text)) + { + return StringUtils.EMPTY; + } + + final StringBuilder tmp = new StringBuilder(text.length() * 6); + char c; + for (int i = 0; i < text.length(); i++) + { + c = text.charAt(i); + if (c < 256) + { + tmp.append("%"); + if (c < 16) + { + tmp.append("0"); + } + tmp.append(Integer.toString(c, 16)); + } + else + { + tmp.append("%u"); + if (c <= 0xfff) + { + // issue#I49JU8@Gitee + tmp.append("0"); + } + tmp.append(Integer.toString(c, 16)); + } + } + return tmp.toString(); + } + + /** + * Escape解码 + * + * @param content 被转义的内容 + * @return 解码后的字符串 + */ + public static String decode(String content) + { + if (StringUtils.isEmpty(content)) + { + return content; + } + + StringBuilder tmp = new StringBuilder(content.length()); + int lastPos = 0, pos = 0; + char ch; + while (lastPos < content.length()) + { + pos = content.indexOf("%", lastPos); + if (pos == lastPos) + { + if (content.charAt(pos + 1) == 'u') + { + ch = (char) Integer.parseInt(content.substring(pos + 2, pos + 6), 16); + tmp.append(ch); + lastPos = pos + 6; + } + else + { + ch = (char) Integer.parseInt(content.substring(pos + 1, pos + 3), 16); + tmp.append(ch); + lastPos = pos + 3; + } + } + else + { + if (pos == -1) + { + tmp.append(content.substring(lastPos)); + lastPos = content.length(); + } + else + { + tmp.append(content.substring(lastPos, pos)); + lastPos = pos; + } + } + } + return tmp.toString(); + } + + public static void main(String[] args) + { + String html = ""; + String escape = EscapeUtil.escape(html); + // String html = "ipt>alert(\"XSS\")ipt>"; + // String html = "<123"; + // String html = "123>"; + System.out.println("clean: " + EscapeUtil.clean(html)); + System.out.println("escape: " + escape); + System.out.println("unescape: " + EscapeUtil.unescape(escape)); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/HTMLFilter.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/HTMLFilter.java new file mode 100644 index 0000000..db069bc --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/html/HTMLFilter.java @@ -0,0 +1,570 @@ +package com.ruoyi.common.utils.html; + +import java.util.ArrayList; +import java.util.Collections; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.ConcurrentMap; +import java.util.regex.Matcher; +import java.util.regex.Pattern; + +/** + * HTML过滤器,用于去除XSS漏洞隐患。 + * + * @author ruoyi + */ +public final class HTMLFilter +{ + /** + * regex flag union representing /si modifiers in php + **/ + private static final int REGEX_FLAGS_SI = Pattern.CASE_INSENSITIVE | Pattern.DOTALL; + private static final Pattern P_COMMENTS = Pattern.compile("", Pattern.DOTALL); + private static final Pattern P_COMMENT = Pattern.compile("^!--(.*)--$", REGEX_FLAGS_SI); + private static final Pattern P_TAGS = Pattern.compile("<(.*?)>", Pattern.DOTALL); + private static final Pattern P_END_TAG = Pattern.compile("^/([a-z0-9]+)", REGEX_FLAGS_SI); + private static final Pattern P_START_TAG = Pattern.compile("^([a-z0-9]+)(.*?)(/?)$", REGEX_FLAGS_SI); + private static final Pattern P_QUOTED_ATTRIBUTES = Pattern.compile("([a-z0-9]+)=([\"'])(.*?)\\2", REGEX_FLAGS_SI); + private static final Pattern P_UNQUOTED_ATTRIBUTES = Pattern.compile("([a-z0-9]+)(=)([^\"\\s']+)", REGEX_FLAGS_SI); + private static final Pattern P_PROTOCOL = Pattern.compile("^([^:]+):", REGEX_FLAGS_SI); + private static final Pattern P_ENTITY = Pattern.compile("&#(\\d+);?"); + private static final Pattern P_ENTITY_UNICODE = Pattern.compile("&#x([0-9a-f]+);?"); + private static final Pattern P_ENCODE = Pattern.compile("%([0-9a-f]{2});?"); + private static final Pattern P_VALID_ENTITIES = Pattern.compile("&([^&;]*)(?=(;|&|$))"); + private static final Pattern P_VALID_QUOTES = Pattern.compile("(>|^)([^<]+?)(<|$)", Pattern.DOTALL); + private static final Pattern P_END_ARROW = Pattern.compile("^>"); + private static final Pattern P_BODY_TO_END = Pattern.compile("<([^>]*?)(?=<|$)"); + private static final Pattern P_XML_CONTENT = Pattern.compile("(^|>)([^<]*?)(?=>)"); + private static final Pattern P_STRAY_LEFT_ARROW = Pattern.compile("<([^>]*?)(?=<|$)"); + private static final Pattern P_STRAY_RIGHT_ARROW = Pattern.compile("(^|>)([^<]*?)(?=>)"); + private static final Pattern P_AMP = Pattern.compile("&"); + private static final Pattern P_QUOTE = Pattern.compile("\""); + private static final Pattern P_LEFT_ARROW = Pattern.compile("<"); + private static final Pattern P_RIGHT_ARROW = Pattern.compile(">"); + private static final Pattern P_BOTH_ARROWS = Pattern.compile("<>"); + + // @xxx could grow large... maybe use sesat's ReferenceMap + private static final ConcurrentMap P_REMOVE_PAIR_BLANKS = new ConcurrentHashMap<>(); + private static final ConcurrentMap P_REMOVE_SELF_BLANKS = new ConcurrentHashMap<>(); + + /** + * set of allowed html elements, along with allowed attributes for each element + **/ + private final Map> vAllowed; + /** + * counts of open tags for each (allowable) html element + **/ + private final Map vTagCounts = new HashMap<>(); + + /** + * html elements which must always be self-closing (e.g. "") + **/ + private final String[] vSelfClosingTags; + /** + * html elements which must always have separate opening and closing tags (e.g. "") + **/ + private final String[] vNeedClosingTags; + /** + * set of disallowed html elements + **/ + private final String[] vDisallowed; + /** + * attributes which should be checked for valid protocols + **/ + private final String[] vProtocolAtts; + /** + * allowed protocols + **/ + private final String[] vAllowedProtocols; + /** + * tags which should be removed if they contain no content (e.g. "" or "") + **/ + private final String[] vRemoveBlanks; + /** + * entities allowed within html markup + **/ + private final String[] vAllowedEntities; + /** + * flag determining whether comments are allowed in input String. + */ + private final boolean stripComment; + private final boolean encodeQuotes; + /** + * flag determining whether to try to make tags when presented with "unbalanced" angle brackets (e.g. "" + * becomes " text "). If set to false, unbalanced angle brackets will be html escaped. + */ + private final boolean alwaysMakeTags; + + /** + * Default constructor. + */ + public HTMLFilter() + { + vAllowed = new HashMap<>(); + + final ArrayList a_atts = new ArrayList<>(); + a_atts.add("href"); + a_atts.add("target"); + vAllowed.put("a", a_atts); + + final ArrayList img_atts = new ArrayList<>(); + img_atts.add("src"); + img_atts.add("width"); + img_atts.add("height"); + img_atts.add("alt"); + vAllowed.put("img", img_atts); + + final ArrayList no_atts = new ArrayList<>(); + vAllowed.put("b", no_atts); + vAllowed.put("strong", no_atts); + vAllowed.put("i", no_atts); + vAllowed.put("em", no_atts); + + vSelfClosingTags = new String[] { "img" }; + vNeedClosingTags = new String[] { "a", "b", "strong", "i", "em" }; + vDisallowed = new String[] {}; + vAllowedProtocols = new String[] { "http", "mailto", "https" }; // no ftp. + vProtocolAtts = new String[] { "src", "href" }; + vRemoveBlanks = new String[] { "a", "b", "strong", "i", "em" }; + vAllowedEntities = new String[] { "amp", "gt", "lt", "quot" }; + stripComment = true; + encodeQuotes = true; + alwaysMakeTags = false; + } + + /** + * Map-parameter configurable constructor. + * + * @param conf map containing configuration. keys match field names. + */ + @SuppressWarnings("unchecked") + public HTMLFilter(final Map conf) + { + + assert conf.containsKey("vAllowed") : "configuration requires vAllowed"; + assert conf.containsKey("vSelfClosingTags") : "configuration requires vSelfClosingTags"; + assert conf.containsKey("vNeedClosingTags") : "configuration requires vNeedClosingTags"; + assert conf.containsKey("vDisallowed") : "configuration requires vDisallowed"; + assert conf.containsKey("vAllowedProtocols") : "configuration requires vAllowedProtocols"; + assert conf.containsKey("vProtocolAtts") : "configuration requires vProtocolAtts"; + assert conf.containsKey("vRemoveBlanks") : "configuration requires vRemoveBlanks"; + assert conf.containsKey("vAllowedEntities") : "configuration requires vAllowedEntities"; + + vAllowed = Collections.unmodifiableMap((HashMap>) conf.get("vAllowed")); + vSelfClosingTags = (String[]) conf.get("vSelfClosingTags"); + vNeedClosingTags = (String[]) conf.get("vNeedClosingTags"); + vDisallowed = (String[]) conf.get("vDisallowed"); + vAllowedProtocols = (String[]) conf.get("vAllowedProtocols"); + vProtocolAtts = (String[]) conf.get("vProtocolAtts"); + vRemoveBlanks = (String[]) conf.get("vRemoveBlanks"); + vAllowedEntities = (String[]) conf.get("vAllowedEntities"); + stripComment = conf.containsKey("stripComment") ? (Boolean) conf.get("stripComment") : true; + encodeQuotes = conf.containsKey("encodeQuotes") ? (Boolean) conf.get("encodeQuotes") : true; + alwaysMakeTags = conf.containsKey("alwaysMakeTags") ? (Boolean) conf.get("alwaysMakeTags") : true; + } + + private void reset() + { + vTagCounts.clear(); + } + + // --------------------------------------------------------------- + // my versions of some PHP library functions + public static String chr(final int decimal) + { + return String.valueOf((char) decimal); + } + + public static String htmlSpecialChars(final String s) + { + String result = s; + result = regexReplace(P_AMP, "&", result); + result = regexReplace(P_QUOTE, """, result); + result = regexReplace(P_LEFT_ARROW, "<", result); + result = regexReplace(P_RIGHT_ARROW, ">", result); + return result; + } + + // --------------------------------------------------------------- + + /** + * given a user submitted input String, filter out any invalid or restricted html. + * + * @param input text (i.e. submitted by a user) than may contain html + * @return "clean" version of input, with only valid, whitelisted html elements allowed + */ + public String filter(final String input) + { + reset(); + String s = input; + + s = escapeComments(s); + + s = balanceHTML(s); + + s = checkTags(s); + + s = processRemoveBlanks(s); + + // s = validateEntities(s); + + return s; + } + + public boolean isAlwaysMakeTags() + { + return alwaysMakeTags; + } + + public boolean isStripComments() + { + return stripComment; + } + + private String escapeComments(final String s) + { + final Matcher m = P_COMMENTS.matcher(s); + final StringBuffer buf = new StringBuffer(); + if (m.find()) + { + final String match = m.group(1); // (.*?) + m.appendReplacement(buf, Matcher.quoteReplacement("")); + } + m.appendTail(buf); + + return buf.toString(); + } + + private String balanceHTML(String s) + { + if (alwaysMakeTags) + { + // + // try and form html + // + s = regexReplace(P_END_ARROW, "", s); + // 不追加结束标签 + s = regexReplace(P_BODY_TO_END, "<$1>", s); + s = regexReplace(P_XML_CONTENT, "$1<$2", s); + + } + else + { + // + // escape stray brackets + // + s = regexReplace(P_STRAY_LEFT_ARROW, "<$1", s); + s = regexReplace(P_STRAY_RIGHT_ARROW, "$1$2><", s); + + // + // the last regexp causes '<>' entities to appear + // (we need to do a lookahead assertion so that the last bracket can + // be used in the next pass of the regexp) + // + s = regexReplace(P_BOTH_ARROWS, "", s); + } + + return s; + } + + private String checkTags(String s) + { + Matcher m = P_TAGS.matcher(s); + + final StringBuffer buf = new StringBuffer(); + while (m.find()) + { + String replaceStr = m.group(1); + replaceStr = processTag(replaceStr); + m.appendReplacement(buf, Matcher.quoteReplacement(replaceStr)); + } + m.appendTail(buf); + + // these get tallied in processTag + // (remember to reset before subsequent calls to filter method) + final StringBuilder sBuilder = new StringBuilder(buf.toString()); + for (String key : vTagCounts.keySet()) + { + for (int ii = 0; ii < vTagCounts.get(key); ii++) + { + sBuilder.append(""); + } + } + s = sBuilder.toString(); + + return s; + } + + private String processRemoveBlanks(final String s) + { + String result = s; + for (String tag : vRemoveBlanks) + { + if (!P_REMOVE_PAIR_BLANKS.containsKey(tag)) + { + P_REMOVE_PAIR_BLANKS.putIfAbsent(tag, Pattern.compile("<" + tag + "(\\s[^>]*)?>")); + } + result = regexReplace(P_REMOVE_PAIR_BLANKS.get(tag), "", result); + if (!P_REMOVE_SELF_BLANKS.containsKey(tag)) + { + P_REMOVE_SELF_BLANKS.putIfAbsent(tag, Pattern.compile("<" + tag + "(\\s[^>]*)?/>")); + } + result = regexReplace(P_REMOVE_SELF_BLANKS.get(tag), "", result); + } + + return result; + } + + private static String regexReplace(final Pattern regex_pattern, final String replacement, final String s) + { + Matcher m = regex_pattern.matcher(s); + return m.replaceAll(replacement); + } + + private String processTag(final String s) + { + // ending tags + Matcher m = P_END_TAG.matcher(s); + if (m.find()) + { + final String name = m.group(1).toLowerCase(); + if (allowed(name)) + { + if (!inArray(name, vSelfClosingTags)) + { + if (vTagCounts.containsKey(name)) + { + vTagCounts.put(name, vTagCounts.get(name) - 1); + return ""; + } + } + } + } + + // starting tags + m = P_START_TAG.matcher(s); + if (m.find()) + { + final String name = m.group(1).toLowerCase(); + final String body = m.group(2); + String ending = m.group(3); + + // debug( "in a starting tag, name='" + name + "'; body='" + body + "'; ending='" + ending + "'" ); + if (allowed(name)) + { + final StringBuilder params = new StringBuilder(); + + final Matcher m2 = P_QUOTED_ATTRIBUTES.matcher(body); + final Matcher m3 = P_UNQUOTED_ATTRIBUTES.matcher(body); + final List paramNames = new ArrayList<>(); + final List paramValues = new ArrayList<>(); + while (m2.find()) + { + paramNames.add(m2.group(1)); // ([a-z0-9]+) + paramValues.add(m2.group(3)); // (.*?) + } + while (m3.find()) + { + paramNames.add(m3.group(1)); // ([a-z0-9]+) + paramValues.add(m3.group(3)); // ([^\"\\s']+) + } + + String paramName, paramValue; + for (int ii = 0; ii < paramNames.size(); ii++) + { + paramName = paramNames.get(ii).toLowerCase(); + paramValue = paramValues.get(ii); + + // debug( "paramName='" + paramName + "'" ); + // debug( "paramValue='" + paramValue + "'" ); + // debug( "allowed? " + vAllowed.get( name ).contains( paramName ) ); + + if (allowedAttribute(name, paramName)) + { + if (inArray(paramName, vProtocolAtts)) + { + paramValue = processParamProtocol(paramValue); + } + params.append(' ').append(paramName).append("=\\\"").append(paramValue).append("\""); + } + } + + if (inArray(name, vSelfClosingTags)) + { + ending = " /"; + } + + if (inArray(name, vNeedClosingTags)) + { + ending = ""; + } + + if (ending == null || ending.length() < 1) + { + if (vTagCounts.containsKey(name)) + { + vTagCounts.put(name, vTagCounts.get(name) + 1); + } + else + { + vTagCounts.put(name, 1); + } + } + else + { + ending = " /"; + } + return "<" + name + params + ending + ">"; + } + else + { + return ""; + } + } + + // comments + m = P_COMMENT.matcher(s); + if (!stripComment && m.find()) + { + return "<" + m.group() + ">"; + } + + return ""; + } + + private String processParamProtocol(String s) + { + s = decodeEntities(s); + final Matcher m = P_PROTOCOL.matcher(s); + if (m.find()) + { + final String protocol = m.group(1); + if (!inArray(protocol, vAllowedProtocols)) + { + // bad protocol, turn into local anchor link instead + s = "#" + s.substring(protocol.length() + 1); + if (s.startsWith("#//")) + { + s = "#" + s.substring(3); + } + } + } + + return s; + } + + private String decodeEntities(String s) + { + StringBuffer buf = new StringBuffer(); + + Matcher m = P_ENTITY.matcher(s); + while (m.find()) + { + final String match = m.group(1); + final int decimal = Integer.decode(match).intValue(); + m.appendReplacement(buf, Matcher.quoteReplacement(chr(decimal))); + } + m.appendTail(buf); + s = buf.toString(); + + buf = new StringBuffer(); + m = P_ENTITY_UNICODE.matcher(s); + while (m.find()) + { + final String match = m.group(1); + final int decimal = Integer.valueOf(match, 16).intValue(); + m.appendReplacement(buf, Matcher.quoteReplacement(chr(decimal))); + } + m.appendTail(buf); + s = buf.toString(); + + buf = new StringBuffer(); + m = P_ENCODE.matcher(s); + while (m.find()) + { + final String match = m.group(1); + final int decimal = Integer.valueOf(match, 16).intValue(); + m.appendReplacement(buf, Matcher.quoteReplacement(chr(decimal))); + } + m.appendTail(buf); + s = buf.toString(); + + s = validateEntities(s); + return s; + } + + private String validateEntities(final String s) + { + StringBuffer buf = new StringBuffer(); + + // validate entities throughout the string + Matcher m = P_VALID_ENTITIES.matcher(s); + while (m.find()) + { + final String one = m.group(1); // ([^&;]*) + final String two = m.group(2); // (?=(;|&|$)) + m.appendReplacement(buf, Matcher.quoteReplacement(checkEntity(one, two))); + } + m.appendTail(buf); + + return encodeQuotes(buf.toString()); + } + + private String encodeQuotes(final String s) + { + if (encodeQuotes) + { + StringBuffer buf = new StringBuffer(); + Matcher m = P_VALID_QUOTES.matcher(s); + while (m.find()) + { + final String one = m.group(1); // (>|^) + final String two = m.group(2); // ([^<]+?) + final String three = m.group(3); // (<|$) + // 不替换双引号为",防止json格式无效 regexReplace(P_QUOTE, """, two) + m.appendReplacement(buf, Matcher.quoteReplacement(one + two + three)); + } + m.appendTail(buf); + return buf.toString(); + } + else + { + return s; + } + } + + private String checkEntity(final String preamble, final String term) + { + + return ";".equals(term) && isValidEntity(preamble) ? '&' + preamble : "&" + preamble; + } + + private boolean isValidEntity(final String entity) + { + return inArray(entity, vAllowedEntities); + } + + private static boolean inArray(final String s, final String[] array) + { + for (String item : array) + { + if (item != null && item.equals(s)) + { + return true; + } + } + return false; + } + + private boolean allowed(final String name) + { + return (vAllowed.isEmpty() || vAllowed.containsKey(name)) && !inArray(name, vDisallowed); + } + + private boolean allowedAttribute(final String name, final String paramName) + { + return allowed(name) && (vAllowed.isEmpty() || vAllowed.get(name).contains(paramName)); + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpHelper.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpHelper.java new file mode 100644 index 0000000..589d123 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpHelper.java @@ -0,0 +1,55 @@ +package com.ruoyi.common.utils.http; + +import java.io.BufferedReader; +import java.io.IOException; +import java.io.InputStream; +import java.io.InputStreamReader; +import java.nio.charset.StandardCharsets; +import javax.servlet.ServletRequest; +import org.apache.commons.lang3.exception.ExceptionUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** + * 通用http工具封装 + * + * @author ruoyi + */ +public class HttpHelper +{ + private static final Logger LOGGER = LoggerFactory.getLogger(HttpHelper.class); + + public static String getBodyString(ServletRequest request) + { + StringBuilder sb = new StringBuilder(); + BufferedReader reader = null; + try (InputStream inputStream = request.getInputStream()) + { + reader = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8)); + String line = ""; + while ((line = reader.readLine()) != null) + { + sb.append(line); + } + } + catch (IOException e) + { + LOGGER.warn("getBodyString出现问题!"); + } + finally + { + if (reader != null) + { + try + { + reader.close(); + } + catch (IOException e) + { + LOGGER.error(ExceptionUtils.getMessage(e)); + } + } + } + return sb.toString(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpUtils.java new file mode 100644 index 0000000..f85c82c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/http/HttpUtils.java @@ -0,0 +1,274 @@ +package com.ruoyi.common.utils.http; + +import java.io.BufferedReader; +import java.io.IOException; +import java.io.InputStream; +import java.io.InputStreamReader; +import java.io.PrintWriter; +import java.net.ConnectException; +import java.net.SocketTimeoutException; +import java.net.URL; +import java.net.URLConnection; +import java.nio.charset.StandardCharsets; +import java.security.cert.X509Certificate; +import javax.net.ssl.HostnameVerifier; +import javax.net.ssl.HttpsURLConnection; +import javax.net.ssl.SSLContext; +import javax.net.ssl.SSLSession; +import javax.net.ssl.TrustManager; +import javax.net.ssl.X509TrustManager; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.utils.StringUtils; + +/** + * 通用http发送方法 + * + * @author ruoyi + */ +public class HttpUtils +{ + private static final Logger log = LoggerFactory.getLogger(HttpUtils.class); + + /** + * 向指定 URL 发送GET方法的请求 + * + * @param url 发送请求的 URL + * @return 所代表远程资源的响应结果 + */ + public static String sendGet(String url) + { + return sendGet(url, StringUtils.EMPTY); + } + + /** + * 向指定 URL 发送GET方法的请求 + * + * @param url 发送请求的 URL + * @param param 请求参数,请求参数应该是 name1=value1&name2=value2 的形式。 + * @return 所代表远程资源的响应结果 + */ + public static String sendGet(String url, String param) + { + return sendGet(url, param, Constants.UTF8); + } + + /** + * 向指定 URL 发送GET方法的请求 + * + * @param url 发送请求的 URL + * @param param 请求参数,请求参数应该是 name1=value1&name2=value2 的形式。 + * @param contentType 编码类型 + * @return 所代表远程资源的响应结果 + */ + public static String sendGet(String url, String param, String contentType) + { + StringBuilder result = new StringBuilder(); + BufferedReader in = null; + try + { + String urlNameString = StringUtils.isNotBlank(param) ? url + "?" + param : url; + log.info("sendGet - {}", urlNameString); + URL realUrl = new URL(urlNameString); + URLConnection connection = realUrl.openConnection(); + connection.setRequestProperty("accept", "*/*"); + connection.setRequestProperty("connection", "Keep-Alive"); + connection.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)"); + connection.connect(); + in = new BufferedReader(new InputStreamReader(connection.getInputStream(), contentType)); + String line; + while ((line = in.readLine()) != null) + { + result.append(line); + } + log.info("recv - {}", result); + } + catch (ConnectException e) + { + log.error("调用HttpUtils.sendGet ConnectException, url=" + url + ",param=" + param, e); + } + catch (SocketTimeoutException e) + { + log.error("调用HttpUtils.sendGet SocketTimeoutException, url=" + url + ",param=" + param, e); + } + catch (IOException e) + { + log.error("调用HttpUtils.sendGet IOException, url=" + url + ",param=" + param, e); + } + catch (Exception e) + { + log.error("调用HttpsUtil.sendGet Exception, url=" + url + ",param=" + param, e); + } + finally + { + try + { + if (in != null) + { + in.close(); + } + } + catch (Exception ex) + { + log.error("调用in.close Exception, url=" + url + ",param=" + param, ex); + } + } + return result.toString(); + } + + /** + * 向指定 URL 发送POST方法的请求 + * + * @param url 发送请求的 URL + * @param param 请求参数,请求参数应该是 name1=value1&name2=value2 的形式。 + * @return 所代表远程资源的响应结果 + */ + public static String sendPost(String url, String param) + { + PrintWriter out = null; + BufferedReader in = null; + StringBuilder result = new StringBuilder(); + try + { + log.info("sendPost - {}", url); + URL realUrl = new URL(url); + URLConnection conn = realUrl.openConnection(); + conn.setRequestProperty("accept", "*/*"); + conn.setRequestProperty("connection", "Keep-Alive"); + conn.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)"); + conn.setRequestProperty("Accept-Charset", "utf-8"); + conn.setRequestProperty("contentType", "utf-8"); + conn.setDoOutput(true); + conn.setDoInput(true); + out = new PrintWriter(conn.getOutputStream()); + out.print(param); + out.flush(); + in = new BufferedReader(new InputStreamReader(conn.getInputStream(), StandardCharsets.UTF_8)); + String line; + while ((line = in.readLine()) != null) + { + result.append(line); + } + log.info("recv - {}", result); + } + catch (ConnectException e) + { + log.error("调用HttpUtils.sendPost ConnectException, url=" + url + ",param=" + param, e); + } + catch (SocketTimeoutException e) + { + log.error("调用HttpUtils.sendPost SocketTimeoutException, url=" + url + ",param=" + param, e); + } + catch (IOException e) + { + log.error("调用HttpUtils.sendPost IOException, url=" + url + ",param=" + param, e); + } + catch (Exception e) + { + log.error("调用HttpsUtil.sendPost Exception, url=" + url + ",param=" + param, e); + } + finally + { + try + { + if (out != null) + { + out.close(); + } + if (in != null) + { + in.close(); + } + } + catch (IOException ex) + { + log.error("调用in.close Exception, url=" + url + ",param=" + param, ex); + } + } + return result.toString(); + } + + public static String sendSSLPost(String url, String param) + { + StringBuilder result = new StringBuilder(); + String urlNameString = url + "?" + param; + try + { + log.info("sendSSLPost - {}", urlNameString); + SSLContext sc = SSLContext.getInstance("SSL"); + sc.init(null, new TrustManager[] { new TrustAnyTrustManager() }, new java.security.SecureRandom()); + URL console = new URL(urlNameString); + HttpsURLConnection conn = (HttpsURLConnection) console.openConnection(); + conn.setRequestProperty("accept", "*/*"); + conn.setRequestProperty("connection", "Keep-Alive"); + conn.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)"); + conn.setRequestProperty("Accept-Charset", "utf-8"); + conn.setRequestProperty("contentType", "utf-8"); + conn.setDoOutput(true); + conn.setDoInput(true); + + conn.setSSLSocketFactory(sc.getSocketFactory()); + conn.setHostnameVerifier(new TrustAnyHostnameVerifier()); + conn.connect(); + InputStream is = conn.getInputStream(); + BufferedReader br = new BufferedReader(new InputStreamReader(is)); + String ret = ""; + while ((ret = br.readLine()) != null) + { + if (ret != null && !"".equals(ret.trim())) + { + result.append(new String(ret.getBytes(StandardCharsets.ISO_8859_1), StandardCharsets.UTF_8)); + } + } + log.info("recv - {}", result); + conn.disconnect(); + br.close(); + } + catch (ConnectException e) + { + log.error("调用HttpUtils.sendSSLPost ConnectException, url=" + url + ",param=" + param, e); + } + catch (SocketTimeoutException e) + { + log.error("调用HttpUtils.sendSSLPost SocketTimeoutException, url=" + url + ",param=" + param, e); + } + catch (IOException e) + { + log.error("调用HttpUtils.sendSSLPost IOException, url=" + url + ",param=" + param, e); + } + catch (Exception e) + { + log.error("调用HttpsUtil.sendSSLPost Exception, url=" + url + ",param=" + param, e); + } + return result.toString(); + } + + private static class TrustAnyTrustManager implements X509TrustManager + { + @Override + public void checkClientTrusted(X509Certificate[] chain, String authType) + { + } + + @Override + public void checkServerTrusted(X509Certificate[] chain, String authType) + { + } + + @Override + public X509Certificate[] getAcceptedIssuers() + { + return new X509Certificate[] {}; + } + } + + private static class TrustAnyHostnameVerifier implements HostnameVerifier + { + @Override + public boolean verify(String hostname, SSLSession session) + { + return true; + } + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/AddressUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/AddressUtils.java new file mode 100644 index 0000000..bc2f4d0 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/AddressUtils.java @@ -0,0 +1,55 @@ +package com.ruoyi.common.utils.ip; + +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.http.HttpUtils; + +/** + * 获取地址类 + * + * @author ruoyi + */ +public class AddressUtils +{ + private static final Logger log = LoggerFactory.getLogger(AddressUtils.class); + + // IP地址查询 + public static final String IP_URL = "http://whois.pconline.com.cn/ipJson.jsp"; + + // 未知地址 + public static final String UNKNOWN = "XX XX"; + + public static String getRealAddressByIP(String ip) + { + // 内网不查询 + if (IpUtils.internalIp(ip)) + { + return "内网IP"; + } + if (RuoYiConfig.isAddressEnabled()) + { + try + { + String rspStr = HttpUtils.sendGet(IP_URL, "ip=" + ip + "&json=true", Constants.GBK); + if (StringUtils.isEmpty(rspStr)) + { + log.error("获取地理位置异常 {}", ip); + return UNKNOWN; + } + JSONObject obj = JSONObject.parseObject(rspStr); + String region = obj.getString("pro"); + String city = obj.getString("city"); + return String.format("%s %s", region, city); + } + catch (Exception e) + { + log.error("获取地理位置异常 {}", ip); + } + } + return UNKNOWN; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/IpUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/IpUtils.java new file mode 100644 index 0000000..c18c56a --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/ip/IpUtils.java @@ -0,0 +1,264 @@ +package com.ruoyi.common.utils.ip; + +import java.net.InetAddress; +import java.net.UnknownHostException; +import javax.servlet.http.HttpServletRequest; +import com.ruoyi.common.utils.StringUtils; + +/** + * 获取IP方法 + * + * @author ruoyi + */ +public class IpUtils +{ + /** + * 获取客户端IP + * + * @param request 请求对象 + * @return IP地址 + */ + public static String getIpAddr(HttpServletRequest request) + { + if (request == null) + { + return "unknown"; + } + String ip = request.getHeader("x-forwarded-for"); + if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) + { + ip = request.getHeader("Proxy-Client-IP"); + } + if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) + { + ip = request.getHeader("X-Forwarded-For"); + } + if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) + { + ip = request.getHeader("WL-Proxy-Client-IP"); + } + if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) + { + ip = request.getHeader("X-Real-IP"); + } + + if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) + { + ip = request.getRemoteAddr(); + } + + return "0:0:0:0:0:0:0:1".equals(ip) ? "127.0.0.1" : getMultistageReverseProxyIp(ip); + } + + /** + * 检查是否为内部IP地址 + * + * @param ip IP地址 + * @return 结果 + */ + public static boolean internalIp(String ip) + { + byte[] addr = textToNumericFormatV4(ip); + return internalIp(addr) || "127.0.0.1".equals(ip); + } + + /** + * 检查是否为内部IP地址 + * + * @param addr byte地址 + * @return 结果 + */ + private static boolean internalIp(byte[] addr) + { + if (StringUtils.isNull(addr) || addr.length < 2) + { + return true; + } + final byte b0 = addr[0]; + final byte b1 = addr[1]; + // 10.x.x.x/8 + final byte SECTION_1 = 0x0A; + // 172.16.x.x/12 + final byte SECTION_2 = (byte) 0xAC; + final byte SECTION_3 = (byte) 0x10; + final byte SECTION_4 = (byte) 0x1F; + // 192.168.x.x/16 + final byte SECTION_5 = (byte) 0xC0; + final byte SECTION_6 = (byte) 0xA8; + switch (b0) + { + case SECTION_1: + return true; + case SECTION_2: + if (b1 >= SECTION_3 && b1 <= SECTION_4) + { + return true; + } + case SECTION_5: + switch (b1) + { + case SECTION_6: + return true; + } + default: + return false; + } + } + + /** + * 将IPv4地址转换成字节 + * + * @param text IPv4地址 + * @return byte 字节 + */ + public static byte[] textToNumericFormatV4(String text) + { + if (text.length() == 0) + { + return null; + } + + byte[] bytes = new byte[4]; + String[] elements = text.split("\\.", -1); + try + { + long l; + int i; + switch (elements.length) + { + case 1: + l = Long.parseLong(elements[0]); + if ((l < 0L) || (l > 4294967295L)) + { + return null; + } + bytes[0] = (byte) (int) (l >> 24 & 0xFF); + bytes[1] = (byte) (int) ((l & 0xFFFFFF) >> 16 & 0xFF); + bytes[2] = (byte) (int) ((l & 0xFFFF) >> 8 & 0xFF); + bytes[3] = (byte) (int) (l & 0xFF); + break; + case 2: + l = Integer.parseInt(elements[0]); + if ((l < 0L) || (l > 255L)) + { + return null; + } + bytes[0] = (byte) (int) (l & 0xFF); + l = Integer.parseInt(elements[1]); + if ((l < 0L) || (l > 16777215L)) + { + return null; + } + bytes[1] = (byte) (int) (l >> 16 & 0xFF); + bytes[2] = (byte) (int) ((l & 0xFFFF) >> 8 & 0xFF); + bytes[3] = (byte) (int) (l & 0xFF); + break; + case 3: + for (i = 0; i < 2; ++i) + { + l = Integer.parseInt(elements[i]); + if ((l < 0L) || (l > 255L)) + { + return null; + } + bytes[i] = (byte) (int) (l & 0xFF); + } + l = Integer.parseInt(elements[2]); + if ((l < 0L) || (l > 65535L)) + { + return null; + } + bytes[2] = (byte) (int) (l >> 8 & 0xFF); + bytes[3] = (byte) (int) (l & 0xFF); + break; + case 4: + for (i = 0; i < 4; ++i) + { + l = Integer.parseInt(elements[i]); + if ((l < 0L) || (l > 255L)) + { + return null; + } + bytes[i] = (byte) (int) (l & 0xFF); + } + break; + default: + return null; + } + } + catch (NumberFormatException e) + { + return null; + } + return bytes; + } + + /** + * 获取IP地址 + * + * @return 本地IP地址 + */ + public static String getHostIp() + { + try + { + return InetAddress.getLocalHost().getHostAddress(); + } + catch (UnknownHostException e) + { + } + return "127.0.0.1"; + } + + /** + * 获取主机名 + * + * @return 本地主机名 + */ + public static String getHostName() + { + try + { + return InetAddress.getLocalHost().getHostName(); + } + catch (UnknownHostException e) + { + } + return "未知"; + } + + /** + * 从多级反向代理中获得第一个非unknown IP地址 + * + * @param ip 获得的IP地址 + * @return 第一个非unknown IP地址 + */ + public static String getMultistageReverseProxyIp(String ip) + { + // 多级反向代理检测 + if (ip != null && ip.indexOf(",") > 0) + { + final String[] ips = ip.trim().split(","); + for (String subIp : ips) + { + if (false == isUnknown(subIp)) + { + ip = subIp; + break; + } + } + } + return ip; + } + + /** + * 检测给定字符串是否为未知,多用于检测HTTP请求相关 + * + * @param checkString 被检测的字符串 + * @return 是否未知 + */ + public static boolean isUnknown(String checkString) + { + return StringUtils.isBlank(checkString) || "unknown".equalsIgnoreCase(checkString); + } +} \ No newline at end of file diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelHandlerAdapter.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelHandlerAdapter.java new file mode 100644 index 0000000..5ea74c1 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelHandlerAdapter.java @@ -0,0 +1,19 @@ +package com.ruoyi.common.utils.poi; + +/** + * Excel数据格式处理适配器 + * + * @author ruoyi + */ +public interface ExcelHandlerAdapter +{ + /** + * 格式化 + * + * @param value 单元格数据值 + * @param args excel注解args参数组 + * + * @return 处理后的值 + */ + Object format(Object value, String[] args); +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelUtil.java new file mode 100644 index 0000000..26a33ce --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/poi/ExcelUtil.java @@ -0,0 +1,1432 @@ +package com.ruoyi.common.utils.poi; + +import java.io.File; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.lang.reflect.Field; +import java.lang.reflect.Method; +import java.math.BigDecimal; +import java.text.DecimalFormat; +import java.time.LocalDate; +import java.time.LocalDateTime; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Comparator; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.UUID; +import java.util.stream.Collectors; +import javax.servlet.http.HttpServletResponse; +import org.apache.commons.lang3.RegExUtils; +import org.apache.poi.hssf.usermodel.HSSFClientAnchor; +import org.apache.poi.hssf.usermodel.HSSFPicture; +import org.apache.poi.hssf.usermodel.HSSFPictureData; +import org.apache.poi.hssf.usermodel.HSSFShape; +import org.apache.poi.hssf.usermodel.HSSFSheet; +import org.apache.poi.hssf.usermodel.HSSFWorkbook; +import org.apache.poi.ooxml.POIXMLDocumentPart; +import org.apache.poi.ss.usermodel.BorderStyle; +import org.apache.poi.ss.usermodel.Cell; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.CellType; +import org.apache.poi.ss.usermodel.ClientAnchor; +import org.apache.poi.ss.usermodel.DataValidation; +import org.apache.poi.ss.usermodel.DataValidationConstraint; +import org.apache.poi.ss.usermodel.DataValidationHelper; +import org.apache.poi.ss.usermodel.DateUtil; +import org.apache.poi.ss.usermodel.Drawing; +import org.apache.poi.ss.usermodel.FillPatternType; +import org.apache.poi.ss.usermodel.Font; +import org.apache.poi.ss.usermodel.HorizontalAlignment; +import org.apache.poi.ss.usermodel.IndexedColors; +import org.apache.poi.ss.usermodel.PictureData; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.apache.poi.ss.usermodel.VerticalAlignment; +import org.apache.poi.ss.usermodel.Workbook; +import org.apache.poi.ss.usermodel.WorkbookFactory; +import org.apache.poi.ss.util.CellRangeAddress; +import org.apache.poi.ss.util.CellRangeAddressList; +import org.apache.poi.util.IOUtils; +import org.apache.poi.xssf.streaming.SXSSFWorkbook; +import org.apache.poi.xssf.usermodel.XSSFClientAnchor; +import org.apache.poi.xssf.usermodel.XSSFDataValidation; +import org.apache.poi.xssf.usermodel.XSSFDrawing; +import org.apache.poi.xssf.usermodel.XSSFPicture; +import org.apache.poi.xssf.usermodel.XSSFShape; +import org.apache.poi.xssf.usermodel.XSSFSheet; +import org.apache.poi.xssf.usermodel.XSSFWorkbook; +import org.openxmlformats.schemas.drawingml.x2006.spreadsheetDrawing.CTMarker; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.annotation.Excel.Type; +import com.ruoyi.common.annotation.Excels; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.exception.UtilException; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.DictUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.file.FileTypeUtils; +import com.ruoyi.common.utils.file.FileUtils; +import com.ruoyi.common.utils.file.ImageUtils; +import com.ruoyi.common.utils.reflect.ReflectUtils; + +/** + * Excel相关处理 + * + * @author ruoyi + */ +public class ExcelUtil +{ + private static final Logger log = LoggerFactory.getLogger(ExcelUtil.class); + + public static final String FORMULA_REGEX_STR = "=|-|\\+|@"; + + public static final String[] FORMULA_STR = { "=", "-", "+", "@" }; + + /** + * Excel sheet最大行数,默认65536 + */ + public static final int sheetSize = 65536; + + /** + * 工作表名称 + */ + private String sheetName; + + /** + * 导出类型(EXPORT:导出数据;IMPORT:导入模板) + */ + private Type type; + + /** + * 工作薄对象 + */ + private Workbook wb; + + /** + * 工作表对象 + */ + private Sheet sheet; + + /** + * 样式列表 + */ + private Map styles; + + /** + * 导入导出数据列表 + */ + private List list; + + /** + * 注解列表 + */ + private List fields; + + /** + * 当前行号 + */ + private int rownum; + + /** + * 标题 + */ + private String title; + + /** + * 最大高度 + */ + private short maxHeight; + + /** + * 统计列表 + */ + private Map statistics = new HashMap(); + + /** + * 数字格式 + */ + private static final DecimalFormat DOUBLE_FORMAT = new DecimalFormat("######0.00"); + + /** + * 实体对象 + */ + public Class clazz; + + public ExcelUtil(Class clazz) + { + this.clazz = clazz; + } + + public void init(List list, String sheetName, String title, Type type) + { + if (list == null) + { + list = new ArrayList(); + } + this.list = list; + this.sheetName = sheetName; + this.type = type; + this.title = title; + createExcelField(); + createWorkbook(); + createTitle(); + } + + /** + * 创建excel第一行标题 + */ + public void createTitle() + { + if (StringUtils.isNotEmpty(title)) + { + Row titleRow = sheet.createRow(rownum == 0 ? rownum++ : 0); + titleRow.setHeightInPoints(30); + Cell titleCell = titleRow.createCell(0); + titleCell.setCellStyle(styles.get("title")); + titleCell.setCellValue(title); + sheet.addMergedRegion(new CellRangeAddress(titleRow.getRowNum(), titleRow.getRowNum(), titleRow.getRowNum(), + this.fields.size() - 1)); + } + } + + /** + * 对excel表单默认第一个索引名转换成list + * + * @param is 输入流 + * @return 转换后集合 + */ + public List importExcel(InputStream is) throws Exception + { + return importExcel(is, 0); + } + + /** + * 对excel表单默认第一个索引名转换成list + * + * @param is 输入流 + * @param titleNum 标题占用行数 + * @return 转换后集合 + */ + public List importExcel(InputStream is, int titleNum) throws Exception + { + return importExcel(StringUtils.EMPTY, is, titleNum); + } + + /** + * 对excel表单指定表格索引名转换成list + * + * @param sheetName 表格索引名 + * @param titleNum 标题占用行数 + * @param is 输入流 + * @return 转换后集合 + */ + public List importExcel(String sheetName, InputStream is, int titleNum) throws Exception + { + this.type = Type.IMPORT; + this.wb = WorkbookFactory.create(is); + List list = new ArrayList(); + // 如果指定sheet名,则取指定sheet中的内容 否则默认指向第1个sheet + Sheet sheet = StringUtils.isNotEmpty(sheetName) ? wb.getSheet(sheetName) : wb.getSheetAt(0); + if (sheet == null) + { + throw new IOException("文件sheet不存在"); + } + boolean isXSSFWorkbook = !(wb instanceof HSSFWorkbook); + Map pictures; + if (isXSSFWorkbook) + { + pictures = getSheetPictures07((XSSFSheet) sheet, (XSSFWorkbook) wb); + } + else + { + pictures = getSheetPictures03((HSSFSheet) sheet, (HSSFWorkbook) wb); + } + // 获取最后一个非空行的行下标,比如总行数为n,则返回的为n-1 + int rows = sheet.getLastRowNum(); + + if (rows > 0) + { + // 定义一个map用于存放excel列的序号和field. + Map cellMap = new HashMap(); + // 获取表头 + Row heard = sheet.getRow(titleNum); + for (int i = 0; i < heard.getPhysicalNumberOfCells(); i++) + { + Cell cell = heard.getCell(i); + if (StringUtils.isNotNull(cell)) + { + String value = this.getCellValue(heard, i).toString(); + cellMap.put(value, i); + } + else + { + cellMap.put(null, i); + } + } + // 有数据时才处理 得到类的所有field. + List fields = this.getFields(); + Map fieldsMap = new HashMap(); + for (Object[] objects : fields) + { + Excel attr = (Excel) objects[1]; + Integer column = cellMap.get(attr.name()); + if (column != null) + { + fieldsMap.put(column, objects); + } + } + for (int i = titleNum + 1; i <= rows; i++) + { + // 从第2行开始取数据,默认第一行是表头. + Row row = sheet.getRow(i); + // 判断当前行是否是空行 + if (isRowEmpty(row)) + { + continue; + } + T entity = null; + for (Map.Entry entry : fieldsMap.entrySet()) + { + Object val = this.getCellValue(row, entry.getKey()); + + // 如果不存在实例则新建. + entity = (entity == null ? clazz.newInstance() : entity); + // 从map中得到对应列的field. + Field field = (Field) entry.getValue()[0]; + Excel attr = (Excel) entry.getValue()[1]; + // 取得类型,并根据对象类型设置值. + Class fieldType = field.getType(); + if (String.class == fieldType) + { + String s = Convert.toStr(val); + if (StringUtils.endsWith(s, ".0")) + { + val = StringUtils.substringBefore(s, ".0"); + } + else + { + String dateFormat = field.getAnnotation(Excel.class).dateFormat(); + if (StringUtils.isNotEmpty(dateFormat)) + { + val = parseDateToStr(dateFormat, val); + } + else + { + val = Convert.toStr(val); + } + } + } + else if ((Integer.TYPE == fieldType || Integer.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) + { + val = Convert.toInt(val); + } + else if ((Long.TYPE == fieldType || Long.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) + { + val = Convert.toLong(val); + } + else if (Double.TYPE == fieldType || Double.class == fieldType) + { + val = Convert.toDouble(val); + } + else if (Float.TYPE == fieldType || Float.class == fieldType) + { + val = Convert.toFloat(val); + } + else if (BigDecimal.class == fieldType) + { + val = Convert.toBigDecimal(val); + } + else if (Date.class == fieldType) + { + if (val instanceof String) + { + val = DateUtils.parseDate(val); + } + else if (val instanceof Double) + { + val = DateUtil.getJavaDate((Double) val); + } + } + else if (Boolean.TYPE == fieldType || Boolean.class == fieldType) + { + val = Convert.toBool(val, false); + } + if (StringUtils.isNotNull(fieldType)) + { + String propertyName = field.getName(); + if (StringUtils.isNotEmpty(attr.targetAttr())) + { + propertyName = field.getName() + "." + attr.targetAttr(); + } + else if (StringUtils.isNotEmpty(attr.readConverterExp())) + { + val = reverseByExp(Convert.toStr(val), attr.readConverterExp(), attr.separator()); + } + else if (StringUtils.isNotEmpty(attr.dictType())) + { + val = reverseDictByExp(Convert.toStr(val), attr.dictType(), attr.separator()); + } + else if (!attr.handler().equals(ExcelHandlerAdapter.class)) + { + val = dataFormatHandlerAdapter(val, attr); + } + else if (ColumnType.IMAGE == attr.cellType() && StringUtils.isNotEmpty(pictures)) + { + PictureData image = pictures.get(row.getRowNum() + "_" + entry.getKey()); + if (image == null) + { + val = ""; + } + else + { + byte[] data = image.getData(); + val = FileUtils.writeImportBytes(data); + } + } + ReflectUtils.invokeSetter(entity, propertyName, val); + } + } + list.add(entity); + } + } + return list; + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param list 导出数据集合 + * @param sheetName 工作表的名称 + * @return 结果 + */ + public AjaxResult exportExcel(List list, String sheetName) + { + return exportExcel(list, sheetName, StringUtils.EMPTY); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param list 导出数据集合 + * @param sheetName 工作表的名称 + * @param title 标题 + * @return 结果 + */ + public AjaxResult exportExcel(List list, String sheetName, String title) + { + this.init(list, sheetName, title, Type.EXPORT); + return exportExcel(); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param response 返回数据 + * @param list 导出数据集合 + * @param sheetName 工作表的名称 + * @return 结果 + */ + public void exportExcel(HttpServletResponse response, List list, String sheetName) + { + exportExcel(response, list, sheetName, StringUtils.EMPTY); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param response 返回数据 + * @param list 导出数据集合 + * @param sheetName 工作表的名称 + * @param title 标题 + * @return 结果 + */ + public void exportExcel(HttpServletResponse response, List list, String sheetName, String title) + { + response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); + response.setCharacterEncoding("utf-8"); + this.init(list, sheetName, title, Type.EXPORT); + exportExcel(response); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param sheetName 工作表的名称 + * @return 结果 + */ + public AjaxResult importTemplateExcel(String sheetName) + { + return importTemplateExcel(sheetName, StringUtils.EMPTY); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param sheetName 工作表的名称 + * @param title 标题 + * @return 结果 + */ + public AjaxResult importTemplateExcel(String sheetName, String title) + { + this.init(null, sheetName, title, Type.IMPORT); + return exportExcel(); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param sheetName 工作表的名称 + * @return 结果 + */ + public void importTemplateExcel(HttpServletResponse response, String sheetName) + { + importTemplateExcel(response, sheetName, StringUtils.EMPTY); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @param sheetName 工作表的名称 + * @param title 标题 + * @return 结果 + */ + public void importTemplateExcel(HttpServletResponse response, String sheetName, String title) + { + response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); + response.setCharacterEncoding("utf-8"); + this.init(null, sheetName, title, Type.IMPORT); + exportExcel(response); + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @return 结果 + */ + public void exportExcel(HttpServletResponse response) + { + try + { + writeSheet(); + wb.write(response.getOutputStream()); + } + catch (Exception e) + { + log.error("导出Excel异常{}", e.getMessage()); + } + finally + { + IOUtils.closeQuietly(wb); + } + } + + /** + * 对list数据源将其里面的数据导入到excel表单 + * + * @return 结果 + */ + public AjaxResult exportExcel() + { + OutputStream out = null; + try + { + writeSheet(); + String filename = encodingFilename(sheetName); + out = new FileOutputStream(getAbsoluteFile(filename)); + wb.write(out); + return AjaxResult.success(filename); + } + catch (Exception e) + { + log.error("导出Excel异常{}", e.getMessage()); + throw new UtilException("导出Excel失败,请联系网站管理员!"); + } + finally + { + IOUtils.closeQuietly(wb); + IOUtils.closeQuietly(out); + } + } + + /** + * 创建写入数据到Sheet + */ + public void writeSheet() + { + // 取出一共有多少个sheet. + int sheetNo = Math.max(1, (int) Math.ceil(list.size() * 1.0 / sheetSize)); + for (int index = 0; index < sheetNo; index++) + { + createSheet(sheetNo, index); + + // 产生一行 + Row row = sheet.createRow(rownum); + int column = 0; + // 写入各个字段的列头名称 + for (Object[] os : fields) + { + Excel excel = (Excel) os[1]; + this.createCell(excel, row, column++); + } + if (Type.EXPORT.equals(type)) + { + fillExcelData(index, row); + addStatisticsRow(); + } + } + } + + /** + * 填充excel数据 + * + * @param index 序号 + * @param row 单元格行 + */ + public void fillExcelData(int index, Row row) + { + int startNo = index * sheetSize; + int endNo = Math.min(startNo + sheetSize, list.size()); + for (int i = startNo; i < endNo; i++) + { + row = sheet.createRow(i + 1 + rownum - startNo); + // 得到导出对象. + T vo = (T) list.get(i); + int column = 0; + for (Object[] os : fields) + { + Field field = (Field) os[0]; + Excel excel = (Excel) os[1]; + this.addCell(excel, row, vo, field, column++); + } + } + } + + /** + * 创建表格样式 + * + * @param wb 工作薄对象 + * @return 样式列表 + */ + private Map createStyles(Workbook wb) + { + // 写入各条记录,每条记录对应excel表中的一行 + Map styles = new HashMap(); + CellStyle style = wb.createCellStyle(); + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + Font titleFont = wb.createFont(); + titleFont.setFontName("Arial"); + titleFont.setFontHeightInPoints((short) 16); + titleFont.setBold(true); + style.setFont(titleFont); + styles.put("title", style); + + style = wb.createCellStyle(); + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + style.setBorderRight(BorderStyle.THIN); + style.setRightBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderLeft(BorderStyle.THIN); + style.setLeftBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderTop(BorderStyle.THIN); + style.setTopBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderBottom(BorderStyle.THIN); + style.setBottomBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + Font dataFont = wb.createFont(); + dataFont.setFontName("Arial"); + dataFont.setFontHeightInPoints((short) 10); + style.setFont(dataFont); + styles.put("data", style); + + style = wb.createCellStyle(); + style.cloneStyleFrom(styles.get("data")); + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + style.setFillForegroundColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setFillPattern(FillPatternType.SOLID_FOREGROUND); + Font headerFont = wb.createFont(); + headerFont.setFontName("Arial"); + headerFont.setFontHeightInPoints((short) 10); + headerFont.setBold(true); + headerFont.setColor(IndexedColors.WHITE.getIndex()); + style.setFont(headerFont); + styles.put("header", style); + + style = wb.createCellStyle(); + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + Font totalFont = wb.createFont(); + totalFont.setFontName("Arial"); + totalFont.setFontHeightInPoints((short) 10); + style.setFont(totalFont); + styles.put("total", style); + + styles.putAll(annotationStyles(wb)); + + return styles; + } + + /** + * 根据Excel注解创建表格样式 + * + * @param wb 工作薄对象 + * @return 自定义样式列表 + */ + private Map annotationStyles(Workbook wb) + { + Map styles = new HashMap(); + for (Object[] os : fields) + { + Excel excel = (Excel) os[1]; + String key = "data_" + excel.align() + "_" + excel.color(); + if (!styles.containsKey(key)) + { + CellStyle style = wb.createCellStyle(); + style = wb.createCellStyle(); + style.setAlignment(excel.align()); + style.setVerticalAlignment(VerticalAlignment.CENTER); + style.setBorderRight(BorderStyle.THIN); + style.setRightBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderLeft(BorderStyle.THIN); + style.setLeftBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderTop(BorderStyle.THIN); + style.setTopBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setBorderBottom(BorderStyle.THIN); + style.setBottomBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); + Font dataFont = wb.createFont(); + dataFont.setFontName("Arial"); + dataFont.setFontHeightInPoints((short) 10); + dataFont.setColor(excel.color().index); + style.setFont(dataFont); + styles.put(key, style); + } + } + return styles; + } + + /** + * 创建单元格 + */ + public Cell createCell(Excel attr, Row row, int column) + { + // 创建列 + Cell cell = row.createCell(column); + // 写入列信息 + cell.setCellValue(attr.name()); + setDataValidation(attr, row, column); + cell.setCellStyle(styles.get("header")); + return cell; + } + + /** + * 设置单元格信息 + * + * @param value 单元格值 + * @param attr 注解相关 + * @param cell 单元格信息 + */ + public void setCellVo(Object value, Excel attr, Cell cell) + { + if (ColumnType.STRING == attr.cellType()) + { + String cellValue = Convert.toStr(value); + // 对于任何以表达式触发字符 =-+@开头的单元格,直接使用tab字符作为前缀,防止CSV注入。 + if (StringUtils.startsWithAny(cellValue, FORMULA_STR)) + { + cellValue = RegExUtils.replaceFirst(cellValue, FORMULA_REGEX_STR, "\t$0"); + } + cell.setCellValue(StringUtils.isNull(cellValue) ? attr.defaultValue() : cellValue + attr.suffix()); + } + else if (ColumnType.NUMERIC == attr.cellType()) + { + if (StringUtils.isNotNull(value)) + { + cell.setCellValue(StringUtils.contains(Convert.toStr(value), ".") ? Convert.toDouble(value) : Convert.toInt(value)); + } + } + else if (ColumnType.IMAGE == attr.cellType()) + { + ClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0, (short) cell.getColumnIndex(), cell.getRow().getRowNum(), (short) (cell.getColumnIndex() + 1), cell.getRow().getRowNum() + 1); + String imagePath = Convert.toStr(value); + if (StringUtils.isNotEmpty(imagePath)) + { + byte[] data = ImageUtils.getImage(imagePath); + getDrawingPatriarch(cell.getSheet()).createPicture(anchor, + cell.getSheet().getWorkbook().addPicture(data, getImageType(data))); + } + } + } + + /** + * 获取画布 + */ + public static Drawing getDrawingPatriarch(Sheet sheet) + { + if (sheet.getDrawingPatriarch() == null) + { + sheet.createDrawingPatriarch(); + } + return sheet.getDrawingPatriarch(); + } + + /** + * 获取图片类型,设置图片插入类型 + */ + public int getImageType(byte[] value) + { + String type = FileTypeUtils.getFileExtendName(value); + if ("JPG".equalsIgnoreCase(type)) + { + return Workbook.PICTURE_TYPE_JPEG; + } + else if ("PNG".equalsIgnoreCase(type)) + { + return Workbook.PICTURE_TYPE_PNG; + } + return Workbook.PICTURE_TYPE_JPEG; + } + + /** + * 创建表格样式 + */ + public void setDataValidation(Excel attr, Row row, int column) + { + if (attr.name().indexOf("注:") >= 0) + { + sheet.setColumnWidth(column, 6000); + } + else + { + // 设置列宽 + sheet.setColumnWidth(column, (int) ((attr.width() + 0.72) * 256)); + } + if (StringUtils.isNotEmpty(attr.prompt()) || attr.combo().length > 0) + { + // 提示信息或只能选择不能输入的列内容. + setPromptOrValidation(sheet, attr.combo(), attr.prompt(), 1, 100, column, column); + } + } + + /** + * 添加单元格 + */ + public Cell addCell(Excel attr, Row row, T vo, Field field, int column) + { + Cell cell = null; + try + { + // 设置行高 + row.setHeight(maxHeight); + // 根据Excel中设置情况决定是否导出,有些情况需要保持为空,希望用户填写这一列. + if (attr.isExport()) + { + // 创建cell + cell = row.createCell(column); + cell.setCellStyle(styles.get("data_" + attr.align() + "_" + attr.color())); + + // 用于读取对象中的属性 + Object value = getTargetValue(vo, field, attr); + String dateFormat = attr.dateFormat(); + String readConverterExp = attr.readConverterExp(); + String separator = attr.separator(); + String dictType = attr.dictType(); + if (StringUtils.isNotEmpty(dateFormat) && StringUtils.isNotNull(value)) + { + cell.setCellValue(parseDateToStr(dateFormat, value)); + } + else if (StringUtils.isNotEmpty(readConverterExp) && StringUtils.isNotNull(value)) + { + cell.setCellValue(convertByExp(Convert.toStr(value), readConverterExp, separator)); + } + else if (StringUtils.isNotEmpty(dictType) && StringUtils.isNotNull(value)) + { + cell.setCellValue(convertDictByExp(Convert.toStr(value), dictType, separator)); + } + else if (value instanceof BigDecimal && -1 != attr.scale()) + { + cell.setCellValue((((BigDecimal) value).setScale(attr.scale(), attr.roundingMode())).toString()); + } + else if (!attr.handler().equals(ExcelHandlerAdapter.class)) + { + cell.setCellValue(dataFormatHandlerAdapter(value, attr)); + } + else + { + // 设置列类型 + setCellVo(value, attr, cell); + } + addStatisticsData(column, Convert.toStr(value), attr); + } + } + catch (Exception e) + { + log.error("导出Excel失败{}", e); + } + return cell; + } + + /** + * 设置 POI XSSFSheet 单元格提示或选择框 + * + * @param sheet 表单 + * @param textlist 下拉框显示的内容 + * @param promptContent 提示内容 + * @param firstRow 开始行 + * @param endRow 结束行 + * @param firstCol 开始列 + * @param endCol 结束列 + */ + public void setPromptOrValidation(Sheet sheet, String[] textlist, String promptContent, int firstRow, int endRow, + int firstCol, int endCol) + { + DataValidationHelper helper = sheet.getDataValidationHelper(); + DataValidationConstraint constraint = textlist.length > 0 ? helper.createExplicitListConstraint(textlist) : helper.createCustomConstraint("DD1"); + CellRangeAddressList regions = new CellRangeAddressList(firstRow, endRow, firstCol, endCol); + DataValidation dataValidation = helper.createValidation(constraint, regions); + if (StringUtils.isNotEmpty(promptContent)) + { + // 如果设置了提示信息则鼠标放上去提示 + dataValidation.createPromptBox("", promptContent); + dataValidation.setShowPromptBox(true); + } + // 处理Excel兼容性问题 + if (dataValidation instanceof XSSFDataValidation) + { + dataValidation.setSuppressDropDownArrow(true); + dataValidation.setShowErrorBox(true); + } + else + { + dataValidation.setSuppressDropDownArrow(false); + } + sheet.addValidationData(dataValidation); + } + + /** + * 解析导出值 0=男,1=女,2=未知 + * + * @param propertyValue 参数值 + * @param converterExp 翻译注解 + * @param separator 分隔符 + * @return 解析后值 + */ + public static String convertByExp(String propertyValue, String converterExp, String separator) + { + StringBuilder propertyString = new StringBuilder(); + String[] convertSource = converterExp.split(","); + for (String item : convertSource) + { + String[] itemArray = item.split("="); + if (StringUtils.containsAny(separator, propertyValue)) + { + for (String value : propertyValue.split(separator)) + { + if (itemArray[0].equals(value)) + { + propertyString.append(itemArray[1] + separator); + break; + } + } + } + else + { + if (itemArray[0].equals(propertyValue)) + { + return itemArray[1]; + } + } + } + return StringUtils.stripEnd(propertyString.toString(), separator); + } + + /** + * 反向解析值 男=0,女=1,未知=2 + * + * @param propertyValue 参数值 + * @param converterExp 翻译注解 + * @param separator 分隔符 + * @return 解析后值 + */ + public static String reverseByExp(String propertyValue, String converterExp, String separator) + { + StringBuilder propertyString = new StringBuilder(); + String[] convertSource = converterExp.split(","); + for (String item : convertSource) + { + String[] itemArray = item.split("="); + if (StringUtils.containsAny(separator, propertyValue)) + { + for (String value : propertyValue.split(separator)) + { + if (itemArray[1].equals(value)) + { + propertyString.append(itemArray[0] + separator); + break; + } + } + } + else + { + if (itemArray[1].equals(propertyValue)) + { + return itemArray[0]; + } + } + } + return StringUtils.stripEnd(propertyString.toString(), separator); + } + + /** + * 解析字典值 + * + * @param dictValue 字典值 + * @param dictType 字典类型 + * @param separator 分隔符 + * @return 字典标签 + */ + public static String convertDictByExp(String dictValue, String dictType, String separator) + { + return DictUtils.getDictLabel(dictType, dictValue, separator); + } + + /** + * 反向解析值字典值 + * + * @param dictLabel 字典标签 + * @param dictType 字典类型 + * @param separator 分隔符 + * @return 字典值 + */ + public static String reverseDictByExp(String dictLabel, String dictType, String separator) + { + return DictUtils.getDictValue(dictType, dictLabel, separator); + } + + /** + * 数据处理器 + * + * @param value 数据值 + * @param excel 数据注解 + * @return + */ + public String dataFormatHandlerAdapter(Object value, Excel excel) + { + try + { + Object instance = excel.handler().newInstance(); + Method formatMethod = excel.handler().getMethod("format", new Class[] { Object.class, String[].class }); + value = formatMethod.invoke(instance, value, excel.args()); + } + catch (Exception e) + { + log.error("不能格式化数据 " + excel.handler(), e.getMessage()); + } + return Convert.toStr(value); + } + + /** + * 合计统计信息 + */ + private void addStatisticsData(Integer index, String text, Excel entity) + { + if (entity != null && entity.isStatistics()) + { + Double temp = 0D; + if (!statistics.containsKey(index)) + { + statistics.put(index, temp); + } + try + { + temp = Double.valueOf(text); + } + catch (NumberFormatException e) + { + } + statistics.put(index, statistics.get(index) + temp); + } + } + + /** + * 创建统计行 + */ + public void addStatisticsRow() + { + if (statistics.size() > 0) + { + Row row = sheet.createRow(sheet.getLastRowNum() + 1); + Set keys = statistics.keySet(); + Cell cell = row.createCell(0); + cell.setCellStyle(styles.get("total")); + cell.setCellValue("合计"); + + for (Integer key : keys) + { + cell = row.createCell(key); + cell.setCellStyle(styles.get("total")); + cell.setCellValue(DOUBLE_FORMAT.format(statistics.get(key))); + } + statistics.clear(); + } + } + + /** + * 编码文件名 + */ + public String encodingFilename(String filename) + { + filename = UUID.randomUUID().toString() + "_" + filename + ".xlsx"; + return filename; + } + + /** + * 获取下载路径 + * + * @param filename 文件名称 + */ + public String getAbsoluteFile(String filename) + { + String downloadPath = RuoYiConfig.getDownloadPath() + filename; + File desc = new File(downloadPath); + if (!desc.getParentFile().exists()) + { + desc.getParentFile().mkdirs(); + } + return downloadPath; + } + + /** + * 获取bean中的属性值 + * + * @param vo 实体对象 + * @param field 字段 + * @param excel 注解 + * @return 最终的属性值 + * @throws Exception + */ + private Object getTargetValue(T vo, Field field, Excel excel) throws Exception + { + Object o = field.get(vo); + if (StringUtils.isNotEmpty(excel.targetAttr())) + { + String target = excel.targetAttr(); + if (target.contains(".")) + { + String[] targets = target.split("[.]"); + for (String name : targets) + { + o = getValue(o, name); + } + } + else + { + o = getValue(o, target); + } + } + return o; + } + + /** + * 以类的属性的get方法方法形式获取值 + * + * @param o + * @param name + * @return value + * @throws Exception + */ + private Object getValue(Object o, String name) throws Exception + { + if (StringUtils.isNotNull(o) && StringUtils.isNotEmpty(name)) + { + Class clazz = o.getClass(); + Field field = clazz.getDeclaredField(name); + field.setAccessible(true); + o = field.get(o); + } + return o; + } + + /** + * 得到所有定义字段 + */ + private void createExcelField() + { + this.fields = getFields(); + this.fields = this.fields.stream().sorted(Comparator.comparing(objects -> ((Excel) objects[1]).sort())).collect(Collectors.toList()); + this.maxHeight = getRowHeight(); + } + + /** + * 获取字段注解信息 + */ + public List getFields() + { + List fields = new ArrayList(); + List tempFields = new ArrayList<>(); + tempFields.addAll(Arrays.asList(clazz.getSuperclass().getDeclaredFields())); + tempFields.addAll(Arrays.asList(clazz.getDeclaredFields())); + for (Field field : tempFields) + { + // 单注解 + if (field.isAnnotationPresent(Excel.class)) + { + Excel attr = field.getAnnotation(Excel.class); + if (attr != null && (attr.type() == Type.ALL || attr.type() == type)) + { + field.setAccessible(true); + fields.add(new Object[] { field, attr }); + } + } + + // 多注解 + if (field.isAnnotationPresent(Excels.class)) + { + Excels attrs = field.getAnnotation(Excels.class); + Excel[] excels = attrs.value(); + for (Excel attr : excels) + { + if (attr != null && (attr.type() == Type.ALL || attr.type() == type)) + { + field.setAccessible(true); + fields.add(new Object[] { field, attr }); + } + } + } + } + return fields; + } + + /** + * 根据注解获取最大行高 + */ + public short getRowHeight() + { + double maxHeight = 0; + for (Object[] os : this.fields) + { + Excel excel = (Excel) os[1]; + maxHeight = Math.max(maxHeight, excel.height()); + } + return (short) (maxHeight * 20); + } + + /** + * 创建一个工作簿 + */ + public void createWorkbook() + { + this.wb = new SXSSFWorkbook(500); + this.sheet = wb.createSheet(); + wb.setSheetName(0, sheetName); + this.styles = createStyles(wb); + } + + /** + * 创建工作表 + * + * @param sheetNo sheet数量 + * @param index 序号 + */ + public void createSheet(int sheetNo, int index) + { + // 设置工作表的名称. + if (sheetNo > 1 && index > 0) + { + this.sheet = wb.createSheet(); + this.createTitle(); + wb.setSheetName(index, sheetName + index); + } + } + + /** + * 获取单元格值 + * + * @param row 获取的行 + * @param column 获取单元格列号 + * @return 单元格值 + */ + public Object getCellValue(Row row, int column) + { + if (row == null) + { + return row; + } + Object val = ""; + try + { + Cell cell = row.getCell(column); + if (StringUtils.isNotNull(cell)) + { + if (cell.getCellType() == CellType.NUMERIC || cell.getCellType() == CellType.FORMULA) + { + val = cell.getNumericCellValue(); + if (DateUtil.isCellDateFormatted(cell)) + { + val = DateUtil.getJavaDate((Double) val); // POI Excel 日期格式转换 + } + else + { + if ((Double) val % 1 != 0) + { + val = new BigDecimal(val.toString()); + } + else + { + val = new DecimalFormat("0").format(val); + } + } + } + else if (cell.getCellType() == CellType.STRING) + { + val = cell.getStringCellValue(); + } + else if (cell.getCellType() == CellType.BOOLEAN) + { + val = cell.getBooleanCellValue(); + } + else if (cell.getCellType() == CellType.ERROR) + { + val = cell.getErrorCellValue(); + } + + } + } + catch (Exception e) + { + return val; + } + return val; + } + + /** + * 判断是否是空行 + * + * @param row 判断的行 + * @return + */ + private boolean isRowEmpty(Row row) + { + if (row == null) + { + return true; + } + for (int i = row.getFirstCellNum(); i < row.getLastCellNum(); i++) + { + Cell cell = row.getCell(i); + if (cell != null && cell.getCellType() != CellType.BLANK) + { + return false; + } + } + return true; + } + + /** + * 获取Excel2003图片 + * + * @param sheet 当前sheet对象 + * @param workbook 工作簿对象 + * @return Map key:图片单元格索引(1_1)String,value:图片流PictureData + */ + public static Map getSheetPictures03(HSSFSheet sheet, HSSFWorkbook workbook) + { + Map sheetIndexPicMap = new HashMap(); + List pictures = workbook.getAllPictures(); + if (!pictures.isEmpty()) + { + for (HSSFShape shape : sheet.getDrawingPatriarch().getChildren()) + { + HSSFClientAnchor anchor = (HSSFClientAnchor) shape.getAnchor(); + if (shape instanceof HSSFPicture) + { + HSSFPicture pic = (HSSFPicture) shape; + int pictureIndex = pic.getPictureIndex() - 1; + HSSFPictureData picData = pictures.get(pictureIndex); + String picIndex = String.valueOf(anchor.getRow1()) + "_" + String.valueOf(anchor.getCol1()); + sheetIndexPicMap.put(picIndex, picData); + } + } + return sheetIndexPicMap; + } + else + { + return sheetIndexPicMap; + } + } + + /** + * 获取Excel2007图片 + * + * @param sheet 当前sheet对象 + * @param workbook 工作簿对象 + * @return Map key:图片单元格索引(1_1)String,value:图片流PictureData + */ + public static Map getSheetPictures07(XSSFSheet sheet, XSSFWorkbook workbook) + { + Map sheetIndexPicMap = new HashMap(); + for (POIXMLDocumentPart dr : sheet.getRelations()) + { + if (dr instanceof XSSFDrawing) + { + XSSFDrawing drawing = (XSSFDrawing) dr; + List shapes = drawing.getShapes(); + for (XSSFShape shape : shapes) + { + if (shape instanceof XSSFPicture) + { + XSSFPicture pic = (XSSFPicture) shape; + XSSFClientAnchor anchor = pic.getPreferredSize(); + CTMarker ctMarker = anchor.getFrom(); + String picIndex = ctMarker.getRow() + "_" + ctMarker.getCol(); + sheetIndexPicMap.put(picIndex, pic.getPictureData()); + } + } + } + } + return sheetIndexPicMap; + } + + /** + * 格式化不同类型的日期对象 + * + * @param dateFormat 日期格式 + * @param val 被格式化的日期对象 + * @return 格式化后的日期字符 + */ + public String parseDateToStr(String dateFormat, Object val) + { + if (val == null) + { + return ""; + } + String str; + if (val instanceof Date) + { + str = DateUtils.parseDateToStr(dateFormat, (Date) val); + } + else if (val instanceof LocalDateTime) + { + str = DateUtils.parseDateToStr(dateFormat, DateUtils.toDate((LocalDateTime) val)); + } + else if (val instanceof LocalDate) + { + str = DateUtils.parseDateToStr(dateFormat, DateUtils.toDate((LocalDate) val)); + } + else + { + str = val.toString(); + } + return str; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/reflect/ReflectUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/reflect/ReflectUtils.java new file mode 100644 index 0000000..b19953e --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/reflect/ReflectUtils.java @@ -0,0 +1,410 @@ +package com.ruoyi.common.utils.reflect; + +import java.lang.reflect.Field; +import java.lang.reflect.InvocationTargetException; +import java.lang.reflect.Method; +import java.lang.reflect.Modifier; +import java.lang.reflect.ParameterizedType; +import java.lang.reflect.Type; +import java.util.Date; +import org.apache.commons.lang3.StringUtils; +import org.apache.commons.lang3.Validate; +import org.apache.poi.ss.usermodel.DateUtil; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.utils.DateUtils; + +/** + * 反射工具类. 提供调用getter/setter方法, 访问私有变量, 调用私有方法, 获取泛型类型Class, 被AOP过的真实类等工具函数. + * + * @author ruoyi + */ +@SuppressWarnings("rawtypes") +public class ReflectUtils +{ + private static final String SETTER_PREFIX = "set"; + + private static final String GETTER_PREFIX = "get"; + + private static final String CGLIB_CLASS_SEPARATOR = "$$"; + + private static Logger logger = LoggerFactory.getLogger(ReflectUtils.class); + + /** + * 调用Getter方法. + * 支持多级,如:对象名.对象名.方法 + */ + @SuppressWarnings("unchecked") + public static E invokeGetter(Object obj, String propertyName) + { + Object object = obj; + for (String name : StringUtils.split(propertyName, ".")) + { + String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(name); + object = invokeMethod(object, getterMethodName, new Class[] {}, new Object[] {}); + } + return (E) object; + } + + /** + * 调用Setter方法, 仅匹配方法名。 + * 支持多级,如:对象名.对象名.方法 + */ + public static void invokeSetter(Object obj, String propertyName, E value) + { + Object object = obj; + String[] names = StringUtils.split(propertyName, "."); + for (int i = 0; i < names.length; i++) + { + if (i < names.length - 1) + { + String getterMethodName = GETTER_PREFIX + StringUtils.capitalize(names[i]); + object = invokeMethod(object, getterMethodName, new Class[] {}, new Object[] {}); + } + else + { + String setterMethodName = SETTER_PREFIX + StringUtils.capitalize(names[i]); + invokeMethodByName(object, setterMethodName, new Object[] { value }); + } + } + } + + /** + * 直接读取对象属性值, 无视private/protected修饰符, 不经过getter函数. + */ + @SuppressWarnings("unchecked") + public static E getFieldValue(final Object obj, final String fieldName) + { + Field field = getAccessibleField(obj, fieldName); + if (field == null) + { + logger.debug("在 [" + obj.getClass() + "] 中,没有找到 [" + fieldName + "] 字段 "); + return null; + } + E result = null; + try + { + result = (E) field.get(obj); + } + catch (IllegalAccessException e) + { + logger.error("不可能抛出的异常{}", e.getMessage()); + } + return result; + } + + /** + * 直接设置对象属性值, 无视private/protected修饰符, 不经过setter函数. + */ + public static void setFieldValue(final Object obj, final String fieldName, final E value) + { + Field field = getAccessibleField(obj, fieldName); + if (field == null) + { + // throw new IllegalArgumentException("在 [" + obj.getClass() + "] 中,没有找到 [" + fieldName + "] 字段 "); + logger.debug("在 [" + obj.getClass() + "] 中,没有找到 [" + fieldName + "] 字段 "); + return; + } + try + { + field.set(obj, value); + } + catch (IllegalAccessException e) + { + logger.error("不可能抛出的异常: {}", e.getMessage()); + } + } + + /** + * 直接调用对象方法, 无视private/protected修饰符. + * 用于一次性调用的情况,否则应使用getAccessibleMethod()函数获得Method后反复调用. + * 同时匹配方法名+参数类型, + */ + @SuppressWarnings("unchecked") + public static E invokeMethod(final Object obj, final String methodName, final Class[] parameterTypes, + final Object[] args) + { + if (obj == null || methodName == null) + { + return null; + } + Method method = getAccessibleMethod(obj, methodName, parameterTypes); + if (method == null) + { + logger.debug("在 [" + obj.getClass() + "] 中,没有找到 [" + methodName + "] 方法 "); + return null; + } + try + { + return (E) method.invoke(obj, args); + } + catch (Exception e) + { + String msg = "method: " + method + ", obj: " + obj + ", args: " + args + ""; + throw convertReflectionExceptionToUnchecked(msg, e); + } + } + + /** + * 直接调用对象方法, 无视private/protected修饰符, + * 用于一次性调用的情况,否则应使用getAccessibleMethodByName()函数获得Method后反复调用. + * 只匹配函数名,如果有多个同名函数调用第一个。 + */ + @SuppressWarnings("unchecked") + public static E invokeMethodByName(final Object obj, final String methodName, final Object[] args) + { + Method method = getAccessibleMethodByName(obj, methodName, args.length); + if (method == null) + { + // 如果为空不报错,直接返回空。 + logger.debug("在 [" + obj.getClass() + "] 中,没有找到 [" + methodName + "] 方法 "); + return null; + } + try + { + // 类型转换(将参数数据类型转换为目标方法参数类型) + Class[] cs = method.getParameterTypes(); + for (int i = 0; i < cs.length; i++) + { + if (args[i] != null && !args[i].getClass().equals(cs[i])) + { + if (cs[i] == String.class) + { + args[i] = Convert.toStr(args[i]); + if (StringUtils.endsWith((String) args[i], ".0")) + { + args[i] = StringUtils.substringBefore((String) args[i], ".0"); + } + } + else if (cs[i] == Integer.class) + { + args[i] = Convert.toInt(args[i]); + } + else if (cs[i] == Long.class) + { + args[i] = Convert.toLong(args[i]); + } + else if (cs[i] == Double.class) + { + args[i] = Convert.toDouble(args[i]); + } + else if (cs[i] == Float.class) + { + args[i] = Convert.toFloat(args[i]); + } + else if (cs[i] == Date.class) + { + if (args[i] instanceof String) + { + args[i] = DateUtils.parseDate(args[i]); + } + else + { + args[i] = DateUtil.getJavaDate((Double) args[i]); + } + } + else if (cs[i] == boolean.class || cs[i] == Boolean.class) + { + args[i] = Convert.toBool(args[i]); + } + } + } + return (E) method.invoke(obj, args); + } + catch (Exception e) + { + String msg = "method: " + method + ", obj: " + obj + ", args: " + args + ""; + throw convertReflectionExceptionToUnchecked(msg, e); + } + } + + /** + * 循环向上转型, 获取对象的DeclaredField, 并强制设置为可访问. + * 如向上转型到Object仍无法找到, 返回null. + */ + public static Field getAccessibleField(final Object obj, final String fieldName) + { + // 为空不报错。直接返回 null + if (obj == null) + { + return null; + } + Validate.notBlank(fieldName, "fieldName can't be blank"); + for (Class superClass = obj.getClass(); superClass != Object.class; superClass = superClass.getSuperclass()) + { + try + { + Field field = superClass.getDeclaredField(fieldName); + makeAccessible(field); + return field; + } + catch (NoSuchFieldException e) + { + continue; + } + } + return null; + } + + /** + * 循环向上转型, 获取对象的DeclaredMethod,并强制设置为可访问. + * 如向上转型到Object仍无法找到, 返回null. + * 匹配函数名+参数类型。 + * 用于方法需要被多次调用的情况. 先使用本函数先取得Method,然后调用Method.invoke(Object obj, Object... args) + */ + public static Method getAccessibleMethod(final Object obj, final String methodName, + final Class... parameterTypes) + { + // 为空不报错。直接返回 null + if (obj == null) + { + return null; + } + Validate.notBlank(methodName, "methodName can't be blank"); + for (Class searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) + { + try + { + Method method = searchType.getDeclaredMethod(methodName, parameterTypes); + makeAccessible(method); + return method; + } + catch (NoSuchMethodException e) + { + continue; + } + } + return null; + } + + /** + * 循环向上转型, 获取对象的DeclaredMethod,并强制设置为可访问. + * 如向上转型到Object仍无法找到, 返回null. + * 只匹配函数名。 + * 用于方法需要被多次调用的情况. 先使用本函数先取得Method,然后调用Method.invoke(Object obj, Object... args) + */ + public static Method getAccessibleMethodByName(final Object obj, final String methodName, int argsNum) + { + // 为空不报错。直接返回 null + if (obj == null) + { + return null; + } + Validate.notBlank(methodName, "methodName can't be blank"); + for (Class searchType = obj.getClass(); searchType != Object.class; searchType = searchType.getSuperclass()) + { + Method[] methods = searchType.getDeclaredMethods(); + for (Method method : methods) + { + if (method.getName().equals(methodName) && method.getParameterTypes().length == argsNum) + { + makeAccessible(method); + return method; + } + } + } + return null; + } + + /** + * 改变private/protected的方法为public,尽量不调用实际改动的语句,避免JDK的SecurityManager抱怨。 + */ + public static void makeAccessible(Method method) + { + if ((!Modifier.isPublic(method.getModifiers()) || !Modifier.isPublic(method.getDeclaringClass().getModifiers())) + && !method.isAccessible()) + { + method.setAccessible(true); + } + } + + /** + * 改变private/protected的成员变量为public,尽量不调用实际改动的语句,避免JDK的SecurityManager抱怨。 + */ + public static void makeAccessible(Field field) + { + if ((!Modifier.isPublic(field.getModifiers()) || !Modifier.isPublic(field.getDeclaringClass().getModifiers()) + || Modifier.isFinal(field.getModifiers())) && !field.isAccessible()) + { + field.setAccessible(true); + } + } + + /** + * 通过反射, 获得Class定义中声明的泛型参数的类型, 注意泛型必须定义在父类处 + * 如无法找到, 返回Object.class. + */ + @SuppressWarnings("unchecked") + public static Class getClassGenricType(final Class clazz) + { + return getClassGenricType(clazz, 0); + } + + /** + * 通过反射, 获得Class定义中声明的父类的泛型参数的类型. + * 如无法找到, 返回Object.class. + */ + public static Class getClassGenricType(final Class clazz, final int index) + { + Type genType = clazz.getGenericSuperclass(); + + if (!(genType instanceof ParameterizedType)) + { + logger.debug(clazz.getSimpleName() + "'s superclass not ParameterizedType"); + return Object.class; + } + + Type[] params = ((ParameterizedType) genType).getActualTypeArguments(); + + if (index >= params.length || index < 0) + { + logger.debug("Index: " + index + ", Size of " + clazz.getSimpleName() + "'s Parameterized Type: " + + params.length); + return Object.class; + } + if (!(params[index] instanceof Class)) + { + logger.debug(clazz.getSimpleName() + " not set the actual class on superclass generic parameter"); + return Object.class; + } + + return (Class) params[index]; + } + + public static Class getUserClass(Object instance) + { + if (instance == null) + { + throw new RuntimeException("Instance must not be null"); + } + Class clazz = instance.getClass(); + if (clazz != null && clazz.getName().contains(CGLIB_CLASS_SEPARATOR)) + { + Class superClass = clazz.getSuperclass(); + if (superClass != null && !Object.class.equals(superClass)) + { + return superClass; + } + } + return clazz; + + } + + /** + * 将反射时的checked exception转换为unchecked exception. + */ + public static RuntimeException convertReflectionExceptionToUnchecked(String msg, Exception e) + { + if (e instanceof IllegalAccessException || e instanceof IllegalArgumentException + || e instanceof NoSuchMethodException) + { + return new IllegalArgumentException(msg, e); + } + else if (e instanceof InvocationTargetException) + { + return new RuntimeException(msg, ((InvocationTargetException) e).getTargetException()); + } + return new RuntimeException(msg, e); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Base64.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Base64.java new file mode 100644 index 0000000..ca1cd92 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Base64.java @@ -0,0 +1,291 @@ +package com.ruoyi.common.utils.sign; + +/** + * Base64工具类 + * + * @author ruoyi + */ +public final class Base64 +{ + static private final int BASELENGTH = 128; + static private final int LOOKUPLENGTH = 64; + static private final int TWENTYFOURBITGROUP = 24; + static private final int EIGHTBIT = 8; + static private final int SIXTEENBIT = 16; + static private final int FOURBYTE = 4; + static private final int SIGN = -128; + static private final char PAD = '='; + static final private byte[] base64Alphabet = new byte[BASELENGTH]; + static final private char[] lookUpBase64Alphabet = new char[LOOKUPLENGTH]; + + static + { + for (int i = 0; i < BASELENGTH; ++i) + { + base64Alphabet[i] = -1; + } + for (int i = 'Z'; i >= 'A'; i--) + { + base64Alphabet[i] = (byte) (i - 'A'); + } + for (int i = 'z'; i >= 'a'; i--) + { + base64Alphabet[i] = (byte) (i - 'a' + 26); + } + + for (int i = '9'; i >= '0'; i--) + { + base64Alphabet[i] = (byte) (i - '0' + 52); + } + + base64Alphabet['+'] = 62; + base64Alphabet['/'] = 63; + + for (int i = 0; i <= 25; i++) + { + lookUpBase64Alphabet[i] = (char) ('A' + i); + } + + for (int i = 26, j = 0; i <= 51; i++, j++) + { + lookUpBase64Alphabet[i] = (char) ('a' + j); + } + + for (int i = 52, j = 0; i <= 61; i++, j++) + { + lookUpBase64Alphabet[i] = (char) ('0' + j); + } + lookUpBase64Alphabet[62] = (char) '+'; + lookUpBase64Alphabet[63] = (char) '/'; + } + + private static boolean isWhiteSpace(char octect) + { + return (octect == 0x20 || octect == 0xd || octect == 0xa || octect == 0x9); + } + + private static boolean isPad(char octect) + { + return (octect == PAD); + } + + private static boolean isData(char octect) + { + return (octect < BASELENGTH && base64Alphabet[octect] != -1); + } + + /** + * Encodes hex octects into Base64 + * + * @param binaryData Array containing binaryData + * @return Encoded Base64 array + */ + public static String encode(byte[] binaryData) + { + if (binaryData == null) + { + return null; + } + + int lengthDataBits = binaryData.length * EIGHTBIT; + if (lengthDataBits == 0) + { + return ""; + } + + int fewerThan24bits = lengthDataBits % TWENTYFOURBITGROUP; + int numberTriplets = lengthDataBits / TWENTYFOURBITGROUP; + int numberQuartet = fewerThan24bits != 0 ? numberTriplets + 1 : numberTriplets; + char encodedData[] = null; + + encodedData = new char[numberQuartet * 4]; + + byte k = 0, l = 0, b1 = 0, b2 = 0, b3 = 0; + + int encodedIndex = 0; + int dataIndex = 0; + + for (int i = 0; i < numberTriplets; i++) + { + b1 = binaryData[dataIndex++]; + b2 = binaryData[dataIndex++]; + b3 = binaryData[dataIndex++]; + + l = (byte) (b2 & 0x0f); + k = (byte) (b1 & 0x03); + + byte val1 = ((b1 & SIGN) == 0) ? (byte) (b1 >> 2) : (byte) ((b1) >> 2 ^ 0xc0); + byte val2 = ((b2 & SIGN) == 0) ? (byte) (b2 >> 4) : (byte) ((b2) >> 4 ^ 0xf0); + byte val3 = ((b3 & SIGN) == 0) ? (byte) (b3 >> 6) : (byte) ((b3) >> 6 ^ 0xfc); + + encodedData[encodedIndex++] = lookUpBase64Alphabet[val1]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[val2 | (k << 4)]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[(l << 2) | val3]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[b3 & 0x3f]; + } + + // form integral number of 6-bit groups + if (fewerThan24bits == EIGHTBIT) + { + b1 = binaryData[dataIndex]; + k = (byte) (b1 & 0x03); + byte val1 = ((b1 & SIGN) == 0) ? (byte) (b1 >> 2) : (byte) ((b1) >> 2 ^ 0xc0); + encodedData[encodedIndex++] = lookUpBase64Alphabet[val1]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[k << 4]; + encodedData[encodedIndex++] = PAD; + encodedData[encodedIndex++] = PAD; + } + else if (fewerThan24bits == SIXTEENBIT) + { + b1 = binaryData[dataIndex]; + b2 = binaryData[dataIndex + 1]; + l = (byte) (b2 & 0x0f); + k = (byte) (b1 & 0x03); + + byte val1 = ((b1 & SIGN) == 0) ? (byte) (b1 >> 2) : (byte) ((b1) >> 2 ^ 0xc0); + byte val2 = ((b2 & SIGN) == 0) ? (byte) (b2 >> 4) : (byte) ((b2) >> 4 ^ 0xf0); + + encodedData[encodedIndex++] = lookUpBase64Alphabet[val1]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[val2 | (k << 4)]; + encodedData[encodedIndex++] = lookUpBase64Alphabet[l << 2]; + encodedData[encodedIndex++] = PAD; + } + return new String(encodedData); + } + + /** + * Decodes Base64 data into octects + * + * @param encoded string containing Base64 data + * @return Array containind decoded data. + */ + public static byte[] decode(String encoded) + { + if (encoded == null) + { + return null; + } + + char[] base64Data = encoded.toCharArray(); + // remove white spaces + int len = removeWhiteSpace(base64Data); + + if (len % FOURBYTE != 0) + { + return null;// should be divisible by four + } + + int numberQuadruple = (len / FOURBYTE); + + if (numberQuadruple == 0) + { + return new byte[0]; + } + + byte decodedData[] = null; + byte b1 = 0, b2 = 0, b3 = 0, b4 = 0; + char d1 = 0, d2 = 0, d3 = 0, d4 = 0; + + int i = 0; + int encodedIndex = 0; + int dataIndex = 0; + decodedData = new byte[(numberQuadruple) * 3]; + + for (; i < numberQuadruple - 1; i++) + { + + if (!isData((d1 = base64Data[dataIndex++])) || !isData((d2 = base64Data[dataIndex++])) + || !isData((d3 = base64Data[dataIndex++])) || !isData((d4 = base64Data[dataIndex++]))) + { + return null; + } // if found "no data" just return null + + b1 = base64Alphabet[d1]; + b2 = base64Alphabet[d2]; + b3 = base64Alphabet[d3]; + b4 = base64Alphabet[d4]; + + decodedData[encodedIndex++] = (byte) (b1 << 2 | b2 >> 4); + decodedData[encodedIndex++] = (byte) (((b2 & 0xf) << 4) | ((b3 >> 2) & 0xf)); + decodedData[encodedIndex++] = (byte) (b3 << 6 | b4); + } + + if (!isData((d1 = base64Data[dataIndex++])) || !isData((d2 = base64Data[dataIndex++]))) + { + return null;// if found "no data" just return null + } + + b1 = base64Alphabet[d1]; + b2 = base64Alphabet[d2]; + + d3 = base64Data[dataIndex++]; + d4 = base64Data[dataIndex++]; + if (!isData((d3)) || !isData((d4))) + {// Check if they are PAD characters + if (isPad(d3) && isPad(d4)) + { + if ((b2 & 0xf) != 0)// last 4 bits should be zero + { + return null; + } + byte[] tmp = new byte[i * 3 + 1]; + System.arraycopy(decodedData, 0, tmp, 0, i * 3); + tmp[encodedIndex] = (byte) (b1 << 2 | b2 >> 4); + return tmp; + } + else if (!isPad(d3) && isPad(d4)) + { + b3 = base64Alphabet[d3]; + if ((b3 & 0x3) != 0)// last 2 bits should be zero + { + return null; + } + byte[] tmp = new byte[i * 3 + 2]; + System.arraycopy(decodedData, 0, tmp, 0, i * 3); + tmp[encodedIndex++] = (byte) (b1 << 2 | b2 >> 4); + tmp[encodedIndex] = (byte) (((b2 & 0xf) << 4) | ((b3 >> 2) & 0xf)); + return tmp; + } + else + { + return null; + } + } + else + { // No PAD e.g 3cQl + b3 = base64Alphabet[d3]; + b4 = base64Alphabet[d4]; + decodedData[encodedIndex++] = (byte) (b1 << 2 | b2 >> 4); + decodedData[encodedIndex++] = (byte) (((b2 & 0xf) << 4) | ((b3 >> 2) & 0xf)); + decodedData[encodedIndex++] = (byte) (b3 << 6 | b4); + + } + return decodedData; + } + + /** + * remove WhiteSpace from MIME containing encoded Base64 data. + * + * @param data the byte array of base64 data (with WS) + * @return the new length + */ + private static int removeWhiteSpace(char[] data) + { + if (data == null) + { + return 0; + } + + // count characters that's not whitespace + int newSize = 0; + int len = data.length; + for (int i = 0; i < len; i++) + { + if (!isWhiteSpace(data[i])) + { + data[newSize++] = data[i]; + } + } + return newSize; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Md5Utils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Md5Utils.java new file mode 100644 index 0000000..c1c58db --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/Md5Utils.java @@ -0,0 +1,67 @@ +package com.ruoyi.common.utils.sign; + +import java.nio.charset.StandardCharsets; +import java.security.MessageDigest; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** + * Md5加密方法 + * + * @author ruoyi + */ +public class Md5Utils +{ + private static final Logger log = LoggerFactory.getLogger(Md5Utils.class); + + private static byte[] md5(String s) + { + MessageDigest algorithm; + try + { + algorithm = MessageDigest.getInstance("MD5"); + algorithm.reset(); + algorithm.update(s.getBytes("UTF-8")); + byte[] messageDigest = algorithm.digest(); + return messageDigest; + } + catch (Exception e) + { + log.error("MD5 Error...", e); + } + return null; + } + + private static final String toHex(byte hash[]) + { + if (hash == null) + { + return null; + } + StringBuffer buf = new StringBuffer(hash.length * 2); + int i; + + for (i = 0; i < hash.length; i++) + { + if ((hash[i] & 0xff) < 0x10) + { + buf.append("0"); + } + buf.append(Long.toString(hash[i] & 0xff, 16)); + } + return buf.toString(); + } + + public static String hash(String s) + { + try + { + return new String(toHex(md5(s)).getBytes(StandardCharsets.UTF_8), StandardCharsets.UTF_8); + } + catch (Exception e) + { + log.error("not supported charset...{}", e); + return s; + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/RsaUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/RsaUtils.java new file mode 100644 index 0000000..f6e3332 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sign/RsaUtils.java @@ -0,0 +1,163 @@ +package com.ruoyi.common.utils.sign; + +import com.alibaba.fastjson.JSON; +import org.apache.commons.codec.binary.Base64; +import javax.crypto.Cipher; +import java.security.*; +import java.security.interfaces.RSAPrivateKey; +import java.security.interfaces.RSAPublicKey; +import java.security.spec.PKCS8EncodedKeySpec; +import java.security.spec.X509EncodedKeySpec; + +/** + * RSA加密解密 + * + * @author ruoyi + **/ +public class RsaUtils +{ + // Rsa 私钥 + public static String privateKey = "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wgg" + + "JbAgEAAoGBAI+Y8TJB3brb1fa43o7mI70KAlR8auS72k5HrrCafeacYCrvM2tLyAJTf4" + + "VjIya1cYzbUNE7bydT7Tl5vsuh5ynGqVUXeU93EZ28Dib4XcXxcSGBgF7b4okTMwS7tRC" + + "3TbX67fyd4qiTyNj54QitMAYOhq/8y/gv7X3oQi2GY9eTAgMBAAECgYATQ5lX/K4aW1Og" + + "/bwdv5Ib0jPt0aMlfG4VAljVKGSrg18DY4OqaUgzx6/b3SjiEIVjOCjnaOnIhOHWTBbkipv" + + "QMyOPGUzeqEKqjbFa8MN7lYDVBCwH0CCsXFuiyAHA5pI8OuNH+Iy9b3VXRhlaJTQkE1c" + + "qJ4pmqOL11LCj20XRgQJBANdruVNus2YFDir48JpmiS71+W0WxfJkxm8EEohLd7T8aLAF" + + "Rc51OXm3G5bDqNIFP5ygqYKW8mVncNUP3t/FeWkCQQCqparD7QcRtZe1ily9hEWVKY32G" + + "ZVMAqNH8IsiveUrsH4Phhj/vdtL1mCTx+DyRygPvKKVj7A07fqu2CTXQQ2bAkAiMC2EaUA" + + "9g7Xg6gs9ZeKFunHwdznJI2c+vTW3vcpsxdZTgTK39nC9uWq+faTwV/blg7YDTLOBlm4" + + "Y0mD7wJppAkAu+HC7gyeZ+Ujr4EzMoXbhqtF3du1AQc2dxJkXk/tbfjYEod2Jr0GIa1" + + "aj9rV1Gp7tTffWgPl4aDTDIZ1qXm6XAkBgWBcbvTrE2bpUYsTXxtZ7eWxTCMdmJxZ3n" + + "4xs0p0RqpU6n8BNJyY3wtJEesRoKLshKFt/NDMsxqVoqICpXxXs"; + + /** + * 私钥解密 + * + * @param text 待解密的文本 + * @return 解密后的文本 + */ + public static String decryptByPrivateKey(String text) throws Exception + { + return decryptByPrivateKey(privateKey, text); + } + + /** + * 公钥解密 + * + * @param publicKeyString 公钥 + * @param text 待解密的信息 + * @return 解密后的文本 + */ + public static String decryptByPublicKey(String publicKeyString, String text) throws Exception + { + X509EncodedKeySpec x509EncodedKeySpec = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString)); + KeyFactory keyFactory = KeyFactory.getInstance("RSA"); + PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec); + Cipher cipher = Cipher.getInstance("RSA"); + cipher.init(Cipher.DECRYPT_MODE, publicKey); + byte[] result = cipher.doFinal(Base64.decodeBase64(text)); + return new String(result); + } + + /** + * 私钥加密 + * + * @param privateKeyString 私钥 + * @param text 待加密的信息 + * @return 加密后的文本 + */ + public static String encryptByPrivateKey(String privateKeyString, String text) throws Exception + { + PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString)); + KeyFactory keyFactory = KeyFactory.getInstance("RSA"); + PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec); + Cipher cipher = Cipher.getInstance("RSA"); + cipher.init(Cipher.ENCRYPT_MODE, privateKey); + byte[] result = cipher.doFinal(text.getBytes()); + return Base64.encodeBase64String(result); + } + + /** + * 私钥解密 + * + * @param privateKeyString 私钥 + * @param text 待解密的文本 + * @return 解密后的文本 + */ + public static String decryptByPrivateKey(String privateKeyString, String text) throws Exception + { + PKCS8EncodedKeySpec pkcs8EncodedKeySpec5 = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString)); + KeyFactory keyFactory = KeyFactory.getInstance("RSA"); + PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec5); + Cipher cipher = Cipher.getInstance("RSA"); + cipher.init(Cipher.DECRYPT_MODE, privateKey); + byte[] result = cipher.doFinal(Base64.decodeBase64(text)); + return new String(result); + } + + /** + * 公钥加密 + * + * @param publicKeyString 公钥 + * @param text 待加密的文本 + * @return 加密后的文本 + */ + public static String encryptByPublicKey(String publicKeyString, String text) throws Exception + { + X509EncodedKeySpec x509EncodedKeySpec2 = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString)); + KeyFactory keyFactory = KeyFactory.getInstance("RSA"); + PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec2); + Cipher cipher = Cipher.getInstance("RSA"); + cipher.init(Cipher.ENCRYPT_MODE, publicKey); + byte[] result = cipher.doFinal(text.getBytes()); + return Base64.encodeBase64String(result); + } + + /** + * 构建RSA密钥对 + * + * @return 生成后的公私钥信息 + */ + public static RsaKeyPair generateKeyPair() throws NoSuchAlgorithmException + { + KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA"); + keyPairGenerator.initialize(1024); + KeyPair keyPair = keyPairGenerator.generateKeyPair(); + RSAPublicKey rsaPublicKey = (RSAPublicKey) keyPair.getPublic(); + RSAPrivateKey rsaPrivateKey = (RSAPrivateKey) keyPair.getPrivate(); + String publicKeyString = Base64.encodeBase64String(rsaPublicKey.getEncoded()); + String privateKeyString = Base64.encodeBase64String(rsaPrivateKey.getEncoded()); + return new RsaKeyPair(publicKeyString, privateKeyString); + } + + public static void main(String[] args) throws NoSuchAlgorithmException { + System.out.println(JSON.toJSONString(generateKeyPair())); + } + + + /** + * RSA密钥对对象 + */ + public static class RsaKeyPair + { + private final String publicKey; + private final String privateKey; + + public RsaKeyPair(String publicKey, String privateKey) + { + this.publicKey = publicKey; + this.privateKey = privateKey; + } + + public String getPublicKey() + { + return publicKey; + } + + public String getPrivateKey() + { + return privateKey; + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3.java new file mode 100644 index 0000000..23f9dcf --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3.java @@ -0,0 +1,359 @@ +package com.ruoyi.common.utils.sm3; + + +public class SM3 +{ + /*public static final byte[] iv = { 0x2C, (byte) 0x91, (byte) 0xB4, 0x01, + (byte) 0xFC, 0x64, (byte) 0xB2, (byte) 0xCE, 0x7C, 0x4E, + (byte) 0xAE, (byte) 0xFB, (byte) 0xB1, 0x3B, (byte) 0xB6, + (byte) 0xD3, 0x17, 0x60, (byte) 0xB6, 0x35, (byte) 0xF3, 0x6F, + 0x13, (byte) 0xEB, (byte) 0xC8, 0x77, (byte) 0xE9, (byte) 0xA0, + (byte) 0xC2, 0x76, (byte) 0xA8, 0x17 };*/ + + public static final byte[] iv = { 0x73, (byte) 0x80, 0x16, 0x6f, 0x49, + 0x14, (byte) 0xb2, (byte) 0xb9, 0x17, 0x24, 0x42, (byte) 0xd7, + (byte) 0xda, (byte) 0x8a, 0x06, 0x00, (byte) 0xa9, 0x6f, 0x30, + (byte) 0xbc, (byte) 0x16, 0x31, 0x38, (byte) 0xaa, (byte) 0xe3, + (byte) 0x8d, (byte) 0xee, 0x4d, (byte) 0xb0, (byte) 0xfb, 0x0e, + 0x4e }; + + public static int[] Tj = new int[64]; + + static + { + for (int i = 0; i < 16; i++) + { + Tj[i] = 0x79cc4519; + } + + for (int i = 16; i < 64; i++) + { + Tj[i] = 0x7a879d8a; + } + } + + public static byte[] CF(byte[] V, byte[] B) + { + int[] v, b; + v = convert(V); + b = convert(B); + return convert(CF(v, b)); + } + + private static int[] convert(byte[] arr) + { + int[] out = new int[arr.length / 4]; + byte[] tmp = new byte[4]; + for (int i = 0; i < arr.length; i += 4) + { + System.arraycopy(arr, i, tmp, 0, 4); + out[i / 4] = bigEndianByteToInt(tmp); + } + return out; + } + + private static byte[] convert(int[] arr) + { + byte[] out = new byte[arr.length * 4]; + byte[] tmp = null; + for (int i = 0; i < arr.length; i++) + { + tmp = bigEndianIntToByte(arr[i]); + System.arraycopy(tmp, 0, out, i * 4, 4); + } + return out; + } + + public static int[] CF(int[] V, int[] B) + { + int a, b, c, d, e, f, g, h; + int ss1, ss2, tt1, tt2; + a = V[0]; + b = V[1]; + c = V[2]; + d = V[3]; + e = V[4]; + f = V[5]; + g = V[6]; + h = V[7]; + + /*System.out.println("IV: "); + System.out.print(Integer.toHexString(a)+" "); + System.out.print(Integer.toHexString(b)+" "); + System.out.print(Integer.toHexString(c)+" "); + System.out.print(Integer.toHexString(d)+" "); + System.out.print(Integer.toHexString(e)+" "); + System.out.print(Integer.toHexString(f)+" "); + System.out.print(Integer.toHexString(g)+" "); + System.out.print(Integer.toHexString(h)+" "); + System.out.println(""); + System.out.println(""); + + System.out.println("填充后的消息: "); + for(int i=0; i= 0 && j <= 15) + { + return FF1j(X, Y, Z); + } + else + { + return FF2j(X, Y, Z); + } + } + + private static int GGj(int X, int Y, int Z, int j) + { + if (j >= 0 && j <= 15) + { + return GG1j(X, Y, Z); + } + else + { + return GG2j(X, Y, Z); + } + } + + // 逻辑位运算函数 + private static int FF1j(int X, int Y, int Z) + { + int tmp = X ^ Y ^ Z; + return tmp; + } + + private static int FF2j(int X, int Y, int Z) + { + int tmp = ((X & Y) | (X & Z) | (Y & Z)); + return tmp; + } + + private static int GG1j(int X, int Y, int Z) + { + int tmp = X ^ Y ^ Z; + return tmp; + } + + private static int GG2j(int X, int Y, int Z) + { + int tmp = (X & Y) | (~X & Z); + return tmp; + } + + private static int P0(int X) + { + int y = rotateLeft(X, 9); + y = bitCycleLeft(X, 9); + int z = rotateLeft(X, 17); + z = bitCycleLeft(X, 17); + int t = X ^ y ^ z; + return t; + } + + private static int P1(int X) + { + int t = X ^ bitCycleLeft(X, 15) ^ bitCycleLeft(X, 23); + return t; + } + + /** + * 对最后一个分组字节数据padding + * + * @param in + * @param bLen + * 分组个数 + * @return + */ + public static byte[] padding(byte[] in, int bLen) + { + int k = 448 - (8 * in.length + 1) % 512; + if (k < 0) + { + k = 960 - (8 * in.length + 1) % 512; + } + k += 1; + byte[] padd = new byte[k / 8]; + padd[0] = (byte) 0x80; + long n = in.length * 8 + bLen * 512; + byte[] out = new byte[in.length + k / 8 + 64 / 8]; + int pos = 0; + System.arraycopy(in, 0, out, 0, in.length); + pos += in.length; + System.arraycopy(padd, 0, out, pos, padd.length); + pos += padd.length; + byte[] tmp = back(Util.longToBytes(n)); + System.arraycopy(tmp, 0, out, pos, tmp.length); + return out; + } + + /** + * 字节数组逆序 + * + * @param in + * @return + */ + private static byte[] back(byte[] in) + { + byte[] out = new byte[in.length]; + for (int i = 0; i < out.length; i++) + { + out[i] = in[out.length - i - 1]; + } + + return out; + } + + public static int rotateLeft(int x, int n) + { + return (x << n) | (x >> (32 - n)); + } + + private static int bitCycleLeft(int n, int bitLen) + { + bitLen %= 32; + byte[] tmp = bigEndianIntToByte(n); + int byteLen = bitLen / 8; + int len = bitLen % 8; + if (byteLen > 0) + { + tmp = byteCycleLeft(tmp, byteLen); + } + + if (len > 0) + { + tmp = bitSmall8CycleLeft(tmp, len); + } + + return bigEndianByteToInt(tmp); + } + + private static byte[] bitSmall8CycleLeft(byte[] in, int len) + { + byte[] tmp = new byte[in.length]; + int t1, t2, t3; + for (int i = 0; i < tmp.length; i++) + { + t1 = (byte) ((in[i] & 0x000000ff) << len); + t2 = (byte) ((in[(i + 1) % tmp.length] & 0x000000ff) >> (8 - len)); + t3 = (byte) (t1 | t2); + tmp[i] = (byte) t3; + } + + return tmp; + } + + private static byte[] byteCycleLeft(byte[] in, int byteLen) + { + byte[] tmp = new byte[in.length]; + System.arraycopy(in, byteLen, tmp, 0, in.length - byteLen); + System.arraycopy(in, 0, tmp, in.length - byteLen, byteLen); + return tmp; + } + + /*private static void print(int[] arr) + { + for (int i = 0; i < arr.length; i++) + { + System.out.print(Integer.toHexString(arr[i]) + " "); + if ((i + 1) % 16 == 0) + { + System.out.println(); + } + } + System.out.println(); + }*/ +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Digest.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Digest.java new file mode 100644 index 0000000..85453d2 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Digest.java @@ -0,0 +1,134 @@ +package com.ruoyi.common.utils.sm3; + +public class SM3Digest +{ + /** SM3值的长度 */ + private static final int BYTE_LENGTH = 32; + + /** SM3分组长度 */ + private static final int BLOCK_LENGTH = 64; + + /** 缓冲区长度 */ + private static final int BUFFER_LENGTH = BLOCK_LENGTH * 1; + + /** 缓冲区 */ + private byte[] xBuf = new byte[BUFFER_LENGTH]; + + /** 缓冲区偏移量 */ + private int xBufOff; + + /** 初始向量 */ + private byte[] V = SM3.iv.clone(); + + private int cntBlock = 0; + + public SM3Digest() { + } + + public SM3Digest(SM3Digest t) + { + System.arraycopy(t.xBuf, 0, this.xBuf, 0, t.xBuf.length); + this.xBufOff = t.xBufOff; + System.arraycopy(t.V, 0, this.V, 0, t.V.length); + } + + /** + * SM3结果输出 + * + * @param out 保存SM3结构的缓冲区 + * @param outOff 缓冲区偏移量 + * @return + */ + public int doFinal(byte[] out, int outOff) + { + byte[] tmp = doFinal(); + System.arraycopy(tmp, 0, out, 0, tmp.length); + return BYTE_LENGTH; + } + + public void reset() + { + xBufOff = 0; + cntBlock = 0; + V = SM3.iv.clone(); + } + + /** + * 明文输入 + * + * @param in + * 明文输入缓冲区 + * @param inOff + * 缓冲区偏移量 + * @param len + * 明文长度 + */ + public void update(byte[] in, int inOff, int len) + { + int partLen = BUFFER_LENGTH - xBufOff; + int inputLen = len; + int dPos = inOff; + if (partLen < inputLen) + { + System.arraycopy(in, dPos, xBuf, xBufOff, partLen); + inputLen -= partLen; + dPos += partLen; + doUpdate(); + while (inputLen > BUFFER_LENGTH) + { + System.arraycopy(in, dPos, xBuf, 0, BUFFER_LENGTH); + inputLen -= BUFFER_LENGTH; + dPos += BUFFER_LENGTH; + doUpdate(); + } + } + + System.arraycopy(in, dPos, xBuf, xBufOff, inputLen); + xBufOff += inputLen; + } + + private void doUpdate() + { + byte[] B = new byte[BLOCK_LENGTH]; + for (int i = 0; i < BUFFER_LENGTH; i += BLOCK_LENGTH) + { + System.arraycopy(xBuf, i, B, 0, B.length); + doHash(B); + } + xBufOff = 0; + } + + private void doHash(byte[] B) + { + byte[] tmp = SM3.CF(V, B); + System.arraycopy(tmp, 0, V, 0, V.length); + cntBlock++; + } + + private byte[] doFinal() + { + byte[] B = new byte[BLOCK_LENGTH]; + byte[] buffer = new byte[xBufOff]; + System.arraycopy(xBuf, 0, buffer, 0, buffer.length); + byte[] tmp = SM3.padding(buffer, cntBlock); + for (int i = 0; i < tmp.length; i += BLOCK_LENGTH) + { + System.arraycopy(tmp, i, B, 0, B.length); + doHash(B); + } + return V; + } + + public void update(byte in) + { + byte[] buffer = new byte[] { in }; + update(buffer, 0, 1); + } + + public int getDigestSize() + { + return BYTE_LENGTH; + } + + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Utils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Utils.java new file mode 100644 index 0000000..76a3c3c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM3Utils.java @@ -0,0 +1,23 @@ +package com.ruoyi.common.utils.sm3; + +import org.bouncycastle.util.encoders.Hex; + +import java.io.IOException; + +public class SM3Utils { + + public static void main(String[] args) throws IOException { + String plainText = "test" + "qawsedrftg" + "1656492199425"; + System.out.println(encode(plainText)); + } + + public static String encode(String plainText) { + byte[] md = new byte[32]; + byte[] msg1 = plainText.getBytes(); + SM3Digest sm3 = new SM3Digest(); + sm3.update(msg1, 0, msg1.length); + sm3.doFinal(md, 0); + String s = new String(Hex.encode(md)); + return s; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4.java new file mode 100644 index 0000000..b54efba --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4.java @@ -0,0 +1,344 @@ +package com.ruoyi.common.utils.sm3; + +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; + +public class SM4 +{ + public static final int SM4_ENCRYPT = 1; + + public static final int SM4_DECRYPT = 0; + + private long GET_ULONG_BE(byte[] b, int i) + { + long n = (long)(b[i] & 0xff) << 24 | (long)((b[i + 1] & 0xff) << 16) | (long)((b[i + 2] & 0xff) << 8) | (long)(b[i + 3] & 0xff) & 0xffffffffL; + return n; + } + + private void PUT_ULONG_BE(long n, byte[] b, int i) + { + b[i] = (byte)(int)(0xFF & n >> 24); + b[i + 1] = (byte)(int)(0xFF & n >> 16); + b[i + 2] = (byte)(int)(0xFF & n >> 8); + b[i + 3] = (byte)(int)(0xFF & n); + } + + private long SHL(long x, int n) + { + return (x & 0xFFFFFFFF) << n; + } + + private long ROTL(long x, int n) + { + return SHL(x, n) | x >> (32 - n); + } + + private void SWAP(long[] sk, int i) + { + long t = sk[i]; + sk[i] = sk[(31 - i)]; + sk[(31 - i)] = t; + } + + public static final byte[] SboxTable = { (byte) 0xd6, (byte) 0x90, (byte) 0xe9, (byte) 0xfe, + (byte) 0xcc, (byte) 0xe1, 0x3d, (byte) 0xb7, 0x16, (byte) 0xb6, + 0x14, (byte) 0xc2, 0x28, (byte) 0xfb, 0x2c, 0x05, 0x2b, 0x67, + (byte) 0x9a, 0x76, 0x2a, (byte) 0xbe, 0x04, (byte) 0xc3, + (byte) 0xaa, 0x44, 0x13, 0x26, 0x49, (byte) 0x86, 0x06, + (byte) 0x99, (byte) 0x9c, 0x42, 0x50, (byte) 0xf4, (byte) 0x91, + (byte) 0xef, (byte) 0x98, 0x7a, 0x33, 0x54, 0x0b, 0x43, + (byte) 0xed, (byte) 0xcf, (byte) 0xac, 0x62, (byte) 0xe4, + (byte) 0xb3, 0x1c, (byte) 0xa9, (byte) 0xc9, 0x08, (byte) 0xe8, + (byte) 0x95, (byte) 0x80, (byte) 0xdf, (byte) 0x94, (byte) 0xfa, + 0x75, (byte) 0x8f, 0x3f, (byte) 0xa6, 0x47, 0x07, (byte) 0xa7, + (byte) 0xfc, (byte) 0xf3, 0x73, 0x17, (byte) 0xba, (byte) 0x83, + 0x59, 0x3c, 0x19, (byte) 0xe6, (byte) 0x85, 0x4f, (byte) 0xa8, + 0x68, 0x6b, (byte) 0x81, (byte) 0xb2, 0x71, 0x64, (byte) 0xda, + (byte) 0x8b, (byte) 0xf8, (byte) 0xeb, 0x0f, 0x4b, 0x70, 0x56, + (byte) 0x9d, 0x35, 0x1e, 0x24, 0x0e, 0x5e, 0x63, 0x58, (byte) 0xd1, + (byte) 0xa2, 0x25, 0x22, 0x7c, 0x3b, 0x01, 0x21, 0x78, (byte) 0x87, + (byte) 0xd4, 0x00, 0x46, 0x57, (byte) 0x9f, (byte) 0xd3, 0x27, + 0x52, 0x4c, 0x36, 0x02, (byte) 0xe7, (byte) 0xa0, (byte) 0xc4, + (byte) 0xc8, (byte) 0x9e, (byte) 0xea, (byte) 0xbf, (byte) 0x8a, + (byte) 0xd2, 0x40, (byte) 0xc7, 0x38, (byte) 0xb5, (byte) 0xa3, + (byte) 0xf7, (byte) 0xf2, (byte) 0xce, (byte) 0xf9, 0x61, 0x15, + (byte) 0xa1, (byte) 0xe0, (byte) 0xae, 0x5d, (byte) 0xa4, + (byte) 0x9b, 0x34, 0x1a, 0x55, (byte) 0xad, (byte) 0x93, 0x32, + 0x30, (byte) 0xf5, (byte) 0x8c, (byte) 0xb1, (byte) 0xe3, 0x1d, + (byte) 0xf6, (byte) 0xe2, 0x2e, (byte) 0x82, 0x66, (byte) 0xca, + 0x60, (byte) 0xc0, 0x29, 0x23, (byte) 0xab, 0x0d, 0x53, 0x4e, 0x6f, + (byte) 0xd5, (byte) 0xdb, 0x37, 0x45, (byte) 0xde, (byte) 0xfd, + (byte) 0x8e, 0x2f, 0x03, (byte) 0xff, 0x6a, 0x72, 0x6d, 0x6c, 0x5b, + 0x51, (byte) 0x8d, 0x1b, (byte) 0xaf, (byte) 0x92, (byte) 0xbb, + (byte) 0xdd, (byte) 0xbc, 0x7f, 0x11, (byte) 0xd9, 0x5c, 0x41, + 0x1f, 0x10, 0x5a, (byte) 0xd8, 0x0a, (byte) 0xc1, 0x31, + (byte) 0x88, (byte) 0xa5, (byte) 0xcd, 0x7b, (byte) 0xbd, 0x2d, + 0x74, (byte) 0xd0, 0x12, (byte) 0xb8, (byte) 0xe5, (byte) 0xb4, + (byte) 0xb0, (byte) 0x89, 0x69, (byte) 0x97, 0x4a, 0x0c, + (byte) 0x96, 0x77, 0x7e, 0x65, (byte) 0xb9, (byte) 0xf1, 0x09, + (byte) 0xc5, 0x6e, (byte) 0xc6, (byte) 0x84, 0x18, (byte) 0xf0, + 0x7d, (byte) 0xec, 0x3a, (byte) 0xdc, 0x4d, 0x20, 0x79, + (byte) 0xee, 0x5f, 0x3e, (byte) 0xd7, (byte) 0xcb, 0x39, 0x48 }; + + public static final int[] FK = { 0xa3b1bac6, 0x56aa3350, 0x677d9197, 0xb27022dc }; + + public static final int[] CK = { 0x00070e15,0x1c232a31,0x383f464d,0x545b6269, + 0x70777e85,0x8c939aa1,0xa8afb6bd,0xc4cbd2d9, + 0xe0e7eef5,0xfc030a11,0x181f262d,0x343b4249, + 0x50575e65,0x6c737a81,0x888f969d,0xa4abb2b9, + 0xc0c7ced5,0xdce3eaf1,0xf8ff060d,0x141b2229, + 0x30373e45,0x4c535a61,0x686f767d,0x848b9299, + 0xa0a7aeb5,0xbcc3cad1,0xd8dfe6ed,0xf4fb0209, + 0x10171e25,0x2c333a41,0x484f565d,0x646b7279 }; + + private byte sm4Sbox(byte inch) + { + int i = inch & 0xFF; + byte retVal = SboxTable[i]; + return retVal; + } + + private long sm4Lt(long ka) + { + long bb = 0L; + long c = 0L; + byte[] a = new byte[4]; + byte[] b = new byte[4]; + PUT_ULONG_BE(ka, a, 0); + b[0] = sm4Sbox(a[0]); + b[1] = sm4Sbox(a[1]); + b[2] = sm4Sbox(a[2]); + b[3] = sm4Sbox(a[3]); + bb = GET_ULONG_BE(b, 0); + c = bb ^ ROTL(bb, 2) ^ ROTL(bb, 10) ^ ROTL(bb, 18) ^ ROTL(bb, 24); + return c; + } + + private long sm4F(long x0, long x1, long x2, long x3, long rk) + { + return x0 ^ sm4Lt(x1 ^ x2 ^ x3 ^ rk); + } + + private long sm4CalciRK(long ka) + { + long bb = 0L; + long rk = 0L; + byte[] a = new byte[4]; + byte[] b = new byte[4]; + PUT_ULONG_BE(ka, a, 0); + b[0] = sm4Sbox(a[0]); + b[1] = sm4Sbox(a[1]); + b[2] = sm4Sbox(a[2]); + b[3] = sm4Sbox(a[3]); + bb = GET_ULONG_BE(b, 0); + rk = bb ^ ROTL(bb, 13) ^ ROTL(bb, 23); + return rk; + } + + private void sm4_setkey(long[] SK, byte[] key) + { + long[] MK = new long[4]; + long[] k = new long[36]; + int i = 0; + MK[0] = GET_ULONG_BE(key, 0); + MK[1] = GET_ULONG_BE(key, 4); + MK[2] = GET_ULONG_BE(key, 8); + MK[3] = GET_ULONG_BE(key, 12); + k[0] = MK[0] ^ (long) FK[0]; + k[1] = MK[1] ^ (long) FK[1]; + k[2] = MK[2] ^ (long) FK[2]; + k[3] = MK[3] ^ (long) FK[3]; + for (; i < 32; i++) + { + k[(i + 4)] = (k[i] ^ sm4CalciRK(k[(i + 1)] ^ k[(i + 2)] ^ k[(i + 3)] ^ (long) CK[i])); + SK[i] = k[(i + 4)]; + } + } + + private void sm4_one_round(long[] sk, byte[] input, byte[] output) + { + int i = 0; + long[] ulbuf = new long[36]; + ulbuf[0] = GET_ULONG_BE(input, 0); + ulbuf[1] = GET_ULONG_BE(input, 4); + ulbuf[2] = GET_ULONG_BE(input, 8); + ulbuf[3] = GET_ULONG_BE(input, 12); + while (i < 32) + { + ulbuf[(i + 4)] = sm4F(ulbuf[i], ulbuf[(i + 1)], ulbuf[(i + 2)], ulbuf[(i + 3)], sk[i]); + i++; + } + PUT_ULONG_BE(ulbuf[35], output, 0); + PUT_ULONG_BE(ulbuf[34], output, 4); + PUT_ULONG_BE(ulbuf[33], output, 8); + PUT_ULONG_BE(ulbuf[32], output, 12); + } + + private byte[] padding(byte[] input, int mode) + { + if (input == null) + { + return null; + } + + byte[] ret = (byte[]) null; + if (mode == SM4_ENCRYPT) + { + int p = 16 - input.length % 16; + ret = new byte[input.length + p]; + System.arraycopy(input, 0, ret, 0, input.length); + for (int i = 0; i < p; i++) + { + ret[input.length + i] = (byte) p; + } + } + else + { + int p = input[input.length - 1]; + ret = new byte[input.length - p]; + System.arraycopy(input, 0, ret, 0, input.length - p); + } + return ret; + } + + public void sm4_setkey_enc(SM4_Context ctx, byte[] key) throws Exception + { + if (ctx == null) + { + throw new Exception("ctx is null!"); + } + + if (key == null || key.length != 16) + { + throw new Exception("key error!"); + } + + ctx.mode = SM4_ENCRYPT; + sm4_setkey(ctx.sk, key); + } + + public void sm4_setkey_dec(SM4_Context ctx, byte[] key) throws Exception + { + if (ctx == null) + { + throw new Exception("ctx is null!"); + } + + if (key == null || key.length != 16) + { + throw new Exception("key error!"); + } + + int i = 0; + ctx.mode = SM4_DECRYPT; + sm4_setkey(ctx.sk, key); + for (i = 0; i < 16; i++) + { + SWAP(ctx.sk, i); + } + } + + public byte[] sm4_crypt_ecb(SM4_Context ctx, byte[] input) throws Exception + { + if (input == null) + { + throw new Exception("input is null!"); + } + + if ((ctx.isPadding) && (ctx.mode == SM4_ENCRYPT)) + { + input = padding(input, SM4_ENCRYPT); + } + + int length = input.length; + ByteArrayInputStream bins = new ByteArrayInputStream(input); + ByteArrayOutputStream bous = new ByteArrayOutputStream(); + for(; length > 0; length -= 16) + { + byte[] in = new byte[16]; + byte[] out = new byte[16]; + bins.read(in); + sm4_one_round(ctx.sk, in, out); + bous.write(out); + } + + byte[] output = bous.toByteArray(); + if (ctx.isPadding && ctx.mode == SM4_DECRYPT) + { + output = padding(output, SM4_DECRYPT); + } + bins.close(); + bous.close(); + return output; + } + + public byte[] sm4_crypt_cbc(SM4_Context ctx, byte[] iv, byte[] input) throws Exception + { + if (iv == null || iv.length != 16) + { + throw new Exception("iv error!"); + } + + if (input == null) + { + throw new Exception("input is null!"); + } + + if (ctx.isPadding && ctx.mode == SM4_ENCRYPT) + { + input = padding(input, SM4_ENCRYPT); + } + + int i = 0; + int length = input.length; + ByteArrayInputStream bins = new ByteArrayInputStream(input); + ByteArrayOutputStream bous = new ByteArrayOutputStream(); + if (ctx.mode == SM4_ENCRYPT) + { + for(; length > 0; length -= 16) + { + byte[] in = new byte[16]; + byte[] out = new byte[16]; + byte[] out1 = new byte[16]; + + bins.read(in); + for (i = 0; i < 16; i++) + { + out[i] = ((byte) (in[i] ^ iv[i])); + } + sm4_one_round(ctx.sk, out, out1); + System.arraycopy(out1, 0, iv, 0, 16); + bous.write(out1); + } + } + else + { + byte[] temp = new byte[16]; + for(; length > 0; length -= 16) + { + byte[] in = new byte[16]; + byte[] out = new byte[16]; + byte[] out1 = new byte[16]; + + bins.read(in); + System.arraycopy(in, 0, temp, 0, 16); + sm4_one_round(ctx.sk, in, out); + for (i = 0; i < 16; i++) + { + out1[i] = ((byte) (out[i] ^ iv[i])); + } + System.arraycopy(temp, 0, iv, 0, 16); + bous.write(out1); + } + } + + byte[] output = bous.toByteArray(); + if (ctx.isPadding && ctx.mode == SM4_DECRYPT) + { + output = padding(output, SM4_DECRYPT); + } + bins.close(); + bous.close(); + return output; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4Utils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4Utils.java new file mode 100644 index 0000000..3c78ae0 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4Utils.java @@ -0,0 +1,186 @@ +//package com.ruoyi.common.utils.sm3; +// +//import sun.misc.*; +// +//import java.util.regex.Matcher; +//import java.util.regex.Pattern; +// +//public class SM4Utils +//{ +// public String getSecretKey() { +// return secretKey; +// } +// +// public void setSecretKey(String secretKey) { +// this.secretKey = secretKey; +// } +// +// public boolean isHexString() { +// return hexString; +// } +// +// public void setHexString(boolean hexString) { +// this.hexString = hexString; +// } +// +// private String secretKey = ""; +// +// public String getIv() { +// return iv; +// } +// +// public void setIv(String iv) { +// this.iv = iv; +// } +// +// private String iv = ""; +// +// private boolean hexString = false; +// +// public SM4Utils() +// { +// } +// +// public String encryptData_ECB(String plainText) +// { +// try +// { +// SM4_Context ctx = new SM4_Context(); +// ctx.isPadding = true; +// ctx.mode = SM4.SM4_ENCRYPT; +// +// byte[] keyBytes; +// if (hexString) +// { +// keyBytes = Util.hexStringToBytes(secretKey); +// } +// else +// { +// keyBytes = secretKey.getBytes(); +// } +// +// SM4 sm4 = new SM4(); +// sm4.sm4_setkey_enc(ctx, keyBytes); +// byte[] encrypted = sm4.sm4_crypt_ecb(ctx, plainText.getBytes("GBK")); +// String cipherText = new BASE64Encoder().encode(encrypted); +// if (cipherText != null && cipherText.trim().length() > 0) +// { +// Pattern p = Pattern.compile("\\s*|\t|\r|\n"); +// Matcher m = p.matcher(cipherText); +// cipherText = m.replaceAll(""); +// } +// return cipherText; +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// return null; +// } +// } +// +// public String decryptData_ECB(String cipherText) +// { +// try +// { +// SM4_Context ctx = new SM4_Context(); +// ctx.isPadding = true; +// ctx.mode = SM4.SM4_DECRYPT; +// +// byte[] keyBytes; +// if (hexString) +// { +// keyBytes = Util.hexStringToBytes(secretKey); +// } +// else +// { +// keyBytes = secretKey.getBytes(); +// } +// +// SM4 sm4 = new SM4(); +// sm4.sm4_setkey_dec(ctx, keyBytes); +// byte[] decrypted = sm4.sm4_crypt_ecb(ctx, new BASE64Decoder().decodeBuffer(cipherText)); +// return new String(decrypted, "GBK"); +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// return null; +// } +// } +// +// public String encryptData_CBC(String plainText) +// { +// try +// { +// SM4_Context ctx = new SM4_Context(); +// ctx.isPadding = true; +// ctx.mode = SM4.SM4_ENCRYPT; +// +// byte[] keyBytes; +// byte[] ivBytes; +// if (hexString) +// { +// keyBytes = Util.hexStringToBytes(secretKey); +// ivBytes = Util.hexStringToBytes(iv); +// } +// else +// { +// keyBytes = secretKey.getBytes(); +// ivBytes = iv.getBytes(); +// } +// +// SM4 sm4 = new SM4(); +// sm4.sm4_setkey_enc(ctx, keyBytes); +// byte[] encrypted = sm4.sm4_crypt_cbc(ctx, ivBytes, plainText.getBytes("GBK")); +// String cipherText = new BASE64Encoder().encode(encrypted); +// if (cipherText != null && cipherText.trim().length() > 0) +// { +// Pattern p = Pattern.compile("\\s*|\t|\r|\n"); +// Matcher m = p.matcher(cipherText); +// cipherText = m.replaceAll(""); +// } +// return cipherText; +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// return null; +// } +// } +// +// public String decryptData_CBC(String cipherText) +// { +// try +// { +// SM4_Context ctx = new SM4_Context(); +// ctx.isPadding = true; +// ctx.mode = SM4.SM4_DECRYPT; +// +// byte[] keyBytes; +// byte[] ivBytes; +// if (hexString) +// { +// keyBytes = Util.hexStringToBytes(secretKey); +// ivBytes = Util.hexStringToBytes(iv); +// } +// else +// { +// keyBytes = secretKey.getBytes(); +// ivBytes = iv.getBytes(); +// } +// +// SM4 sm4 = new SM4(); +// sm4.sm4_setkey_dec(ctx, keyBytes); +// byte[] decrypted = sm4.sm4_crypt_cbc(ctx, ivBytes, new BASE64Decoder().decodeBuffer(cipherText)); +// return new String(decrypted, "GBK"); +// } +// catch (Exception e) +// { +// e.printStackTrace(); +// return null; +// } +// } +// +// +// +//} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4_Context.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4_Context.java new file mode 100644 index 0000000..95cb87e --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/SM4_Context.java @@ -0,0 +1,17 @@ +package com.ruoyi.common.utils.sm3; + +public class SM4_Context +{ + public int mode; + + public long[] sk; + + public boolean isPadding; + + public SM4_Context() + { + this.mode = 1; + this.isPadding = true; + this.sk = new long[32]; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/Util.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/Util.java new file mode 100644 index 0000000..bd20f8c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sm3/Util.java @@ -0,0 +1,662 @@ +package com.ruoyi.common.utils.sm3; + +import java.math.BigInteger; + +public class Util +{ + /** + * 整形转换成网络传输的字节流(字节数组)型数据 + * + * @param num 一个整型数据 + * @return 4个字节的自己数组 + */ + public static byte[] intToBytes(int num) + { + byte[] bytes = new byte[4]; + bytes[0] = (byte) (0xff & (num >> 0)); + bytes[1] = (byte) (0xff & (num >> 8)); + bytes[2] = (byte) (0xff & (num >> 16)); + bytes[3] = (byte) (0xff & (num >> 24)); + return bytes; + } + + /** + * 四个字节的字节数据转换成一个整形数据 + * + * @param bytes 4个字节的字节数组 + * @return 一个整型数据 + */ + public static int byteToInt(byte[] bytes) + { + int num = 0; + int temp; + temp = (0x000000ff & (bytes[0])) << 0; + num = num | temp; + temp = (0x000000ff & (bytes[1])) << 8; + num = num | temp; + temp = (0x000000ff & (bytes[2])) << 16; + num = num | temp; + temp = (0x000000ff & (bytes[3])) << 24; + num = num | temp; + return num; + } + + /** + * 长整形转换成网络传输的字节流(字节数组)型数据 + * + * @param num 一个长整型数据 + * @return 4个字节的自己数组 + */ + public static byte[] longToBytes(long num) + { + byte[] bytes = new byte[8]; + for (int i = 0; i < 8; i++) + { + bytes[i] = (byte) (0xff & (num >> (i * 8))); + } + + return bytes; + } + + /** + * 大数字转换字节流(字节数组)型数据 + * + * @param n + * @return + */ + public static byte[] byteConvert32Bytes(BigInteger n) + { + byte tmpd[] = (byte[])null; + if(n == null) + { + return null; + } + + if(n.toByteArray().length == 33) + { + tmpd = new byte[32]; + System.arraycopy(n.toByteArray(), 1, tmpd, 0, 32); + } + else if(n.toByteArray().length == 32) + { + tmpd = n.toByteArray(); + } + else + { + tmpd = new byte[32]; + for(int i = 0; i < 32 - n.toByteArray().length; i++) + { + tmpd[i] = 0; + } + System.arraycopy(n.toByteArray(), 0, tmpd, 32 - n.toByteArray().length, n.toByteArray().length); + } + return tmpd; + } + + /** + * 换字节流(字节数组)型数据转大数字 + * + * @param b + * @return + */ + public static BigInteger byteConvertInteger(byte[] b) + { + if (b[0] < 0) + { + byte[] temp = new byte[b.length + 1]; + temp[0] = 0; + System.arraycopy(b, 0, temp, 1, b.length); + return new BigInteger(temp); + } + return new BigInteger(b); + } + + /** + * 根据字节数组获得值(十六进制数字) + * + * @param bytes + * @return + */ + public static String getHexString(byte[] bytes) + { + return getHexString(bytes, true); + } + + /** + * 根据字节数组获得值(十六进制数字) + * + * @param bytes + * @param upperCase + * @return + */ + public static String getHexString(byte[] bytes, boolean upperCase) + { + String ret = ""; + for (int i = 0; i < bytes.length; i++) + { + ret += Integer.toString((bytes[i] & 0xff) + 0x100, 16).substring(1); + } + return upperCase ? ret.toUpperCase() : ret; + } + + /** + * 打印十六进制字符串 + * + * @param bytes + */ + public static void printHexString(byte[] bytes) + { + for (int i = 0; i < bytes.length; i++) + { + String hex = Integer.toHexString(bytes[i] & 0xFF); + if (hex.length() == 1) + { + hex = '0' + hex; + } + System.out.print("0x" + hex.toUpperCase() + ","); + } + System.out.println(""); + } + + /** + * Convert hex string to byte[] + * + * @param hexString + * the hex string + * @return byte[] + */ + public static byte[] hexStringToBytes(String hexString) + { + if (hexString == null || hexString.equals("")) + { + return null; + } + + hexString = hexString.toUpperCase(); + int length = hexString.length() / 2; + char[] hexChars = hexString.toCharArray(); + byte[] d = new byte[length]; + for (int i = 0; i < length; i++) + { + int pos = i * 2; + d[i] = (byte) (charToByte(hexChars[pos]) << 4 | charToByte(hexChars[pos + 1])); + } + return d; + } + + /** + * Convert char to byte + * + * @param c + * char + * @return byte + */ + public static byte charToByte(char c) + { + return (byte) "0123456789ABCDEF".indexOf(c); + } + + /** + * 用于建立十六进制字符的输出的小写字符数组 + */ + private static final char[] DIGITS_LOWER = {'0', '1', '2', '3', '4', '5', + '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'}; + + /** + * 用于建立十六进制字符的输出的大写字符数组 + */ + private static final char[] DIGITS_UPPER = {'0', '1', '2', '3', '4', '5', + '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'}; + + /** + * 将字节数组转换为十六进制字符数组 + * + * @param data byte[] + * @return 十六进制char[] + */ + public static char[] encodeHex(byte[] data) { + return encodeHex(data, true); + } + + /** + * 将字节数组转换为十六进制字符数组 + * + * @param data byte[] + * @param toLowerCase true 传换成小写格式 , false 传换成大写格式 + * @return 十六进制char[] + */ + public static char[] encodeHex(byte[] data, boolean toLowerCase) { + return encodeHex(data, toLowerCase ? DIGITS_LOWER : DIGITS_UPPER); + } + + /** + * 将字节数组转换为十六进制字符数组 + * + * @param data byte[] + * @param toDigits 用于控制输出的char[] + * @return 十六进制char[] + */ + protected static char[] encodeHex(byte[] data, char[] toDigits) { + int l = data.length; + char[] out = new char[l << 1]; + // two characters form the hex value. + for (int i = 0, j = 0; i < l; i++) { + out[j++] = toDigits[(0xF0 & data[i]) >>> 4]; + out[j++] = toDigits[0x0F & data[i]]; + } + return out; + } + + /** + * 将字节数组转换为十六进制字符串 + * + * @param data byte[] + * @return 十六进制String + */ + public static String encodeHexString(byte[] data) { + return encodeHexString(data, true); + } + + /** + * 将字节数组转换为十六进制字符串 + * + * @param data byte[] + * @param toLowerCase true 传换成小写格式 , false 传换成大写格式 + * @return 十六进制String + */ + public static String encodeHexString(byte[] data, boolean toLowerCase) { + return encodeHexString(data, toLowerCase ? DIGITS_LOWER : DIGITS_UPPER); + } + + /** + * 将字节数组转换为十六进制字符串 + * + * @param data byte[] + * @param toDigits 用于控制输出的char[] + * @return 十六进制String + */ + protected static String encodeHexString(byte[] data, char[] toDigits) { + return new String(encodeHex(data, toDigits)); + } + + /** + * 将十六进制字符数组转换为字节数组 + * + * @param data 十六进制char[] + * @return byte[] + * @throws RuntimeException 如果源十六进制字符数组是一个奇怪的长度,将抛出运行时异常 + */ + public static byte[] decodeHex(char[] data) { + int len = data.length; + + if ((len & 0x01) != 0) { + throw new RuntimeException("Odd number of characters."); + } + + byte[] out = new byte[len >> 1]; + + // two characters form the hex value. + for (int i = 0, j = 0; j < len; i++) { + int f = toDigit(data[j], j) << 4; + j++; + f = f | toDigit(data[j], j); + j++; + out[i] = (byte) (f & 0xFF); + } + + return out; + } + + /** + * 将十六进制字符转换成一个整数 + * + * @param ch 十六进制char + * @param index 十六进制字符在字符数组中的位置 + * @return 一个整数 + * @throws RuntimeException 当ch不是一个合法的十六进制字符时,抛出运行时异常 + */ + protected static int toDigit(char ch, int index) { + int digit = Character.digit(ch, 16); + if (digit == -1) { + throw new RuntimeException("Illegal hexadecimal character " + ch + + " at index " + index); + } + return digit; + } + + /** + * 数字字符串转ASCII码字符串 + * + * @param String + * 字符串 + * @return ASCII字符串 + */ + public static String StringToAsciiString(String content) { + String result = ""; + int max = content.length(); + for (int i = 0; i < max; i++) { + char c = content.charAt(i); + String b = Integer.toHexString(c); + result = result + b; + } + return result; + } + + /** + * 十六进制转字符串 + * + * @param hexString + * 十六进制字符串 + * @param encodeType + * 编码类型4:Unicode,2:普通编码 + * @return 字符串 + */ + public static String hexStringToString(String hexString, int encodeType) { + String result = ""; + int max = hexString.length() / encodeType; + for (int i = 0; i < max; i++) { + char c = (char) hexStringToAlgorism(hexString + .substring(i * encodeType, (i + 1) * encodeType)); + result += c; + } + return result; + } + + /** + * 十六进制字符串装十进制 + * + * @param hex + * 十六进制字符串 + * @return 十进制数值 + */ + public static int hexStringToAlgorism(String hex) { + hex = hex.toUpperCase(); + int max = hex.length(); + int result = 0; + for (int i = max; i > 0; i--) { + char c = hex.charAt(i - 1); + int algorism = 0; + if (c >= '0' && c <= '9') { + algorism = c - '0'; + } else { + algorism = c - 55; + } + result += Math.pow(16, max - i) * algorism; + } + return result; + } + + /** + * 十六转二进制 + * + * @param hex + * 十六进制字符串 + * @return 二进制字符串 + */ + public static String hexStringToBinary(String hex) { + hex = hex.toUpperCase(); + String result = ""; + int max = hex.length(); + for (int i = 0; i < max; i++) { + char c = hex.charAt(i); + switch (c) { + case '0': + result += "0000"; + break; + case '1': + result += "0001"; + break; + case '2': + result += "0010"; + break; + case '3': + result += "0011"; + break; + case '4': + result += "0100"; + break; + case '5': + result += "0101"; + break; + case '6': + result += "0110"; + break; + case '7': + result += "0111"; + break; + case '8': + result += "1000"; + break; + case '9': + result += "1001"; + break; + case 'A': + result += "1010"; + break; + case 'B': + result += "1011"; + break; + case 'C': + result += "1100"; + break; + case 'D': + result += "1101"; + break; + case 'E': + result += "1110"; + break; + case 'F': + result += "1111"; + break; + } + } + return result; + } + + /** + * ASCII码字符串转数字字符串 + * + * @param String + * ASCII字符串 + * @return 字符串 + */ + public static String AsciiStringToString(String content) { + String result = ""; + int length = content.length() / 2; + for (int i = 0; i < length; i++) { + String c = content.substring(i * 2, i * 2 + 2); + int a = hexStringToAlgorism(c); + char b = (char) a; + String d = String.valueOf(b); + result += d; + } + return result; + } + + /** + * 将十进制转换为指定长度的十六进制字符串 + * + * @param algorism + * int 十进制数字 + * @param maxLength + * int 转换后的十六进制字符串长度 + * @return String 转换后的十六进制字符串 + */ + public static String algorismToHexString(int algorism, int maxLength) { + String result = ""; + result = Integer.toHexString(algorism); + + if (result.length() % 2 == 1) { + result = "0" + result; + } + return patchHexString(result.toUpperCase(), maxLength); + } + + /** + * 字节数组转为普通字符串(ASCII对应的字符) + * + * @param bytearray + * byte[] + * @return String + */ + public static String byteToString(byte[] bytearray) { + String result = ""; + char temp; + + int length = bytearray.length; + for (int i = 0; i < length; i++) { + temp = (char) bytearray[i]; + result += temp; + } + return result; + } + + /** + * 二进制字符串转十进制 + * + * @param binary + * 二进制字符串 + * @return 十进制数值 + */ + public static int binaryToAlgorism(String binary) { + int max = binary.length(); + int result = 0; + for (int i = max; i > 0; i--) { + char c = binary.charAt(i - 1); + int algorism = c - '0'; + result += Math.pow(2, max - i) * algorism; + } + return result; + } + + /** + * 十进制转换为十六进制字符串 + * + * @param algorism + * int 十进制的数字 + * @return String 对应的十六进制字符串 + */ + public static String algorismToHEXString(int algorism) { + String result = ""; + result = Integer.toHexString(algorism); + + if (result.length() % 2 == 1) { + result = "0" + result; + + } + result = result.toUpperCase(); + + return result; + } + + /** + * HEX字符串前补0,主要用于长度位数不足。 + * + * @param str + * String 需要补充长度的十六进制字符串 + * @param maxLength + * int 补充后十六进制字符串的长度 + * @return 补充结果 + */ + static public String patchHexString(String str, int maxLength) { + String temp = ""; + for (int i = 0; i < maxLength - str.length(); i++) { + temp = "0" + temp; + } + str = (temp + str).substring(0, maxLength); + return str; + } + + /** + * 将一个字符串转换为int + * + * @param s + * String 要转换的字符串 + * @param defaultInt + * int 如果出现异常,默认返回的数字 + * @param radix + * int 要转换的字符串是什么进制的,如16 8 10. + * @return int 转换后的数字 + */ + public static int parseToInt(String s, int defaultInt, int radix) { + int i = 0; + try { + i = Integer.parseInt(s, radix); + } catch (NumberFormatException ex) { + i = defaultInt; + } + return i; + } + + /** + * 将一个十进制形式的数字字符串转换为int + * + * @param s + * String 要转换的字符串 + * @param defaultInt + * int 如果出现异常,默认返回的数字 + * @return int 转换后的数字 + */ + public static int parseToInt(String s, int defaultInt) { + int i = 0; + try { + i = Integer.parseInt(s); + } catch (NumberFormatException ex) { + i = defaultInt; + } + return i; + } + + /** + * 十六进制串转化为byte数组 + * + * @return the array of byte + */ + public static byte[] hexToByte(String hex) + throws IllegalArgumentException { + if (hex.length() % 2 != 0) { + throw new IllegalArgumentException(); + } + char[] arr = hex.toCharArray(); + byte[] b = new byte[hex.length() / 2]; + for (int i = 0, j = 0, l = hex.length(); i < l; i++, j++) { + String swap = "" + arr[i++] + arr[i]; + int byteint = Integer.parseInt(swap, 16) & 0xFF; + b[j] = new Integer(byteint).byteValue(); + } + return b; + } + + /** + * 字节数组转换为十六进制字符串 + * + * @param b + * byte[] 需要转换的字节数组 + * @return String 十六进制字符串 + */ + public static String byteToHex(byte b[]) { + if (b == null) { + throw new IllegalArgumentException( + "Argument b ( byte array ) is null! "); + } + String hs = ""; + String stmp = ""; + for (int n = 0; n < b.length; n++) { + stmp = Integer.toHexString(b[n] & 0xff); + if (stmp.length() == 1) { + hs = hs + "0" + stmp; + } else { + hs = hs + stmp; + } + } + return hs.toUpperCase(); + } + + public static byte[] subByte(byte[] input, int startIndex, int length) { + byte[] bt = new byte[length]; + for (int i = 0; i < length; i++) { + bt[i] = input[i + startIndex]; + } + return bt; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/spring/SpringUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/spring/SpringUtils.java new file mode 100644 index 0000000..9993b98 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/spring/SpringUtils.java @@ -0,0 +1,146 @@ +package com.ruoyi.common.utils.spring; + +import org.springframework.aop.framework.AopContext; +import org.springframework.beans.BeansException; +import org.springframework.beans.factory.NoSuchBeanDefinitionException; +import org.springframework.beans.factory.config.BeanFactoryPostProcessor; +import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; +import org.springframework.context.ApplicationContext; +import org.springframework.context.ApplicationContextAware; +import org.springframework.stereotype.Component; +import com.ruoyi.common.utils.StringUtils; + +/** + * spring工具类 方便在非spring管理环境中获取bean + * + * @author ruoyi + */ +@Component +public final class SpringUtils implements BeanFactoryPostProcessor, ApplicationContextAware +{ + /** Spring应用上下文环境 */ + private static ConfigurableListableBeanFactory beanFactory; + + private static ApplicationContext applicationContext; + + @Override + public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException + { + SpringUtils.beanFactory = beanFactory; + } + + @Override + public void setApplicationContext(ApplicationContext applicationContext) throws BeansException + { + SpringUtils.applicationContext = applicationContext; + } + + /** + * 获取对象 + * + * @param name + * @return Object 一个以所给名字注册的bean的实例 + * @throws org.springframework.beans.BeansException + * + */ + @SuppressWarnings("unchecked") + public static T getBean(String name) throws BeansException + { + return (T) beanFactory.getBean(name); + } + + /** + * 获取类型为requiredType的对象 + * + * @param clz + * @return + * @throws org.springframework.beans.BeansException + * + */ + public static T getBean(Class clz) throws BeansException + { + T result = (T) beanFactory.getBean(clz); + return result; + } + + /** + * 如果BeanFactory包含一个与所给名称匹配的bean定义,则返回true + * + * @param name + * @return boolean + */ + public static boolean containsBean(String name) + { + return beanFactory.containsBean(name); + } + + /** + * 判断以给定名字注册的bean定义是一个singleton还是一个prototype。 如果与给定名字相应的bean定义没有被找到,将会抛出一个异常(NoSuchBeanDefinitionException) + * + * @param name + * @return boolean + * @throws org.springframework.beans.factory.NoSuchBeanDefinitionException + * + */ + public static boolean isSingleton(String name) throws NoSuchBeanDefinitionException + { + return beanFactory.isSingleton(name); + } + + /** + * @param name + * @return Class 注册对象的类型 + * @throws org.springframework.beans.factory.NoSuchBeanDefinitionException + * + */ + public static Class getType(String name) throws NoSuchBeanDefinitionException + { + return beanFactory.getType(name); + } + + /** + * 如果给定的bean名字在bean定义中有别名,则返回这些别名 + * + * @param name + * @return + * @throws org.springframework.beans.factory.NoSuchBeanDefinitionException + * + */ + public static String[] getAliases(String name) throws NoSuchBeanDefinitionException + { + return beanFactory.getAliases(name); + } + + /** + * 获取aop代理对象 + * + * @param invoker + * @return + */ + @SuppressWarnings("unchecked") + public static T getAopProxy(T invoker) + { + return (T) AopContext.currentProxy(); + } + + /** + * 获取当前的环境配置,无配置返回null + * + * @return 当前的环境配置 + */ + public static String[] getActiveProfiles() + { + return applicationContext.getEnvironment().getActiveProfiles(); + } + + /** + * 获取当前的环境配置,当有多个环境配置时,只获取第一个 + * + * @return 当前的环境配置 + */ + public static String getActiveProfile() + { + final String[] activeProfiles = getActiveProfiles(); + return StringUtils.isNotEmpty(activeProfiles) ? activeProfiles[0] : null; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/sql/SqlUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sql/SqlUtil.java new file mode 100644 index 0000000..246a9cf --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/sql/SqlUtil.java @@ -0,0 +1,61 @@ +package com.ruoyi.common.utils.sql; + +import com.ruoyi.common.exception.UtilException; +import com.ruoyi.common.utils.StringUtils; + +/** + * sql操作工具类 + * + * @author ruoyi + */ +public class SqlUtil +{ + /** + * 定义常用的 sql关键字 + */ + public static String SQL_REGEX = "select |insert |delete |update |drop |count |exec |chr |mid |master |truncate |char |and |declare "; + + /** + * 仅支持字母、数字、下划线、空格、逗号、小数点(支持多个字段排序) + */ + public static String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+"; + + /** + * 检查字符,防止注入绕过 + */ + public static String escapeOrderBySql(String value) + { + if (StringUtils.isNotEmpty(value) && !isValidOrderBySql(value)) + { + throw new UtilException("参数不符合规范,不能进行查询"); + } + return value; + } + + /** + * 验证 order by 语法是否符合规范 + */ + public static boolean isValidOrderBySql(String value) + { + return value.matches(SQL_PATTERN); + } + + /** + * SQL关键字检查 + */ + public static void filterKeyword(String value) + { + if (StringUtils.isEmpty(value)) + { + return; + } + String[] sqlKeywords = StringUtils.split(SQL_REGEX, "\\|"); + for (String sqlKeyword : sqlKeywords) + { + if (StringUtils.indexOfIgnoreCase(value, sqlKeyword) > -1) + { + throw new UtilException("参数存在SQL注入风险"); + } + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/IdUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/IdUtils.java new file mode 100644 index 0000000..2c84427 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/IdUtils.java @@ -0,0 +1,49 @@ +package com.ruoyi.common.utils.uuid; + +/** + * ID生成器工具类 + * + * @author ruoyi + */ +public class IdUtils +{ + /** + * 获取随机UUID + * + * @return 随机UUID + */ + public static String randomUUID() + { + return UUID.randomUUID().toString(); + } + + /** + * 简化的UUID,去掉了横线 + * + * @return 简化的UUID,去掉了横线 + */ + public static String simpleUUID() + { + return UUID.randomUUID().toString(true); + } + + /** + * 获取随机UUID,使用性能更好的ThreadLocalRandom生成UUID + * + * @return 随机UUID + */ + public static String fastUUID() + { + return UUID.fastUUID().toString(); + } + + /** + * 简化的UUID,去掉了横线,使用性能更好的ThreadLocalRandom生成UUID + * + * @return 简化的UUID,去掉了横线 + */ + public static String fastSimpleUUID() + { + return UUID.fastUUID().toString(true); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/Seq.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/Seq.java new file mode 100644 index 0000000..528f3c1 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/Seq.java @@ -0,0 +1,86 @@ +package com.ruoyi.common.utils.uuid; + +import java.util.concurrent.atomic.AtomicInteger; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.StringUtils; + +/** + * @author ruoyi 序列生成类 + */ +public class Seq +{ + // 通用序列类型 + public static final String commSeqType = "COMMON"; + + // 上传序列类型 + public static final String uploadSeqType = "UPLOAD"; + + // 通用接口序列数 + private static AtomicInteger commSeq = new AtomicInteger(1); + + // 上传接口序列数 + private static AtomicInteger uploadSeq = new AtomicInteger(1); + + // 机器标识 + private static String machineCode = "A"; + + /** + * 获取通用序列号 + * + * @return 序列值 + */ + public static String getId() + { + return getId(commSeqType); + } + + /** + * 默认16位序列号 yyMMddHHmmss + 一位机器标识 + 3长度循环递增字符串 + * + * @return 序列值 + */ + public static String getId(String type) + { + AtomicInteger atomicInt = commSeq; + if (uploadSeqType.equals(type)) + { + atomicInt = uploadSeq; + } + return getId(atomicInt, 3); + } + + /** + * 通用接口序列号 yyMMddHHmmss + 一位机器标识 + length长度循环递增字符串 + * + * @param atomicInt 序列数 + * @param length 数值长度 + * @return 序列值 + */ + public static String getId(AtomicInteger atomicInt, int length) + { + String result = DateUtils.dateTimeNow(); + result += machineCode; + result += getSeq(atomicInt, length); + return result; + } + + /** + * 序列循环递增字符串[1, 10 的 (length)幂次方), 用0左补齐length位数 + * + * @return 序列值 + */ + private synchronized static String getSeq(AtomicInteger atomicInt, int length) + { + // 先取值再+1 + int value = atomicInt.getAndIncrement(); + + // 如果更新后值>=10 的 (length)幂次方则重置为1 + int maxSeq = (int) Math.pow(10, length); + if (atomicInt.get() >= maxSeq) + { + atomicInt.set(1); + } + // 转字符串,用0左补齐 + return StringUtils.padl(value, length); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/UUID.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/UUID.java new file mode 100644 index 0000000..dfda46c --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/uuid/UUID.java @@ -0,0 +1,484 @@ +package com.ruoyi.common.utils.uuid; + +import java.security.MessageDigest; +import java.security.NoSuchAlgorithmException; +import java.security.SecureRandom; +import java.util.Random; +import java.util.concurrent.ThreadLocalRandom; +import com.ruoyi.common.exception.UtilException; + +/** + * 提供通用唯一识别码(universally unique identifier)(UUID)实现 + * + * @author ruoyi + */ +public final class UUID implements java.io.Serializable, Comparable +{ + private static final long serialVersionUID = -1185015143654744140L; + + /** + * SecureRandom 的单例 + * + */ + private static class Holder + { + static final SecureRandom numberGenerator = getSecureRandom(); + } + + /** 此UUID的最高64有效位 */ + private final long mostSigBits; + + /** 此UUID的最低64有效位 */ + private final long leastSigBits; + + /** + * 私有构造 + * + * @param data 数据 + */ + private UUID(byte[] data) + { + long msb = 0; + long lsb = 0; + assert data.length == 16 : "data must be 16 bytes in length"; + for (int i = 0; i < 8; i++) + { + msb = (msb << 8) | (data[i] & 0xff); + } + for (int i = 8; i < 16; i++) + { + lsb = (lsb << 8) | (data[i] & 0xff); + } + this.mostSigBits = msb; + this.leastSigBits = lsb; + } + + /** + * 使用指定的数据构造新的 UUID。 + * + * @param mostSigBits 用于 {@code UUID} 的最高有效 64 位 + * @param leastSigBits 用于 {@code UUID} 的最低有效 64 位 + */ + public UUID(long mostSigBits, long leastSigBits) + { + this.mostSigBits = mostSigBits; + this.leastSigBits = leastSigBits; + } + + /** + * 获取类型 4(伪随机生成的)UUID 的静态工厂。 使用加密的本地线程伪随机数生成器生成该 UUID。 + * + * @return 随机生成的 {@code UUID} + */ + public static UUID fastUUID() + { + return randomUUID(false); + } + + /** + * 获取类型 4(伪随机生成的)UUID 的静态工厂。 使用加密的强伪随机数生成器生成该 UUID。 + * + * @return 随机生成的 {@code UUID} + */ + public static UUID randomUUID() + { + return randomUUID(true); + } + + /** + * 获取类型 4(伪随机生成的)UUID 的静态工厂。 使用加密的强伪随机数生成器生成该 UUID。 + * + * @param isSecure 是否使用{@link SecureRandom}如果是可以获得更安全的随机码,否则可以得到更好的性能 + * @return 随机生成的 {@code UUID} + */ + public static UUID randomUUID(boolean isSecure) + { + final Random ng = isSecure ? Holder.numberGenerator : getRandom(); + + byte[] randomBytes = new byte[16]; + ng.nextBytes(randomBytes); + randomBytes[6] &= 0x0f; /* clear version */ + randomBytes[6] |= 0x40; /* set to version 4 */ + randomBytes[8] &= 0x3f; /* clear variant */ + randomBytes[8] |= 0x80; /* set to IETF variant */ + return new UUID(randomBytes); + } + + /** + * 根据指定的字节数组获取类型 3(基于名称的)UUID 的静态工厂。 + * + * @param name 用于构造 UUID 的字节数组。 + * + * @return 根据指定数组生成的 {@code UUID} + */ + public static UUID nameUUIDFromBytes(byte[] name) + { + MessageDigest md; + try + { + md = MessageDigest.getInstance("MD5"); + } + catch (NoSuchAlgorithmException nsae) + { + throw new InternalError("MD5 not supported"); + } + byte[] md5Bytes = md.digest(name); + md5Bytes[6] &= 0x0f; /* clear version */ + md5Bytes[6] |= 0x30; /* set to version 3 */ + md5Bytes[8] &= 0x3f; /* clear variant */ + md5Bytes[8] |= 0x80; /* set to IETF variant */ + return new UUID(md5Bytes); + } + + /** + * 根据 {@link #toString()} 方法中描述的字符串标准表示形式创建{@code UUID}。 + * + * @param name 指定 {@code UUID} 字符串 + * @return 具有指定值的 {@code UUID} + * @throws IllegalArgumentException 如果 name 与 {@link #toString} 中描述的字符串表示形式不符抛出此异常 + * + */ + public static UUID fromString(String name) + { + String[] components = name.split("-"); + if (components.length != 5) + { + throw new IllegalArgumentException("Invalid UUID string: " + name); + } + for (int i = 0; i < 5; i++) + { + components[i] = "0x" + components[i]; + } + + long mostSigBits = Long.decode(components[0]).longValue(); + mostSigBits <<= 16; + mostSigBits |= Long.decode(components[1]).longValue(); + mostSigBits <<= 16; + mostSigBits |= Long.decode(components[2]).longValue(); + + long leastSigBits = Long.decode(components[3]).longValue(); + leastSigBits <<= 48; + leastSigBits |= Long.decode(components[4]).longValue(); + + return new UUID(mostSigBits, leastSigBits); + } + + /** + * 返回此 UUID 的 128 位值中的最低有效 64 位。 + * + * @return 此 UUID 的 128 位值中的最低有效 64 位。 + */ + public long getLeastSignificantBits() + { + return leastSigBits; + } + + /** + * 返回此 UUID 的 128 位值中的最高有效 64 位。 + * + * @return 此 UUID 的 128 位值中最高有效 64 位。 + */ + public long getMostSignificantBits() + { + return mostSigBits; + } + + /** + * 与此 {@code UUID} 相关联的版本号. 版本号描述此 {@code UUID} 是如何生成的。 + *

+ * 版本号具有以下含意: + *

    + *
  • 1 基于时间的 UUID + *
  • 2 DCE 安全 UUID + *
  • 3 基于名称的 UUID + *
  • 4 随机生成的 UUID + *
+ * + * @return 此 {@code UUID} 的版本号 + */ + public int version() + { + // Version is bits masked by 0x000000000000F000 in MS long + return (int) ((mostSigBits >> 12) & 0x0f); + } + + /** + * 与此 {@code UUID} 相关联的变体号。变体号描述 {@code UUID} 的布局。 + *

+ * 变体号具有以下含意: + *

    + *
  • 0 为 NCS 向后兼容保留 + *
  • 2 IETF RFC 4122(Leach-Salz), 用于此类 + *
  • 6 保留,微软向后兼容 + *
  • 7 保留供以后定义使用 + *
+ * + * @return 此 {@code UUID} 相关联的变体号 + */ + public int variant() + { + // This field is composed of a varying number of bits. + // 0 - - Reserved for NCS backward compatibility + // 1 0 - The IETF aka Leach-Salz variant (used by this class) + // 1 1 0 Reserved, Microsoft backward compatibility + // 1 1 1 Reserved for future definition. + return (int) ((leastSigBits >>> (64 - (leastSigBits >>> 62))) & (leastSigBits >> 63)); + } + + /** + * 与此 UUID 相关联的时间戳值。 + * + *

+ * 60 位的时间戳值根据此 {@code UUID} 的 time_low、time_mid 和 time_hi 字段构造。
+ * 所得到的时间戳以 100 毫微秒为单位,从 UTC(通用协调时间) 1582 年 10 月 15 日零时开始。 + * + *

+ * 时间戳值仅在在基于时间的 UUID(其 version 类型为 1)中才有意义。
+ * 如果此 {@code UUID} 不是基于时间的 UUID,则此方法抛出 UnsupportedOperationException。 + * + * @throws UnsupportedOperationException 如果此 {@code UUID} 不是 version 为 1 的 UUID。 + */ + public long timestamp() throws UnsupportedOperationException + { + checkTimeBase(); + return (mostSigBits & 0x0FFFL) << 48// + | ((mostSigBits >> 16) & 0x0FFFFL) << 32// + | mostSigBits >>> 32; + } + + /** + * 与此 UUID 相关联的时钟序列值。 + * + *

+ * 14 位的时钟序列值根据此 UUID 的 clock_seq 字段构造。clock_seq 字段用于保证在基于时间的 UUID 中的时间唯一性。 + *

+ * {@code clockSequence} 值仅在基于时间的 UUID(其 version 类型为 1)中才有意义。 如果此 UUID 不是基于时间的 UUID,则此方法抛出 + * UnsupportedOperationException。 + * + * @return 此 {@code UUID} 的时钟序列 + * + * @throws UnsupportedOperationException 如果此 UUID 的 version 不为 1 + */ + public int clockSequence() throws UnsupportedOperationException + { + checkTimeBase(); + return (int) ((leastSigBits & 0x3FFF000000000000L) >>> 48); + } + + /** + * 与此 UUID 相关的节点值。 + * + *

+ * 48 位的节点值根据此 UUID 的 node 字段构造。此字段旨在用于保存机器的 IEEE 802 地址,该地址用于生成此 UUID 以保证空间唯一性。 + *

+ * 节点值仅在基于时间的 UUID(其 version 类型为 1)中才有意义。
+ * 如果此 UUID 不是基于时间的 UUID,则此方法抛出 UnsupportedOperationException。 + * + * @return 此 {@code UUID} 的节点值 + * + * @throws UnsupportedOperationException 如果此 UUID 的 version 不为 1 + */ + public long node() throws UnsupportedOperationException + { + checkTimeBase(); + return leastSigBits & 0x0000FFFFFFFFFFFFL; + } + + /** + * 返回此{@code UUID} 的字符串表现形式。 + * + *

+ * UUID 的字符串表示形式由此 BNF 描述: + * + *

+     * {@code
+     * UUID                   = ----
+     * time_low               = 4*
+     * time_mid               = 2*
+     * time_high_and_version  = 2*
+     * variant_and_sequence   = 2*
+     * node                   = 6*
+     * hexOctet               = 
+     * hexDigit               = [0-9a-fA-F]
+     * }
+     * 
+ * + * + * + * @return 此{@code UUID} 的字符串表现形式 + * @see #toString(boolean) + */ + @Override + public String toString() + { + return toString(false); + } + + /** + * 返回此{@code UUID} 的字符串表现形式。 + * + *

+ * UUID 的字符串表示形式由此 BNF 描述: + * + *

+     * {@code
+     * UUID                   = ----
+     * time_low               = 4*
+     * time_mid               = 2*
+     * time_high_and_version  = 2*
+     * variant_and_sequence   = 2*
+     * node                   = 6*
+     * hexOctet               = 
+     * hexDigit               = [0-9a-fA-F]
+     * }
+     * 
+ * + * + * + * @param isSimple 是否简单模式,简单模式为不带'-'的UUID字符串 + * @return 此{@code UUID} 的字符串表现形式 + */ + public String toString(boolean isSimple) + { + final StringBuilder builder = new StringBuilder(isSimple ? 32 : 36); + // time_low + builder.append(digits(mostSigBits >> 32, 8)); + if (!isSimple) + { + builder.append('-'); + } + // time_mid + builder.append(digits(mostSigBits >> 16, 4)); + if (!isSimple) + { + builder.append('-'); + } + // time_high_and_version + builder.append(digits(mostSigBits, 4)); + if (!isSimple) + { + builder.append('-'); + } + // variant_and_sequence + builder.append(digits(leastSigBits >> 48, 4)); + if (!isSimple) + { + builder.append('-'); + } + // node + builder.append(digits(leastSigBits, 12)); + + return builder.toString(); + } + + /** + * 返回此 UUID 的哈希码。 + * + * @return UUID 的哈希码值。 + */ + @Override + public int hashCode() + { + long hilo = mostSigBits ^ leastSigBits; + return ((int) (hilo >> 32)) ^ (int) hilo; + } + + /** + * 将此对象与指定对象比较。 + *

+ * 当且仅当参数不为 {@code null}、而是一个 UUID 对象、具有与此 UUID 相同的 varriant、包含相同的值(每一位均相同)时,结果才为 {@code true}。 + * + * @param obj 要与之比较的对象 + * + * @return 如果对象相同,则返回 {@code true};否则返回 {@code false} + */ + @Override + public boolean equals(Object obj) + { + if ((null == obj) || (obj.getClass() != UUID.class)) + { + return false; + } + UUID id = (UUID) obj; + return (mostSigBits == id.mostSigBits && leastSigBits == id.leastSigBits); + } + + // Comparison Operations + + /** + * 将此 UUID 与指定的 UUID 比较。 + * + *

+ * 如果两个 UUID 不同,且第一个 UUID 的最高有效字段大于第二个 UUID 的对应字段,则第一个 UUID 大于第二个 UUID。 + * + * @param val 与此 UUID 比较的 UUID + * + * @return 在此 UUID 小于、等于或大于 val 时,分别返回 -1、0 或 1。 + * + */ + @Override + public int compareTo(UUID val) + { + // The ordering is intentionally set up so that the UUIDs + // can simply be numerically compared as two numbers + return (this.mostSigBits < val.mostSigBits ? -1 : // + (this.mostSigBits > val.mostSigBits ? 1 : // + (this.leastSigBits < val.leastSigBits ? -1 : // + (this.leastSigBits > val.leastSigBits ? 1 : // + 0)))); + } + + // ------------------------------------------------------------------------------------------------------------------- + // Private method start + /** + * 返回指定数字对应的hex值 + * + * @param val 值 + * @param digits 位 + * @return 值 + */ + private static String digits(long val, int digits) + { + long hi = 1L << (digits * 4); + return Long.toHexString(hi | (val & (hi - 1))).substring(1); + } + + /** + * 检查是否为time-based版本UUID + */ + private void checkTimeBase() + { + if (version() != 1) + { + throw new UnsupportedOperationException("Not a time-based UUID"); + } + } + + /** + * 获取{@link SecureRandom},类提供加密的强随机数生成器 (RNG) + * + * @return {@link SecureRandom} + */ + public static SecureRandom getSecureRandom() + { + try + { + return SecureRandom.getInstance("SHA1PRNG"); + } + catch (NoSuchAlgorithmException e) + { + throw new UtilException(e); + } + } + + /** + * 获取随机数生成器对象
+ * ThreadLocalRandom是JDK 7之后提供并发产生随机数,能够解决多个线程发生的竞争争夺。 + * + * @return {@link ThreadLocalRandom} + */ + public static ThreadLocalRandom getRandom() + { + return ThreadLocalRandom.current(); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/validator/ValidatorUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/validator/ValidatorUtils.java new file mode 100644 index 0000000..648a8d6 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/validator/ValidatorUtils.java @@ -0,0 +1,84 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + *

+ * https://www.renren.io + *

+ * 版权所有,侵权必究! + */ + +package com.ruoyi.common.validator; + + +import com.ruoyi.common.exception.RYException; + +import javax.validation.ConstraintViolation; +import javax.validation.Validation; +import javax.validation.Validator; +import java.util.Set; + +/** + * hibernate-validator校验工具类 + *

+ * 参考文档:http://docs.jboss.org/hibernate/validator/5.4/reference/en-US/html_single/ + * + * @author Mark sunlightcs@gmail.com + */ +public class ValidatorUtils { + private static Validator validator; + + static { + validator = Validation.buildDefaultValidatorFactory().getValidator(); + } + + /** + * 校验对象 + * + * @param object 待校验对象 + * @param groups 待校验的组 + * @throws RYException 校验不通过,则报RYException异常 + */ + public static void validateEntity(Object object, Class... groups) + throws RYException { + Set> constraintViolations = validator.validate(object, groups); + if (!constraintViolations.isEmpty()) { + ConstraintViolation constraint = (ConstraintViolation) constraintViolations.iterator().next(); + throw new RYException(constraint.getMessage()); + } + } + + /** + * 校验对象,版本2 + * + * @param object 待校验对象 + * @param groups 待校验的组 + * @throws RYException 校验不通过,则报RYException异常 + */ + public static void validateV2(Object object, Class... groups) + throws RYException { + validateV2(0, object, groups); + } + + /** + * 校验对象,版本2 + * + * @param i 第几条校验 + * @param object 待校验对象 + * @param groups 待校验的组 + * @throws RYException 校验不通过,则报RYException异常 + */ + public static void validateV2(int i, Object object, Class... groups) + throws RYException { + Set> validateSet = validator.validate(object, groups); + if (!validateSet.isEmpty()) { + String messages = validateSet.stream() + .map(ConstraintViolation::getMessage) + .reduce((m1, m2) -> m1 + ";" + m2) + .orElse("参数输入有误!"); + + if (i > 0) { + messages = "第" + i + "条数据:" + messages; + } + throw new RYException(messages, 11); + } + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/xss/SQLFilter.java b/ruoyi-common/src/main/java/com/ruoyi/common/xss/SQLFilter.java new file mode 100644 index 0000000..ea386db --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/xss/SQLFilter.java @@ -0,0 +1,51 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.common.xss; + + +import com.ruoyi.common.exception.RYException; +import org.apache.commons.lang3.StringUtils; + +/** + * SQL过滤 + * + * @author Mark sunlightcs@gmail.com + */ +public class SQLFilter { + + /** + * SQL注入过滤 + * @param str 待验证的字符串 + */ + public static String sqlInject(String str){ + if(StringUtils.isBlank(str)){ + return null; + } + //去掉'|"|;|\字符 + str = StringUtils.replace(str, "'", ""); + str = StringUtils.replace(str, "\"", ""); + str = StringUtils.replace(str, ";", ""); + str = StringUtils.replace(str, "\\", ""); + + //转换成小写 + str = str.toLowerCase(); + + //非法字符 + String[] keywords = {"master", "truncate", "insert", "select", "delete", "update", "declare", "alter", "drop"}; + + //判断是否包含非法字符 + for(String keyword : keywords){ + if(str.indexOf(keyword) != -1){ + throw new RYException("包含非法字符"); + } + } + + return str; + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/xss/Xss.java b/ruoyi-common/src/main/java/com/ruoyi/common/xss/Xss.java new file mode 100644 index 0000000..7bfdf04 --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/xss/Xss.java @@ -0,0 +1,27 @@ +package com.ruoyi.common.xss; + +import javax.validation.Constraint; +import javax.validation.Payload; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 自定义xss校验注解 + * + * @author ruoyi + */ +@Retention(RetentionPolicy.RUNTIME) +@Target(value = { ElementType.METHOD, ElementType.FIELD, ElementType.CONSTRUCTOR, ElementType.PARAMETER }) +@Constraint(validatedBy = { XssValidator.class }) +public @interface Xss +{ + String message() + + default "不允许任何脚本运行"; + + Class[] groups() default {}; + + Class[] payload() default {}; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/xss/XssValidator.java b/ruoyi-common/src/main/java/com/ruoyi/common/xss/XssValidator.java new file mode 100644 index 0000000..ed9ec1f --- /dev/null +++ b/ruoyi-common/src/main/java/com/ruoyi/common/xss/XssValidator.java @@ -0,0 +1,34 @@ +package com.ruoyi.common.xss; + +import com.ruoyi.common.utils.StringUtils; +import javax.validation.ConstraintValidator; +import javax.validation.ConstraintValidatorContext; +import java.util.regex.Matcher; +import java.util.regex.Pattern; + +/** + * 自定义xss校验注解实现 + * + * @author ruoyi + */ +public class XssValidator implements ConstraintValidator +{ + private static final String HTML_PATTERN = "<(\\S*?)[^>]*>.*?|<.*? />"; + + @Override + public boolean isValid(String value, ConstraintValidatorContext constraintValidatorContext) + { + if (StringUtils.isBlank(value)) + { + return true; + } + return !containsHtml(value); + } + + public static boolean containsHtml(String value) + { + Pattern pattern = Pattern.compile(HTML_PATTERN); + Matcher matcher = pattern.matcher(value); + return matcher.matches(); + } +} \ No newline at end of file diff --git a/ruoyi-framework/pom.xml b/ruoyi-framework/pom.xml new file mode 100644 index 0000000..9339f11 --- /dev/null +++ b/ruoyi-framework/pom.xml @@ -0,0 +1,69 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-framework + + + framework框架核心 + + + + + + + org.springframework.boot + spring-boot-starter-web + + + + + org.springframework.boot + spring-boot-starter-aop + + + + + com.alibaba + druid-spring-boot-starter + + + + + com.github.penggle + kaptcha + + + javax.servlet-api + javax.servlet + + + + + + + com.github.oshi + oshi-core + + + + + com.ruoyi + ruoyi-system + + + org.jodd + jodd-http + 6.2.1 + + + + + diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataScopeAspect.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataScopeAspect.java new file mode 100644 index 0000000..48f8d72 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataScopeAspect.java @@ -0,0 +1,149 @@ +package com.ruoyi.framework.aspectj; + +import org.aspectj.lang.JoinPoint; +import org.aspectj.lang.annotation.Aspect; +import org.aspectj.lang.annotation.Before; +import org.springframework.stereotype.Component; +import com.ruoyi.common.annotation.DataScope; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.SecurityUtils; + +/** + * 数据过滤处理 + * + * @author ruoyi + */ +@Aspect +@Component +public class DataScopeAspect +{ + /** + * 全部数据权限 + */ + public static final String DATA_SCOPE_ALL = "1"; + + /** + * 自定数据权限 + */ + public static final String DATA_SCOPE_CUSTOM = "2"; + + /** + * 部门数据权限 + */ + public static final String DATA_SCOPE_DEPT = "3"; + + /** + * 部门及以下数据权限 + */ + public static final String DATA_SCOPE_DEPT_AND_CHILD = "4"; + + /** + * 仅本人数据权限 + */ + public static final String DATA_SCOPE_SELF = "5"; + + /** + * 数据权限过滤关键字 + */ + public static final String DATA_SCOPE = "dataScope"; + + @Before("@annotation(controllerDataScope)") + public void doBefore(JoinPoint point, DataScope controllerDataScope) throws Throwable + { + clearDataScope(point); + handleDataScope(point, controllerDataScope); + } + + protected void handleDataScope(final JoinPoint joinPoint, DataScope controllerDataScope) + { + // 获取当前的用户 + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (StringUtils.isNotNull(loginUser)) + { + SysUser currentUser = loginUser.getUser(); + // 如果是超级管理员,则不过滤数据 + if (StringUtils.isNotNull(currentUser) && !currentUser.isAdmin()) + { + dataScopeFilter(joinPoint, currentUser, controllerDataScope.deptAlias(), + controllerDataScope.userAlias()); + } + } + } + + /** + * 数据范围过滤 + * + * @param joinPoint 切点 + * @param user 用户 + * @param userAlias 别名 + */ + public static void dataScopeFilter(JoinPoint joinPoint, SysUser user, String deptAlias, String userAlias) + { + StringBuilder sqlString = new StringBuilder(); + + for (SysRole role : user.getRoles()) + { + String dataScope = role.getDataScope(); + if (DATA_SCOPE_ALL.equals(dataScope)) + { + sqlString = new StringBuilder(); + break; + } + else if (DATA_SCOPE_CUSTOM.equals(dataScope)) + { + sqlString.append(StringUtils.format( + " OR {}.dept_id IN ( SELECT dept_id FROM sys_role_dept WHERE role_id = {} ) ", deptAlias, + role.getRoleId())); + } + else if (DATA_SCOPE_DEPT.equals(dataScope)) + { + sqlString.append(StringUtils.format(" OR {}.dept_id = {} ", deptAlias, user.getDeptId())); + } + else if (DATA_SCOPE_DEPT_AND_CHILD.equals(dataScope)) + { + sqlString.append(StringUtils.format( + " OR {}.dept_id IN ( SELECT dept_id FROM sys_dept WHERE dept_id = {} or find_in_set( {} , ancestors ) )", + deptAlias, user.getDeptId(), user.getDeptId())); + } + else if (DATA_SCOPE_SELF.equals(dataScope)) + { + if (StringUtils.isNotBlank(userAlias)) + { + sqlString.append(StringUtils.format(" OR {}.user_id = {} ", userAlias, user.getUserId())); + } + else + { + // 数据权限为仅本人且没有userAlias别名不查询任何数据 + sqlString.append(" OR 1=0 "); + } + } + } + + if (StringUtils.isNotBlank(sqlString.toString())) + { + Object params = joinPoint.getArgs()[0]; + if (StringUtils.isNotNull(params) && params instanceof BaseEntity) + { + BaseEntity baseEntity = (BaseEntity) params; + baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")"); + } + } + } + + /** + * 拼接权限sql前先清空params.dataScope参数防止注入 + */ + private void clearDataScope(final JoinPoint joinPoint) + { + Object params = joinPoint.getArgs()[0]; + if (StringUtils.isNotNull(params) && params instanceof BaseEntity) + { + BaseEntity baseEntity = (BaseEntity) params; + baseEntity.getParams().put(DATA_SCOPE, ""); + } + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataSourceAspect.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataSourceAspect.java new file mode 100644 index 0000000..8c2c9f4 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataSourceAspect.java @@ -0,0 +1,72 @@ +package com.ruoyi.framework.aspectj; + +import java.util.Objects; +import org.aspectj.lang.ProceedingJoinPoint; +import org.aspectj.lang.annotation.Around; +import org.aspectj.lang.annotation.Aspect; +import org.aspectj.lang.annotation.Pointcut; +import org.aspectj.lang.reflect.MethodSignature; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.core.annotation.AnnotationUtils; +import org.springframework.core.annotation.Order; +import org.springframework.stereotype.Component; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.datasource.DynamicDataSourceContextHolder; + +/** + * 多数据源处理 + * + * @author ruoyi + */ +@Aspect +@Order(1) +@Component +public class DataSourceAspect +{ + protected Logger logger = LoggerFactory.getLogger(getClass()); + + @Pointcut("@annotation(com.ruoyi.common.annotation.DataSource)" + + "|| @within(com.ruoyi.common.annotation.DataSource)") + public void dsPointCut() + { + + } + + @Around("dsPointCut()") + public Object around(ProceedingJoinPoint point) throws Throwable + { + DataSource dataSource = getDataSource(point); + + if (StringUtils.isNotNull(dataSource)) + { + DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name()); + } + + try + { + return point.proceed(); + } + finally + { + // 销毁数据源 在执行方法之后 + DynamicDataSourceContextHolder.clearDataSourceType(); + } + } + + /** + * 获取需要切换的数据源 + */ + public DataSource getDataSource(ProceedingJoinPoint point) + { + MethodSignature signature = (MethodSignature) point.getSignature(); + DataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(), DataSource.class); + if (Objects.nonNull(dataSource)) + { + return dataSource; + } + + return AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/LogAspect.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/LogAspect.java new file mode 100644 index 0000000..7f5a257 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/LogAspect.java @@ -0,0 +1,217 @@ +package com.ruoyi.framework.aspectj; + +import java.util.Collection; +import java.util.Map; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.aspectj.lang.JoinPoint; +import org.aspectj.lang.annotation.AfterReturning; +import org.aspectj.lang.annotation.AfterThrowing; +import org.aspectj.lang.annotation.Aspect; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.stereotype.Component; +import org.springframework.validation.BindingResult; +import org.springframework.web.multipart.MultipartFile; +import org.springframework.web.servlet.HandlerMapping; +import com.alibaba.fastjson.JSON; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.enums.BusinessStatus; +import com.ruoyi.common.enums.HttpMethod; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.system.domain.SysOperLog; + +/** + * 操作日志记录处理 + * + * @author ruoyi + */ +@Aspect +@Component +public class LogAspect +{ + private static final Logger log = LoggerFactory.getLogger(LogAspect.class); + + /** + * 处理完请求后执行 + * + * @param joinPoint 切点 + */ + @AfterReturning(pointcut = "@annotation(controllerLog)", returning = "jsonResult") + public void doAfterReturning(JoinPoint joinPoint, Log controllerLog, Object jsonResult) + { + handleLog(joinPoint, controllerLog, null, jsonResult); + } + + /** + * 拦截异常操作 + * + * @param joinPoint 切点 + * @param e 异常 + */ + @AfterThrowing(value = "@annotation(controllerLog)", throwing = "e") + public void doAfterThrowing(JoinPoint joinPoint, Log controllerLog, Exception e) + { + handleLog(joinPoint, controllerLog, e, null); + } + + protected void handleLog(final JoinPoint joinPoint, Log controllerLog, final Exception e, Object jsonResult) + { + try + { + // 获取当前的用户 + LoginUser loginUser = SecurityUtils.getLoginUser(); + + // *========数据库日志=========*// + SysOperLog operLog = new SysOperLog(); + operLog.setStatus(BusinessStatus.SUCCESS.ordinal()); + // 请求的地址 + String ip = IpUtils.getIpAddr(ServletUtils.getRequest()); + operLog.setOperIp(ip); + operLog.setOperUrl(ServletUtils.getRequest().getRequestURI()); + if (loginUser != null) + { + operLog.setOperName(loginUser.getUsername()); + } + + if (e != null) + { + operLog.setStatus(BusinessStatus.FAIL.ordinal()); + operLog.setErrorMsg(StringUtils.substring(e.getMessage(), 0, 2000)); + } + // 设置方法名称 + String className = joinPoint.getTarget().getClass().getName(); + String methodName = joinPoint.getSignature().getName(); + operLog.setMethod(className + "." + methodName + "()"); + // 设置请求方式 + operLog.setRequestMethod(ServletUtils.getRequest().getMethod()); + // 处理设置注解上的参数 + getControllerMethodDescription(joinPoint, controllerLog, operLog, jsonResult); + // 保存数据库 + AsyncManager.me().execute(AsyncFactory.recordOper(operLog)); + } + catch (Exception exp) + { + // 记录本地异常日志 + log.error("==前置通知异常=="); + log.error("异常信息:{}", exp.getMessage()); + exp.printStackTrace(); + } + } + + /** + * 获取注解中对方法的描述信息 用于Controller层注解 + * + * @param log 日志 + * @param operLog 操作日志 + * @throws Exception + */ + public void getControllerMethodDescription(JoinPoint joinPoint, Log log, SysOperLog operLog, Object jsonResult) throws Exception + { + // 设置action动作 + operLog.setBusinessType(log.businessType().ordinal()); + // 设置标题 + operLog.setTitle(log.title()); + // 设置操作人类别 + operLog.setOperatorType(log.operatorType().ordinal()); + // 是否需要保存request,参数和值 + if (log.isSaveRequestData()) + { + // 获取参数的信息,传入到数据库中。 + setRequestValue(joinPoint, operLog); + } + // 是否需要保存response,参数和值 + if (log.isSaveResponseData() && StringUtils.isNotNull(jsonResult)) + { + operLog.setJsonResult(StringUtils.substring(JSON.toJSONString(jsonResult), 0, 2000)); + } + } + + /** + * 获取请求的参数,放到log中 + * + * @param operLog 操作日志 + * @throws Exception 异常 + */ + private void setRequestValue(JoinPoint joinPoint, SysOperLog operLog) throws Exception + { + String requestMethod = operLog.getRequestMethod(); + if (HttpMethod.PUT.name().equals(requestMethod) || HttpMethod.POST.name().equals(requestMethod)) + { + String params = argsArrayToString(joinPoint.getArgs()); + operLog.setOperParam(StringUtils.substring(params, 0, 2000)); + } + else + { + Map paramsMap = (Map) ServletUtils.getRequest().getAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE); + operLog.setOperParam(StringUtils.substring(paramsMap.toString(), 0, 2000)); + } + } + + /** + * 参数拼装 + */ + private String argsArrayToString(Object[] paramsArray) + { + String params = ""; + if (paramsArray != null && paramsArray.length > 0) + { + for (Object o : paramsArray) + { + if (StringUtils.isNotNull(o) && !isFilterObject(o)) + { + try + { + Object jsonObj = JSON.toJSON(o); + params += jsonObj.toString() + " "; + } + catch (Exception e) + { + } + } + } + } + return params.trim(); + } + + /** + * 判断是否需要过滤的对象。 + * + * @param o 对象信息。 + * @return 如果是需要过滤的对象,则返回true;否则返回false。 + */ + @SuppressWarnings("rawtypes") + public boolean isFilterObject(final Object o) + { + Class clazz = o.getClass(); + if (clazz.isArray()) + { + return clazz.getComponentType().isAssignableFrom(MultipartFile.class); + } + else if (Collection.class.isAssignableFrom(clazz)) + { + Collection collection = (Collection) o; + for (Object value : collection) + { + return value instanceof MultipartFile; + } + } + else if (Map.class.isAssignableFrom(clazz)) + { + Map map = (Map) o; + for (Object value : map.entrySet()) + { + Map.Entry entry = (Map.Entry) value; + return entry.getValue() instanceof MultipartFile; + } + } + return o instanceof MultipartFile || o instanceof HttpServletRequest || o instanceof HttpServletResponse + || o instanceof BindingResult; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java new file mode 100644 index 0000000..0d1b62f --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java @@ -0,0 +1,91 @@ +package com.ruoyi.framework.aspectj; + +import java.lang.reflect.Method; +import java.util.Collections; +import java.util.List; +import org.aspectj.lang.JoinPoint; +import org.aspectj.lang.annotation.Aspect; +import org.aspectj.lang.annotation.Before; +import org.aspectj.lang.reflect.MethodSignature; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.redis.core.RedisTemplate; +import org.springframework.data.redis.core.script.RedisScript; +import org.springframework.stereotype.Component; +import com.ruoyi.common.annotation.RateLimiter; +import com.ruoyi.common.enums.LimitType; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.ip.IpUtils; + +/** + * 限流处理 + * + * @author ruoyi + */ +@Aspect +@Component +public class RateLimiterAspect +{ + private static final Logger log = LoggerFactory.getLogger(RateLimiterAspect.class); + + private RedisTemplate redisTemplate; + + private RedisScript limitScript; + + @Autowired + public void setRedisTemplate1(RedisTemplate redisTemplate) + { + this.redisTemplate = redisTemplate; + } + + @Autowired + public void setLimitScript(RedisScript limitScript) + { + this.limitScript = limitScript; + } + + @Before("@annotation(rateLimiter)") + public void doBefore(JoinPoint point, RateLimiter rateLimiter) throws Throwable + { + String key = rateLimiter.key(); + int time = rateLimiter.time(); + int count = rateLimiter.count(); + + String combineKey = getCombineKey(rateLimiter, point); + List keys = Collections.singletonList(combineKey); + try + { + Long number = redisTemplate.execute(limitScript, keys, count, time); + if (StringUtils.isNull(number) || number.intValue() > count) + { + throw new ServiceException("访问过于频繁,请稍候再试"); + } + log.info("限制请求'{}',当前请求'{}',缓存key'{}'", count, number.intValue(), key); + } + catch (ServiceException e) + { + throw e; + } + catch (Exception e) + { + throw new RuntimeException("服务器限流异常,请稍候再试"); + } + } + + public String getCombineKey(RateLimiter rateLimiter, JoinPoint point) + { + StringBuffer stringBuffer = new StringBuffer(rateLimiter.key()); + if (rateLimiter.limitType() == LimitType.IP) + { + stringBuffer.append(IpUtils.getIpAddr(ServletUtils.getRequest())).append("-"); + } + MethodSignature signature = (MethodSignature) point.getSignature(); + Method method = signature.getMethod(); + Class targetClass = method.getDeclaringClass(); + stringBuffer.append(targetClass.getName()).append("-").append(method.getName()); + return stringBuffer.toString(); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ApplicationConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ApplicationConfig.java new file mode 100644 index 0000000..1d4dc1f --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ApplicationConfig.java @@ -0,0 +1,30 @@ +package com.ruoyi.framework.config; + +import java.util.TimeZone; +import org.mybatis.spring.annotation.MapperScan; +import org.springframework.boot.autoconfigure.jackson.Jackson2ObjectMapperBuilderCustomizer; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.context.annotation.EnableAspectJAutoProxy; + +/** + * 程序注解配置 + * + * @author ruoyi + */ +@Configuration +// 表示通过aop框架暴露该代理对象,AopContext能够访问 +@EnableAspectJAutoProxy(exposeProxy = true) +// 指定要扫描的Mapper类的包的路径 +@MapperScan("com.ruoyi.**.mapper") +public class ApplicationConfig +{ + /** + * 时区配置 + */ + @Bean + public Jackson2ObjectMapperBuilderCustomizer jacksonObjectMapperCustomization() + { + return jacksonObjectMapperBuilder -> jacksonObjectMapperBuilder.timeZone(TimeZone.getDefault()); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/CaptchaConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/CaptchaConfig.java new file mode 100644 index 0000000..43e78ae --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/CaptchaConfig.java @@ -0,0 +1,83 @@ +package com.ruoyi.framework.config; + +import java.util.Properties; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import com.google.code.kaptcha.impl.DefaultKaptcha; +import com.google.code.kaptcha.util.Config; +import static com.google.code.kaptcha.Constants.*; + +/** + * 验证码配置 + * + * @author ruoyi + */ +@Configuration +public class CaptchaConfig +{ + @Bean(name = "captchaProducer") + public DefaultKaptcha getKaptchaBean() + { + DefaultKaptcha defaultKaptcha = new DefaultKaptcha(); + Properties properties = new Properties(); + // 是否有边框 默认为true 我们可以自己设置yes,no + properties.setProperty(KAPTCHA_BORDER, "yes"); + // 验证码文本字符颜色 默认为Color.BLACK + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_COLOR, "black"); + // 验证码图片宽度 默认为200 + properties.setProperty(KAPTCHA_IMAGE_WIDTH, "160"); + // 验证码图片高度 默认为50 + properties.setProperty(KAPTCHA_IMAGE_HEIGHT, "60"); + // 验证码文本字符大小 默认为40 + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_SIZE, "38"); + // KAPTCHA_SESSION_KEY + properties.setProperty(KAPTCHA_SESSION_CONFIG_KEY, "kaptchaCode"); + // 验证码文本字符长度 默认为5 + properties.setProperty(KAPTCHA_TEXTPRODUCER_CHAR_LENGTH, "4"); + // 验证码文本字体样式 默认为new Font("Arial", 1, fontSize), new Font("Courier", 1, fontSize) + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_NAMES, "Arial,Courier"); + // 图片样式 水纹com.google.code.kaptcha.impl.WaterRipple 鱼眼com.google.code.kaptcha.impl.FishEyeGimpy 阴影com.google.code.kaptcha.impl.ShadowGimpy + properties.setProperty(KAPTCHA_OBSCURIFICATOR_IMPL, "com.google.code.kaptcha.impl.ShadowGimpy"); + Config config = new Config(properties); + defaultKaptcha.setConfig(config); + return defaultKaptcha; + } + + @Bean(name = "captchaProducerMath") + public DefaultKaptcha getKaptchaBeanMath() + { + DefaultKaptcha defaultKaptcha = new DefaultKaptcha(); + Properties properties = new Properties(); + // 是否有边框 默认为true 我们可以自己设置yes,no + properties.setProperty(KAPTCHA_BORDER, "yes"); + // 边框颜色 默认为Color.BLACK + properties.setProperty(KAPTCHA_BORDER_COLOR, "105,179,90"); + // 验证码文本字符颜色 默认为Color.BLACK + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_COLOR, "blue"); + // 验证码图片宽度 默认为200 + properties.setProperty(KAPTCHA_IMAGE_WIDTH, "160"); + // 验证码图片高度 默认为50 + properties.setProperty(KAPTCHA_IMAGE_HEIGHT, "60"); + // 验证码文本字符大小 默认为40 + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_SIZE, "35"); + // KAPTCHA_SESSION_KEY + properties.setProperty(KAPTCHA_SESSION_CONFIG_KEY, "kaptchaCodeMath"); + // 验证码文本生成器 + properties.setProperty(KAPTCHA_TEXTPRODUCER_IMPL, "com.ruoyi.framework.config.KaptchaTextCreator"); + // 验证码文本字符间距 默认为2 + properties.setProperty(KAPTCHA_TEXTPRODUCER_CHAR_SPACE, "3"); + // 验证码文本字符长度 默认为5 + properties.setProperty(KAPTCHA_TEXTPRODUCER_CHAR_LENGTH, "6"); + // 验证码文本字体样式 默认为new Font("Arial", 1, fontSize), new Font("Courier", 1, fontSize) + properties.setProperty(KAPTCHA_TEXTPRODUCER_FONT_NAMES, "Arial,Courier"); + // 验证码噪点颜色 默认为Color.BLACK + properties.setProperty(KAPTCHA_NOISE_COLOR, "white"); + // 干扰实现类 + properties.setProperty(KAPTCHA_NOISE_IMPL, "com.google.code.kaptcha.impl.NoNoise"); + // 图片样式 水纹com.google.code.kaptcha.impl.WaterRipple 鱼眼com.google.code.kaptcha.impl.FishEyeGimpy 阴影com.google.code.kaptcha.impl.ShadowGimpy + properties.setProperty(KAPTCHA_OBSCURIFICATOR_IMPL, "com.google.code.kaptcha.impl.ShadowGimpy"); + Config config = new Config(properties); + defaultKaptcha.setConfig(config); + return defaultKaptcha; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/DruidConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/DruidConfig.java new file mode 100644 index 0000000..0918e8e --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/DruidConfig.java @@ -0,0 +1,138 @@ +package com.ruoyi.framework.config; + +import java.io.IOException; +import java.util.HashMap; +import java.util.Map; +import javax.servlet.Filter; +import javax.servlet.FilterChain; +import javax.servlet.ServletException; +import javax.servlet.ServletRequest; +import javax.servlet.ServletResponse; +import javax.sql.DataSource; +import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.boot.web.servlet.FilterRegistrationBean; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.context.annotation.Primary; +import com.alibaba.druid.pool.DruidDataSource; +import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; +import com.alibaba.druid.spring.boot.autoconfigure.properties.DruidStatProperties; +import com.alibaba.druid.util.Utils; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.framework.config.properties.DruidProperties; +import com.ruoyi.framework.datasource.DynamicDataSource; + +/** + * druid 配置多数据源 + * + * @author ruoyi + */ +@Configuration +public class DruidConfig +{ + @Bean + @ConfigurationProperties("spring.datasource.druid.master") + public DataSource masterDataSource(DruidProperties druidProperties) + { + DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); + return druidProperties.dataSource(dataSource); + } + + @Bean + @ConfigurationProperties("spring.datasource.druid.slave") + @ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true") + public DataSource slaveDataSource(DruidProperties druidProperties) + { + DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); + return druidProperties.dataSource(dataSource); + } + + @Bean + @ConfigurationProperties("spring.datasource.druid.shardsource") + @ConditionalOnProperty(prefix = "spring.datasource.druid.shardsource", name = "enabled", havingValue = "true") + public DataSource shardDataSource(DruidProperties druidProperties) + { + DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); + return druidProperties.dataSource(dataSource); + } + + + + @Bean(name = "dynamicDataSource") + @Primary + public DynamicDataSource dataSource(DataSource masterDataSource) + { + Map targetDataSources = new HashMap<>(); + targetDataSources.put(DataSourceType.MASTER.name(), masterDataSource); + setDataSource(targetDataSources, DataSourceType.SLAVE.name(), "slaveDataSource"); + setDataSource(targetDataSources, DataSourceType.SHARDING.name(), "shardDataSource"); + return new DynamicDataSource(masterDataSource, targetDataSources); + } + + /** + * 设置数据源 + * + * @param targetDataSources 备选数据源集合 + * @param sourceName 数据源名称 + * @param beanName bean名称 + */ + public void setDataSource(Map targetDataSources, String sourceName, String beanName) + { + try + { + DataSource dataSource = SpringUtils.getBean(beanName); + targetDataSources.put(sourceName, dataSource); + } + catch (Exception e) + { + } + } + + /** + * 去除监控页面底部的广告 + */ + @SuppressWarnings({ "rawtypes", "unchecked" }) + @Bean + @ConditionalOnProperty(name = "spring.datasource.druid.statViewServlet.enabled", havingValue = "true") + public FilterRegistrationBean removeDruidFilterRegistrationBean(DruidStatProperties properties) + { + // 获取web监控页面的参数 + DruidStatProperties.StatViewServlet config = properties.getStatViewServlet(); + // 提取common.js的配置路径 + String pattern = config.getUrlPattern() != null ? config.getUrlPattern() : "/druid/*"; + String commonJsPattern = pattern.replaceAll("\\*", "js/common.js"); + final String filePath = "support/http/resources/js/common.js"; + // 创建filter进行过滤 + Filter filter = new Filter() + { + @Override + public void init(javax.servlet.FilterConfig filterConfig) throws ServletException + { + } + @Override + public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) + throws IOException, ServletException + { + chain.doFilter(request, response); + // 重置缓冲区,响应头不会被重置 + response.resetBuffer(); + // 获取common.js + String text = Utils.readFromResource(filePath); + // 正则替换banner, 除去底部的广告信息 + text = text.replaceAll("
", ""); + text = text.replaceAll("powered.*?shrek.wang", ""); + response.getWriter().write(text); + } + @Override + public void destroy() + { + } + }; + FilterRegistrationBean registrationBean = new FilterRegistrationBean(); + registrationBean.setFilter(filter); + registrationBean.addUrlPatterns(commonJsPattern); + return registrationBean; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FastJson2JsonRedisSerializer.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FastJson2JsonRedisSerializer.java new file mode 100644 index 0000000..59812ea --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FastJson2JsonRedisSerializer.java @@ -0,0 +1,71 @@ +package com.ruoyi.framework.config; + +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.serializer.SerializerFeature; +import com.fasterxml.jackson.databind.JavaType; +import com.fasterxml.jackson.databind.ObjectMapper; +import com.fasterxml.jackson.databind.type.TypeFactory; +import org.springframework.data.redis.serializer.RedisSerializer; +import org.springframework.data.redis.serializer.SerializationException; +import com.alibaba.fastjson.parser.ParserConfig; +import org.springframework.util.Assert; +import java.nio.charset.Charset; + +/** + * Redis使用FastJson序列化 + * + * @author ruoyi + */ +public class FastJson2JsonRedisSerializer implements RedisSerializer +{ + @SuppressWarnings("unused") + private ObjectMapper objectMapper = new ObjectMapper(); + + public static final Charset DEFAULT_CHARSET = Charset.forName("UTF-8"); + + private Class clazz; + + static + { + ParserConfig.getGlobalInstance().setAutoTypeSupport(true); + } + + public FastJson2JsonRedisSerializer(Class clazz) + { + super(); + this.clazz = clazz; + } + + @Override + public byte[] serialize(T t) throws SerializationException + { + if (t == null) + { + return new byte[0]; + } + return JSON.toJSONString(t, SerializerFeature.WriteClassName).getBytes(DEFAULT_CHARSET); + } + + @Override + public T deserialize(byte[] bytes) throws SerializationException + { + if (bytes == null || bytes.length <= 0) + { + return null; + } + String str = new String(bytes, DEFAULT_CHARSET); + + return JSON.parseObject(str, clazz); + } + + public void setObjectMapper(ObjectMapper objectMapper) + { + Assert.notNull(objectMapper, "'objectMapper' must not be null"); + this.objectMapper = objectMapper; + } + + protected JavaType getJavaType(Class clazz) + { + return TypeFactory.defaultInstance().constructType(clazz); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FilterConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FilterConfig.java new file mode 100644 index 0000000..bb14c04 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/FilterConfig.java @@ -0,0 +1,58 @@ +package com.ruoyi.framework.config; + +import java.util.HashMap; +import java.util.Map; +import javax.servlet.DispatcherType; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; +import org.springframework.boot.web.servlet.FilterRegistrationBean; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import com.ruoyi.common.filter.RepeatableFilter; +import com.ruoyi.common.filter.XssFilter; +import com.ruoyi.common.utils.StringUtils; + +/** + * Filter配置 + * + * @author ruoyi + */ +@Configuration +public class FilterConfig +{ + @Value("${xss.excludes}") + private String excludes; + + @Value("${xss.urlPatterns}") + private String urlPatterns; + + @SuppressWarnings({ "rawtypes", "unchecked" }) + @Bean + @ConditionalOnProperty(value = "xss.enabled", havingValue = "true") + public FilterRegistrationBean xssFilterRegistration() + { + FilterRegistrationBean registration = new FilterRegistrationBean(); + registration.setDispatcherTypes(DispatcherType.REQUEST); + registration.setFilter(new XssFilter()); + registration.addUrlPatterns(StringUtils.split(urlPatterns, ",")); + registration.setName("xssFilter"); + registration.setOrder(FilterRegistrationBean.HIGHEST_PRECEDENCE); + Map initParameters = new HashMap(); + initParameters.put("excludes", excludes); + registration.setInitParameters(initParameters); + return registration; + } + + @SuppressWarnings({ "rawtypes", "unchecked" }) + @Bean + public FilterRegistrationBean someFilterRegistration() + { + FilterRegistrationBean registration = new FilterRegistrationBean(); + registration.setFilter(new RepeatableFilter()); + registration.addUrlPatterns("/*"); + registration.setName("repeatableFilter"); + registration.setOrder(FilterRegistrationBean.LOWEST_PRECEDENCE); + return registration; + } + +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/KaptchaTextCreator.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/KaptchaTextCreator.java new file mode 100644 index 0000000..3e74580 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/KaptchaTextCreator.java @@ -0,0 +1,75 @@ +package com.ruoyi.framework.config; + +import java.util.Random; +import com.google.code.kaptcha.text.impl.DefaultTextCreator; + +/** + * 验证码文本生成器 + * + * @author ruoyi + */ +public class KaptchaTextCreator extends DefaultTextCreator +{ + private static final String[] CNUMBERS = "0,1,2,3,4,5,6,7,8,9,10".split(","); + + @Override + public String getText() + { + Integer result = 0; + Random random = new Random(); + int x = random.nextInt(10); + int y = random.nextInt(10); + StringBuilder suChinese = new StringBuilder(); + int randomoperands = (int) Math.round(Math.random() * 2); + if (randomoperands == 0) + { + result = x * y; + suChinese.append(CNUMBERS[x]); + suChinese.append("*"); + suChinese.append(CNUMBERS[y]); + } + else if (randomoperands == 1) + { + if (!(x == 0) && y % x == 0) + { + result = y / x; + suChinese.append(CNUMBERS[y]); + suChinese.append("/"); + suChinese.append(CNUMBERS[x]); + } + else + { + result = x + y; + suChinese.append(CNUMBERS[x]); + suChinese.append("+"); + suChinese.append(CNUMBERS[y]); + } + } + else if (randomoperands == 2) + { + if (x >= y) + { + result = x - y; + suChinese.append(CNUMBERS[x]); + suChinese.append("-"); + suChinese.append(CNUMBERS[y]); + } + else + { + result = y - x; + suChinese.append(CNUMBERS[y]); + suChinese.append("-"); + suChinese.append(CNUMBERS[x]); + } + } + else + { + result = x + y; + suChinese.append(CNUMBERS[x]); + suChinese.append("+"); + suChinese.append(CNUMBERS[y]); + } + suChinese.append("=?@" + result); + return suChinese.toString(); + } +} \ No newline at end of file diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/MybatisPlusConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/MybatisPlusConfig.java new file mode 100644 index 0000000..f712180 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/MybatisPlusConfig.java @@ -0,0 +1,62 @@ +package com.ruoyi.framework.config; + +import com.baomidou.mybatisplus.annotation.DbType; +import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor; +import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor; +import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor; +import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.transaction.annotation.EnableTransactionManagement; + +/** + * Mybatis Plus 配置 + * + * @author ruoyi + */ +@EnableTransactionManagement(proxyTargetClass = true) +@Configuration +public class MybatisPlusConfig +{ + @Bean + public MybatisPlusInterceptor mybatisPlusInterceptor() + { + MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); + // 分页插件 + interceptor.addInnerInterceptor(paginationInnerInterceptor()); + // 乐观锁插件 + interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor()); + // 阻断插件 + interceptor.addInnerInterceptor(blockAttackInnerInterceptor()); + return interceptor; + } + + /** + * 分页插件,自动识别数据库类型 https://baomidou.com/guide/interceptor-pagination.html + */ + public PaginationInnerInterceptor paginationInnerInterceptor() + { + PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor(); + // 设置数据库类型为mysql + paginationInnerInterceptor.setDbType(DbType.MYSQL); + // 设置最大单页限制数量,默认 500 条,-1 不受限制 + paginationInnerInterceptor.setMaxLimit(-1L); + return paginationInnerInterceptor; + } + + /** + * 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html + */ + public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor() + { + return new OptimisticLockerInnerInterceptor(); + } + + /** + * 如果是对全表的删除或更新操作,就会终止该操作 https://baomidou.com/guide/interceptor-block-attack.html + */ + public BlockAttackInnerInterceptor blockAttackInnerInterceptor() + { + return new BlockAttackInnerInterceptor(); + } +} \ No newline at end of file diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java new file mode 100644 index 0000000..833f219 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java @@ -0,0 +1,79 @@ +package com.ruoyi.framework.config; + +import org.springframework.cache.annotation.CachingConfigurerSupport; +import org.springframework.cache.annotation.EnableCaching; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.data.redis.connection.RedisConnectionFactory; +import org.springframework.data.redis.core.RedisTemplate; +import org.springframework.data.redis.core.script.DefaultRedisScript; +import org.springframework.data.redis.serializer.StringRedisSerializer; +import com.fasterxml.jackson.annotation.JsonAutoDetect; +import com.fasterxml.jackson.annotation.JsonTypeInfo; +import com.fasterxml.jackson.annotation.PropertyAccessor; +import com.fasterxml.jackson.databind.ObjectMapper; +import com.fasterxml.jackson.databind.jsontype.impl.LaissezFaireSubTypeValidator; + +/** + * redis配置 + * + * @author ruoyi + */ +@Configuration +@EnableCaching +public class RedisConfig extends CachingConfigurerSupport +{ + @Bean + @SuppressWarnings(value = { "unchecked", "rawtypes" }) + public RedisTemplate redisTemplate(RedisConnectionFactory connectionFactory) + { + RedisTemplate template = new RedisTemplate<>(); + template.setConnectionFactory(connectionFactory); + + FastJson2JsonRedisSerializer serializer = new FastJson2JsonRedisSerializer(Object.class); + + ObjectMapper mapper = new ObjectMapper(); + mapper.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY); + mapper.activateDefaultTyping(LaissezFaireSubTypeValidator.instance, ObjectMapper.DefaultTyping.NON_FINAL, JsonTypeInfo.As.PROPERTY); + serializer.setObjectMapper(mapper); + + // 使用StringRedisSerializer来序列化和反序列化redis的key值 + template.setKeySerializer(new StringRedisSerializer()); + template.setValueSerializer(serializer); + + // Hash的key也采用StringRedisSerializer的序列化方式 + template.setHashKeySerializer(new StringRedisSerializer()); + template.setHashValueSerializer(serializer); + + template.afterPropertiesSet(); + return template; + } + + @Bean + public DefaultRedisScript limitScript() + { + DefaultRedisScript redisScript = new DefaultRedisScript<>(); + redisScript.setScriptText(limitScriptText()); + redisScript.setResultType(Long.class); + return redisScript; + } + + /** + * 限流脚本 + */ + private String limitScriptText() + { + return "local key = KEYS[1]\n" + + "local count = tonumber(ARGV[1])\n" + + "local time = tonumber(ARGV[2])\n" + + "local current = redis.call('get', key);\n" + + "if current and tonumber(current) > count then\n" + + " return tonumber(current);\n" + + "end\n" + + "current = redis.call('incr', key)\n" + + "if tonumber(current) == 1 then\n" + + " redis.call('expire', key, time)\n" + + "end\n" + + "return tonumber(current);"; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ResourcesConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ResourcesConfig.java new file mode 100644 index 0000000..50ad496 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ResourcesConfig.java @@ -0,0 +1,70 @@ +package com.ruoyi.framework.config; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.web.cors.CorsConfiguration; +import org.springframework.web.cors.UrlBasedCorsConfigurationSource; +import org.springframework.web.filter.CorsFilter; +import org.springframework.web.servlet.config.annotation.InterceptorRegistry; +import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry; +import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.framework.interceptor.RepeatSubmitInterceptor; + +/** + * 通用配置 + * + * @author ruoyi + */ +@Configuration +public class ResourcesConfig implements WebMvcConfigurer +{ + @Autowired + private RepeatSubmitInterceptor repeatSubmitInterceptor; + + @Override + public void addResourceHandlers(ResourceHandlerRegistry registry) + { + /** 本地文件上传路径 */ + registry.addResourceHandler(Constants.RESOURCE_PREFIX + "/**") + .addResourceLocations("file:" + RuoYiConfig.getProfile() + "/"); + + /** swagger配置 */ + registry.addResourceHandler("/swagger-ui/**") + .addResourceLocations("classpath:/META-INF/resources/webjars/springfox-swagger-ui/"); + } + + /** + * 自定义拦截规则 + */ + @Override + public void addInterceptors(InterceptorRegistry registry) + { + registry.addInterceptor(repeatSubmitInterceptor).addPathPatterns("/**"); + } + + /** + * 跨域配置 + */ + @Bean + public CorsFilter corsFilter() + { + CorsConfiguration config = new CorsConfiguration(); + config.setAllowCredentials(true); + // 设置访问源地址 + config.addAllowedOriginPattern("*"); + // 设置访问源请求头 + config.addAllowedHeader("*"); + // 设置访问源请求方法 + config.addAllowedMethod("*"); + // 有效期 1800秒 + config.setMaxAge(1800L); + // 添加映射路径,拦截一切请求 + UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource(); + source.registerCorsConfiguration("/**", config); + // 返回新的CorsFilter + return new CorsFilter(source); + } +} \ No newline at end of file diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java new file mode 100644 index 0000000..458a8cf --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java @@ -0,0 +1,145 @@ +package com.ruoyi.framework.config; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Bean; +import org.springframework.http.HttpMethod; +import org.springframework.security.authentication.AuthenticationManager; +import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder; +import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity; +import org.springframework.security.config.annotation.web.builders.HttpSecurity; +import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter; +import org.springframework.security.config.http.SessionCreationPolicy; +import org.springframework.security.core.userdetails.UserDetailsService; +import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder; +import org.springframework.security.web.authentication.UsernamePasswordAuthenticationFilter; +import org.springframework.security.web.authentication.logout.LogoutFilter; +import org.springframework.web.filter.CorsFilter; +import com.ruoyi.framework.security.filter.JwtAuthenticationTokenFilter; +import com.ruoyi.framework.security.handle.AuthenticationEntryPointImpl; +import com.ruoyi.framework.security.handle.LogoutSuccessHandlerImpl; + +/** + * spring security配置 + * + * @author ruoyi + */ +@EnableGlobalMethodSecurity(prePostEnabled = true, securedEnabled = true) +public class SecurityConfig extends WebSecurityConfigurerAdapter +{ + /** + * 自定义用户认证逻辑 + */ + @Autowired + private UserDetailsService userDetailsService; + + /** + * 认证失败处理类 + */ + @Autowired + private AuthenticationEntryPointImpl unauthorizedHandler; + + /** + * 退出处理类 + */ + @Autowired + private LogoutSuccessHandlerImpl logoutSuccessHandler; + + /** + * token认证过滤器 + */ + @Autowired + private JwtAuthenticationTokenFilter authenticationTokenFilter; + + /** + * 跨域过滤器 + */ + @Autowired + private CorsFilter corsFilter; + + /** + * 解决 无法直接注入 AuthenticationManager + * + * @return + * @throws Exception + */ + @Bean + @Override + public AuthenticationManager authenticationManagerBean() throws Exception + { + return super.authenticationManagerBean(); + } + + /** + * anyRequest | 匹配所有请求路径 + * access | SpringEl表达式结果为true时可以访问 + * anonymous | 匿名可以访问 + * denyAll | 用户不能访问 + * fullyAuthenticated | 用户完全认证可以访问(非remember-me下自动登录) + * hasAnyAuthority | 如果有参数,参数表示权限,则其中任何一个权限可以访问 + * hasAnyRole | 如果有参数,参数表示角色,则其中任何一个角色可以访问 + * hasAuthority | 如果有参数,参数表示权限,则其权限可以访问 + * hasIpAddress | 如果有参数,参数表示IP地址,如果用户IP和参数匹配,则可以访问 + * hasRole | 如果有参数,参数表示角色,则其角色可以访问 + * permitAll | 用户可以任意访问 + * rememberMe | 允许通过remember-me登录的用户访问 + * authenticated | 用户登录后可访问 + */ + @Override + protected void configure(HttpSecurity httpSecurity) throws Exception + { + httpSecurity + // CSRF禁用,因为不使用session + .csrf().disable() + // 认证失败处理类 + .exceptionHandling().authenticationEntryPoint(unauthorizedHandler).and() + // 基于token,所以不需要session + .sessionManagement().sessionCreationPolicy(SessionCreationPolicy.STATELESS).and() + // 过滤请求 + .authorizeRequests() + // 对于登录login 注册register 验证码captchaImage 允许匿名访问 + .antMatchers("/login","/loginSso", "/register", "/captchaImage").anonymous() + .antMatchers( + HttpMethod.GET, + "/", + "/*.html", + "/**/*.html", + "/**/*.css", + "/**/*.js", + "/profile/**" + ).permitAll() + .antMatchers("/swagger-ui.html").anonymous() + .antMatchers("/swagger-resources/**").anonymous() + .antMatchers("/webjars/**").anonymous() + .antMatchers("/*/api-docs").anonymous() + .antMatchers("/druid/**").anonymous() + .antMatchers("/api/**").anonymous() + // 除上面外的所有请求全部需要鉴权认证 + .anyRequest().authenticated() + .and() + .headers().frameOptions().disable(); + httpSecurity.logout().logoutUrl("/logout").logoutSuccessHandler(logoutSuccessHandler); + // 添加JWT filter + httpSecurity.addFilterBefore(authenticationTokenFilter, UsernamePasswordAuthenticationFilter.class); + // 添加CORS filter + httpSecurity.addFilterBefore(corsFilter, JwtAuthenticationTokenFilter.class); + httpSecurity.addFilterBefore(corsFilter, LogoutFilter.class); + } + + /** + * 强散列哈希加密实现 + */ + @Bean + public BCryptPasswordEncoder bCryptPasswordEncoder() + { + return new BCryptPasswordEncoder(); + } + + /** + * 身份认证接口 + */ + @Override + protected void configure(AuthenticationManagerBuilder auth) throws Exception + { + auth.userDetailsService(userDetailsService).passwordEncoder(bCryptPasswordEncoder()); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ServerConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ServerConfig.java new file mode 100644 index 0000000..b5b7de3 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ServerConfig.java @@ -0,0 +1,32 @@ +package com.ruoyi.framework.config; + +import javax.servlet.http.HttpServletRequest; +import org.springframework.stereotype.Component; +import com.ruoyi.common.utils.ServletUtils; + +/** + * 服务相关配置 + * + * @author ruoyi + */ +@Component +public class ServerConfig +{ + /** + * 获取完整的请求路径,包括:域名,端口,上下文访问路径 + * + * @return 服务地址 + */ + public String getUrl() + { + HttpServletRequest request = ServletUtils.getRequest(); + return getDomain(request); + } + + public static String getDomain(HttpServletRequest request) + { + StringBuffer url = request.getRequestURL(); + String contextPath = request.getServletContext().getContextPath(); + return url.delete(url.length() - request.getRequestURI().length(), url.length()).append(contextPath).toString(); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SlaveDataSourceConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SlaveDataSourceConfig.java new file mode 100644 index 0000000..b07ebf2 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/SlaveDataSourceConfig.java @@ -0,0 +1,50 @@ +//package com.ruoyi.framework.config; +// +//import com.alibaba.druid.pool.DruidDataSource; +//import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; +//import com.ruoyi.framework.config.properties.DruidProperties; +//import org.springframework.beans.factory.annotation.Qualifier; +//import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; +//import org.springframework.boot.context.properties.ConfigurationProperties; +//import org.springframework.context.annotation.Bean; +//import org.springframework.context.annotation.Configuration; +// +//import javax.sql.DataSource; +//import java.sql.SQLException; +//import java.util.HashMap; +//import java.util.Map; +//import java.util.Properties; +// +///** +// * @Author al +// * @Date 2024/12/27 14:30 +// * @Description: TODO +// * @Version +// */ +/////** +//// * sharding 配置信息 +//// * +//// * @author ruoyi +//// */ +//@Configuration +//public class SlaveDataSourceConfig +//{ +// @Bean +// @ConfigurationProperties("spring.datasource.druid.shardsource") +// @ConditionalOnProperty(prefix = "spring.datasource.druid.shardsource", name = "enabled", havingValue = "true") +// public DataSource shardDataSource(DruidProperties druidProperties) +// { +// DruidDataSource dataSource = DruidDataSourceBuilder.create().build(); +// return druidProperties.dataSource(dataSource); +// } +// +// /** +// * 系统参数配置 +// */ +// private Properties getProperties() +// { +// Properties shardingProperties = new Properties(); +// shardingProperties.put("sql.show", true); +// return shardingProperties; +// } +//} \ No newline at end of file diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ThreadPoolConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ThreadPoolConfig.java new file mode 100644 index 0000000..7840141 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/ThreadPoolConfig.java @@ -0,0 +1,63 @@ +package com.ruoyi.framework.config; + +import com.ruoyi.common.utils.Threads; +import org.apache.commons.lang3.concurrent.BasicThreadFactory; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; +import java.util.concurrent.ScheduledExecutorService; +import java.util.concurrent.ScheduledThreadPoolExecutor; +import java.util.concurrent.ThreadPoolExecutor; + +/** + * 线程池配置 + * + * @author ruoyi + **/ +@Configuration +public class ThreadPoolConfig +{ + // 核心线程池大小 + private int corePoolSize = 50; + + // 最大可创建的线程数 + private int maxPoolSize = 200; + + // 队列最大长度 + private int queueCapacity = 1000; + + // 线程池维护线程所允许的空闲时间 + private int keepAliveSeconds = 300; + + @Bean(name = "threadPoolTaskExecutor") + public ThreadPoolTaskExecutor threadPoolTaskExecutor() + { + ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); + executor.setMaxPoolSize(maxPoolSize); + executor.setCorePoolSize(corePoolSize); + executor.setQueueCapacity(queueCapacity); + executor.setKeepAliveSeconds(keepAliveSeconds); + // 线程池对拒绝任务(无线程可用)的处理策略 + executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); + return executor; + } + + /** + * 执行周期性或定时任务 + */ + @Bean(name = "scheduledExecutorService") + protected ScheduledExecutorService scheduledExecutorService() + { + return new ScheduledThreadPoolExecutor(corePoolSize, + new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(), + new ThreadPoolExecutor.CallerRunsPolicy()) + { + @Override + protected void afterExecute(Runnable r, Throwable t) + { + super.afterExecute(r, t); + Threads.printException(r, t); + } + }; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/properties/DruidProperties.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/properties/DruidProperties.java new file mode 100644 index 0000000..84f7e00 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/properties/DruidProperties.java @@ -0,0 +1,77 @@ +package com.ruoyi.framework.config.properties; + +import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.annotation.Configuration; +import com.alibaba.druid.pool.DruidDataSource; + +/** + * druid 配置属性 + * + * @author ruoyi + */ +@Configuration +public class DruidProperties +{ + @Value("${spring.datasource.druid.initialSize}") + private int initialSize; + + @Value("${spring.datasource.druid.minIdle}") + private int minIdle; + + @Value("${spring.datasource.druid.maxActive}") + private int maxActive; + + @Value("${spring.datasource.druid.maxWait}") + private int maxWait; + + @Value("${spring.datasource.druid.timeBetweenEvictionRunsMillis}") + private int timeBetweenEvictionRunsMillis; + + @Value("${spring.datasource.druid.minEvictableIdleTimeMillis}") + private int minEvictableIdleTimeMillis; + + @Value("${spring.datasource.druid.maxEvictableIdleTimeMillis}") + private int maxEvictableIdleTimeMillis; + + @Value("${spring.datasource.druid.validationQuery}") + private String validationQuery; + + @Value("${spring.datasource.druid.testWhileIdle}") + private boolean testWhileIdle; + + @Value("${spring.datasource.druid.testOnBorrow}") + private boolean testOnBorrow; + + @Value("${spring.datasource.druid.testOnReturn}") + private boolean testOnReturn; + + public DruidDataSource dataSource(DruidDataSource datasource) + { + /** 配置初始化大小、最小、最大 */ + datasource.setInitialSize(initialSize); + datasource.setMaxActive(maxActive); + datasource.setMinIdle(minIdle); + + /** 配置获取连接等待超时的时间 */ + datasource.setMaxWait(maxWait); + + /** 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 */ + datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis); + + /** 配置一个连接在池中最小、最大生存的时间,单位是毫秒 */ + datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis); + datasource.setMaxEvictableIdleTimeMillis(maxEvictableIdleTimeMillis); + + /** + * 用来检测连接是否有效的sql,要求是一个查询语句,常用select 'x'。如果validationQuery为null,testOnBorrow、testOnReturn、testWhileIdle都不会起作用。 + */ + datasource.setValidationQuery(validationQuery); + /** 建议配置为true,不影响性能,并且保证安全性。申请连接的时候检测,如果空闲时间大于timeBetweenEvictionRunsMillis,执行validationQuery检测连接是否有效。 */ + datasource.setTestWhileIdle(testWhileIdle); + /** 申请连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 */ + datasource.setTestOnBorrow(testOnBorrow); + /** 归还连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 */ + datasource.setTestOnReturn(testOnReturn); + return datasource; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSource.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSource.java new file mode 100644 index 0000000..e70b8cf --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSource.java @@ -0,0 +1,26 @@ +package com.ruoyi.framework.datasource; + +import java.util.Map; +import javax.sql.DataSource; +import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; + +/** + * 动态数据源 + * + * @author ruoyi + */ +public class DynamicDataSource extends AbstractRoutingDataSource +{ + public DynamicDataSource(DataSource defaultTargetDataSource, Map targetDataSources) + { + super.setDefaultTargetDataSource(defaultTargetDataSource); + super.setTargetDataSources(targetDataSources); + super.afterPropertiesSet(); + } + + @Override + protected Object determineCurrentLookupKey() + { + return DynamicDataSourceContextHolder.getDataSourceType(); + } +} \ No newline at end of file diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSourceContextHolder.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSourceContextHolder.java new file mode 100644 index 0000000..3572db9 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/datasource/DynamicDataSourceContextHolder.java @@ -0,0 +1,45 @@ +package com.ruoyi.framework.datasource; + +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** + * 数据源切换处理 + * + * @author ruoyi + */ +public class DynamicDataSourceContextHolder +{ + public static final Logger log = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class); + + /** + * 使用ThreadLocal维护变量,ThreadLocal为每个使用该变量的线程提供独立的变量副本, + * 所以每一个线程都可以独立地改变自己的副本,而不会影响其它线程所对应的副本。 + */ + private static final ThreadLocal CONTEXT_HOLDER = new ThreadLocal<>(); + + /** + * 设置数据源的变量 + */ + public static void setDataSourceType(String dsType) + { + log.info("切换到{}数据源", dsType); + CONTEXT_HOLDER.set(dsType); + } + + /** + * 获得数据源的变量 + */ + public static String getDataSourceType() + { + return CONTEXT_HOLDER.get(); + } + + /** + * 清空数据源变量 + */ + public static void clearDataSourceType() + { + CONTEXT_HOLDER.remove(); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/RepeatSubmitInterceptor.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/RepeatSubmitInterceptor.java new file mode 100644 index 0000000..7575748 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/RepeatSubmitInterceptor.java @@ -0,0 +1,55 @@ +package com.ruoyi.framework.interceptor; + +import java.lang.reflect.Method; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.springframework.stereotype.Component; +import org.springframework.web.method.HandlerMethod; +import org.springframework.web.servlet.HandlerInterceptor; +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.annotation.RepeatSubmit; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.utils.ServletUtils; + +/** + * 防止重复提交拦截器 + * + * @author ruoyi + */ +@Component +public abstract class RepeatSubmitInterceptor implements HandlerInterceptor +{ + @Override + public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception + { + if (handler instanceof HandlerMethod) + { + HandlerMethod handlerMethod = (HandlerMethod) handler; + Method method = handlerMethod.getMethod(); + RepeatSubmit annotation = method.getAnnotation(RepeatSubmit.class); + if (annotation != null) + { + if (this.isRepeatSubmit(request, annotation)) + { + AjaxResult ajaxResult = AjaxResult.error(annotation.message()); + ServletUtils.renderString(response, JSONObject.toJSONString(ajaxResult)); + return false; + } + } + return true; + } + else + { + return true; + } + } + + /** + * 验证是否重复提交由子类实现具体的防重复提交的规则 + * + * @param request + * @return + * @throws Exception + */ + public abstract boolean isRepeatSubmit(HttpServletRequest request, RepeatSubmit annotation); +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/impl/SameUrlDataInterceptor.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/impl/SameUrlDataInterceptor.java new file mode 100644 index 0000000..0e1ff68 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/impl/SameUrlDataInterceptor.java @@ -0,0 +1,110 @@ +package com.ruoyi.framework.interceptor.impl; + +import java.util.HashMap; +import java.util.Map; +import java.util.concurrent.TimeUnit; +import javax.servlet.http.HttpServletRequest; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Component; +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.annotation.RepeatSubmit; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.filter.RepeatedlyRequestWrapper; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.http.HttpHelper; +import com.ruoyi.framework.interceptor.RepeatSubmitInterceptor; + +/** + * 判断请求url和数据是否和上一次相同, + * 如果和上次相同,则是重复提交表单。 有效时间为10秒内。 + * + * @author ruoyi + */ +@Component +public class SameUrlDataInterceptor extends RepeatSubmitInterceptor +{ + public final String REPEAT_PARAMS = "repeatParams"; + + public final String REPEAT_TIME = "repeatTime"; + + // 令牌自定义标识 + @Value("${token.header}") + private String header; + + @Autowired + private RedisCache redisCache; + + @SuppressWarnings("unchecked") + @Override + public boolean isRepeatSubmit(HttpServletRequest request, RepeatSubmit annotation) + { + String nowParams = ""; + if (request instanceof RepeatedlyRequestWrapper) + { + RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request; + nowParams = HttpHelper.getBodyString(repeatedlyRequest); + } + + // body参数为空,获取Parameter的数据 + if (StringUtils.isEmpty(nowParams)) + { + nowParams = JSONObject.toJSONString(request.getParameterMap()); + } + Map nowDataMap = new HashMap(); + nowDataMap.put(REPEAT_PARAMS, nowParams); + nowDataMap.put(REPEAT_TIME, System.currentTimeMillis()); + + // 请求地址(作为存放cache的key值) + String url = request.getRequestURI(); + + // 唯一值(没有消息头则使用请求地址) + String submitKey = StringUtils.trimToEmpty(request.getHeader(header)); + + // 唯一标识(指定key + url + 消息头) + String cacheRepeatKey = Constants.REPEAT_SUBMIT_KEY + url + submitKey; + + Object sessionObj = redisCache.getCacheObject(cacheRepeatKey); + if (sessionObj != null) + { + Map sessionMap = (Map) sessionObj; + if (sessionMap.containsKey(url)) + { + Map preDataMap = (Map) sessionMap.get(url); + if (compareParams(nowDataMap, preDataMap) && compareTime(nowDataMap, preDataMap, annotation.interval())) + { + return true; + } + } + } + Map cacheMap = new HashMap(); + cacheMap.put(url, nowDataMap); + redisCache.setCacheObject(cacheRepeatKey, cacheMap, annotation.interval(), TimeUnit.MILLISECONDS); + return false; + } + + /** + * 判断参数是否相同 + */ + private boolean compareParams(Map nowMap, Map preMap) + { + String nowParams = (String) nowMap.get(REPEAT_PARAMS); + String preParams = (String) preMap.get(REPEAT_PARAMS); + return nowParams.equals(preParams); + } + + /** + * 判断两次间隔时间 + */ + private boolean compareTime(Map nowMap, Map preMap, int interval) + { + long time1 = (Long) nowMap.get(REPEAT_TIME); + long time2 = (Long) preMap.get(REPEAT_TIME); + if ((time1 - time2) < interval) + { + return true; + } + return false; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/AsyncManager.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/AsyncManager.java new file mode 100644 index 0000000..7387a02 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/AsyncManager.java @@ -0,0 +1,55 @@ +package com.ruoyi.framework.manager; + +import java.util.TimerTask; +import java.util.concurrent.ScheduledExecutorService; +import java.util.concurrent.TimeUnit; +import com.ruoyi.common.utils.Threads; +import com.ruoyi.common.utils.spring.SpringUtils; + +/** + * 异步任务管理器 + * + * @author ruoyi + */ +public class AsyncManager +{ + /** + * 操作延迟10毫秒 + */ + private final int OPERATE_DELAY_TIME = 10; + + /** + * 异步操作任务调度线程池 + */ + private ScheduledExecutorService executor = SpringUtils.getBean("scheduledExecutorService"); + + /** + * 单例模式 + */ + private AsyncManager(){} + + private static AsyncManager me = new AsyncManager(); + + public static AsyncManager me() + { + return me; + } + + /** + * 执行任务 + * + * @param task 任务 + */ + public void execute(TimerTask task) + { + executor.schedule(task, OPERATE_DELAY_TIME, TimeUnit.MILLISECONDS); + } + + /** + * 停止任务线程池 + */ + public void shutdown() + { + Threads.shutdownAndAwaitTermination(executor); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/ShutdownManager.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/ShutdownManager.java new file mode 100644 index 0000000..e36ca3c --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/ShutdownManager.java @@ -0,0 +1,39 @@ +package com.ruoyi.framework.manager; + +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.stereotype.Component; +import javax.annotation.PreDestroy; + +/** + * 确保应用退出时能关闭后台线程 + * + * @author ruoyi + */ +@Component +public class ShutdownManager +{ + private static final Logger logger = LoggerFactory.getLogger("sys-user"); + + @PreDestroy + public void destroy() + { + shutdownAsyncManager(); + } + + /** + * 停止异步执行任务 + */ + private void shutdownAsyncManager() + { + try + { + logger.info("====关闭后台任务任务线程池===="); + AsyncManager.me().shutdown(); + } + catch (Exception e) + { + logger.error(e.getMessage(), e); + } + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/factory/AsyncFactory.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/factory/AsyncFactory.java new file mode 100644 index 0000000..23d0230 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/manager/factory/AsyncFactory.java @@ -0,0 +1,102 @@ +package com.ruoyi.framework.manager.factory; + +import java.util.TimerTask; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.utils.LogUtils; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.ip.AddressUtils; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.system.domain.SysLogininfor; +import com.ruoyi.system.domain.SysOperLog; +import com.ruoyi.system.service.ISysLogininforService; +import com.ruoyi.system.service.ISysOperLogService; +import eu.bitwalker.useragentutils.UserAgent; + +/** + * 异步工厂(产生任务用) + * + * @author ruoyi + */ +public class AsyncFactory +{ + private static final Logger sys_user_logger = LoggerFactory.getLogger("sys-user"); + + /** + * 记录登录信息 + * + * @param username 用户名 + * @param status 状态 + * @param message 消息 + * @param args 列表 + * @return 任务task + */ + public static TimerTask recordLogininfor(final String username, final String status, final String message, + final Object... args) + { + final UserAgent userAgent = UserAgent.parseUserAgentString(ServletUtils.getRequest().getHeader("User-Agent")); + final String ip = IpUtils.getIpAddr(ServletUtils.getRequest()); + return new TimerTask() + { + @Override + public void run() + { + String address = AddressUtils.getRealAddressByIP(ip); + StringBuilder s = new StringBuilder(); + s.append(LogUtils.getBlock(ip)); + s.append(address); + s.append(LogUtils.getBlock(username)); + s.append(LogUtils.getBlock(status)); + s.append(LogUtils.getBlock(message)); + // 打印信息到日志 + sys_user_logger.info(s.toString(), args); + // 获取客户端操作系统 + String os = userAgent.getOperatingSystem().getName(); + // 获取客户端浏览器 + String browser = userAgent.getBrowser().getName(); + // 封装对象 + SysLogininfor logininfor = new SysLogininfor(); + logininfor.setUserName(username); + logininfor.setIpaddr(ip); + logininfor.setLoginLocation(address); + logininfor.setBrowser(browser); + logininfor.setOs(os); + logininfor.setMsg(message); + // 日志状态 + if (StringUtils.equalsAny(status, Constants.LOGIN_SUCCESS, Constants.LOGOUT, Constants.REGISTER)) + { + logininfor.setStatus(Constants.SUCCESS); + } + else if (Constants.LOGIN_FAIL.equals(status)) + { + logininfor.setStatus(Constants.FAIL); + } + // 插入数据 + SpringUtils.getBean(ISysLogininforService.class).insertLogininfor(logininfor); + } + }; + } + + /** + * 操作日志记录 + * + * @param operLog 操作日志信息 + * @return 任务task + */ + public static TimerTask recordOper(final SysOperLog operLog) + { + return new TimerTask() + { + @Override + public void run() + { + // 远程查询操作地点 + operLog.setOperLocation(AddressUtils.getRealAddressByIP(operLog.getOperIp())); + SpringUtils.getBean(ISysOperLogService.class).insertOperlog(operLog); + } + }; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/security/filter/JwtAuthenticationTokenFilter.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/filter/JwtAuthenticationTokenFilter.java new file mode 100644 index 0000000..3eb2495 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/filter/JwtAuthenticationTokenFilter.java @@ -0,0 +1,44 @@ +package com.ruoyi.framework.security.filter; + +import java.io.IOException; +import javax.servlet.FilterChain; +import javax.servlet.ServletException; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.security.web.authentication.WebAuthenticationDetailsSource; +import org.springframework.stereotype.Component; +import org.springframework.web.filter.OncePerRequestFilter; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.web.service.TokenService; + +/** + * token过滤器 验证token有效性 + * + * @author ruoyi + */ +@Component +public class JwtAuthenticationTokenFilter extends OncePerRequestFilter +{ + @Autowired + private TokenService tokenService; + + @Override + protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain) + throws ServletException, IOException + { + LoginUser loginUser = tokenService.getLoginUser(request); + if (StringUtils.isNotNull(loginUser) && StringUtils.isNull(SecurityUtils.getAuthentication())) + { + tokenService.verifyToken(loginUser); + UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(loginUser, null, loginUser.getAuthorities()); + authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request)); + SecurityContextHolder.getContext().setAuthentication(authenticationToken); + } + chain.doFilter(request, response); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/AuthenticationEntryPointImpl.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/AuthenticationEntryPointImpl.java new file mode 100644 index 0000000..c22dd32 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/AuthenticationEntryPointImpl.java @@ -0,0 +1,34 @@ +package com.ruoyi.framework.security.handle; + +import java.io.IOException; +import java.io.Serializable; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.springframework.security.core.AuthenticationException; +import org.springframework.security.web.AuthenticationEntryPoint; +import org.springframework.stereotype.Component; +import com.alibaba.fastjson.JSON; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; + +/** + * 认证失败处理类 返回未授权 + * + * @author ruoyi + */ +@Component +public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable +{ + private static final long serialVersionUID = -8970718410437077606L; + + @Override + public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e) + throws IOException + { + int code = HttpStatus.UNAUTHORIZED; + String msg = StringUtils.format("请求访问:{},认证失败,无法访问系统资源", request.getRequestURI()); + ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code, msg))); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/LogoutSuccessHandlerImpl.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/LogoutSuccessHandlerImpl.java new file mode 100644 index 0000000..e5fc11d --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/security/handle/LogoutSuccessHandlerImpl.java @@ -0,0 +1,53 @@ +package com.ruoyi.framework.security.handle; + +import java.io.IOException; +import javax.servlet.ServletException; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Configuration; +import org.springframework.security.core.Authentication; +import org.springframework.security.web.authentication.logout.LogoutSuccessHandler; +import com.alibaba.fastjson.JSON; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.framework.web.service.TokenService; + +/** + * 自定义退出处理类 返回成功 + * + * @author ruoyi + */ +@Configuration +public class LogoutSuccessHandlerImpl implements LogoutSuccessHandler +{ + @Autowired + private TokenService tokenService; + + /** + * 退出处理 + * + * @return + */ + @Override + public void onLogoutSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) + throws IOException, ServletException + { + LoginUser loginUser = tokenService.getLoginUser(request); + if (StringUtils.isNotNull(loginUser)) + { + String userName = loginUser.getUsername(); + // 删除用户缓存记录 + tokenService.delLoginUser(loginUser.getToken()); + // 记录用户退出日志 + AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGOUT, "退出成功")); + } + ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(HttpStatus.SUCCESS, "退出成功"))); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/Server.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/Server.java new file mode 100644 index 0000000..63b03da --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/Server.java @@ -0,0 +1,240 @@ +package com.ruoyi.framework.web.domain; + +import java.net.UnknownHostException; +import java.util.LinkedList; +import java.util.List; +import java.util.Properties; +import com.ruoyi.common.utils.Arith; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.framework.web.domain.server.Cpu; +import com.ruoyi.framework.web.domain.server.Jvm; +import com.ruoyi.framework.web.domain.server.Mem; +import com.ruoyi.framework.web.domain.server.Sys; +import com.ruoyi.framework.web.domain.server.SysFile; +import oshi.SystemInfo; +import oshi.hardware.CentralProcessor; +import oshi.hardware.CentralProcessor.TickType; +import oshi.hardware.GlobalMemory; +import oshi.hardware.HardwareAbstractionLayer; +import oshi.software.os.FileSystem; +import oshi.software.os.OSFileStore; +import oshi.software.os.OperatingSystem; +import oshi.util.Util; + +/** + * 服务器相关信息 + * + * @author ruoyi + */ +public class Server +{ + private static final int OSHI_WAIT_SECOND = 1000; + + /** + * CPU相关信息 + */ + private Cpu cpu = new Cpu(); + + /** + * 內存相关信息 + */ + private Mem mem = new Mem(); + + /** + * JVM相关信息 + */ + private Jvm jvm = new Jvm(); + + /** + * 服务器相关信息 + */ + private Sys sys = new Sys(); + + /** + * 磁盘相关信息 + */ + private List sysFiles = new LinkedList(); + + public Cpu getCpu() + { + return cpu; + } + + public void setCpu(Cpu cpu) + { + this.cpu = cpu; + } + + public Mem getMem() + { + return mem; + } + + public void setMem(Mem mem) + { + this.mem = mem; + } + + public Jvm getJvm() + { + return jvm; + } + + public void setJvm(Jvm jvm) + { + this.jvm = jvm; + } + + public Sys getSys() + { + return sys; + } + + public void setSys(Sys sys) + { + this.sys = sys; + } + + public List getSysFiles() + { + return sysFiles; + } + + public void setSysFiles(List sysFiles) + { + this.sysFiles = sysFiles; + } + + public void copyTo() throws Exception + { + SystemInfo si = new SystemInfo(); + HardwareAbstractionLayer hal = si.getHardware(); + + setCpuInfo(hal.getProcessor()); + + setMemInfo(hal.getMemory()); + + setSysInfo(); + + setJvmInfo(); + + setSysFiles(si.getOperatingSystem()); + } + + /** + * 设置CPU信息 + */ + private void setCpuInfo(CentralProcessor processor) + { + // CPU信息 + long[] prevTicks = processor.getSystemCpuLoadTicks(); + Util.sleep(OSHI_WAIT_SECOND); + long[] ticks = processor.getSystemCpuLoadTicks(); + long nice = ticks[TickType.NICE.getIndex()] - prevTicks[TickType.NICE.getIndex()]; + long irq = ticks[TickType.IRQ.getIndex()] - prevTicks[TickType.IRQ.getIndex()]; + long softirq = ticks[TickType.SOFTIRQ.getIndex()] - prevTicks[TickType.SOFTIRQ.getIndex()]; + long steal = ticks[TickType.STEAL.getIndex()] - prevTicks[TickType.STEAL.getIndex()]; + long cSys = ticks[TickType.SYSTEM.getIndex()] - prevTicks[TickType.SYSTEM.getIndex()]; + long user = ticks[TickType.USER.getIndex()] - prevTicks[TickType.USER.getIndex()]; + long iowait = ticks[TickType.IOWAIT.getIndex()] - prevTicks[TickType.IOWAIT.getIndex()]; + long idle = ticks[TickType.IDLE.getIndex()] - prevTicks[TickType.IDLE.getIndex()]; + long totalCpu = user + nice + cSys + idle + iowait + irq + softirq + steal; + cpu.setCpuNum(processor.getLogicalProcessorCount()); + cpu.setTotal(totalCpu); + cpu.setSys(cSys); + cpu.setUsed(user); + cpu.setWait(iowait); + cpu.setFree(idle); + } + + /** + * 设置内存信息 + */ + private void setMemInfo(GlobalMemory memory) + { + mem.setTotal(memory.getTotal()); + mem.setUsed(memory.getTotal() - memory.getAvailable()); + mem.setFree(memory.getAvailable()); + } + + /** + * 设置服务器信息 + */ + private void setSysInfo() + { + Properties props = System.getProperties(); + sys.setComputerName(IpUtils.getHostName()); + sys.setComputerIp(IpUtils.getHostIp()); + sys.setOsName(props.getProperty("os.name")); + sys.setOsArch(props.getProperty("os.arch")); + sys.setUserDir(props.getProperty("user.dir")); + } + + /** + * 设置Java虚拟机 + */ + private void setJvmInfo() throws UnknownHostException + { + Properties props = System.getProperties(); + jvm.setTotal(Runtime.getRuntime().totalMemory()); + jvm.setMax(Runtime.getRuntime().maxMemory()); + jvm.setFree(Runtime.getRuntime().freeMemory()); + jvm.setVersion(props.getProperty("java.version")); + jvm.setHome(props.getProperty("java.home")); + } + + /** + * 设置磁盘信息 + */ + private void setSysFiles(OperatingSystem os) + { + FileSystem fileSystem = os.getFileSystem(); + List fsArray = fileSystem.getFileStores(); + for (OSFileStore fs : fsArray) + { + long free = fs.getUsableSpace(); + long total = fs.getTotalSpace(); + long used = total - free; + SysFile sysFile = new SysFile(); + sysFile.setDirName(fs.getMount()); + sysFile.setSysTypeName(fs.getType()); + sysFile.setTypeName(fs.getName()); + sysFile.setTotal(convertFileSize(total)); + sysFile.setFree(convertFileSize(free)); + sysFile.setUsed(convertFileSize(used)); + sysFile.setUsage(Arith.mul(Arith.div(used, total, 4), 100)); + sysFiles.add(sysFile); + } + } + + /** + * 字节转换 + * + * @param size 字节大小 + * @return 转换后值 + */ + public String convertFileSize(long size) + { + long kb = 1024; + long mb = kb * 1024; + long gb = mb * 1024; + if (size >= gb) + { + return String.format("%.1f GB", (float) size / gb); + } + else if (size >= mb) + { + float f = (float) size / mb; + return String.format(f > 100 ? "%.0f MB" : "%.1f MB", f); + } + else if (size >= kb) + { + float f = (float) size / kb; + return String.format(f > 100 ? "%.0f KB" : "%.1f KB", f); + } + else + { + return String.format("%d B", size); + } + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Cpu.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Cpu.java new file mode 100644 index 0000000..a13a66c --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Cpu.java @@ -0,0 +1,101 @@ +package com.ruoyi.framework.web.domain.server; + +import com.ruoyi.common.utils.Arith; + +/** + * CPU相关信息 + * + * @author ruoyi + */ +public class Cpu +{ + /** + * 核心数 + */ + private int cpuNum; + + /** + * CPU总的使用率 + */ + private double total; + + /** + * CPU系统使用率 + */ + private double sys; + + /** + * CPU用户使用率 + */ + private double used; + + /** + * CPU当前等待率 + */ + private double wait; + + /** + * CPU当前空闲率 + */ + private double free; + + public int getCpuNum() + { + return cpuNum; + } + + public void setCpuNum(int cpuNum) + { + this.cpuNum = cpuNum; + } + + public double getTotal() + { + return Arith.round(Arith.mul(total, 100), 2); + } + + public void setTotal(double total) + { + this.total = total; + } + + public double getSys() + { + return Arith.round(Arith.mul(sys / total, 100), 2); + } + + public void setSys(double sys) + { + this.sys = sys; + } + + public double getUsed() + { + return Arith.round(Arith.mul(used / total, 100), 2); + } + + public void setUsed(double used) + { + this.used = used; + } + + public double getWait() + { + return Arith.round(Arith.mul(wait / total, 100), 2); + } + + public void setWait(double wait) + { + this.wait = wait; + } + + public double getFree() + { + return Arith.round(Arith.mul(free / total, 100), 2); + } + + public void setFree(double free) + { + this.free = free; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Jvm.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Jvm.java new file mode 100644 index 0000000..444b280 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Jvm.java @@ -0,0 +1,130 @@ +package com.ruoyi.framework.web.domain.server; + +import java.lang.management.ManagementFactory; +import com.ruoyi.common.utils.Arith; +import com.ruoyi.common.utils.DateUtils; + +/** + * JVM相关信息 + * + * @author ruoyi + */ +public class Jvm +{ + /** + * 当前JVM占用的内存总数(M) + */ + private double total; + + /** + * JVM最大可用内存总数(M) + */ + private double max; + + /** + * JVM空闲内存(M) + */ + private double free; + + /** + * JDK版本 + */ + private String version; + + /** + * JDK路径 + */ + private String home; + + public double getTotal() + { + return Arith.div(total, (1024 * 1024), 2); + } + + public void setTotal(double total) + { + this.total = total; + } + + public double getMax() + { + return Arith.div(max, (1024 * 1024), 2); + } + + public void setMax(double max) + { + this.max = max; + } + + public double getFree() + { + return Arith.div(free, (1024 * 1024), 2); + } + + public void setFree(double free) + { + this.free = free; + } + + public double getUsed() + { + return Arith.div(total - free, (1024 * 1024), 2); + } + + public double getUsage() + { + return Arith.mul(Arith.div(total - free, total, 4), 100); + } + + /** + * 获取JDK名称 + */ + public String getName() + { + return ManagementFactory.getRuntimeMXBean().getVmName(); + } + + public String getVersion() + { + return version; + } + + public void setVersion(String version) + { + this.version = version; + } + + public String getHome() + { + return home; + } + + public void setHome(String home) + { + this.home = home; + } + + /** + * JDK启动时间 + */ + public String getStartTime() + { + return DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD_HH_MM_SS, DateUtils.getServerStartDate()); + } + + /** + * JDK运行时间 + */ + public String getRunTime() + { + return DateUtils.getDatePoor(DateUtils.getNowDate(), DateUtils.getServerStartDate()); + } + + /** + * 运行参数 + */ + public String getInputArgs() + { + return ManagementFactory.getRuntimeMXBean().getInputArguments().toString(); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Mem.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Mem.java new file mode 100644 index 0000000..13eec52 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Mem.java @@ -0,0 +1,61 @@ +package com.ruoyi.framework.web.domain.server; + +import com.ruoyi.common.utils.Arith; + +/** + * 內存相关信息 + * + * @author ruoyi + */ +public class Mem +{ + /** + * 内存总量 + */ + private double total; + + /** + * 已用内存 + */ + private double used; + + /** + * 剩余内存 + */ + private double free; + + public double getTotal() + { + return Arith.div(total, (1024 * 1024 * 1024), 2); + } + + public void setTotal(long total) + { + this.total = total; + } + + public double getUsed() + { + return Arith.div(used, (1024 * 1024 * 1024), 2); + } + + public void setUsed(long used) + { + this.used = used; + } + + public double getFree() + { + return Arith.div(free, (1024 * 1024 * 1024), 2); + } + + public void setFree(long free) + { + this.free = free; + } + + public double getUsage() + { + return Arith.mul(Arith.div(used, total, 4), 100); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Sys.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Sys.java new file mode 100644 index 0000000..45d64d9 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/Sys.java @@ -0,0 +1,84 @@ +package com.ruoyi.framework.web.domain.server; + +/** + * 系统相关信息 + * + * @author ruoyi + */ +public class Sys +{ + /** + * 服务器名称 + */ + private String computerName; + + /** + * 服务器Ip + */ + private String computerIp; + + /** + * 项目路径 + */ + private String userDir; + + /** + * 操作系统 + */ + private String osName; + + /** + * 系统架构 + */ + private String osArch; + + public String getComputerName() + { + return computerName; + } + + public void setComputerName(String computerName) + { + this.computerName = computerName; + } + + public String getComputerIp() + { + return computerIp; + } + + public void setComputerIp(String computerIp) + { + this.computerIp = computerIp; + } + + public String getUserDir() + { + return userDir; + } + + public void setUserDir(String userDir) + { + this.userDir = userDir; + } + + public String getOsName() + { + return osName; + } + + public void setOsName(String osName) + { + this.osName = osName; + } + + public String getOsArch() + { + return osArch; + } + + public void setOsArch(String osArch) + { + this.osArch = osArch; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/SysFile.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/SysFile.java new file mode 100644 index 0000000..1320cde --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/domain/server/SysFile.java @@ -0,0 +1,114 @@ +package com.ruoyi.framework.web.domain.server; + +/** + * 系统文件相关信息 + * + * @author ruoyi + */ +public class SysFile +{ + /** + * 盘符路径 + */ + private String dirName; + + /** + * 盘符类型 + */ + private String sysTypeName; + + /** + * 文件类型 + */ + private String typeName; + + /** + * 总大小 + */ + private String total; + + /** + * 剩余大小 + */ + private String free; + + /** + * 已经使用量 + */ + private String used; + + /** + * 资源的使用率 + */ + private double usage; + + public String getDirName() + { + return dirName; + } + + public void setDirName(String dirName) + { + this.dirName = dirName; + } + + public String getSysTypeName() + { + return sysTypeName; + } + + public void setSysTypeName(String sysTypeName) + { + this.sysTypeName = sysTypeName; + } + + public String getTypeName() + { + return typeName; + } + + public void setTypeName(String typeName) + { + this.typeName = typeName; + } + + public String getTotal() + { + return total; + } + + public void setTotal(String total) + { + this.total = total; + } + + public String getFree() + { + return free; + } + + public void setFree(String free) + { + this.free = free; + } + + public String getUsed() + { + return used; + } + + public void setUsed(String used) + { + this.used = used; + } + + public double getUsage() + { + return usage; + } + + public void setUsage(double usage) + { + this.usage = usage; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/exception/GlobalExceptionHandler.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/exception/GlobalExceptionHandler.java new file mode 100644 index 0000000..51dd8c5 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/exception/GlobalExceptionHandler.java @@ -0,0 +1,114 @@ +package com.ruoyi.framework.web.exception; + +import javax.servlet.http.HttpServletRequest; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.security.access.AccessDeniedException; +import org.springframework.validation.BindException; +import org.springframework.web.HttpRequestMethodNotSupportedException; +import org.springframework.web.bind.MethodArgumentNotValidException; +import org.springframework.web.bind.annotation.ExceptionHandler; +import org.springframework.web.bind.annotation.RestControllerAdvice; +import com.ruoyi.common.constant.HttpStatus; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.exception.DemoModeException; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.StringUtils; + +/** + * 全局异常处理器 + * + * @author ruoyi + */ +@RestControllerAdvice +public class GlobalExceptionHandler +{ + private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class); + + /** + * 权限校验异常 + */ + @ExceptionHandler(AccessDeniedException.class) + public AjaxResult handleAccessDeniedException(AccessDeniedException e, HttpServletRequest request) + { + String requestURI = request.getRequestURI(); + log.error("请求地址'{}',权限校验失败'{}'", requestURI, e.getMessage()); + return AjaxResult.error(HttpStatus.FORBIDDEN, "没有权限,请联系管理员授权"); + } + + /** + * 请求方式不支持 + */ + @ExceptionHandler(HttpRequestMethodNotSupportedException.class) + public AjaxResult handleHttpRequestMethodNotSupported(HttpRequestMethodNotSupportedException e, + HttpServletRequest request) + { + String requestURI = request.getRequestURI(); + log.error("请求地址'{}',不支持'{}'请求", requestURI, e.getMethod()); + return AjaxResult.error(e.getMessage()); + } + + /** + * 业务异常 + */ + @ExceptionHandler(ServiceException.class) + public AjaxResult handleServiceException(ServiceException e, HttpServletRequest request) + { + log.error(e.getMessage(), e); + Integer code = e.getCode(); + return StringUtils.isNotNull(code) ? AjaxResult.error(code, e.getMessage()) : AjaxResult.error(e.getMessage()); + } + + /** + * 拦截未知的运行时异常 + */ + @ExceptionHandler(RuntimeException.class) + public AjaxResult handleRuntimeException(RuntimeException e, HttpServletRequest request) + { + String requestURI = request.getRequestURI(); + log.error("请求地址'{}',发生未知异常.", requestURI, e); + return AjaxResult.error(e.getMessage()); + } + + /** + * 系统异常 + */ + @ExceptionHandler(Exception.class) + public AjaxResult handleException(Exception e, HttpServletRequest request) + { + String requestURI = request.getRequestURI(); + log.error("请求地址'{}',发生系统异常.", requestURI, e); + return AjaxResult.error(e.getMessage()); + } + + /** + * 自定义验证异常 + */ + @ExceptionHandler(BindException.class) + public AjaxResult handleBindException(BindException e) + { + log.error(e.getMessage(), e); + String message = e.getAllErrors().get(0).getDefaultMessage(); + return AjaxResult.error(message); + } + + /** + * 自定义验证异常 + */ + @ExceptionHandler(MethodArgumentNotValidException.class) + public Object handleMethodArgumentNotValidException(MethodArgumentNotValidException e) + { + log.error(e.getMessage(), e); + String message = e.getBindingResult().getFieldError().getDefaultMessage(); + return AjaxResult.error(message); + } + + /** + * 演示模式异常 + */ + @ExceptionHandler(DemoModeException.class) + public AjaxResult handleDemoModeException(DemoModeException e) + { + return AjaxResult.error("演示模式,不允许操作"); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/PermissionService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/PermissionService.java new file mode 100644 index 0000000..68b1d69 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/PermissionService.java @@ -0,0 +1,165 @@ +package com.ruoyi.framework.web.service; + +import java.util.Set; +import org.springframework.stereotype.Service; +import org.springframework.util.CollectionUtils; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; + +/** + * RuoYi首创 自定义权限实现,ss取自SpringSecurity首字母 + * + * @author ruoyi + */ +@Service("ss") +public class PermissionService +{ + /** 所有权限标识 */ + private static final String ALL_PERMISSION = "*:*:*"; + + /** 管理员角色权限标识 */ + private static final String SUPER_ADMIN = "admin"; + + private static final String ROLE_DELIMETER = ","; + + private static final String PERMISSION_DELIMETER = ","; + + /** + * 验证用户是否具备某权限 + * + * @param permission 权限字符串 + * @return 用户是否具备某权限 + */ + public boolean hasPermi(String permission) + { + if (StringUtils.isEmpty(permission)) + { + return false; + } + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) + { + return false; + } + return hasPermissions(loginUser.getPermissions(), permission); + } + + /** + * 验证用户是否不具备某权限,与 hasPermi逻辑相反 + * + * @param permission 权限字符串 + * @return 用户是否不具备某权限 + */ + public boolean lacksPermi(String permission) + { + return hasPermi(permission) != true; + } + + /** + * 验证用户是否具有以下任意一个权限 + * + * @param permissions 以 PERMISSION_NAMES_DELIMETER 为分隔符的权限列表 + * @return 用户是否具有以下任意一个权限 + */ + public boolean hasAnyPermi(String permissions) + { + if (StringUtils.isEmpty(permissions)) + { + return false; + } + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) + { + return false; + } + Set authorities = loginUser.getPermissions(); + for (String permission : permissions.split(PERMISSION_DELIMETER)) + { + if (permission != null && hasPermissions(authorities, permission)) + { + return true; + } + } + return false; + } + + /** + * 判断用户是否拥有某个角色 + * + * @param role 角色字符串 + * @return 用户是否具备某角色 + */ + public boolean hasRole(String role) + { + if (StringUtils.isEmpty(role)) + { + return false; + } + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) + { + return false; + } + for (SysRole sysRole : loginUser.getUser().getRoles()) + { + String roleKey = sysRole.getRoleKey(); + if (SUPER_ADMIN.equals(roleKey) || roleKey.equals(StringUtils.trim(role))) + { + return true; + } + } + return false; + } + + /** + * 验证用户是否不具备某角色,与 isRole逻辑相反。 + * + * @param role 角色名称 + * @return 用户是否不具备某角色 + */ + public boolean lacksRole(String role) + { + return hasRole(role) != true; + } + + /** + * 验证用户是否具有以下任意一个角色 + * + * @param roles 以 ROLE_NAMES_DELIMETER 为分隔符的角色列表 + * @return 用户是否具有以下任意一个角色 + */ + public boolean hasAnyRoles(String roles) + { + if (StringUtils.isEmpty(roles)) + { + return false; + } + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) + { + return false; + } + for (String role : roles.split(ROLE_DELIMETER)) + { + if (hasRole(role)) + { + return true; + } + } + return false; + } + + /** + * 判断是否包含权限 + * + * @param permissions 权限列表 + * @param permission 权限字符串 + * @return 用户是否具备某权限 + */ + private boolean hasPermissions(Set permissions, String permission) + { + return permissions.contains(ALL_PERMISSION) || permissions.contains(StringUtils.trim(permission)); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java new file mode 100644 index 0000000..e9e051b --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java @@ -0,0 +1,134 @@ +package com.ruoyi.framework.web.service; + +import javax.annotation.Resource; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.authentication.AuthenticationManager; +import org.springframework.security.authentication.BadCredentialsException; +import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; +import org.springframework.security.core.Authentication; +import org.springframework.stereotype.Component; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.exception.user.CaptchaException; +import com.ruoyi.common.exception.user.CaptchaExpireException; +import com.ruoyi.common.exception.user.UserPasswordNotMatchException; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.MessageUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.system.service.ISysConfigService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 登录校验方法 + * + * @author ruoyi + */ +@Component +public class SysLoginService +{ + @Autowired + private TokenService tokenService; + + @Resource + private AuthenticationManager authenticationManager; + + @Autowired + private RedisCache redisCache; + + @Autowired + private ISysUserService userService; + + @Autowired + private ISysConfigService configService; + + /** + * 登录验证 + * + * @param username 用户名 + * @param password 密码 + * @param code 验证码 + * @param uuid 唯一标识 + * @return 结果 + */ + public String login(String username, String password, String code, String uuid) + { + boolean captchaOnOff = configService.selectCaptchaOnOff(); + // 验证码开关 + if (captchaOnOff) + { + validateCaptcha(username, code, uuid); + } + // 用户验证 + Authentication authentication = null; + try + { + // 该方法会去调用UserDetailsServiceImpl.loadUserByUsername + authentication = authenticationManager + .authenticate(new UsernamePasswordAuthenticationToken(username, password)); + } + catch (Exception e) + { + if (e instanceof BadCredentialsException) + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match"))); + throw new UserPasswordNotMatchException(); + } + else + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, e.getMessage())); + throw new ServiceException(e.getMessage()); + } + } + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success"))); + LoginUser loginUser = (LoginUser) authentication.getPrincipal(); + recordLoginInfo(loginUser.getUserId()); + // 生成token + return tokenService.createToken(loginUser); + } + + /** + * 校验验证码 + * + * @param username 用户名 + * @param code 验证码 + * @param uuid 唯一标识 + * @return 结果 + */ + public void validateCaptcha(String username, String code, String uuid) + { + String verifyKey = Constants.CAPTCHA_CODE_KEY + StringUtils.nvl(uuid, ""); + String captcha = redisCache.getCacheObject(verifyKey); + redisCache.deleteObject(verifyKey); + if (captcha == null) + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.expire"))); + throw new CaptchaExpireException(); + } + if (!code.equalsIgnoreCase(captcha)) + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.error"))); + throw new CaptchaException(); + } + } + + /** + * 记录登录信息 + * + * @param userId 用户ID + */ + public void recordLoginInfo(Long userId) + { + SysUser sysUser = new SysUser(); + sysUser.setUserId(userId); + sysUser.setLoginIp(IpUtils.getIpAddr(ServletUtils.getRequest())); + sysUser.setLoginDate(DateUtils.getNowDate()); + userService.updateUserProfile(sysUser); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysPermissionService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysPermissionService.java new file mode 100644 index 0000000..feb8038 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysPermissionService.java @@ -0,0 +1,66 @@ +package com.ruoyi.framework.web.service; + +import java.util.HashSet; +import java.util.Set; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.system.service.ISysMenuService; +import com.ruoyi.system.service.ISysRoleService; + +/** + * 用户权限处理 + * + * @author ruoyi + */ +@Component +public class SysPermissionService +{ + @Autowired + private ISysRoleService roleService; + + @Autowired + private ISysMenuService menuService; + + /** + * 获取角色数据权限 + * + * @param user 用户信息 + * @return 角色权限信息 + */ + public Set getRolePermission(SysUser user) + { + Set roles = new HashSet(); + // 管理员拥有所有权限 + if (user.isAdmin()) + { + roles.add("admin"); + } + else + { + roles.addAll(roleService.selectRolePermissionByUserId(user.getUserId())); + } + return roles; + } + + /** + * 获取菜单数据权限 + * + * @param user 用户信息 + * @return 菜单权限信息 + */ + public Set getMenuPermission(SysUser user) + { + Set perms = new HashSet(); + // 管理员拥有所有权限 + if (user.isAdmin()) + { + perms.add("*:*:*"); + } + else + { + perms.addAll(menuService.selectMenuPermsByUserId(user.getUserId())); + } + return perms; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysRegisterService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysRegisterService.java new file mode 100644 index 0000000..8cc5ced --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysRegisterService.java @@ -0,0 +1,115 @@ +package com.ruoyi.framework.web.service; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.RegisterBody; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.exception.user.CaptchaException; +import com.ruoyi.common.exception.user.CaptchaExpireException; +import com.ruoyi.common.utils.MessageUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.system.service.ISysConfigService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 注册校验方法 + * + * @author ruoyi + */ +@Component +public class SysRegisterService +{ + @Autowired + private ISysUserService userService; + + @Autowired + private ISysConfigService configService; + + @Autowired + private RedisCache redisCache; + + /** + * 注册 + */ + public String register(RegisterBody registerBody) + { + String msg = "", username = registerBody.getUsername(), password = registerBody.getPassword(); + + boolean captchaOnOff = configService.selectCaptchaOnOff(); + // 验证码开关 + if (captchaOnOff) + { + validateCaptcha(username, registerBody.getCode(), registerBody.getUuid()); + } + + if (StringUtils.isEmpty(username)) + { + msg = "用户名不能为空"; + } + else if (StringUtils.isEmpty(password)) + { + msg = "用户密码不能为空"; + } + else if (username.length() < UserConstants.USERNAME_MIN_LENGTH + || username.length() > UserConstants.USERNAME_MAX_LENGTH) + { + msg = "账户长度必须在2到20个字符之间"; + } + else if (password.length() < UserConstants.PASSWORD_MIN_LENGTH + || password.length() > UserConstants.PASSWORD_MAX_LENGTH) + { + msg = "密码长度必须在5到20个字符之间"; + } + else if (UserConstants.NOT_UNIQUE.equals(userService.checkUserNameUnique(username))) + { + msg = "保存用户'" + username + "'失败,注册账号已存在"; + } + else + { + SysUser sysUser = new SysUser(); + sysUser.setUserName(username); + sysUser.setNickName(username); + sysUser.setPassword(SecurityUtils.encryptPassword(registerBody.getPassword())); + boolean regFlag = userService.registerUser(sysUser); + if (!regFlag) + { + msg = "注册失败,请联系系统管理人员"; + } + else + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.REGISTER, + MessageUtils.message("user.register.success"))); + } + } + return msg; + } + + /** + * 校验验证码 + * + * @param username 用户名 + * @param code 验证码 + * @param uuid 唯一标识 + * @return 结果 + */ + public void validateCaptcha(String username, String code, String uuid) + { + String verifyKey = Constants.CAPTCHA_CODE_KEY + StringUtils.nvl(uuid, ""); + String captcha = redisCache.getCacheObject(verifyKey); + redisCache.deleteObject(verifyKey); + if (captcha == null) + { + throw new CaptchaExpireException(); + } + if (!code.equalsIgnoreCase(captcha)) + { + throw new CaptchaException(); + } + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/ThirdLoginService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/ThirdLoginService.java new file mode 100644 index 0000000..e3e4f99 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/ThirdLoginService.java @@ -0,0 +1,172 @@ +package com.ruoyi.framework.web.service; + +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.exception.user.UserPasswordNotMatchException; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.MessageUtils; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.framework.manager.AsyncManager; +import com.ruoyi.framework.manager.factory.AsyncFactory; +import com.ruoyi.system.service.ISysUserService; +import jodd.http.HttpRequest; +import jodd.http.HttpResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.http.HttpHeaders; +import org.springframework.security.authentication.AuthenticationManager; +import org.springframework.security.authentication.BadCredentialsException; +import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; +import org.springframework.security.core.Authentication; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.security.core.userdetails.UserDetailsService; +import org.springframework.stereotype.Service; + +import javax.annotation.Resource; +import java.util.Optional; + +/** + * @projectName: RuoYi-Vue + * @package: com.ruoyi.framework.web.service + * @className: ThirdLoginService + * @author: lsyong + * @description: TODO + * @date: 2023/2/22 7:53 + * @version: 1.0 + */ +@Service +public class ThirdLoginService { + + @Autowired + private ISysUserService userService; + + @Autowired + private TokenService tokenService; + + @Resource + private AuthenticationManager authenticationManager; + @Autowired + private UserDetailsService userDetailsService; + + /** + * 无密码登录 + * @param userName + * @return + */ + public String noPwdLogin(String userName,String token){ + LoginUser loginUser + = (LoginUser)userDetailsService.loadUserByUsername(userName); + // 记录登陆信息 + AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGIN_SUCCESS, + MessageUtils.message("user.login.success"))); + recordLoginInfo(loginUser.getUserId()); + return tokenService.createToken(loginUser); + } + + /** + * 不加验证码登录 + * + * @param username 用户名 + * @param password 密码 + * @param uuid 唯一标识 + * @return 结果 + */ + public String loginNoCode(String username, String password, String uuid) + { + // 用户验证 + Authentication authentication = null; + try + { + // 该方法会去调用UserDetailsServiceImpl.loadUserByUsername + UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(username, password); + SecurityContextHolder.getContext().setAuthentication(authenticationToken); + authentication = authenticationManager.authenticate(authenticationToken); + } + catch (Exception e) + { + if (e instanceof BadCredentialsException) + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match"))); + throw new UserPasswordNotMatchException(); + } + else + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, e.getMessage())); + throw new ServiceException(e.getMessage()); + } + } + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success"))); + LoginUser loginUser = + (LoginUser) authentication.getPrincipal(); + recordLoginInfo(loginUser.getUserId()); + // 生成token + return tokenService.createToken(loginUser); + } + + /** + * @param token: + * @return String + * @title thirdLogin + * @descriptions 三方平台登陆 + */ + public String thirdLogin(String token) { + //调用验证token的方法(正式用的时候使用checkJToken()方法返回用户名 + JSONObject jsonObject = checkJToken(token); + String code = jsonObject.getString("code"); + if(!code.equals("200")){ + return null; + }else { + String loginName = jsonObject.getString("data"); + //根据token获取用户相关信息 + SysUser sysUser = userService.selectUserByUserName(loginName); + Optional.ofNullable(sysUser).orElseThrow(()->new ServiceException("用户不存在,请联系管理员")); + return this.noPwdLogin(sysUser.getUserName(),token); + } + } + + + /** + * 记录登录信息 + * + * @param userId 用户ID + */ + public void recordLoginInfo(Long userId) + { + SysUser sysUser = new SysUser(); + sysUser.setUserId(userId); + sysUser.setLoginIp(IpUtils.getIpAddr(ServletUtils.getRequest())); + sysUser.setLoginDate(DateUtils.getNowDate()); + userService.updateUserProfile(sysUser); + } + + + @Value("${ssoip}") + private String ssoip; + @Value("${ssoport}") + private String ssoport; + + /** + * 通过第三方平台接口,鉴定token合法性,并返回userName等登录信息 + * 这个方法需要根据实际需要进行修改 + * @param token + * @return + */ + public JSONObject checkJToken(String token) { + JSONObject jsonObject = new JSONObject(); + //测试环境 + String baseUrl = "http://"+ssoip+":"+ssoport+"/sso/getUserNameByToken?token="+token;//根据实际地址进行修改 + HttpRequest request = HttpRequest.get(baseUrl) + .header(HttpHeaders.CONTENT_TYPE, "application/json") + .header(HttpHeaders.ACCEPT, "application/json") + .header("X-Requested-With","XMLHttpRequest") + .header("satoken",token); + + HttpResponse response = request.send(); + System.out.println(JSONObject.parse(response.bodyText())); + return (JSONObject) JSONObject.parse(response.bodyText()); + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/TokenService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/TokenService.java new file mode 100644 index 0000000..3220c99 --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/TokenService.java @@ -0,0 +1,245 @@ +package com.ruoyi.framework.web.service; + +import java.util.HashMap; +import java.util.Map; +import java.util.concurrent.TimeUnit; +import javax.servlet.http.HttpServletRequest; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Component; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.utils.ServletUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.ip.AddressUtils; +import com.ruoyi.common.utils.ip.IpUtils; +import com.ruoyi.common.utils.uuid.IdUtils; +import eu.bitwalker.useragentutils.UserAgent; +import io.jsonwebtoken.Claims; +import io.jsonwebtoken.Jwts; +import io.jsonwebtoken.SignatureAlgorithm; + +/** + * token验证处理 + * + * @author ruoyi + */ +@Component +public class TokenService +{ + // 令牌自定义标识 + @Value("${token.header}") + private String header; + + // 令牌秘钥 + @Value("${token.secret}") + private String secret; + + // 令牌有效期(默认30分钟) + @Value("${token.expireTime}") + private int expireTime; + + protected static final long MILLIS_SECOND = 1000; + + protected static final long MILLIS_MINUTE = 60 * MILLIS_SECOND; + + private static final Long MILLIS_MINUTE_TEN = 20 * 60 * 1000L; + + @Autowired + private RedisCache redisCache; + + /** + * 获取用户身份信息 + * + * @return 用户信息 + */ + public LoginUser getLoginUser(HttpServletRequest request) + { + // 获取请求携带的令牌 + String token = getToken(request); + if (StringUtils.isNotEmpty(token)) + { + try + { + Claims claims = parseToken(token); + // 解析对应的权限以及用户信息 + String uuid = (String) claims.get(Constants.LOGIN_USER_KEY); + String userKey = getTokenKey(uuid); + LoginUser user = redisCache.getCacheObject(userKey); + return user; + } + catch (Exception e) + { + } + } + return null; + } + + + public LoginUser getThirdLoginUser(String token) + { + if (StringUtils.isNotEmpty(token)) + { + try + { + Claims claims = parseToken(token); + // 解析对应的权限以及用户信息 + String uuid = (String) claims.get(Constants.LOGIN_USER_KEY); + String userKey = getTokenKey(uuid); + LoginUser user = redisCache.getCacheObject(userKey); + return user; + } + catch (Exception e) + { + } + } + return null; + } + /** + * 设置用户身份信息 + */ + public void setLoginUser(LoginUser loginUser) + { + if (StringUtils.isNotNull(loginUser) && StringUtils.isNotEmpty(loginUser.getToken())) + { + refreshToken(loginUser); + } + } + + /** + * 删除用户身份信息 + */ + public void delLoginUser(String token) + { + if (StringUtils.isNotEmpty(token)) + { + String userKey = getTokenKey(token); + redisCache.deleteObject(userKey); + } + } + + /** + * 创建令牌 + * + * @param loginUser 用户信息 + * @return 令牌 + */ + public String createToken(LoginUser loginUser) + { + String token = IdUtils.fastUUID(); + loginUser.setToken(token); + setUserAgent(loginUser); + refreshToken(loginUser); + + Map claims = new HashMap<>(); + claims.put(Constants.LOGIN_USER_KEY, token); + return createToken(claims); + } + + /** + * 验证令牌有效期,相差不足20分钟,自动刷新缓存 + * + * @param loginUser + * @return 令牌 + */ + public void verifyToken(LoginUser loginUser) + { + long expireTime = loginUser.getExpireTime(); + long currentTime = System.currentTimeMillis(); + if (expireTime - currentTime <= MILLIS_MINUTE_TEN) + { + refreshToken(loginUser); + } + } + + /** + * 刷新令牌有效期 + * + * @param loginUser 登录信息 + */ + public void refreshToken(LoginUser loginUser) + { + loginUser.setLoginTime(System.currentTimeMillis()); + loginUser.setExpireTime(loginUser.getLoginTime() + expireTime * MILLIS_MINUTE); + // 根据uuid将loginUser缓存 + String userKey = getTokenKey(loginUser.getToken()); + redisCache.setCacheObject(userKey, loginUser, expireTime, TimeUnit.MINUTES); + } + + /** + * 设置用户代理信息 + * + * @param loginUser 登录信息 + */ + public void setUserAgent(LoginUser loginUser) + { + UserAgent userAgent = UserAgent.parseUserAgentString(ServletUtils.getRequest().getHeader("User-Agent")); + String ip = IpUtils.getIpAddr(ServletUtils.getRequest()); + loginUser.setIpaddr(ip); + loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip)); + loginUser.setBrowser(userAgent.getBrowser().getName()); + loginUser.setOs(userAgent.getOperatingSystem().getName()); + } + + /** + * 从数据声明生成令牌 + * + * @param claims 数据声明 + * @return 令牌 + */ + private String createToken(Map claims) + { + String token = Jwts.builder() + .setClaims(claims) + .signWith(SignatureAlgorithm.HS512, secret).compact(); + return token; + } + + /** + * 从令牌中获取数据声明 + * + * @param token 令牌 + * @return 数据声明 + */ + private Claims parseToken(String token) + { + return Jwts.parser() + .setSigningKey(secret) + .parseClaimsJws(token) + .getBody(); + } + + /** + * 从令牌中获取用户名 + * + * @param token 令牌 + * @return 用户名 + */ + public String getUsernameFromToken(String token) + { + Claims claims = parseToken(token); + return claims.getSubject(); + } + + /** + * 获取请求token + * + * @param request + * @return token + */ + private String getToken(HttpServletRequest request) + { + String token = request.getHeader(header); + if (StringUtils.isNotEmpty(token) && token.startsWith(Constants.TOKEN_PREFIX)) + { + token = token.replace(Constants.TOKEN_PREFIX, ""); + } + return token; + } + + private String getTokenKey(String uuid) + { + return Constants.LOGIN_TOKEN_KEY + uuid; + } +} diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/UserDetailsServiceImpl.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/UserDetailsServiceImpl.java new file mode 100644 index 0000000..c8b1c7b --- /dev/null +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/UserDetailsServiceImpl.java @@ -0,0 +1,60 @@ +package com.ruoyi.framework.web.service; + +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.core.userdetails.UserDetails; +import org.springframework.security.core.userdetails.UserDetailsService; +import org.springframework.security.core.userdetails.UsernameNotFoundException; +import org.springframework.stereotype.Service; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.enums.UserStatus; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.service.ISysUserService; + +/** + * 用户验证处理 + * + * @author ruoyi + */ +@Service +public class UserDetailsServiceImpl implements UserDetailsService +{ + private static final Logger log = LoggerFactory.getLogger(UserDetailsServiceImpl.class); + + @Autowired + private ISysUserService userService; + + @Autowired + private SysPermissionService permissionService; + + @Override + public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException + { + SysUser user = userService.selectUserByUserName(username); + if (StringUtils.isNull(user)) + { + log.info("登录用户:{} 不存在.", username); + throw new ServiceException("登录用户:" + username + " 不存在"); + } + else if (UserStatus.DELETED.getCode().equals(user.getDelFlag())) + { + log.info("登录用户:{} 已被删除.", username); + throw new ServiceException("对不起,您的账号:" + username + " 已被删除"); + } + else if (UserStatus.DISABLE.getCode().equals(user.getStatus())) + { + log.info("登录用户:{} 已被停用.", username); + throw new ServiceException("对不起,您的账号:" + username + " 已停用"); + } + + return createLoginUser(user); + } + + public UserDetails createLoginUser(SysUser user) + { + return new LoginUser(user.getUserId(), user.getDeptId(), user, permissionService.getMenuPermission(user)); + } +} diff --git a/ruoyi-generator/pom.xml b/ruoyi-generator/pom.xml new file mode 100644 index 0000000..f173cc8 --- /dev/null +++ b/ruoyi-generator/pom.xml @@ -0,0 +1,40 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-generator + + + generator代码生成 + + + + + + + org.apache.velocity + velocity-engine-core + + + + + commons-collections + commons-collections + + + + + com.ruoyi + ruoyi-common + + + + + diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/config/GenConfig.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/config/GenConfig.java new file mode 100644 index 0000000..cc4cd14 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/config/GenConfig.java @@ -0,0 +1,73 @@ +package com.ruoyi.generator.config; + +import org.springframework.beans.factory.annotation.Value; +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.context.annotation.PropertySource; +import org.springframework.stereotype.Component; + +/** + * 读取代码生成相关配置 + * + * @author ruoyi + */ +@Component +@ConfigurationProperties(prefix = "gen") +@PropertySource(value = { "classpath:generator.yml" }) +public class GenConfig +{ + /** 作者 */ + public static String author; + + /** 生成包路径 */ + public static String packageName; + + /** 自动去除表前缀,默认是false */ + public static boolean autoRemovePre; + + /** 表前缀(类名不会包含表前缀) */ + public static String tablePrefix; + + public static String getAuthor() + { + return author; + } + + @Value("${author}") + public void setAuthor(String author) + { + GenConfig.author = author; + } + + public static String getPackageName() + { + return packageName; + } + + @Value("${packageName}") + public void setPackageName(String packageName) + { + GenConfig.packageName = packageName; + } + + public static boolean getAutoRemovePre() + { + return autoRemovePre; + } + + @Value("${autoRemovePre}") + public void setAutoRemovePre(boolean autoRemovePre) + { + GenConfig.autoRemovePre = autoRemovePre; + } + + public static String getTablePrefix() + { + return tablePrefix; + } + + @Value("${tablePrefix}") + public void setTablePrefix(String tablePrefix) + { + GenConfig.tablePrefix = tablePrefix; + } +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/controller/GenController.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/controller/GenController.java new file mode 100644 index 0000000..8521b53 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/controller/GenController.java @@ -0,0 +1,214 @@ +package com.ruoyi.generator.controller; + +import java.io.IOException; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import javax.servlet.http.HttpServletResponse; +import org.apache.commons.io.IOUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.validation.annotation.Validated; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.generator.domain.GenTable; +import com.ruoyi.generator.domain.GenTableColumn; +import com.ruoyi.generator.service.IGenTableColumnService; +import com.ruoyi.generator.service.IGenTableService; + +/** + * 代码生成 操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/tool/gen") +public class GenController extends BaseController +{ + @Autowired + private IGenTableService genTableService; + + @Autowired + private IGenTableColumnService genTableColumnService; + + /** + * 查询代码生成列表 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:list')") + @GetMapping("/list") + public TableDataInfo genList(GenTable genTable) + { + startPage(); + List list = genTableService.selectGenTableList(genTable); + return getDataTable(list); + } + + /** + * 修改代码生成业务 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:query')") + @GetMapping(value = "/{tableId}") + public AjaxResult getInfo(@PathVariable Long tableId) + { + GenTable table = genTableService.selectGenTableById(tableId); + List tables = genTableService.selectGenTableAll(); + List list = genTableColumnService.selectGenTableColumnListByTableId(tableId); + Map map = new HashMap(); + map.put("info", table); + map.put("rows", list); + map.put("tables", tables); + return AjaxResult.success(map); + } + + /** + * 查询数据库列表 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:list')") + @GetMapping("/db/list") + public TableDataInfo dataList(GenTable genTable) + { + startPage(); + List list = genTableService.selectDbTableList(genTable); + return getDataTable(list); + } + + /** + * 查询数据表字段列表 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:list')") + @GetMapping(value = "/column/{tableId}") + public TableDataInfo columnList(Long tableId) + { + TableDataInfo dataInfo = new TableDataInfo(); + List list = genTableColumnService.selectGenTableColumnListByTableId(tableId); + dataInfo.setRows(list); + dataInfo.setTotal(list.size()); + return dataInfo; + } + + /** + * 导入表结构(保存) + */ + @PreAuthorize("@ss.hasPermi('tool:gen:import')") + @Log(title = "代码生成", businessType = BusinessType.IMPORT) + @PostMapping("/importTable") + public AjaxResult importTableSave(String tables) + { + String[] tableNames = Convert.toStrArray(tables); + // 查询表信息 + List tableList = genTableService.selectDbTableListByNames(tableNames); + genTableService.importGenTable(tableList); + return AjaxResult.success(); + } + + /** + * 修改保存代码生成业务 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:edit')") + @Log(title = "代码生成", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult editSave(@Validated @RequestBody GenTable genTable) + { + genTableService.validateEdit(genTable); + genTableService.updateGenTable(genTable); + return AjaxResult.success(); + } + + /** + * 删除代码生成 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:remove')") + @Log(title = "代码生成", businessType = BusinessType.DELETE) + @DeleteMapping("/{tableIds}") + public AjaxResult remove(@PathVariable Long[] tableIds) + { + genTableService.deleteGenTableByIds(tableIds); + return AjaxResult.success(); + } + + /** + * 预览代码 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:preview')") + @GetMapping("/preview/{tableId}") + public AjaxResult preview(@PathVariable("tableId") Long tableId) throws IOException + { + Map dataMap = genTableService.previewCode(tableId); + return AjaxResult.success(dataMap); + } + + /** + * 生成代码(下载方式) + */ + @PreAuthorize("@ss.hasPermi('tool:gen:code')") + @Log(title = "代码生成", businessType = BusinessType.GENCODE) + @GetMapping("/download/{tableName}") + public void download(HttpServletResponse response, @PathVariable("tableName") String tableName) throws IOException + { + byte[] data = genTableService.downloadCode(tableName); + genCode(response, data); + } + + /** + * 生成代码(自定义路径) + */ + @PreAuthorize("@ss.hasPermi('tool:gen:code')") + @Log(title = "代码生成", businessType = BusinessType.GENCODE) + @GetMapping("/genCode/{tableName}") + public AjaxResult genCode(@PathVariable("tableName") String tableName) + { + genTableService.generatorCode(tableName); + return AjaxResult.success(); + } + + /** + * 同步数据库 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:edit')") + @Log(title = "代码生成", businessType = BusinessType.UPDATE) + @GetMapping("/synchDb/{tableName}") + public AjaxResult synchDb(@PathVariable("tableName") String tableName) + { + genTableService.synchDb(tableName); + return AjaxResult.success(); + } + + /** + * 批量生成代码 + */ + @PreAuthorize("@ss.hasPermi('tool:gen:code')") + @Log(title = "代码生成", businessType = BusinessType.GENCODE) + @GetMapping("/batchGenCode") + public void batchGenCode(HttpServletResponse response, String tables) throws IOException + { + String[] tableNames = Convert.toStrArray(tables); + byte[] data = genTableService.downloadCode(tableNames); + genCode(response, data); + } + + /** + * 生成zip文件 + */ + private void genCode(HttpServletResponse response, byte[] data) throws IOException + { + response.reset(); + response.addHeader("Access-Control-Allow-Origin", "*"); + response.addHeader("Access-Control-Expose-Headers", "Content-Disposition"); + response.setHeader("Content-Disposition", "attachment; filename=\"ruoyi.zip\""); + response.addHeader("Content-Length", "" + data.length); + response.setContentType("application/octet-stream; charset=UTF-8"); + IOUtils.write(data, response.getOutputStream()); + } +} \ No newline at end of file diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTable.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTable.java new file mode 100644 index 0000000..269779c --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTable.java @@ -0,0 +1,372 @@ +package com.ruoyi.generator.domain; + +import java.util.List; +import javax.validation.Valid; +import javax.validation.constraints.NotBlank; +import org.apache.commons.lang3.ArrayUtils; +import com.ruoyi.common.constant.GenConstants; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.utils.StringUtils; + +/** + * 业务表 gen_table + * + * @author ruoyi + */ +public class GenTable extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 编号 */ + private Long tableId; + + /** 表名称 */ + @NotBlank(message = "表名称不能为空") + private String tableName; + + /** 表描述 */ + @NotBlank(message = "表描述不能为空") + private String tableComment; + + /** 关联父表的表名 */ + private String subTableName; + + /** 本表关联父表的外键名 */ + private String subTableFkName; + + /** 实体类名称(首字母大写) */ + @NotBlank(message = "实体类名称不能为空") + private String className; + + /** 使用的模板(crud单表操作 tree树表操作 sub主子表操作) */ + private String tplCategory; + + /** 生成包路径 */ + @NotBlank(message = "生成包路径不能为空") + private String packageName; + + /** 生成模块名 */ + @NotBlank(message = "生成模块名不能为空") + private String moduleName; + + /** 生成业务名 */ + @NotBlank(message = "生成业务名不能为空") + private String businessName; + + /** 生成功能名 */ + @NotBlank(message = "生成功能名不能为空") + private String functionName; + + /** 生成作者 */ + @NotBlank(message = "作者不能为空") + private String functionAuthor; + + /** 生成代码方式(0zip压缩包 1自定义路径) */ + private String genType; + + /** 生成路径(不填默认项目路径) */ + private String genPath; + + /** 主键信息 */ + private GenTableColumn pkColumn; + + /** 子表信息 */ + private GenTable subTable; + + /** 表列信息 */ + @Valid + private List columns; + + /** 其它生成选项 */ + private String options; + + /** 树编码字段 */ + private String treeCode; + + /** 树父编码字段 */ + private String treeParentCode; + + /** 树名称字段 */ + private String treeName; + + /** 上级菜单ID字段 */ + private String parentMenuId; + + /** 上级菜单名称字段 */ + private String parentMenuName; + + public Long getTableId() + { + return tableId; + } + + public void setTableId(Long tableId) + { + this.tableId = tableId; + } + + public String getTableName() + { + return tableName; + } + + public void setTableName(String tableName) + { + this.tableName = tableName; + } + + public String getTableComment() + { + return tableComment; + } + + public void setTableComment(String tableComment) + { + this.tableComment = tableComment; + } + + public String getSubTableName() + { + return subTableName; + } + + public void setSubTableName(String subTableName) + { + this.subTableName = subTableName; + } + + public String getSubTableFkName() + { + return subTableFkName; + } + + public void setSubTableFkName(String subTableFkName) + { + this.subTableFkName = subTableFkName; + } + + public String getClassName() + { + return className; + } + + public void setClassName(String className) + { + this.className = className; + } + + public String getTplCategory() + { + return tplCategory; + } + + public void setTplCategory(String tplCategory) + { + this.tplCategory = tplCategory; + } + + public String getPackageName() + { + return packageName; + } + + public void setPackageName(String packageName) + { + this.packageName = packageName; + } + + public String getModuleName() + { + return moduleName; + } + + public void setModuleName(String moduleName) + { + this.moduleName = moduleName; + } + + public String getBusinessName() + { + return businessName; + } + + public void setBusinessName(String businessName) + { + this.businessName = businessName; + } + + public String getFunctionName() + { + return functionName; + } + + public void setFunctionName(String functionName) + { + this.functionName = functionName; + } + + public String getFunctionAuthor() + { + return functionAuthor; + } + + public void setFunctionAuthor(String functionAuthor) + { + this.functionAuthor = functionAuthor; + } + + public String getGenType() + { + return genType; + } + + public void setGenType(String genType) + { + this.genType = genType; + } + + public String getGenPath() + { + return genPath; + } + + public void setGenPath(String genPath) + { + this.genPath = genPath; + } + + public GenTableColumn getPkColumn() + { + return pkColumn; + } + + public void setPkColumn(GenTableColumn pkColumn) + { + this.pkColumn = pkColumn; + } + + public GenTable getSubTable() + { + return subTable; + } + + public void setSubTable(GenTable subTable) + { + this.subTable = subTable; + } + + public List getColumns() + { + return columns; + } + + public void setColumns(List columns) + { + this.columns = columns; + } + + public String getOptions() + { + return options; + } + + public void setOptions(String options) + { + this.options = options; + } + + public String getTreeCode() + { + return treeCode; + } + + public void setTreeCode(String treeCode) + { + this.treeCode = treeCode; + } + + public String getTreeParentCode() + { + return treeParentCode; + } + + public void setTreeParentCode(String treeParentCode) + { + this.treeParentCode = treeParentCode; + } + + public String getTreeName() + { + return treeName; + } + + public void setTreeName(String treeName) + { + this.treeName = treeName; + } + + public String getParentMenuId() + { + return parentMenuId; + } + + public void setParentMenuId(String parentMenuId) + { + this.parentMenuId = parentMenuId; + } + + public String getParentMenuName() + { + return parentMenuName; + } + + public void setParentMenuName(String parentMenuName) + { + this.parentMenuName = parentMenuName; + } + + public boolean isSub() + { + return isSub(this.tplCategory); + } + + public static boolean isSub(String tplCategory) + { + return tplCategory != null && StringUtils.equals(GenConstants.TPL_SUB, tplCategory); + } + + public boolean isTree() + { + return isTree(this.tplCategory); + } + + public static boolean isTree(String tplCategory) + { + return tplCategory != null && StringUtils.equals(GenConstants.TPL_TREE, tplCategory); + } + + public boolean isCrud() + { + return isCrud(this.tplCategory); + } + + public static boolean isCrud(String tplCategory) + { + return tplCategory != null && StringUtils.equals(GenConstants.TPL_CRUD, tplCategory); + } + + public boolean isSuperColumn(String javaField) + { + return isSuperColumn(this.tplCategory, javaField); + } + + public static boolean isSuperColumn(String tplCategory, String javaField) + { + if (isTree(tplCategory)) + { + return StringUtils.equalsAnyIgnoreCase(javaField, + ArrayUtils.addAll(GenConstants.TREE_ENTITY, GenConstants.BASE_ENTITY)); + } + return StringUtils.equalsAnyIgnoreCase(javaField, GenConstants.BASE_ENTITY); + } +} \ No newline at end of file diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTableColumn.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTableColumn.java new file mode 100644 index 0000000..d1733b6 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTableColumn.java @@ -0,0 +1,373 @@ +package com.ruoyi.generator.domain; + +import javax.validation.constraints.NotBlank; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.utils.StringUtils; + +/** + * 代码生成业务字段表 gen_table_column + * + * @author ruoyi + */ +public class GenTableColumn extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 编号 */ + private Long columnId; + + /** 归属表编号 */ + private Long tableId; + + /** 列名称 */ + private String columnName; + + /** 列描述 */ + private String columnComment; + + /** 列类型 */ + private String columnType; + + /** JAVA类型 */ + private String javaType; + + /** JAVA字段名 */ + @NotBlank(message = "Java属性不能为空") + private String javaField; + + /** 是否主键(1是) */ + private String isPk; + + /** 是否自增(1是) */ + private String isIncrement; + + /** 是否必填(1是) */ + private String isRequired; + + /** 是否为插入字段(1是) */ + private String isInsert; + + /** 是否编辑字段(1是) */ + private String isEdit; + + /** 是否列表字段(1是) */ + private String isList; + + /** 是否查询字段(1是) */ + private String isQuery; + + /** 查询方式(EQ等于、NE不等于、GT大于、LT小于、LIKE模糊、BETWEEN范围) */ + private String queryType; + + /** 显示类型(input文本框、textarea文本域、select下拉框、checkbox复选框、radio单选框、datetime日期控件、image图片上传控件、upload文件上传控件、editor富文本控件) */ + private String htmlType; + + /** 字典类型 */ + private String dictType; + + /** 排序 */ + private Integer sort; + + public void setColumnId(Long columnId) + { + this.columnId = columnId; + } + + public Long getColumnId() + { + return columnId; + } + + public void setTableId(Long tableId) + { + this.tableId = tableId; + } + + public Long getTableId() + { + return tableId; + } + + public void setColumnName(String columnName) + { + this.columnName = columnName; + } + + public String getColumnName() + { + return columnName; + } + + public void setColumnComment(String columnComment) + { + this.columnComment = columnComment; + } + + public String getColumnComment() + { + return columnComment; + } + + public void setColumnType(String columnType) + { + this.columnType = columnType; + } + + public String getColumnType() + { + return columnType; + } + + public void setJavaType(String javaType) + { + this.javaType = javaType; + } + + public String getJavaType() + { + return javaType; + } + + public void setJavaField(String javaField) + { + this.javaField = javaField; + } + + public String getJavaField() + { + return javaField; + } + + public String getCapJavaField() + { + return StringUtils.capitalize(javaField); + } + + public void setIsPk(String isPk) + { + this.isPk = isPk; + } + + public String getIsPk() + { + return isPk; + } + + public boolean isPk() + { + return isPk(this.isPk); + } + + public boolean isPk(String isPk) + { + return isPk != null && StringUtils.equals("1", isPk); + } + + public String getIsIncrement() + { + return isIncrement; + } + + public void setIsIncrement(String isIncrement) + { + this.isIncrement = isIncrement; + } + + public boolean isIncrement() + { + return isIncrement(this.isIncrement); + } + + public boolean isIncrement(String isIncrement) + { + return isIncrement != null && StringUtils.equals("1", isIncrement); + } + + public void setIsRequired(String isRequired) + { + this.isRequired = isRequired; + } + + public String getIsRequired() + { + return isRequired; + } + + public boolean isRequired() + { + return isRequired(this.isRequired); + } + + public boolean isRequired(String isRequired) + { + return isRequired != null && StringUtils.equals("1", isRequired); + } + + public void setIsInsert(String isInsert) + { + this.isInsert = isInsert; + } + + public String getIsInsert() + { + return isInsert; + } + + public boolean isInsert() + { + return isInsert(this.isInsert); + } + + public boolean isInsert(String isInsert) + { + return isInsert != null && StringUtils.equals("1", isInsert); + } + + public void setIsEdit(String isEdit) + { + this.isEdit = isEdit; + } + + public String getIsEdit() + { + return isEdit; + } + + public boolean isEdit() + { + return isInsert(this.isEdit); + } + + public boolean isEdit(String isEdit) + { + return isEdit != null && StringUtils.equals("1", isEdit); + } + + public void setIsList(String isList) + { + this.isList = isList; + } + + public String getIsList() + { + return isList; + } + + public boolean isList() + { + return isList(this.isList); + } + + public boolean isList(String isList) + { + return isList != null && StringUtils.equals("1", isList); + } + + public void setIsQuery(String isQuery) + { + this.isQuery = isQuery; + } + + public String getIsQuery() + { + return isQuery; + } + + public boolean isQuery() + { + return isQuery(this.isQuery); + } + + public boolean isQuery(String isQuery) + { + return isQuery != null && StringUtils.equals("1", isQuery); + } + + public void setQueryType(String queryType) + { + this.queryType = queryType; + } + + public String getQueryType() + { + return queryType; + } + + public String getHtmlType() + { + return htmlType; + } + + public void setHtmlType(String htmlType) + { + this.htmlType = htmlType; + } + + public void setDictType(String dictType) + { + this.dictType = dictType; + } + + public String getDictType() + { + return dictType; + } + + public void setSort(Integer sort) + { + this.sort = sort; + } + + public Integer getSort() + { + return sort; + } + + public boolean isSuperColumn() + { + return isSuperColumn(this.javaField); + } + + public static boolean isSuperColumn(String javaField) + { + return StringUtils.equalsAnyIgnoreCase(javaField, + // BaseEntity + "createBy", "createTime", "updateBy", "updateTime", "remark", + // TreeEntity + "parentName", "parentId", "orderNum", "ancestors"); + } + + public boolean isUsableColumn() + { + return isUsableColumn(javaField); + } + + public static boolean isUsableColumn(String javaField) + { + // isSuperColumn()中的名单用于避免生成多余Domain属性,若某些属性在生成页面时需要用到不能忽略,则放在此处白名单 + return StringUtils.equalsAnyIgnoreCase(javaField, "parentId", "orderNum", "remark"); + } + + public String readConverterExp() + { + String remarks = StringUtils.substringBetween(this.columnComment, "(", ")"); + StringBuffer sb = new StringBuffer(); + if (StringUtils.isNotEmpty(remarks)) + { + for (String value : remarks.split(" ")) + { + if (StringUtils.isNotEmpty(value)) + { + Object startStr = value.subSequence(0, 1); + String endStr = value.substring(1); + sb.append("").append(startStr).append("=").append(endStr).append(","); + } + } + return sb.deleteCharAt(sb.length() - 1).toString(); + } + else + { + return this.columnComment; + } + } +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableColumnMapper.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableColumnMapper.java new file mode 100644 index 0000000..951e166 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableColumnMapper.java @@ -0,0 +1,60 @@ +package com.ruoyi.generator.mapper; + +import java.util.List; +import com.ruoyi.generator.domain.GenTableColumn; + +/** + * 业务字段 数据层 + * + * @author ruoyi + */ +public interface GenTableColumnMapper +{ + /** + * 根据表名称查询列信息 + * + * @param tableName 表名称 + * @return 列信息 + */ + public List selectDbTableColumnsByName(String tableName); + + /** + * 查询业务字段列表 + * + * @param tableId 业务字段编号 + * @return 业务字段集合 + */ + public List selectGenTableColumnListByTableId(Long tableId); + + /** + * 新增业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + public int insertGenTableColumn(GenTableColumn genTableColumn); + + /** + * 修改业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + public int updateGenTableColumn(GenTableColumn genTableColumn); + + /** + * 删除业务字段 + * + * @param genTableColumns 列数据 + * @return 结果 + */ + public int deleteGenTableColumns(List genTableColumns); + + /** + * 批量删除业务字段 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteGenTableColumnByIds(Long[] ids); +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableMapper.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableMapper.java new file mode 100644 index 0000000..9b330df --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/mapper/GenTableMapper.java @@ -0,0 +1,83 @@ +package com.ruoyi.generator.mapper; + +import java.util.List; +import com.ruoyi.generator.domain.GenTable; + +/** + * 业务 数据层 + * + * @author ruoyi + */ +public interface GenTableMapper +{ + /** + * 查询业务列表 + * + * @param genTable 业务信息 + * @return 业务集合 + */ + public List selectGenTableList(GenTable genTable); + + /** + * 查询据库列表 + * + * @param genTable 业务信息 + * @return 数据库表集合 + */ + public List selectDbTableList(GenTable genTable); + + /** + * 查询据库列表 + * + * @param tableNames 表名称组 + * @return 数据库表集合 + */ + public List selectDbTableListByNames(String[] tableNames); + + /** + * 查询所有表信息 + * + * @return 表信息集合 + */ + public List selectGenTableAll(); + + /** + * 查询表ID业务信息 + * + * @param id 业务ID + * @return 业务信息 + */ + public GenTable selectGenTableById(Long id); + + /** + * 查询表名称业务信息 + * + * @param tableName 表名称 + * @return 业务信息 + */ + public GenTable selectGenTableByName(String tableName); + + /** + * 新增业务 + * + * @param genTable 业务信息 + * @return 结果 + */ + public int insertGenTable(GenTable genTable); + + /** + * 修改业务 + * + * @param genTable 业务信息 + * @return 结果 + */ + public int updateGenTable(GenTable genTable); + + /** + * 批量删除业务 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteGenTableByIds(Long[] ids); +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableColumnServiceImpl.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableColumnServiceImpl.java new file mode 100644 index 0000000..0679689 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableColumnServiceImpl.java @@ -0,0 +1,68 @@ +package com.ruoyi.generator.service; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.generator.domain.GenTableColumn; +import com.ruoyi.generator.mapper.GenTableColumnMapper; + +/** + * 业务字段 服务层实现 + * + * @author ruoyi + */ +@Service +public class GenTableColumnServiceImpl implements IGenTableColumnService +{ + @Autowired + private GenTableColumnMapper genTableColumnMapper; + + /** + * 查询业务字段列表 + * + * @param tableId 业务字段编号 + * @return 业务字段集合 + */ + @Override + public List selectGenTableColumnListByTableId(Long tableId) + { + return genTableColumnMapper.selectGenTableColumnListByTableId(tableId); + } + + /** + * 新增业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + @Override + public int insertGenTableColumn(GenTableColumn genTableColumn) + { + return genTableColumnMapper.insertGenTableColumn(genTableColumn); + } + + /** + * 修改业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + @Override + public int updateGenTableColumn(GenTableColumn genTableColumn) + { + return genTableColumnMapper.updateGenTableColumn(genTableColumn); + } + + /** + * 删除业务字段对象 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + @Override + public int deleteGenTableColumnByIds(String ids) + { + return genTableColumnMapper.deleteGenTableColumnByIds(Convert.toLongArray(ids)); + } +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java new file mode 100644 index 0000000..168b4c8 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java @@ -0,0 +1,521 @@ +package com.ruoyi.generator.service; + +import java.io.ByteArrayOutputStream; +import java.io.File; +import java.io.IOException; +import java.io.StringWriter; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import java.util.function.Function; +import java.util.stream.Collectors; +import java.util.zip.ZipEntry; +import java.util.zip.ZipOutputStream; +import org.apache.commons.io.FileUtils; +import org.apache.commons.io.IOUtils; +import org.apache.velocity.Template; +import org.apache.velocity.VelocityContext; +import org.apache.velocity.app.Velocity; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.GenConstants; +import com.ruoyi.common.core.text.CharsetKit; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.generator.domain.GenTable; +import com.ruoyi.generator.domain.GenTableColumn; +import com.ruoyi.generator.mapper.GenTableColumnMapper; +import com.ruoyi.generator.mapper.GenTableMapper; +import com.ruoyi.generator.util.GenUtils; +import com.ruoyi.generator.util.VelocityInitializer; +import com.ruoyi.generator.util.VelocityUtils; + +/** + * 业务 服务层实现 + * + * @author ruoyi + */ +@Service +public class GenTableServiceImpl implements IGenTableService +{ + private static final Logger log = LoggerFactory.getLogger(GenTableServiceImpl.class); + + @Autowired + private GenTableMapper genTableMapper; + + @Autowired + private GenTableColumnMapper genTableColumnMapper; + + /** + * 查询业务信息 + * + * @param id 业务ID + * @return 业务信息 + */ + @Override + public GenTable selectGenTableById(Long id) + { + GenTable genTable = genTableMapper.selectGenTableById(id); + setTableFromOptions(genTable); + return genTable; + } + + /** + * 查询业务列表 + * + * @param genTable 业务信息 + * @return 业务集合 + */ + @Override + public List selectGenTableList(GenTable genTable) + { + return genTableMapper.selectGenTableList(genTable); + } + + /** + * 查询据库列表 + * + * @param genTable 业务信息 + * @return 数据库表集合 + */ + @Override + public List selectDbTableList(GenTable genTable) + { + return genTableMapper.selectDbTableList(genTable); + } + + /** + * 查询据库列表 + * + * @param tableNames 表名称组 + * @return 数据库表集合 + */ + @Override + public List selectDbTableListByNames(String[] tableNames) + { + return genTableMapper.selectDbTableListByNames(tableNames); + } + + /** + * 查询所有表信息 + * + * @return 表信息集合 + */ + @Override + public List selectGenTableAll() + { + return genTableMapper.selectGenTableAll(); + } + + /** + * 修改业务 + * + * @param genTable 业务信息 + * @return 结果 + */ + @Override + @Transactional + public void updateGenTable(GenTable genTable) + { + String options = JSON.toJSONString(genTable.getParams()); + genTable.setOptions(options); + int row = genTableMapper.updateGenTable(genTable); + if (row > 0) + { + for (GenTableColumn cenTableColumn : genTable.getColumns()) + { + genTableColumnMapper.updateGenTableColumn(cenTableColumn); + } + } + } + + /** + * 删除业务对象 + * + * @param tableIds 需要删除的数据ID + * @return 结果 + */ + @Override + @Transactional + public void deleteGenTableByIds(Long[] tableIds) + { + genTableMapper.deleteGenTableByIds(tableIds); + genTableColumnMapper.deleteGenTableColumnByIds(tableIds); + } + + /** + * 导入表结构 + * + * @param tableList 导入表列表 + */ + @Override + @Transactional + public void importGenTable(List tableList) + { + String operName = SecurityUtils.getUsername(); + try + { + for (GenTable table : tableList) + { + String tableName = table.getTableName(); + GenUtils.initTable(table, operName); + int row = genTableMapper.insertGenTable(table); + if (row > 0) + { + // 保存列信息 + List genTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName); + for (GenTableColumn column : genTableColumns) + { + GenUtils.initColumnField(column, table); + genTableColumnMapper.insertGenTableColumn(column); + } + } + } + } + catch (Exception e) + { + throw new ServiceException("导入失败:" + e.getMessage()); + } + } + + /** + * 预览代码 + * + * @param tableId 表编号 + * @return 预览数据列表 + */ + @Override + public Map previewCode(Long tableId) + { + Map dataMap = new LinkedHashMap<>(); + // 查询表信息 + GenTable table = genTableMapper.selectGenTableById(tableId); + // 设置主子表信息 + setSubTable(table); + // 设置主键列信息 + setPkColumn(table); + VelocityInitializer.initVelocity(); + + VelocityContext context = VelocityUtils.prepareContext(table); + + // 获取模板列表 + List templates = VelocityUtils.getTemplateList(table.getTplCategory()); + for (String template : templates) + { + // 渲染模板 + StringWriter sw = new StringWriter(); + Template tpl = Velocity.getTemplate(template, Constants.UTF8); + tpl.merge(context, sw); + dataMap.put(template, sw.toString()); + } + return dataMap; + } + + /** + * 生成代码(下载方式) + * + * @param tableName 表名称 + * @return 数据 + */ + @Override + public byte[] downloadCode(String tableName) + { + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); + ZipOutputStream zip = new ZipOutputStream(outputStream); + generatorCode(tableName, zip); + IOUtils.closeQuietly(zip); + return outputStream.toByteArray(); + } + + /** + * 生成代码(自定义路径) + * + * @param tableName 表名称 + */ + @Override + public void generatorCode(String tableName) + { + // 查询表信息 + GenTable table = genTableMapper.selectGenTableByName(tableName); + // 设置主子表信息 + setSubTable(table); + // 设置主键列信息 + setPkColumn(table); + + VelocityInitializer.initVelocity(); + + VelocityContext context = VelocityUtils.prepareContext(table); + + // 获取模板列表 + List templates = VelocityUtils.getTemplateList(table.getTplCategory()); + for (String template : templates) + { + if (!StringUtils.containsAny(template, "sql.vm", "api.js.vm", "index.vue.vm", "index-tree.vue.vm")) + { + // 渲染模板 + StringWriter sw = new StringWriter(); + Template tpl = Velocity.getTemplate(template, Constants.UTF8); + tpl.merge(context, sw); + try + { + String path = getGenPath(table, template); + FileUtils.writeStringToFile(new File(path), sw.toString(), CharsetKit.UTF_8); + } + catch (IOException e) + { + throw new ServiceException("渲染模板失败,表名:" + table.getTableName()); + } + } + } + } + + /** + * 同步数据库 + * + * @param tableName 表名称 + */ + @Override + @Transactional + public void synchDb(String tableName) + { + GenTable table = genTableMapper.selectGenTableByName(tableName); + List tableColumns = table.getColumns(); + Map tableColumnMap = tableColumns.stream().collect(Collectors.toMap(GenTableColumn::getColumnName, Function.identity())); + + List dbTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName); + if (StringUtils.isEmpty(dbTableColumns)) + { + throw new ServiceException("同步数据失败,原表结构不存在"); + } + List dbTableColumnNames = dbTableColumns.stream().map(GenTableColumn::getColumnName).collect(Collectors.toList()); + + dbTableColumns.forEach(column -> { + GenUtils.initColumnField(column, table); + if (tableColumnMap.containsKey(column.getColumnName())) + { + GenTableColumn prevColumn = tableColumnMap.get(column.getColumnName()); + column.setColumnId(prevColumn.getColumnId()); + if (column.isList()) + { + // 如果是列表,继续保留查询方式/字典类型选项 + column.setDictType(prevColumn.getDictType()); + column.setQueryType(prevColumn.getQueryType()); + } + if (StringUtils.isNotEmpty(prevColumn.getIsRequired()) && !column.isPk() + && (column.isInsert() || column.isEdit()) + && ((column.isUsableColumn()) || (!column.isSuperColumn()))) + { + // 如果是(新增/修改&非主键/非忽略及父属性),继续保留必填/显示类型选项 + column.setIsRequired(prevColumn.getIsRequired()); + column.setHtmlType(prevColumn.getHtmlType()); + } + genTableColumnMapper.updateGenTableColumn(column); + } + else + { + genTableColumnMapper.insertGenTableColumn(column); + } + }); + + List delColumns = tableColumns.stream().filter(column -> !dbTableColumnNames.contains(column.getColumnName())).collect(Collectors.toList()); + if (StringUtils.isNotEmpty(delColumns)) + { + genTableColumnMapper.deleteGenTableColumns(delColumns); + } + } + + /** + * 批量生成代码(下载方式) + * + * @param tableNames 表数组 + * @return 数据 + */ + @Override + public byte[] downloadCode(String[] tableNames) + { + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); + ZipOutputStream zip = new ZipOutputStream(outputStream); + for (String tableName : tableNames) + { + generatorCode(tableName, zip); + } + IOUtils.closeQuietly(zip); + return outputStream.toByteArray(); + } + + /** + * 查询表信息并生成代码 + */ + private void generatorCode(String tableName, ZipOutputStream zip) + { + // 查询表信息 + GenTable table = genTableMapper.selectGenTableByName(tableName); + // 设置主子表信息 + setSubTable(table); + // 设置主键列信息 + setPkColumn(table); + + VelocityInitializer.initVelocity(); + + VelocityContext context = VelocityUtils.prepareContext(table); + + // 获取模板列表 + List templates = VelocityUtils.getTemplateList(table.getTplCategory()); + for (String template : templates) + { + // 渲染模板 + StringWriter sw = new StringWriter(); + Template tpl = Velocity.getTemplate(template, Constants.UTF8); + tpl.merge(context, sw); + try + { + // 添加到zip + zip.putNextEntry(new ZipEntry(VelocityUtils.getFileName(template, table))); + IOUtils.write(sw.toString(), zip, Constants.UTF8); + IOUtils.closeQuietly(sw); + zip.flush(); + zip.closeEntry(); + } + catch (IOException e) + { + log.error("渲染模板失败,表名:" + table.getTableName(), e); + } + } + } + + /** + * 修改保存参数校验 + * + * @param genTable 业务信息 + */ + @Override + public void validateEdit(GenTable genTable) + { + if (GenConstants.TPL_TREE.equals(genTable.getTplCategory())) + { + String options = JSON.toJSONString(genTable.getParams()); + JSONObject paramsObj = JSONObject.parseObject(options); + if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_CODE))) + { + throw new ServiceException("树编码字段不能为空"); + } + else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_PARENT_CODE))) + { + throw new ServiceException("树父编码字段不能为空"); + } + else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_NAME))) + { + throw new ServiceException("树名称字段不能为空"); + } + else if (GenConstants.TPL_SUB.equals(genTable.getTplCategory())) + { + if (StringUtils.isEmpty(genTable.getSubTableName())) + { + throw new ServiceException("关联子表的表名不能为空"); + } + else if (StringUtils.isEmpty(genTable.getSubTableFkName())) + { + throw new ServiceException("子表关联的外键名不能为空"); + } + } + } + } + + /** + * 设置主键列信息 + * + * @param table 业务表信息 + */ + public void setPkColumn(GenTable table) + { + for (GenTableColumn column : table.getColumns()) + { + if (column.isPk()) + { + table.setPkColumn(column); + break; + } + } + if (StringUtils.isNull(table.getPkColumn())) + { + table.setPkColumn(table.getColumns().get(0)); + } + if (GenConstants.TPL_SUB.equals(table.getTplCategory())) + { + for (GenTableColumn column : table.getSubTable().getColumns()) + { + if (column.isPk()) + { + table.getSubTable().setPkColumn(column); + break; + } + } + if (StringUtils.isNull(table.getSubTable().getPkColumn())) + { + table.getSubTable().setPkColumn(table.getSubTable().getColumns().get(0)); + } + } + } + + /** + * 设置主子表信息 + * + * @param table 业务表信息 + */ + public void setSubTable(GenTable table) + { + String subTableName = table.getSubTableName(); + if (StringUtils.isNotEmpty(subTableName)) + { + table.setSubTable(genTableMapper.selectGenTableByName(subTableName)); + } + } + + /** + * 设置代码生成其他选项值 + * + * @param genTable 设置后的生成对象 + */ + public void setTableFromOptions(GenTable genTable) + { + JSONObject paramsObj = JSONObject.parseObject(genTable.getOptions()); + if (StringUtils.isNotNull(paramsObj)) + { + String treeCode = paramsObj.getString(GenConstants.TREE_CODE); + String treeParentCode = paramsObj.getString(GenConstants.TREE_PARENT_CODE); + String treeName = paramsObj.getString(GenConstants.TREE_NAME); + String parentMenuId = paramsObj.getString(GenConstants.PARENT_MENU_ID); + String parentMenuName = paramsObj.getString(GenConstants.PARENT_MENU_NAME); + + genTable.setTreeCode(treeCode); + genTable.setTreeParentCode(treeParentCode); + genTable.setTreeName(treeName); + genTable.setParentMenuId(parentMenuId); + genTable.setParentMenuName(parentMenuName); + } + } + + /** + * 获取代码生成地址 + * + * @param table 业务表信息 + * @param template 模板文件路径 + * @return 生成地址 + */ + public static String getGenPath(GenTable table, String template) + { + String genPath = table.getGenPath(); + if (StringUtils.equals(genPath, "/")) + { + return System.getProperty("user.dir") + File.separator + "src" + File.separator + VelocityUtils.getFileName(template, table); + } + return genPath + File.separator + VelocityUtils.getFileName(template, table); + } +} \ No newline at end of file diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableColumnService.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableColumnService.java new file mode 100644 index 0000000..3037f70 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableColumnService.java @@ -0,0 +1,44 @@ +package com.ruoyi.generator.service; + +import java.util.List; +import com.ruoyi.generator.domain.GenTableColumn; + +/** + * 业务字段 服务层 + * + * @author ruoyi + */ +public interface IGenTableColumnService +{ + /** + * 查询业务字段列表 + * + * @param tableId 业务字段编号 + * @return 业务字段集合 + */ + public List selectGenTableColumnListByTableId(Long tableId); + + /** + * 新增业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + public int insertGenTableColumn(GenTableColumn genTableColumn); + + /** + * 修改业务字段 + * + * @param genTableColumn 业务字段信息 + * @return 结果 + */ + public int updateGenTableColumn(GenTableColumn genTableColumn); + + /** + * 删除业务字段信息 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteGenTableColumnByIds(String ids); +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableService.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableService.java new file mode 100644 index 0000000..9d53f95 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/IGenTableService.java @@ -0,0 +1,121 @@ +package com.ruoyi.generator.service; + +import java.util.List; +import java.util.Map; +import com.ruoyi.generator.domain.GenTable; + +/** + * 业务 服务层 + * + * @author ruoyi + */ +public interface IGenTableService +{ + /** + * 查询业务列表 + * + * @param genTable 业务信息 + * @return 业务集合 + */ + public List selectGenTableList(GenTable genTable); + + /** + * 查询据库列表 + * + * @param genTable 业务信息 + * @return 数据库表集合 + */ + public List selectDbTableList(GenTable genTable); + + /** + * 查询据库列表 + * + * @param tableNames 表名称组 + * @return 数据库表集合 + */ + public List selectDbTableListByNames(String[] tableNames); + + /** + * 查询所有表信息 + * + * @return 表信息集合 + */ + public List selectGenTableAll(); + + /** + * 查询业务信息 + * + * @param id 业务ID + * @return 业务信息 + */ + public GenTable selectGenTableById(Long id); + + /** + * 修改业务 + * + * @param genTable 业务信息 + * @return 结果 + */ + public void updateGenTable(GenTable genTable); + + /** + * 删除业务信息 + * + * @param tableIds 需要删除的表数据ID + * @return 结果 + */ + public void deleteGenTableByIds(Long[] tableIds); + + /** + * 导入表结构 + * + * @param tableList 导入表列表 + */ + public void importGenTable(List tableList); + + /** + * 预览代码 + * + * @param tableId 表编号 + * @return 预览数据列表 + */ + public Map previewCode(Long tableId); + + /** + * 生成代码(下载方式) + * + * @param tableName 表名称 + * @return 数据 + */ + public byte[] downloadCode(String tableName); + + /** + * 生成代码(自定义路径) + * + * @param tableName 表名称 + * @return 数据 + */ + public void generatorCode(String tableName); + + /** + * 同步数据库 + * + * @param tableName 表名称 + */ + public void synchDb(String tableName); + + /** + * 批量生成代码(下载方式) + * + * @param tableNames 表数组 + * @return 数据 + */ + public byte[] downloadCode(String[] tableNames); + + /** + * 修改保存参数校验 + * + * @param genTable 业务信息 + */ + public void validateEdit(GenTable genTable); +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/util/GenUtils.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/GenUtils.java new file mode 100644 index 0000000..e7ebc20 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/GenUtils.java @@ -0,0 +1,257 @@ +package com.ruoyi.generator.util; + +import java.util.Arrays; +import org.apache.commons.lang3.RegExUtils; +import com.ruoyi.common.constant.GenConstants; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.generator.config.GenConfig; +import com.ruoyi.generator.domain.GenTable; +import com.ruoyi.generator.domain.GenTableColumn; + +/** + * 代码生成器 工具类 + * + * @author ruoyi + */ +public class GenUtils +{ + /** + * 初始化表信息 + */ + public static void initTable(GenTable genTable, String operName) + { + genTable.setClassName(convertClassName(genTable.getTableName())); + genTable.setPackageName(GenConfig.getPackageName()); + genTable.setModuleName(getModuleName(GenConfig.getPackageName())); + genTable.setBusinessName(getBusinessName(genTable.getTableName())); + genTable.setFunctionName(replaceText(genTable.getTableComment())); + genTable.setFunctionAuthor(GenConfig.getAuthor()); + genTable.setCreateBy(operName); + } + + /** + * 初始化列属性字段 + */ + public static void initColumnField(GenTableColumn column, GenTable table) + { + String dataType = getDbType(column.getColumnType()); + String columnName = column.getColumnName(); + column.setTableId(table.getTableId()); + column.setCreateBy(table.getCreateBy()); + // 设置java字段名 + column.setJavaField(StringUtils.toCamelCase(columnName)); + // 设置默认类型 + column.setJavaType(GenConstants.TYPE_STRING); + column.setQueryType(GenConstants.QUERY_EQ); + + if (arraysContains(GenConstants.COLUMNTYPE_STR, dataType) || arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType)) + { + // 字符串长度超过500设置为文本域 + Integer columnLength = getColumnLength(column.getColumnType()); + String htmlType = columnLength >= 500 || arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType) ? GenConstants.HTML_TEXTAREA : GenConstants.HTML_INPUT; + column.setHtmlType(htmlType); + } + else if (arraysContains(GenConstants.COLUMNTYPE_TIME, dataType)) + { + column.setJavaType(GenConstants.TYPE_DATE); + column.setHtmlType(GenConstants.HTML_DATETIME); + } + else if (arraysContains(GenConstants.COLUMNTYPE_NUMBER, dataType)) + { + column.setHtmlType(GenConstants.HTML_INPUT); + + // 如果是浮点型 统一用BigDecimal + String[] str = StringUtils.split(StringUtils.substringBetween(column.getColumnType(), "(", ")"), ","); + if (str != null && str.length == 2 && Integer.parseInt(str[1]) > 0) + { + column.setJavaType(GenConstants.TYPE_BIGDECIMAL); + } + // 如果是整形 + else if (str != null && str.length == 1 && Integer.parseInt(str[0]) <= 10) + { + column.setJavaType(GenConstants.TYPE_INTEGER); + } + // 长整形 + else + { + column.setJavaType(GenConstants.TYPE_LONG); + } + } + + // 插入字段(默认所有字段都需要插入) + column.setIsInsert(GenConstants.REQUIRE); + + // 编辑字段 + if (!arraysContains(GenConstants.COLUMNNAME_NOT_EDIT, columnName) && !column.isPk()) + { + column.setIsEdit(GenConstants.REQUIRE); + } + // 列表字段 + if (!arraysContains(GenConstants.COLUMNNAME_NOT_LIST, columnName) && !column.isPk()) + { + column.setIsList(GenConstants.REQUIRE); + } + // 查询字段 + if (!arraysContains(GenConstants.COLUMNNAME_NOT_QUERY, columnName) && !column.isPk()) + { + column.setIsQuery(GenConstants.REQUIRE); + } + + // 查询字段类型 + if (StringUtils.endsWithIgnoreCase(columnName, "name")) + { + column.setQueryType(GenConstants.QUERY_LIKE); + } + // 状态字段设置单选框 + if (StringUtils.endsWithIgnoreCase(columnName, "status")) + { + column.setHtmlType(GenConstants.HTML_RADIO); + } + // 类型&性别字段设置下拉框 + else if (StringUtils.endsWithIgnoreCase(columnName, "type") + || StringUtils.endsWithIgnoreCase(columnName, "sex")) + { + column.setHtmlType(GenConstants.HTML_SELECT); + } + // 图片字段设置图片上传控件 + else if (StringUtils.endsWithIgnoreCase(columnName, "image")) + { + column.setHtmlType(GenConstants.HTML_IMAGE_UPLOAD); + } + // 文件字段设置文件上传控件 + else if (StringUtils.endsWithIgnoreCase(columnName, "file")) + { + column.setHtmlType(GenConstants.HTML_FILE_UPLOAD); + } + // 内容字段设置富文本控件 + else if (StringUtils.endsWithIgnoreCase(columnName, "content")) + { + column.setHtmlType(GenConstants.HTML_EDITOR); + } + } + + /** + * 校验数组是否包含指定值 + * + * @param arr 数组 + * @param targetValue 值 + * @return 是否包含 + */ + public static boolean arraysContains(String[] arr, String targetValue) + { + return Arrays.asList(arr).contains(targetValue); + } + + /** + * 获取模块名 + * + * @param packageName 包名 + * @return 模块名 + */ + public static String getModuleName(String packageName) + { + int lastIndex = packageName.lastIndexOf("."); + int nameLength = packageName.length(); + return StringUtils.substring(packageName, lastIndex + 1, nameLength); + } + + /** + * 获取业务名 + * + * @param tableName 表名 + * @return 业务名 + */ + public static String getBusinessName(String tableName) + { + int lastIndex = tableName.lastIndexOf("_"); + int nameLength = tableName.length(); + return StringUtils.substring(tableName, lastIndex + 1, nameLength); + } + + /** + * 表名转换成Java类名 + * + * @param tableName 表名称 + * @return 类名 + */ + public static String convertClassName(String tableName) + { + boolean autoRemovePre = GenConfig.getAutoRemovePre(); + String tablePrefix = GenConfig.getTablePrefix(); + if (autoRemovePre && StringUtils.isNotEmpty(tablePrefix)) + { + String[] searchList = StringUtils.split(tablePrefix, ","); + tableName = replaceFirst(tableName, searchList); + } + return StringUtils.convertToCamelCase(tableName); + } + + /** + * 批量替换前缀 + * + * @param replacementm 替换值 + * @param searchList 替换列表 + * @return + */ + public static String replaceFirst(String replacementm, String[] searchList) + { + String text = replacementm; + for (String searchString : searchList) + { + if (replacementm.startsWith(searchString)) + { + text = replacementm.replaceFirst(searchString, ""); + break; + } + } + return text; + } + + /** + * 关键字替换 + * + * @param text 需要被替换的名字 + * @return 替换后的名字 + */ + public static String replaceText(String text) + { + return RegExUtils.replaceAll(text, "(?:表|若依)", ""); + } + + /** + * 获取数据库类型字段 + * + * @param columnType 列类型 + * @return 截取后的列类型 + */ + public static String getDbType(String columnType) + { + if (StringUtils.indexOf(columnType, "(") > 0) + { + return StringUtils.substringBefore(columnType, "("); + } + else + { + return columnType; + } + } + + /** + * 获取字段长度 + * + * @param columnType 列类型 + * @return 截取后的列类型 + */ + public static Integer getColumnLength(String columnType) + { + if (StringUtils.indexOf(columnType, "(") > 0) + { + String length = StringUtils.substringBetween(columnType, "(", ")"); + return Integer.valueOf(length); + } + else + { + return 0; + } + } +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityInitializer.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityInitializer.java new file mode 100644 index 0000000..9f69403 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityInitializer.java @@ -0,0 +1,34 @@ +package com.ruoyi.generator.util; + +import java.util.Properties; +import org.apache.velocity.app.Velocity; +import com.ruoyi.common.constant.Constants; + +/** + * VelocityEngine工厂 + * + * @author ruoyi + */ +public class VelocityInitializer +{ + /** + * 初始化vm方法 + */ + public static void initVelocity() + { + Properties p = new Properties(); + try + { + // 加载classpath目录下的vm文件 + p.setProperty("resource.loader.file.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader"); + // 定义字符集 + p.setProperty(Velocity.INPUT_ENCODING, Constants.UTF8); + // 初始化Velocity引擎,指定配置Properties + Velocity.init(p); + } + catch (Exception e) + { + throw new RuntimeException(e); + } + } +} diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java new file mode 100644 index 0000000..9a05b69 --- /dev/null +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java @@ -0,0 +1,401 @@ +package com.ruoyi.generator.util; + +import java.util.ArrayList; +import java.util.HashSet; +import java.util.List; +import java.util.Set; +import org.apache.velocity.VelocityContext; +import com.alibaba.fastjson.JSONObject; +import com.ruoyi.common.constant.GenConstants; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.generator.domain.GenTable; +import com.ruoyi.generator.domain.GenTableColumn; + +/** + * 模板处理工具类 + * + * @author ruoyi + */ +public class VelocityUtils +{ + /** 项目空间路径 */ + private static final String PROJECT_PATH = "main/java"; + + /** mybatis空间路径 */ + private static final String MYBATIS_PATH = "main/resources/mapper"; + + /** 默认上级菜单,系统工具 */ + private static final String DEFAULT_PARENT_MENU_ID = "3"; + + /** + * 设置模板变量信息 + * + * @return 模板列表 + */ + public static VelocityContext prepareContext(GenTable genTable) + { + String moduleName = genTable.getModuleName(); + String businessName = genTable.getBusinessName(); + String packageName = genTable.getPackageName(); + String tplCategory = genTable.getTplCategory(); + String functionName = genTable.getFunctionName(); + + VelocityContext velocityContext = new VelocityContext(); + velocityContext.put("tplCategory", genTable.getTplCategory()); + velocityContext.put("tableName", genTable.getTableName()); + velocityContext.put("functionName", StringUtils.isNotEmpty(functionName) ? functionName : "【请填写功能名称】"); + velocityContext.put("ClassName", genTable.getClassName()); + velocityContext.put("className", StringUtils.uncapitalize(genTable.getClassName())); + velocityContext.put("moduleName", genTable.getModuleName()); + velocityContext.put("BusinessName", StringUtils.capitalize(genTable.getBusinessName())); + velocityContext.put("businessName", genTable.getBusinessName()); + velocityContext.put("basePackage", getPackagePrefix(packageName)); + velocityContext.put("packageName", packageName); + velocityContext.put("author", genTable.getFunctionAuthor()); + velocityContext.put("datetime", DateUtils.getDate()); + velocityContext.put("pkColumn", genTable.getPkColumn()); + velocityContext.put("importList", getImportList(genTable)); + velocityContext.put("permissionPrefix", getPermissionPrefix(moduleName, businessName)); + velocityContext.put("columns", genTable.getColumns()); + velocityContext.put("table", genTable); + velocityContext.put("dicts", getDicts(genTable)); + setMenuVelocityContext(velocityContext, genTable); + if (GenConstants.TPL_TREE.equals(tplCategory)) + { + setTreeVelocityContext(velocityContext, genTable); + } + if (GenConstants.TPL_SUB.equals(tplCategory)) + { + setSubVelocityContext(velocityContext, genTable); + } + return velocityContext; + } + + public static void setMenuVelocityContext(VelocityContext context, GenTable genTable) + { + String options = genTable.getOptions(); + JSONObject paramsObj = JSONObject.parseObject(options); + String parentMenuId = getParentMenuId(paramsObj); + context.put("parentMenuId", parentMenuId); + } + + public static void setTreeVelocityContext(VelocityContext context, GenTable genTable) + { + String options = genTable.getOptions(); + JSONObject paramsObj = JSONObject.parseObject(options); + String treeCode = getTreecode(paramsObj); + String treeParentCode = getTreeParentCode(paramsObj); + String treeName = getTreeName(paramsObj); + + context.put("treeCode", treeCode); + context.put("treeParentCode", treeParentCode); + context.put("treeName", treeName); + context.put("expandColumn", getExpandColumn(genTable)); + if (paramsObj.containsKey(GenConstants.TREE_PARENT_CODE)) + { + context.put("tree_parent_code", paramsObj.getString(GenConstants.TREE_PARENT_CODE)); + } + if (paramsObj.containsKey(GenConstants.TREE_NAME)) + { + context.put("tree_name", paramsObj.getString(GenConstants.TREE_NAME)); + } + } + + public static void setSubVelocityContext(VelocityContext context, GenTable genTable) + { + GenTable subTable = genTable.getSubTable(); + String subTableName = genTable.getSubTableName(); + String subTableFkName = genTable.getSubTableFkName(); + String subClassName = genTable.getSubTable().getClassName(); + String subTableFkClassName = StringUtils.convertToCamelCase(subTableFkName); + + context.put("subTable", subTable); + context.put("subTableName", subTableName); + context.put("subTableFkName", subTableFkName); + context.put("subTableFkClassName", subTableFkClassName); + context.put("subTableFkclassName", StringUtils.uncapitalize(subTableFkClassName)); + context.put("subClassName", subClassName); + context.put("subclassName", StringUtils.uncapitalize(subClassName)); + context.put("subImportList", getImportList(genTable.getSubTable())); + } + + /** + * 获取模板信息 + * + * @return 模板列表 + */ + public static List getTemplateList(String tplCategory) + { + List templates = new ArrayList(); + templates.add("vm/java/domain.java.vm"); + templates.add("vm/java/mapper.java.vm"); + templates.add("vm/java/service.java.vm"); + templates.add("vm/java/serviceImpl.java.vm"); + templates.add("vm/java/controller.java.vm"); + templates.add("vm/xml/mapper.xml.vm"); + templates.add("vm/sql/sql.vm"); + templates.add("vm/js/api.js.vm"); + if (GenConstants.TPL_CRUD.equals(tplCategory)) + { + templates.add("vm/vue/index.vue.vm"); + } + else if (GenConstants.TPL_TREE.equals(tplCategory)) + { + templates.add("vm/vue/index-tree.vue.vm"); + } + else if (GenConstants.TPL_SUB.equals(tplCategory)) + { + templates.add("vm/vue/index.vue.vm"); + templates.add("vm/java/sub-domain.java.vm"); + } + return templates; + } + + /** + * 获取文件名 + */ + public static String getFileName(String template, GenTable genTable) + { + // 文件名称 + String fileName = ""; + // 包路径 + String packageName = genTable.getPackageName(); + // 模块名 + String moduleName = genTable.getModuleName(); + // 大写类名 + String className = genTable.getClassName(); + // 业务名称 + String businessName = genTable.getBusinessName(); + + String javaPath = PROJECT_PATH + "/" + StringUtils.replace(packageName, ".", "/"); + String mybatisPath = MYBATIS_PATH + "/" + moduleName; + String vuePath = "vue"; + + if (template.contains("domain.java.vm")) + { + fileName = StringUtils.format("{}/domain/{}.java", javaPath, className); + } + if (template.contains("sub-domain.java.vm") && StringUtils.equals(GenConstants.TPL_SUB, genTable.getTplCategory())) + { + fileName = StringUtils.format("{}/domain/{}.java", javaPath, genTable.getSubTable().getClassName()); + } + else if (template.contains("mapper.java.vm")) + { + fileName = StringUtils.format("{}/mapper/{}Mapper.java", javaPath, className); + } + else if (template.contains("service.java.vm")) + { + fileName = StringUtils.format("{}/service/I{}Service.java", javaPath, className); + } + else if (template.contains("serviceImpl.java.vm")) + { + fileName = StringUtils.format("{}/service/impl/{}ServiceImpl.java", javaPath, className); + } + else if (template.contains("controller.java.vm")) + { + fileName = StringUtils.format("{}/controller/{}Controller.java", javaPath, className); + } + else if (template.contains("mapper.xml.vm")) + { + fileName = StringUtils.format("{}/{}Mapper.xml", mybatisPath, className); + } + else if (template.contains("sql.vm")) + { + fileName = businessName + "Menu.sql"; + } + else if (template.contains("api.js.vm")) + { + fileName = StringUtils.format("{}/api/{}/{}.js", vuePath, moduleName, businessName); + } + else if (template.contains("index.vue.vm")) + { + fileName = StringUtils.format("{}/views/{}/{}/index.vue", vuePath, moduleName, businessName); + } + else if (template.contains("index-tree.vue.vm")) + { + fileName = StringUtils.format("{}/views/{}/{}/index.vue", vuePath, moduleName, businessName); + } + return fileName; + } + + /** + * 获取包前缀 + * + * @param packageName 包名称 + * @return 包前缀名称 + */ + public static String getPackagePrefix(String packageName) + { + int lastIndex = packageName.lastIndexOf("."); + return StringUtils.substring(packageName, 0, lastIndex); + } + + /** + * 根据列类型获取导入包 + * + * @param genTable 业务表对象 + * @return 返回需要导入的包列表 + */ + public static HashSet getImportList(GenTable genTable) + { + List columns = genTable.getColumns(); + GenTable subGenTable = genTable.getSubTable(); + HashSet importList = new HashSet(); + if (StringUtils.isNotNull(subGenTable)) + { + importList.add("java.util.List"); + } + for (GenTableColumn column : columns) + { + if (!column.isSuperColumn() && GenConstants.TYPE_DATE.equals(column.getJavaType())) + { + importList.add("java.util.Date"); + importList.add("com.fasterxml.jackson.annotation.JsonFormat"); + } + else if (!column.isSuperColumn() && GenConstants.TYPE_BIGDECIMAL.equals(column.getJavaType())) + { + importList.add("java.math.BigDecimal"); + } + } + return importList; + } + + /** + * 根据列类型获取字典组 + * + * @param genTable 业务表对象 + * @return 返回字典组 + */ + public static String getDicts(GenTable genTable) + { + List columns = genTable.getColumns(); + Set dicts = new HashSet(); + addDicts(dicts, columns); + if (StringUtils.isNotNull(genTable.getSubTable())) + { + List subColumns = genTable.getSubTable().getColumns(); + addDicts(dicts, subColumns); + } + return StringUtils.join(dicts, ", "); + } + + /** + * 添加字典列表 + * + * @param dicts 字典列表 + * @param columns 列集合 + */ + public static void addDicts(Set dicts, List columns) + { + for (GenTableColumn column : columns) + { + if (!column.isSuperColumn() && StringUtils.isNotEmpty(column.getDictType()) && StringUtils.equalsAny( + column.getHtmlType(), + new String[] { GenConstants.HTML_SELECT, GenConstants.HTML_RADIO, GenConstants.HTML_CHECKBOX })) + { + dicts.add("'" + column.getDictType() + "'"); + } + } + } + + /** + * 获取权限前缀 + * + * @param moduleName 模块名称 + * @param businessName 业务名称 + * @return 返回权限前缀 + */ + public static String getPermissionPrefix(String moduleName, String businessName) + { + return StringUtils.format("{}:{}", moduleName, businessName); + } + + /** + * 获取上级菜单ID字段 + * + * @param paramsObj 生成其他选项 + * @return 上级菜单ID字段 + */ + public static String getParentMenuId(JSONObject paramsObj) + { + if (StringUtils.isNotEmpty(paramsObj) && paramsObj.containsKey(GenConstants.PARENT_MENU_ID) + && StringUtils.isNotEmpty(paramsObj.getString(GenConstants.PARENT_MENU_ID))) + { + return paramsObj.getString(GenConstants.PARENT_MENU_ID); + } + return DEFAULT_PARENT_MENU_ID; + } + + /** + * 获取树编码 + * + * @param paramsObj 生成其他选项 + * @return 树编码 + */ + public static String getTreecode(JSONObject paramsObj) + { + if (paramsObj.containsKey(GenConstants.TREE_CODE)) + { + return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_CODE)); + } + return StringUtils.EMPTY; + } + + /** + * 获取树父编码 + * + * @param paramsObj 生成其他选项 + * @return 树父编码 + */ + public static String getTreeParentCode(JSONObject paramsObj) + { + if (paramsObj.containsKey(GenConstants.TREE_PARENT_CODE)) + { + return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_PARENT_CODE)); + } + return StringUtils.EMPTY; + } + + /** + * 获取树名称 + * + * @param paramsObj 生成其他选项 + * @return 树名称 + */ + public static String getTreeName(JSONObject paramsObj) + { + if (paramsObj.containsKey(GenConstants.TREE_NAME)) + { + return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_NAME)); + } + return StringUtils.EMPTY; + } + + /** + * 获取需要在哪一列上面显示展开按钮 + * + * @param genTable 业务表对象 + * @return 展开按钮列序号 + */ + public static int getExpandColumn(GenTable genTable) + { + String options = genTable.getOptions(); + JSONObject paramsObj = JSONObject.parseObject(options); + String treeName = paramsObj.getString(GenConstants.TREE_NAME); + int num = 0; + for (GenTableColumn column : genTable.getColumns()) + { + if (column.isList()) + { + num++; + String columnName = column.getColumnName(); + if (columnName.equals(treeName)) + { + break; + } + } + } + return num; + } +} diff --git a/ruoyi-generator/src/main/resources/generator.yml b/ruoyi-generator/src/main/resources/generator.yml new file mode 100644 index 0000000..5bd3dd6 --- /dev/null +++ b/ruoyi-generator/src/main/resources/generator.yml @@ -0,0 +1,10 @@ +# 代码生成 +gen: + # 作者 + author: ruoyi + # 默认生成包路径 system 需改成自己的模块名称 如 system monitor tool + packageName: com.ruoyi.system + # 自动去除表前缀,默认是false + autoRemovePre: false + # 表前缀(生成类名不会包含表前缀,多个用逗号分隔) + tablePrefix: sys_ \ No newline at end of file diff --git a/ruoyi-generator/src/main/resources/mapper/generator/GenTableColumnMapper.xml b/ruoyi-generator/src/main/resources/mapper/generator/GenTableColumnMapper.xml new file mode 100644 index 0000000..5fa790f --- /dev/null +++ b/ruoyi-generator/src/main/resources/mapper/generator/GenTableColumnMapper.xml @@ -0,0 +1,127 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + select column_id, table_id, column_name, column_comment, column_type, java_type, java_field, is_pk, is_increment, is_required, is_insert, is_edit, is_list, is_query, query_type, html_type, dict_type, sort, create_by, create_time, update_by, update_time from gen_table_column + + + + + + + + insert into gen_table_column ( + table_id, + column_name, + column_comment, + column_type, + java_type, + java_field, + is_pk, + is_increment, + is_required, + is_insert, + is_edit, + is_list, + is_query, + query_type, + html_type, + dict_type, + sort, + create_by, + create_time + )values( + #{tableId}, + #{columnName}, + #{columnComment}, + #{columnType}, + #{javaType}, + #{javaField}, + #{isPk}, + #{isIncrement}, + #{isRequired}, + #{isInsert}, + #{isEdit}, + #{isList}, + #{isQuery}, + #{queryType}, + #{htmlType}, + #{dictType}, + #{sort}, + #{createBy}, + sysdate() + ) + + + + update gen_table_column + + column_comment = #{columnComment}, + java_type = #{javaType}, + java_field = #{javaField}, + is_insert = #{isInsert}, + is_edit = #{isEdit}, + is_list = #{isList}, + is_query = #{isQuery}, + is_required = #{isRequired}, + query_type = #{queryType}, + html_type = #{htmlType}, + dict_type = #{dictType}, + sort = #{sort}, + update_by = #{updateBy}, + update_time = sysdate() + + where column_id = #{columnId} + + + + delete from gen_table_column where table_id in + + #{tableId} + + + + + delete from gen_table_column where column_id in + + #{item.columnId} + + + + \ No newline at end of file diff --git a/ruoyi-generator/src/main/resources/mapper/generator/GenTableMapper.xml b/ruoyi-generator/src/main/resources/mapper/generator/GenTableMapper.xml new file mode 100644 index 0000000..b605e90 --- /dev/null +++ b/ruoyi-generator/src/main/resources/mapper/generator/GenTableMapper.xml @@ -0,0 +1,202 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + select table_id, table_name, table_comment, sub_table_name, sub_table_fk_name, class_name, tpl_category, package_name, module_name, business_name, function_name, function_author, gen_type, gen_path, options, create_by, create_time, update_by, update_time, remark from gen_table + + + + + + + + + + + + + + + + + + insert into gen_table ( + table_name, + table_comment, + class_name, + tpl_category, + package_name, + module_name, + business_name, + function_name, + function_author, + gen_type, + gen_path, + remark, + create_by, + create_time + )values( + #{tableName}, + #{tableComment}, + #{className}, + #{tplCategory}, + #{packageName}, + #{moduleName}, + #{businessName}, + #{functionName}, + #{functionAuthor}, + #{genType}, + #{genPath}, + #{remark}, + #{createBy}, + sysdate() + ) + + + + update gen_table + + table_name = #{tableName}, + table_comment = #{tableComment}, + sub_table_name = #{subTableName}, + sub_table_fk_name = #{subTableFkName}, + class_name = #{className}, + function_author = #{functionAuthor}, + gen_type = #{genType}, + gen_path = #{genPath}, + tpl_category = #{tplCategory}, + package_name = #{packageName}, + module_name = #{moduleName}, + business_name = #{businessName}, + function_name = #{functionName}, + options = #{options}, + update_by = #{updateBy}, + remark = #{remark}, + update_time = sysdate() + + where table_id = #{tableId} + + + + delete from gen_table where table_id in + + #{tableId} + + + + \ No newline at end of file diff --git a/ruoyi-generator/src/main/resources/vm/java/controller.java.vm b/ruoyi-generator/src/main/resources/vm/java/controller.java.vm new file mode 100644 index 0000000..00e0dd9 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/controller.java.vm @@ -0,0 +1,106 @@ +package ${packageName}.controller; + +import java.util.List; +import java.util.Arrays; +import java.util.Map; +import javax.servlet.http.HttpServletResponse; +import javax.annotation.Resource; +import com.ruoyi.common.core.page.R; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.enums.BusinessType; +import ${packageName}.domain.${ClassName}; +import ${packageName}.service.I${ClassName}Service; +import com.ruoyi.common.utils.poi.ExcelUtil; +/** + * ${functionName}Controller + * + * @author ${author} + * @date ${datetime} + */ +@RestController +@RequestMapping("/${moduleName}/${businessName}") +public class ${ClassName}Controller extends BaseController +{ + @Resource + private I${ClassName}Service ${className}Service; + + /** + * 查询列表 + */ + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:list')") + @RequestMapping("/list") + public R list(@RequestParam Map params){ + return ${className}Service.queryPage(params); + } + + /** + * 获取详细信息 + */ + @RequestMapping("/info/{${pkColumn.javaField}}") + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:query')") + public R info(@PathVariable("${pkColumn.javaField}") ${pkColumn.javaType} ${pkColumn.javaField}){ + ${ClassName} ${className} = ${className}Service.getById(${pkColumn.javaField}); + + return R.ok().put("data", ${className}); + } + + + /** + * 导出列表 + */ + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:export')") + @Log(title = "列表", businessType = BusinessType.EXPORT) + @PostMapping("/export") + public void export(HttpServletResponse response, ${ClassName} ${className}) + { + List<${ClassName}> list = ${className}Service.list(); + ExcelUtil<${ClassName}> util = new ExcelUtil<${ClassName}>(${ClassName}.class); + util.exportExcel(response, list, "列表数据"); + } + + /** + * 新增 + */ + @RequestMapping("/add") + @Log(title = "${ClassName}", businessType = BusinessType.INSERT) + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:add')") + public R add(@RequestBody ${ClassName} ${className}){ + ${className}Service.save(${className}); + return R.ok(); + } + + /** + * 修改 + */ + @RequestMapping("/edit") + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:edit')") + @Log(title = "${ClassName}", businessType = BusinessType.UPDATE) + public R edit(@RequestBody ${ClassName} ${className}){ + ${className}Service.updateById(${className}); + return R.ok(); + } + + /** + * 删除 + */ + @RequestMapping("/delete/{${pkColumn.javaField}s}") + @PreAuthorize("@ss.hasPermi('${permissionPrefix}:remove')") + @Log(title = "${ClassName}", businessType = BusinessType.DELETE) + public R delete(@PathVariable ${pkColumn.javaType}[] ${pkColumn.javaField}s){ + ${className}Service.removeByIds(Arrays.asList(${pkColumn.javaField}s)); + return R.ok(); + } + +} diff --git a/ruoyi-generator/src/main/resources/vm/java/domain.java.vm b/ruoyi-generator/src/main/resources/vm/java/domain.java.vm new file mode 100644 index 0000000..bc1a28f --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/domain.java.vm @@ -0,0 +1,52 @@ +package ${packageName}.domain; + +#foreach ($import in $importList) +import ${import}; +#end +import com.baomidou.mybatisplus.annotation.IdType; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.baomidou.mybatisplus.annotation.TableName; +import com.baomidou.mybatisplus.annotation.TableId; +import com.ruoyi.common.annotation.Excel; +import java.io.Serializable; +import lombok.Data; + + +/** + * ${functionName}对象 ${tableName} + * + * @author ${author} + * @date ${datetime} + */ +@Data +@TableName("${tableName}") +public class ${ClassName} implements Serializable +{ + private static final long serialVersionUID = 1L; + +#foreach ($column in $columns) +#if(!$table.isSuperColumn($column.javaField)) + /** $column.columnComment */ +#if($column.list) +#set($parentheseIndex=$column.columnComment.indexOf("(")) +#if($parentheseIndex != -1) +#set($comment=$column.columnComment.substring(0, $parentheseIndex)) +#else +#set($comment=$column.columnComment) +#end +#end + #if($column.columnName == $pkColumn.columnName) + @TableId(type = IdType.INPUT) + #end + private $column.javaType $column.javaField; + +#end +#end +#if($table.sub) + /** $table.subTable.functionName信息 */ + private List<${subClassName}> ${subclassName}List; + +#end + +} diff --git a/ruoyi-generator/src/main/resources/vm/java/mapper.java.vm b/ruoyi-generator/src/main/resources/vm/java/mapper.java.vm new file mode 100644 index 0000000..ee91132 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/mapper.java.vm @@ -0,0 +1,18 @@ +package ${packageName}.mapper; + +import java.util.List; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import org.apache.ibatis.annotations.Mapper; +import ${packageName}.domain.${ClassName}; + +/** + * Mapper接口 + * + * @author ${author} + * @date ${datetime} + */ +@Mapper +public interface ${ClassName}Mapper extends BaseMapper<${ClassName}> +{ + +} diff --git a/ruoyi-generator/src/main/resources/vm/java/service.java.vm b/ruoyi-generator/src/main/resources/vm/java/service.java.vm new file mode 100644 index 0000000..5485a3d --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/service.java.vm @@ -0,0 +1,21 @@ +package ${packageName}.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import java.util.Map; +import ${packageName}.domain.${ClassName}; + +/** + * Service接口 + * + * @author ${author} + * @date ${datetime} + */ +public interface I${ClassName}Service extends IService<${ClassName}> +{ + /** + * 查询 + */ + R queryPage(Map params); + +} diff --git a/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm b/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm new file mode 100644 index 0000000..aa8667c --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm @@ -0,0 +1,33 @@ +package ${packageName}.service.impl; + +import java.util.Map; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.utils.Query; +import org.springframework.stereotype.Service; +import ${packageName}.mapper.${ClassName}Mapper; +import ${packageName}.domain.${ClassName}; +import ${packageName}.service.I${ClassName}Service; + +/** + * Service业务层处理 + * + * @author ${author} + * @date ${datetime} + */ +@Service("${className}Service") +public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${ClassName}> implements I${ClassName}Service +{ + @Override + public R queryPage(Map params) { + IPage<${ClassName}> page = this.page( + new Query<${ClassName}>().getPage(params), + new QueryWrapper<${ClassName}>() + ); + + return R.ok().put("count", page.getTotal()).put("data", page.getRecords()); + } + +} diff --git a/ruoyi-generator/src/main/resources/vm/java/sub-domain.java.vm b/ruoyi-generator/src/main/resources/vm/java/sub-domain.java.vm new file mode 100644 index 0000000..a3f53eb --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/java/sub-domain.java.vm @@ -0,0 +1,76 @@ +package ${packageName}.domain; + +#foreach ($import in $subImportList) +import ${import}; +#end +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * ${subTable.functionName}对象 ${subTableName} + * + * @author ${author} + * @date ${datetime} + */ +public class ${subClassName} extends BaseEntity +{ + private static final long serialVersionUID = 1L; + +#foreach ($column in $subTable.columns) +#if(!$table.isSuperColumn($column.javaField)) + /** $column.columnComment */ +#if($column.list) +#set($parentheseIndex=$column.columnComment.indexOf("(")) +#if($parentheseIndex != -1) +#set($comment=$column.columnComment.substring(0, $parentheseIndex)) +#else +#set($comment=$column.columnComment) +#end +#if($parentheseIndex != -1) + @Excel(name = "${comment}", readConverterExp = "$column.readConverterExp()") +#elseif($column.javaType == 'Date') + @JsonFormat(pattern = "yyyy-MM-dd") + @Excel(name = "${comment}", width = 30, dateFormat = "yyyy-MM-dd") +#else + @Excel(name = "${comment}") +#end +#end + private $column.javaType $column.javaField; + +#end +#end +#foreach ($column in $subTable.columns) +#if(!$table.isSuperColumn($column.javaField)) +#if($column.javaField.length() > 2 && $column.javaField.substring(1,2).matches("[A-Z]")) +#set($AttrName=$column.javaField) +#else +#set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)}) +#end + public void set${AttrName}($column.javaType $column.javaField) + { + this.$column.javaField = $column.javaField; + } + + public $column.javaType get${AttrName}() + { + return $column.javaField; + } +#end +#end + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) +#foreach ($column in $subTable.columns) +#if($column.javaField.length() > 2 && $column.javaField.substring(1,2).matches("[A-Z]")) +#set($AttrName=$column.javaField) +#else +#set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)}) +#end + .append("${column.javaField}", get${AttrName}()) +#end + .toString(); + } +} diff --git a/ruoyi-generator/src/main/resources/vm/js/api.js.vm b/ruoyi-generator/src/main/resources/vm/js/api.js.vm new file mode 100644 index 0000000..9295524 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/js/api.js.vm @@ -0,0 +1,44 @@ +import request from '@/utils/request' + +// 查询${functionName}列表 +export function list${BusinessName}(query) { + return request({ + url: '/${moduleName}/${businessName}/list', + method: 'get', + params: query + }) +} + +// 查询${functionName}详细 +export function get${BusinessName}(${pkColumn.javaField}) { + return request({ + url: '/${moduleName}/${businessName}/' + ${pkColumn.javaField}, + method: 'get' + }) +} + +// 新增${functionName} +export function add${BusinessName}(data) { + return request({ + url: '/${moduleName}/${businessName}', + method: 'post', + data: data + }) +} + +// 修改${functionName} +export function update${BusinessName}(data) { + return request({ + url: '/${moduleName}/${businessName}', + method: 'put', + data: data + }) +} + +// 删除${functionName} +export function del${BusinessName}(${pkColumn.javaField}) { + return request({ + url: '/${moduleName}/${businessName}/' + ${pkColumn.javaField}, + method: 'delete' + }) +} diff --git a/ruoyi-generator/src/main/resources/vm/sql/sql.vm b/ruoyi-generator/src/main/resources/vm/sql/sql.vm new file mode 100644 index 0000000..0575583 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/sql/sql.vm @@ -0,0 +1,22 @@ +-- 菜单 SQL +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}', '${parentMenuId}', '1', '${businessName}', '${moduleName}/${businessName}/index', 1, 0, 'C', '0', '0', '${permissionPrefix}:list', '#', 'admin', sysdate(), '', null, '${functionName}菜单'); + +-- 按钮父菜单ID +SELECT @parentId := LAST_INSERT_ID(); + +-- 按钮 SQL +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}查询', @parentId, '1', '#', '', 1, 0, 'F', '0', '0', '${permissionPrefix}:query', '#', 'admin', sysdate(), '', null, ''); + +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}新增', @parentId, '2', '#', '', 1, 0, 'F', '0', '0', '${permissionPrefix}:add', '#', 'admin', sysdate(), '', null, ''); + +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}修改', @parentId, '3', '#', '', 1, 0, 'F', '0', '0', '${permissionPrefix}:edit', '#', 'admin', sysdate(), '', null, ''); + +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}删除', @parentId, '4', '#', '', 1, 0, 'F', '0', '0', '${permissionPrefix}:remove', '#', 'admin', sysdate(), '', null, ''); + +insert into sys_menu (menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time, update_by, update_time, remark) +values('${functionName}导出', @parentId, '5', '#', '', 1, 0, 'F', '0', '0', '${permissionPrefix}:export', '#', 'admin', sysdate(), '', null, ''); \ No newline at end of file diff --git a/ruoyi-generator/src/main/resources/vm/vue/index-tree.vue.vm b/ruoyi-generator/src/main/resources/vm/vue/index-tree.vue.vm new file mode 100644 index 0000000..d9a5c45 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/vue/index-tree.vue.vm @@ -0,0 +1,480 @@ + + + diff --git a/ruoyi-generator/src/main/resources/vm/vue/index.vue.vm b/ruoyi-generator/src/main/resources/vm/vue/index.vue.vm new file mode 100644 index 0000000..4a6e907 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/vue/index.vue.vm @@ -0,0 +1,541 @@ + + + diff --git a/ruoyi-generator/src/main/resources/vm/vue/v3/index-tree.vue.vm b/ruoyi-generator/src/main/resources/vm/vue/v3/index-tree.vue.vm new file mode 100644 index 0000000..d57bbdf --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/vue/v3/index-tree.vue.vm @@ -0,0 +1,464 @@ + + + diff --git a/ruoyi-generator/src/main/resources/vm/vue/v3/index.vue.vm b/ruoyi-generator/src/main/resources/vm/vue/v3/index.vue.vm new file mode 100644 index 0000000..a363ef3 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/vue/v3/index.vue.vm @@ -0,0 +1,596 @@ + + + diff --git a/ruoyi-generator/src/main/resources/vm/vue/v3/readme.txt b/ruoyi-generator/src/main/resources/vm/vue/v3/readme.txt new file mode 100644 index 0000000..99239bb --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/vue/v3/readme.txt @@ -0,0 +1 @@ +ʹõRuoYi-Vue3ǰˣôҪһ´Ŀ¼ģindex.vue.vmindex-tree.vue.vmļϼvueĿ¼ \ No newline at end of file diff --git a/ruoyi-generator/src/main/resources/vm/xml/mapper.xml.vm b/ruoyi-generator/src/main/resources/vm/xml/mapper.xml.vm new file mode 100644 index 0000000..9fb48d9 --- /dev/null +++ b/ruoyi-generator/src/main/resources/vm/xml/mapper.xml.vm @@ -0,0 +1,7 @@ + + + + + diff --git a/ruoyi-quartz/pom.xml b/ruoyi-quartz/pom.xml new file mode 100644 index 0000000..1e1f855 --- /dev/null +++ b/ruoyi-quartz/pom.xml @@ -0,0 +1,46 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-quartz + + + quartz定时任务 + + + + + + + org.quartz-scheduler + quartz + + + com.mchange + c3p0 + + + + + + + com.ruoyi + ruoyi-common + + + + com.ruoyi + ruoyi-api + + + + + + diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/config/ScheduleConfig.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/config/ScheduleConfig.java new file mode 100644 index 0000000..a558170 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/config/ScheduleConfig.java @@ -0,0 +1,57 @@ +//package com.ruoyi.quartz.config; +// +//import org.springframework.context.annotation.Bean; +//import org.springframework.context.annotation.Configuration; +//import org.springframework.scheduling.quartz.SchedulerFactoryBean; +//import javax.sql.DataSource; +//import java.util.Properties; +// +///** +// * 定时任务配置(单机部署建议删除此类和qrtz数据库表,默认走内存会最高效) +// * +// * @author ruoyi +// */ +//@Configuration +//public class ScheduleConfig +//{ +// @Bean +// public SchedulerFactoryBean schedulerFactoryBean(DataSource dataSource) +// { +// SchedulerFactoryBean factory = new SchedulerFactoryBean(); +// factory.setDataSource(dataSource); +// +// // quartz参数 +// Properties prop = new Properties(); +// prop.put("org.quartz.scheduler.instanceName", "RuoyiScheduler"); +// prop.put("org.quartz.scheduler.instanceId", "AUTO"); +// // 线程池配置 +// prop.put("org.quartz.threadPool.class", "org.quartz.simpl.SimpleThreadPool"); +// prop.put("org.quartz.threadPool.threadCount", "20"); +// prop.put("org.quartz.threadPool.threadPriority", "5"); +// // JobStore配置 +// prop.put("org.quartz.jobStore.class", "org.springframework.scheduling.quartz.LocalDataSourceJobStore"); +// // 集群配置 +// prop.put("org.quartz.jobStore.isClustered", "true"); +// prop.put("org.quartz.jobStore.clusterCheckinInterval", "15000"); +// prop.put("org.quartz.jobStore.maxMisfiresToHandleAtATime", "1"); +// prop.put("org.quartz.jobStore.txIsolationLevelSerializable", "true"); +// +// // sqlserver 启用 +// // prop.put("org.quartz.jobStore.selectWithLockSQL", "SELECT * FROM {0}LOCKS UPDLOCK WHERE LOCK_NAME = ?"); +// prop.put("org.quartz.jobStore.misfireThreshold", "12000"); +// prop.put("org.quartz.jobStore.tablePrefix", "QRTZ_"); +// factory.setQuartzProperties(prop); +// +// factory.setSchedulerName("RuoyiScheduler"); +// // 延时启动 +// factory.setStartupDelay(1); +// factory.setApplicationContextSchedulerContextKey("applicationContextKey"); +// // 可选,QuartzScheduler +// // 启动时更新己存在的Job,这样就不用每次修改targetObject后删除qrtz_job_details表对应记录了 +// factory.setOverwriteExistingJobs(true); +// // 设置自动启动,默认为true +// factory.setAutoStartup(true); +// +// return factory; +// } +//} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobController.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobController.java new file mode 100644 index 0000000..f248430 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobController.java @@ -0,0 +1,185 @@ +package com.ruoyi.quartz.controller; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.quartz.SchedulerException; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.exception.job.TaskException; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.quartz.domain.SysJob; +import com.ruoyi.quartz.service.ISysJobService; +import com.ruoyi.quartz.util.CronUtils; +import com.ruoyi.quartz.util.ScheduleUtils; + +/** + * 调度任务信息操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/job") +public class SysJobController extends BaseController +{ + @Autowired + private ISysJobService jobService; + + /** + * 查询定时任务列表 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:list')") + @GetMapping("/list") + public TableDataInfo list(SysJob sysJob) + { + startPage(); + List list = jobService.selectJobList(sysJob); + return getDataTable(list); + } + + /** + * 导出定时任务列表 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:export')") + @Log(title = "定时任务", businessType = BusinessType.EXPORT) + @PostMapping("/export") + public void export(HttpServletResponse response, SysJob sysJob) + { + List list = jobService.selectJobList(sysJob); + ExcelUtil util = new ExcelUtil(SysJob.class); + util.exportExcel(response, list, "定时任务"); + } + + /** + * 获取定时任务详细信息 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:query')") + @GetMapping(value = "/{jobId}") + public AjaxResult getInfo(@PathVariable("jobId") Long jobId) + { + return AjaxResult.success(jobService.selectJobById(jobId)); + } + + /** + * 新增定时任务 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:add')") + @Log(title = "定时任务", businessType = BusinessType.INSERT) + @PostMapping + public AjaxResult add(@RequestBody SysJob job) throws SchedulerException, TaskException + { + if (!CronUtils.isValid(job.getCronExpression())) + { + return error("新增任务'" + job.getJobName() + "'失败,Cron表达式不正确"); + } + else if (StringUtils.containsIgnoreCase(job.getInvokeTarget(), Constants.LOOKUP_RMI)) + { + return error("新增任务'" + job.getJobName() + "'失败,目标字符串不允许'rmi'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), new String[] { Constants.LOOKUP_LDAP, Constants.LOOKUP_LDAPS })) + { + return error("新增任务'" + job.getJobName() + "'失败,目标字符串不允许'ldap(s)'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), new String[] { Constants.HTTP, Constants.HTTPS })) + { + return error("新增任务'" + job.getJobName() + "'失败,目标字符串不允许'http(s)'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), Constants.JOB_ERROR_STR)) + { + return error("新增任务'" + job.getJobName() + "'失败,目标字符串存在违规"); + } + else if (!ScheduleUtils.whiteList(job.getInvokeTarget())) + { + return error("新增任务'" + job.getJobName() + "'失败,目标字符串不在白名单内"); + } + job.setCreateBy(getUsername()); + return toAjax(jobService.insertJob(job)); + } + + /** + * 修改定时任务 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:edit')") + @Log(title = "定时任务", businessType = BusinessType.UPDATE) + @PutMapping + public AjaxResult edit(@RequestBody SysJob job) throws SchedulerException, TaskException + { + if (!CronUtils.isValid(job.getCronExpression())) + { + return error("修改任务'" + job.getJobName() + "'失败,Cron表达式不正确"); + } + else if (StringUtils.containsIgnoreCase(job.getInvokeTarget(), Constants.LOOKUP_RMI)) + { + return error("修改任务'" + job.getJobName() + "'失败,目标字符串不允许'rmi'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), new String[] { Constants.LOOKUP_LDAP, Constants.LOOKUP_LDAPS })) + { + return error("修改任务'" + job.getJobName() + "'失败,目标字符串不允许'ldap(s)'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), new String[] { Constants.HTTP, Constants.HTTPS })) + { + return error("修改任务'" + job.getJobName() + "'失败,目标字符串不允许'http(s)'调用"); + } + else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), Constants.JOB_ERROR_STR)) + { + return error("修改任务'" + job.getJobName() + "'失败,目标字符串存在违规"); + } + else if (!ScheduleUtils.whiteList(job.getInvokeTarget())) + { + return error("修改任务'" + job.getJobName() + "'失败,目标字符串不在白名单内"); + } + job.setUpdateBy(getUsername()); + return toAjax(jobService.updateJob(job)); + } + + /** + * 定时任务状态修改 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:changeStatus')") + @Log(title = "定时任务", businessType = BusinessType.UPDATE) + @PutMapping("/changeStatus") + public AjaxResult changeStatus(@RequestBody SysJob job) throws SchedulerException + { + SysJob newJob = jobService.selectJobById(job.getJobId()); + newJob.setStatus(job.getStatus()); + return toAjax(jobService.changeStatus(newJob)); + } + + /** + * 定时任务立即执行一次 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:changeStatus')") + @Log(title = "定时任务", businessType = BusinessType.UPDATE) + @PutMapping("/run") + public AjaxResult run(@RequestBody SysJob job) throws SchedulerException + { + jobService.run(job); + return AjaxResult.success(); + } + + /** + * 删除定时任务 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:remove')") + @Log(title = "定时任务", businessType = BusinessType.DELETE) + @DeleteMapping("/{jobIds}") + public AjaxResult remove(@PathVariable Long[] jobIds) throws SchedulerException, TaskException + { + jobService.deleteJobByIds(jobIds); + return AjaxResult.success(); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobLogController.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobLogController.java new file mode 100644 index 0000000..d27100d --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/controller/SysJobLogController.java @@ -0,0 +1,92 @@ +package com.ruoyi.quartz.controller; + +import java.util.List; +import javax.servlet.http.HttpServletResponse; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import com.ruoyi.common.annotation.Log; +import com.ruoyi.common.core.controller.BaseController; +import com.ruoyi.common.core.domain.AjaxResult; +import com.ruoyi.common.core.page.TableDataInfo; +import com.ruoyi.common.enums.BusinessType; +import com.ruoyi.common.utils.poi.ExcelUtil; +import com.ruoyi.quartz.domain.SysJobLog; +import com.ruoyi.quartz.service.ISysJobLogService; + +/** + * 调度日志操作处理 + * + * @author ruoyi + */ +@RestController +@RequestMapping("/monitor/jobLog") +public class SysJobLogController extends BaseController +{ + @Autowired + private ISysJobLogService jobLogService; + + /** + * 查询定时任务调度日志列表 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:list')") + @GetMapping("/list") + public TableDataInfo list(SysJobLog sysJobLog) + { + startPage(); + List list = jobLogService.selectJobLogList(sysJobLog); + return getDataTable(list); + } + + /** + * 导出定时任务调度日志列表 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:export')") + @Log(title = "任务调度日志", businessType = BusinessType.EXPORT) + @PostMapping("/export") + public void export(HttpServletResponse response, SysJobLog sysJobLog) + { + List list = jobLogService.selectJobLogList(sysJobLog); + ExcelUtil util = new ExcelUtil(SysJobLog.class); + util.exportExcel(response, list, "调度日志"); + } + + /** + * 根据调度编号获取详细信息 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:query')") + @GetMapping(value = "/{configId}") + public AjaxResult getInfo(@PathVariable Long jobLogId) + { + return AjaxResult.success(jobLogService.selectJobLogById(jobLogId)); + } + + + /** + * 删除定时任务调度日志 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:remove')") + @Log(title = "定时任务调度日志", businessType = BusinessType.DELETE) + @DeleteMapping("/{jobLogIds}") + public AjaxResult remove(@PathVariable Long[] jobLogIds) + { + return toAjax(jobLogService.deleteJobLogByIds(jobLogIds)); + } + + /** + * 清空定时任务调度日志 + */ + @PreAuthorize("@ss.hasPermi('monitor:job:remove')") + @Log(title = "调度日志", businessType = BusinessType.CLEAN) + @DeleteMapping("/clean") + public AjaxResult clean() + { + jobLogService.cleanJobLog(); + return AjaxResult.success(); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJob.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJob.java new file mode 100644 index 0000000..1f49695 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJob.java @@ -0,0 +1,171 @@ +package com.ruoyi.quartz.domain; + +import java.util.Date; +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.fasterxml.jackson.annotation.JsonFormat; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.constant.ScheduleConstants; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.quartz.util.CronUtils; + +/** + * 定时任务调度表 sys_job + * + * @author ruoyi + */ +public class SysJob extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 任务ID */ + @Excel(name = "任务序号", cellType = ColumnType.NUMERIC) + private Long jobId; + + /** 任务名称 */ + @Excel(name = "任务名称") + private String jobName; + + /** 任务组名 */ + @Excel(name = "任务组名") + private String jobGroup; + + /** 调用目标字符串 */ + @Excel(name = "调用目标字符串") + private String invokeTarget; + + /** cron执行表达式 */ + @Excel(name = "执行表达式 ") + private String cronExpression; + + /** cron计划策略 */ + @Excel(name = "计划策略 ", readConverterExp = "0=默认,1=立即触发执行,2=触发一次执行,3=不触发立即执行") + private String misfirePolicy = ScheduleConstants.MISFIRE_DEFAULT; + + /** 是否并发执行(0允许 1禁止) */ + @Excel(name = "并发执行", readConverterExp = "0=允许,1=禁止") + private String concurrent; + + /** 任务状态(0正常 1暂停) */ + @Excel(name = "任务状态", readConverterExp = "0=正常,1=暂停") + private String status; + + public Long getJobId() + { + return jobId; + } + + public void setJobId(Long jobId) + { + this.jobId = jobId; + } + + @NotBlank(message = "任务名称不能为空") + @Size(min = 0, max = 64, message = "任务名称不能超过64个字符") + public String getJobName() + { + return jobName; + } + + public void setJobName(String jobName) + { + this.jobName = jobName; + } + + public String getJobGroup() + { + return jobGroup; + } + + public void setJobGroup(String jobGroup) + { + this.jobGroup = jobGroup; + } + + @NotBlank(message = "调用目标字符串不能为空") + @Size(min = 0, max = 500, message = "调用目标字符串长度不能超过500个字符") + public String getInvokeTarget() + { + return invokeTarget; + } + + public void setInvokeTarget(String invokeTarget) + { + this.invokeTarget = invokeTarget; + } + + @NotBlank(message = "Cron执行表达式不能为空") + @Size(min = 0, max = 255, message = "Cron执行表达式不能超过255个字符") + public String getCronExpression() + { + return cronExpression; + } + + public void setCronExpression(String cronExpression) + { + this.cronExpression = cronExpression; + } + + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + public Date getNextValidTime() + { + if (StringUtils.isNotEmpty(cronExpression)) + { + return CronUtils.getNextExecution(cronExpression); + } + return null; + } + + public String getMisfirePolicy() + { + return misfirePolicy; + } + + public void setMisfirePolicy(String misfirePolicy) + { + this.misfirePolicy = misfirePolicy; + } + + public String getConcurrent() + { + return concurrent; + } + + public void setConcurrent(String concurrent) + { + this.concurrent = concurrent; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("jobId", getJobId()) + .append("jobName", getJobName()) + .append("jobGroup", getJobGroup()) + .append("cronExpression", getCronExpression()) + .append("nextValidTime", getNextValidTime()) + .append("misfirePolicy", getMisfirePolicy()) + .append("concurrent", getConcurrent()) + .append("status", getStatus()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJobLog.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJobLog.java new file mode 100644 index 0000000..121c035 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/domain/SysJobLog.java @@ -0,0 +1,155 @@ +package com.ruoyi.quartz.domain; + +import java.util.Date; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 定时任务调度日志表 sys_job_log + * + * @author ruoyi + */ +public class SysJobLog extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** ID */ + @Excel(name = "日志序号") + private Long jobLogId; + + /** 任务名称 */ + @Excel(name = "任务名称") + private String jobName; + + /** 任务组名 */ + @Excel(name = "任务组名") + private String jobGroup; + + /** 调用目标字符串 */ + @Excel(name = "调用目标字符串") + private String invokeTarget; + + /** 日志信息 */ + @Excel(name = "日志信息") + private String jobMessage; + + /** 执行状态(0正常 1失败) */ + @Excel(name = "执行状态", readConverterExp = "0=正常,1=失败") + private String status; + + /** 异常信息 */ + @Excel(name = "异常信息") + private String exceptionInfo; + + /** 开始时间 */ + private Date startTime; + + /** 停止时间 */ + private Date stopTime; + + public Long getJobLogId() + { + return jobLogId; + } + + public void setJobLogId(Long jobLogId) + { + this.jobLogId = jobLogId; + } + + public String getJobName() + { + return jobName; + } + + public void setJobName(String jobName) + { + this.jobName = jobName; + } + + public String getJobGroup() + { + return jobGroup; + } + + public void setJobGroup(String jobGroup) + { + this.jobGroup = jobGroup; + } + + public String getInvokeTarget() + { + return invokeTarget; + } + + public void setInvokeTarget(String invokeTarget) + { + this.invokeTarget = invokeTarget; + } + + public String getJobMessage() + { + return jobMessage; + } + + public void setJobMessage(String jobMessage) + { + this.jobMessage = jobMessage; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getExceptionInfo() + { + return exceptionInfo; + } + + public void setExceptionInfo(String exceptionInfo) + { + this.exceptionInfo = exceptionInfo; + } + + public Date getStartTime() + { + return startTime; + } + + public void setStartTime(Date startTime) + { + this.startTime = startTime; + } + + public Date getStopTime() + { + return stopTime; + } + + public void setStopTime(Date stopTime) + { + this.stopTime = stopTime; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("jobLogId", getJobLogId()) + .append("jobName", getJobName()) + .append("jobGroup", getJobGroup()) + .append("jobMessage", getJobMessage()) + .append("status", getStatus()) + .append("exceptionInfo", getExceptionInfo()) + .append("startTime", getStartTime()) + .append("stopTime", getStopTime()) + .toString(); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobLogMapper.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobLogMapper.java new file mode 100644 index 0000000..727d916 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobLogMapper.java @@ -0,0 +1,64 @@ +package com.ruoyi.quartz.mapper; + +import java.util.List; +import com.ruoyi.quartz.domain.SysJobLog; + +/** + * 调度任务日志信息 数据层 + * + * @author ruoyi + */ +public interface SysJobLogMapper +{ + /** + * 获取quartz调度器日志的计划任务 + * + * @param jobLog 调度日志信息 + * @return 调度任务日志集合 + */ + public List selectJobLogList(SysJobLog jobLog); + + /** + * 查询所有调度任务日志 + * + * @return 调度任务日志列表 + */ + public List selectJobLogAll(); + + /** + * 通过调度任务日志ID查询调度信息 + * + * @param jobLogId 调度任务日志ID + * @return 调度任务日志对象信息 + */ + public SysJobLog selectJobLogById(Long jobLogId); + + /** + * 新增任务日志 + * + * @param jobLog 调度日志信息 + * @return 结果 + */ + public int insertJobLog(SysJobLog jobLog); + + /** + * 批量删除调度日志信息 + * + * @param logIds 需要删除的数据ID + * @return 结果 + */ + public int deleteJobLogByIds(Long[] logIds); + + /** + * 删除任务日志 + * + * @param jobId 调度日志ID + * @return 结果 + */ + public int deleteJobLogById(Long jobId); + + /** + * 清空任务日志 + */ + public void cleanJobLog(); +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobMapper.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobMapper.java new file mode 100644 index 0000000..20f45db --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/mapper/SysJobMapper.java @@ -0,0 +1,67 @@ +package com.ruoyi.quartz.mapper; + +import java.util.List; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 调度任务信息 数据层 + * + * @author ruoyi + */ +public interface SysJobMapper +{ + /** + * 查询调度任务日志集合 + * + * @param job 调度信息 + * @return 操作日志集合 + */ + public List selectJobList(SysJob job); + + /** + * 查询所有调度任务 + * + * @return 调度任务列表 + */ + public List selectJobAll(); + + /** + * 通过调度ID查询调度任务信息 + * + * @param jobId 调度ID + * @return 角色对象信息 + */ + public SysJob selectJobById(Long jobId); + + /** + * 通过调度ID删除调度任务信息 + * + * @param jobId 调度ID + * @return 结果 + */ + public int deleteJobById(Long jobId); + + /** + * 批量删除调度任务信息 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteJobByIds(Long[] ids); + + /** + * 修改调度任务信息 + * + * @param job 调度任务信息 + * @return 结果 + */ + public int updateJob(SysJob job); + + /** + * 新增调度任务信息 + * + * @param job 调度任务信息 + * @return 结果 + */ + public int insertJob(SysJob job); +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobLogService.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobLogService.java new file mode 100644 index 0000000..8546792 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobLogService.java @@ -0,0 +1,56 @@ +package com.ruoyi.quartz.service; + +import java.util.List; +import com.ruoyi.quartz.domain.SysJobLog; + +/** + * 定时任务调度日志信息信息 服务层 + * + * @author ruoyi + */ +public interface ISysJobLogService +{ + /** + * 获取quartz调度器日志的计划任务 + * + * @param jobLog 调度日志信息 + * @return 调度任务日志集合 + */ + public List selectJobLogList(SysJobLog jobLog); + + /** + * 通过调度任务日志ID查询调度信息 + * + * @param jobLogId 调度任务日志ID + * @return 调度任务日志对象信息 + */ + public SysJobLog selectJobLogById(Long jobLogId); + + /** + * 新增任务日志 + * + * @param jobLog 调度日志信息 + */ + public void addJobLog(SysJobLog jobLog); + + /** + * 批量删除调度日志信息 + * + * @param logIds 需要删除的日志ID + * @return 结果 + */ + public int deleteJobLogByIds(Long[] logIds); + + /** + * 删除任务日志 + * + * @param jobId 调度日志ID + * @return 结果 + */ + public int deleteJobLogById(Long jobId); + + /** + * 清空任务日志 + */ + public void cleanJobLog(); +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobService.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobService.java new file mode 100644 index 0000000..6d62661 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/ISysJobService.java @@ -0,0 +1,102 @@ +package com.ruoyi.quartz.service; + +import java.util.List; +import org.quartz.SchedulerException; +import com.ruoyi.common.exception.job.TaskException; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 定时任务调度信息信息 服务层 + * + * @author ruoyi + */ +public interface ISysJobService +{ + /** + * 获取quartz调度器的计划任务 + * + * @param job 调度信息 + * @return 调度任务集合 + */ + public List selectJobList(SysJob job); + + /** + * 通过调度任务ID查询调度信息 + * + * @param jobId 调度任务ID + * @return 调度任务对象信息 + */ + public SysJob selectJobById(Long jobId); + + /** + * 暂停任务 + * + * @param job 调度信息 + * @return 结果 + */ + public int pauseJob(SysJob job) throws SchedulerException; + + /** + * 恢复任务 + * + * @param job 调度信息 + * @return 结果 + */ + public int resumeJob(SysJob job) throws SchedulerException; + + /** + * 删除任务后,所对应的trigger也将被删除 + * + * @param job 调度信息 + * @return 结果 + */ + public int deleteJob(SysJob job) throws SchedulerException; + + /** + * 批量删除调度信息 + * + * @param jobIds 需要删除的任务ID + * @return 结果 + */ + public void deleteJobByIds(Long[] jobIds) throws SchedulerException; + + /** + * 任务调度状态修改 + * + * @param job 调度信息 + * @return 结果 + */ + public int changeStatus(SysJob job) throws SchedulerException; + + /** + * 立即运行任务 + * + * @param job 调度信息 + * @return 结果 + */ + public void run(SysJob job) throws SchedulerException; + + /** + * 新增任务 + * + * @param job 调度信息 + * @return 结果 + */ + public int insertJob(SysJob job) throws SchedulerException, TaskException; + + /** + * 更新任务 + * + * @param job 调度信息 + * @return 结果 + */ + public int updateJob(SysJob job) throws SchedulerException, TaskException; + + /** + * 校验cron表达式是否有效 + * + * @param cronExpression 表达式 + * @return 结果 + */ + public boolean checkCronExpressionIsValid(String cronExpression); +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobLogServiceImpl.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobLogServiceImpl.java new file mode 100644 index 0000000..812eed7 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobLogServiceImpl.java @@ -0,0 +1,87 @@ +package com.ruoyi.quartz.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.quartz.domain.SysJobLog; +import com.ruoyi.quartz.mapper.SysJobLogMapper; +import com.ruoyi.quartz.service.ISysJobLogService; + +/** + * 定时任务调度日志信息 服务层 + * + * @author ruoyi + */ +@Service +public class SysJobLogServiceImpl implements ISysJobLogService +{ + @Autowired + private SysJobLogMapper jobLogMapper; + + /** + * 获取quartz调度器日志的计划任务 + * + * @param jobLog 调度日志信息 + * @return 调度任务日志集合 + */ + @Override + public List selectJobLogList(SysJobLog jobLog) + { + return jobLogMapper.selectJobLogList(jobLog); + } + + /** + * 通过调度任务日志ID查询调度信息 + * + * @param jobLogId 调度任务日志ID + * @return 调度任务日志对象信息 + */ + @Override + public SysJobLog selectJobLogById(Long jobLogId) + { + return jobLogMapper.selectJobLogById(jobLogId); + } + + /** + * 新增任务日志 + * + * @param jobLog 调度日志信息 + */ + @Override + public void addJobLog(SysJobLog jobLog) + { + jobLogMapper.insertJobLog(jobLog); + } + + /** + * 批量删除调度日志信息 + * + * @param logIds 需要删除的数据ID + * @return 结果 + */ + @Override + public int deleteJobLogByIds(Long[] logIds) + { + return jobLogMapper.deleteJobLogByIds(logIds); + } + + /** + * 删除任务日志 + * + * @param jobId 调度日志ID + */ + @Override + public int deleteJobLogById(Long jobId) + { + return jobLogMapper.deleteJobLogById(jobId); + } + + /** + * 清空任务日志 + */ + @Override + public void cleanJobLog() + { + jobLogMapper.cleanJobLog(); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobServiceImpl.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobServiceImpl.java new file mode 100644 index 0000000..0d9a53d --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/service/impl/SysJobServiceImpl.java @@ -0,0 +1,254 @@ +package com.ruoyi.quartz.service.impl; + +import java.util.List; +import javax.annotation.PostConstruct; +import org.quartz.JobDataMap; +import org.quartz.JobKey; +import org.quartz.Scheduler; +import org.quartz.SchedulerException; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import com.ruoyi.common.constant.ScheduleConstants; +import com.ruoyi.common.exception.job.TaskException; +import com.ruoyi.quartz.domain.SysJob; +import com.ruoyi.quartz.mapper.SysJobMapper; +import com.ruoyi.quartz.service.ISysJobService; +import com.ruoyi.quartz.util.CronUtils; +import com.ruoyi.quartz.util.ScheduleUtils; + +/** + * 定时任务调度信息 服务层 + * + * @author ruoyi + */ +@Service +public class SysJobServiceImpl implements ISysJobService +{ + @Autowired + private Scheduler scheduler; + + @Autowired + private SysJobMapper jobMapper; + + /** + * 项目启动时,初始化定时器 主要是防止手动修改数据库导致未同步到定时任务处理(注:不能手动修改数据库ID和任务组名,否则会导致脏数据) + */ + @PostConstruct + public void init() throws SchedulerException, TaskException + { + scheduler.clear(); + List jobList = jobMapper.selectJobAll(); + for (SysJob job : jobList) + { + ScheduleUtils.createScheduleJob(scheduler, job); + } + } + + /** + * 获取quartz调度器的计划任务列表 + * + * @param job 调度信息 + * @return + */ + @Override + public List selectJobList(SysJob job) + { + return jobMapper.selectJobList(job); + } + + /** + * 通过调度任务ID查询调度信息 + * + * @param jobId 调度任务ID + * @return 调度任务对象信息 + */ + @Override + public SysJob selectJobById(Long jobId) + { + return jobMapper.selectJobById(jobId); + } + + /** + * 暂停任务 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int pauseJob(SysJob job) throws SchedulerException + { + Long jobId = job.getJobId(); + String jobGroup = job.getJobGroup(); + job.setStatus(ScheduleConstants.Status.PAUSE.getValue()); + int rows = jobMapper.updateJob(job); + if (rows > 0) + { + scheduler.pauseJob(ScheduleUtils.getJobKey(jobId, jobGroup)); + } + return rows; + } + + /** + * 恢复任务 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int resumeJob(SysJob job) throws SchedulerException + { + Long jobId = job.getJobId(); + String jobGroup = job.getJobGroup(); + job.setStatus(ScheduleConstants.Status.NORMAL.getValue()); + int rows = jobMapper.updateJob(job); + if (rows > 0) + { + scheduler.resumeJob(ScheduleUtils.getJobKey(jobId, jobGroup)); + } + return rows; + } + + /** + * 删除任务后,所对应的trigger也将被删除 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int deleteJob(SysJob job) throws SchedulerException + { + Long jobId = job.getJobId(); + String jobGroup = job.getJobGroup(); + int rows = jobMapper.deleteJobById(jobId); + if (rows > 0) + { + scheduler.deleteJob(ScheduleUtils.getJobKey(jobId, jobGroup)); + } + return rows; + } + + /** + * 批量删除调度信息 + * + * @param jobIds 需要删除的任务ID + * @return 结果 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void deleteJobByIds(Long[] jobIds) throws SchedulerException + { + for (Long jobId : jobIds) + { + SysJob job = jobMapper.selectJobById(jobId); + deleteJob(job); + } + } + + /** + * 任务调度状态修改 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int changeStatus(SysJob job) throws SchedulerException + { + int rows = 0; + String status = job.getStatus(); + if (ScheduleConstants.Status.NORMAL.getValue().equals(status)) + { + rows = resumeJob(job); + } + else if (ScheduleConstants.Status.PAUSE.getValue().equals(status)) + { + rows = pauseJob(job); + } + return rows; + } + + /** + * 立即运行任务 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void run(SysJob job) throws SchedulerException + { + Long jobId = job.getJobId(); + String jobGroup = job.getJobGroup(); + SysJob properties = selectJobById(job.getJobId()); + // 参数 + JobDataMap dataMap = new JobDataMap(); + dataMap.put(ScheduleConstants.TASK_PROPERTIES, properties); + scheduler.triggerJob(ScheduleUtils.getJobKey(jobId, jobGroup), dataMap); + } + + /** + * 新增任务 + * + * @param job 调度信息 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int insertJob(SysJob job) throws SchedulerException, TaskException + { + job.setStatus(ScheduleConstants.Status.PAUSE.getValue()); + int rows = jobMapper.insertJob(job); + if (rows > 0) + { + ScheduleUtils.createScheduleJob(scheduler, job); + } + return rows; + } + + /** + * 更新任务的时间表达式 + * + * @param job 调度信息 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public int updateJob(SysJob job) throws SchedulerException, TaskException + { + SysJob properties = selectJobById(job.getJobId()); + int rows = jobMapper.updateJob(job); + if (rows > 0) + { + updateSchedulerJob(job, properties.getJobGroup()); + } + return rows; + } + + /** + * 更新任务 + * + * @param job 任务对象 + * @param jobGroup 任务组名 + */ + public void updateSchedulerJob(SysJob job, String jobGroup) throws SchedulerException, TaskException + { + Long jobId = job.getJobId(); + // 判断是否存在 + JobKey jobKey = ScheduleUtils.getJobKey(jobId, jobGroup); + if (scheduler.checkExists(jobKey)) + { + // 防止创建时存在数据问题 先移除,然后在执行创建操作 + scheduler.deleteJob(jobKey); + } + ScheduleUtils.createScheduleJob(scheduler, job); + } + + /** + * 校验cron表达式是否有效 + * + * @param cronExpression 表达式 + * @return 结果 + */ + @Override + public boolean checkCronExpressionIsValid(String cronExpression) + { + return CronUtils.isValid(cronExpression); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/task/RyTask.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/task/RyTask.java new file mode 100644 index 0000000..f360097 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/task/RyTask.java @@ -0,0 +1,29 @@ +package com.ruoyi.quartz.task; + +import com.ruoyi.common.utils.StringUtils; +import org.springframework.stereotype.Component; + +/** + * 定时任务调度测试 + * + * @author ruoyi + */ +@Component("ryTask") +public class RyTask +{ + + public void ryMultipleParams(String s, Boolean b, Long l, Double d, Integer i) + { + System.out.println(StringUtils.format("执行多参方法: 字符串类型{},布尔类型{},长整型{},浮点型{},整形{}", s, b, l, d, i)); + } + + public void ryParams(String params) + { + System.out.println("执行有参方法:" + params); + } + + public void ryNoParams() + { + System.out.println("执行无参方法"); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/AbstractQuartzJob.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/AbstractQuartzJob.java new file mode 100644 index 0000000..731a5eb --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/AbstractQuartzJob.java @@ -0,0 +1,107 @@ +package com.ruoyi.quartz.util; + +import java.util.Date; +import org.quartz.Job; +import org.quartz.JobExecutionContext; +import org.quartz.JobExecutionException; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.ScheduleConstants; +import com.ruoyi.common.utils.ExceptionUtil; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.quartz.domain.SysJob; +import com.ruoyi.quartz.domain.SysJobLog; +import com.ruoyi.quartz.service.ISysJobLogService; + +/** + * 抽象quartz调用 + * + * @author ruoyi + */ +public abstract class AbstractQuartzJob implements Job +{ + private static final Logger log = LoggerFactory.getLogger(AbstractQuartzJob.class); + + /** + * 线程本地变量 + */ + private static ThreadLocal threadLocal = new ThreadLocal<>(); + + @Override + public void execute(JobExecutionContext context) throws JobExecutionException + { + SysJob sysJob = new SysJob(); + BeanUtils.copyBeanProp(sysJob, context.getMergedJobDataMap().get(ScheduleConstants.TASK_PROPERTIES)); + try + { + before(context, sysJob); + if (sysJob != null) + { + doExecute(context, sysJob); + } + after(context, sysJob, null); + } + catch (Exception e) + { + log.error("任务执行异常 - :", e); + after(context, sysJob, e); + } + } + + /** + * 执行前 + * + * @param context 工作执行上下文对象 + * @param sysJob 系统计划任务 + */ + protected void before(JobExecutionContext context, SysJob sysJob) + { + threadLocal.set(new Date()); + } + + /** + * 执行后 + * + * @param context 工作执行上下文对象 + * @param sysJob 系统计划任务 + */ + protected void after(JobExecutionContext context, SysJob sysJob, Exception e) + { + Date startTime = threadLocal.get(); + threadLocal.remove(); + + final SysJobLog sysJobLog = new SysJobLog(); + sysJobLog.setJobName(sysJob.getJobName()); + sysJobLog.setJobGroup(sysJob.getJobGroup()); + sysJobLog.setInvokeTarget(sysJob.getInvokeTarget()); + sysJobLog.setStartTime(startTime); + sysJobLog.setStopTime(new Date()); + long runMs = sysJobLog.getStopTime().getTime() - sysJobLog.getStartTime().getTime(); + sysJobLog.setJobMessage(sysJobLog.getJobName() + " 总共耗时:" + runMs + "毫秒"); + if (e != null) + { + sysJobLog.setStatus(Constants.FAIL); + String errorMsg = StringUtils.substring(ExceptionUtil.getExceptionMessage(e), 0, 2000); + sysJobLog.setExceptionInfo(errorMsg); + } + else + { + sysJobLog.setStatus(Constants.SUCCESS); + } + + // 写入数据库当中 + SpringUtils.getBean(ISysJobLogService.class).addJobLog(sysJobLog); + } + + /** + * 执行方法,由子类重载 + * + * @param context 工作执行上下文对象 + * @param sysJob 系统计划任务 + * @throws Exception 执行过程中的异常 + */ + protected abstract void doExecute(JobExecutionContext context, SysJob sysJob) throws Exception; +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/CronUtils.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/CronUtils.java new file mode 100644 index 0000000..dd53839 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/CronUtils.java @@ -0,0 +1,63 @@ +package com.ruoyi.quartz.util; + +import java.text.ParseException; +import java.util.Date; +import org.quartz.CronExpression; + +/** + * cron表达式工具类 + * + * @author ruoyi + * + */ +public class CronUtils +{ + /** + * 返回一个布尔值代表一个给定的Cron表达式的有效性 + * + * @param cronExpression Cron表达式 + * @return boolean 表达式是否有效 + */ + public static boolean isValid(String cronExpression) + { + return CronExpression.isValidExpression(cronExpression); + } + + /** + * 返回一个字符串值,表示该消息无效Cron表达式给出有效性 + * + * @param cronExpression Cron表达式 + * @return String 无效时返回表达式错误描述,如果有效返回null + */ + public static String getInvalidMessage(String cronExpression) + { + try + { + new CronExpression(cronExpression); + return null; + } + catch (ParseException pe) + { + return pe.getMessage(); + } + } + + /** + * 返回下一个执行时间根据给定的Cron表达式 + * + * @param cronExpression Cron表达式 + * @return Date 下次Cron表达式执行时间 + */ + public static Date getNextExecution(String cronExpression) + { + try + { + CronExpression cron = new CronExpression(cronExpression); + return cron.getNextValidTimeAfter(new Date(System.currentTimeMillis())); + } + catch (ParseException e) + { + throw new IllegalArgumentException(e.getMessage()); + } + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/JobInvokeUtil.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/JobInvokeUtil.java new file mode 100644 index 0000000..e2e420e --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/JobInvokeUtil.java @@ -0,0 +1,182 @@ +package com.ruoyi.quartz.util; + +import java.lang.reflect.InvocationTargetException; +import java.lang.reflect.Method; +import java.util.LinkedList; +import java.util.List; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 任务执行工具 + * + * @author ruoyi + */ +public class JobInvokeUtil +{ + /** + * 执行方法 + * + * @param sysJob 系统任务 + */ + public static void invokeMethod(SysJob sysJob) throws Exception + { + String invokeTarget = sysJob.getInvokeTarget(); + String beanName = getBeanName(invokeTarget); + String methodName = getMethodName(invokeTarget); + List methodParams = getMethodParams(invokeTarget); + + if (!isValidClassName(beanName)) + { + Object bean = SpringUtils.getBean(beanName); + invokeMethod(bean, methodName, methodParams); + } + else + { + Object bean = Class.forName(beanName).newInstance(); + invokeMethod(bean, methodName, methodParams); + } + } + + /** + * 调用任务方法 + * + * @param bean 目标对象 + * @param methodName 方法名称 + * @param methodParams 方法参数 + */ + private static void invokeMethod(Object bean, String methodName, List methodParams) + throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, + InvocationTargetException + { + if (StringUtils.isNotNull(methodParams) && methodParams.size() > 0) + { + Method method = bean.getClass().getDeclaredMethod(methodName, getMethodParamsType(methodParams)); + method.invoke(bean, getMethodParamsValue(methodParams)); + } + else + { + Method method = bean.getClass().getDeclaredMethod(methodName); + method.invoke(bean); + } + } + + /** + * 校验是否为为class包名 + * + * @param invokeTarget 名称 + * @return true是 false否 + */ + public static boolean isValidClassName(String invokeTarget) + { + return StringUtils.countMatches(invokeTarget, ".") > 1; + } + + /** + * 获取bean名称 + * + * @param invokeTarget 目标字符串 + * @return bean名称 + */ + public static String getBeanName(String invokeTarget) + { + String beanName = StringUtils.substringBefore(invokeTarget, "("); + return StringUtils.substringBeforeLast(beanName, "."); + } + + /** + * 获取bean方法 + * + * @param invokeTarget 目标字符串 + * @return method方法 + */ + public static String getMethodName(String invokeTarget) + { + String methodName = StringUtils.substringBefore(invokeTarget, "("); + return StringUtils.substringAfterLast(methodName, "."); + } + + /** + * 获取method方法参数相关列表 + * + * @param invokeTarget 目标字符串 + * @return method方法相关参数列表 + */ + public static List getMethodParams(String invokeTarget) + { + String methodStr = StringUtils.substringBetween(invokeTarget, "(", ")"); + if (StringUtils.isEmpty(methodStr)) + { + return null; + } + String[] methodParams = methodStr.split(",(?=([^\"']*[\"'][^\"']*[\"'])*[^\"']*$)"); + List classs = new LinkedList<>(); + for (int i = 0; i < methodParams.length; i++) + { + String str = StringUtils.trimToEmpty(methodParams[i]); + // String字符串类型,以'或"开头 + if (StringUtils.startsWithAny(str, "'", "\"")) + { + classs.add(new Object[] { StringUtils.substring(str, 1, str.length() - 1), String.class }); + } + // boolean布尔类型,等于true或者false + else if ("true".equalsIgnoreCase(str) || "false".equalsIgnoreCase(str)) + { + classs.add(new Object[] { Boolean.valueOf(str), Boolean.class }); + } + // long长整形,以L结尾 + else if (StringUtils.endsWith(str, "L")) + { + classs.add(new Object[] { Long.valueOf(StringUtils.substring(str, 0, str.length() - 1)), Long.class }); + } + // double浮点类型,以D结尾 + else if (StringUtils.endsWith(str, "D")) + { + classs.add(new Object[] { Double.valueOf(StringUtils.substring(str, 0, str.length() - 1)), Double.class }); + } + // 其他类型归类为整形 + else + { + classs.add(new Object[] { Integer.valueOf(str), Integer.class }); + } + } + return classs; + } + + /** + * 获取参数类型 + * + * @param methodParams 参数相关列表 + * @return 参数类型列表 + */ + public static Class[] getMethodParamsType(List methodParams) + { + Class[] classs = new Class[methodParams.size()]; + int index = 0; + for (Object[] os : methodParams) + { + classs[index] = (Class) os[1]; + index++; + } + return classs; + } + + /** + * 获取参数值 + * + * @param methodParams 参数相关列表 + * @return 参数值列表 + */ + public static Object[] getMethodParamsValue(List methodParams) + { + Object[] classs = new Object[methodParams.size()]; + int index = 0; + for (Object[] os : methodParams) + { + classs[index] = (Object) os[0]; + index++; + } + return classs; + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzDisallowConcurrentExecution.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzDisallowConcurrentExecution.java new file mode 100644 index 0000000..5e13558 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzDisallowConcurrentExecution.java @@ -0,0 +1,21 @@ +package com.ruoyi.quartz.util; + +import org.quartz.DisallowConcurrentExecution; +import org.quartz.JobExecutionContext; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 定时任务处理(禁止并发执行) + * + * @author ruoyi + * + */ +@DisallowConcurrentExecution +public class QuartzDisallowConcurrentExecution extends AbstractQuartzJob +{ + @Override + protected void doExecute(JobExecutionContext context, SysJob sysJob) throws Exception + { + JobInvokeUtil.invokeMethod(sysJob); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzJobExecution.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzJobExecution.java new file mode 100644 index 0000000..e975326 --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/QuartzJobExecution.java @@ -0,0 +1,19 @@ +package com.ruoyi.quartz.util; + +import org.quartz.JobExecutionContext; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 定时任务处理(允许并发执行) + * + * @author ruoyi + * + */ +public class QuartzJobExecution extends AbstractQuartzJob +{ + @Override + protected void doExecute(JobExecutionContext context, SysJob sysJob) throws Exception + { + JobInvokeUtil.invokeMethod(sysJob); + } +} diff --git a/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/ScheduleUtils.java b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/ScheduleUtils.java new file mode 100644 index 0000000..5d00c2c --- /dev/null +++ b/ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/ScheduleUtils.java @@ -0,0 +1,134 @@ +package com.ruoyi.quartz.util; + +import org.quartz.CronScheduleBuilder; +import org.quartz.CronTrigger; +import org.quartz.Job; +import org.quartz.JobBuilder; +import org.quartz.JobDetail; +import org.quartz.JobKey; +import org.quartz.Scheduler; +import org.quartz.SchedulerException; +import org.quartz.TriggerBuilder; +import org.quartz.TriggerKey; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.ScheduleConstants; +import com.ruoyi.common.exception.job.TaskException; +import com.ruoyi.common.exception.job.TaskException.Code; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.quartz.domain.SysJob; + +/** + * 定时任务工具类 + * + * @author ruoyi + * + */ +public class ScheduleUtils +{ + /** + * 得到quartz任务类 + * + * @param sysJob 执行计划 + * @return 具体执行任务类 + */ + private static Class getQuartzJobClass(SysJob sysJob) + { + boolean isConcurrent = "0".equals(sysJob.getConcurrent()); + return isConcurrent ? QuartzJobExecution.class : QuartzDisallowConcurrentExecution.class; + } + + /** + * 构建任务触发对象 + */ + public static TriggerKey getTriggerKey(Long jobId, String jobGroup) + { + return TriggerKey.triggerKey(ScheduleConstants.TASK_CLASS_NAME + jobId, jobGroup); + } + + /** + * 构建任务键对象 + */ + public static JobKey getJobKey(Long jobId, String jobGroup) + { + return JobKey.jobKey(ScheduleConstants.TASK_CLASS_NAME + jobId, jobGroup); + } + + /** + * 创建定时任务 + */ + public static void createScheduleJob(Scheduler scheduler, SysJob job) throws SchedulerException, TaskException + { + Class jobClass = getQuartzJobClass(job); + // 构建job信息 + Long jobId = job.getJobId(); + String jobGroup = job.getJobGroup(); + JobDetail jobDetail = JobBuilder.newJob(jobClass).withIdentity(getJobKey(jobId, jobGroup)).build(); + + // 表达式调度构建器 + CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(job.getCronExpression()); + cronScheduleBuilder = handleCronScheduleMisfirePolicy(job, cronScheduleBuilder); + + // 按新的cronExpression表达式构建一个新的trigger + CronTrigger trigger = TriggerBuilder.newTrigger().withIdentity(getTriggerKey(jobId, jobGroup)) + .withSchedule(cronScheduleBuilder).build(); + + // 放入参数,运行时的方法可以获取 + jobDetail.getJobDataMap().put(ScheduleConstants.TASK_PROPERTIES, job); + + // 判断是否存在 + if (scheduler.checkExists(getJobKey(jobId, jobGroup))) + { + // 防止创建时存在数据问题 先移除,然后在执行创建操作 + scheduler.deleteJob(getJobKey(jobId, jobGroup)); + } + + scheduler.scheduleJob(jobDetail, trigger); + + // 暂停任务 + if (job.getStatus().equals(ScheduleConstants.Status.PAUSE.getValue())) + { + scheduler.pauseJob(ScheduleUtils.getJobKey(jobId, jobGroup)); + } + } + + /** + * 设置定时任务策略 + */ + public static CronScheduleBuilder handleCronScheduleMisfirePolicy(SysJob job, CronScheduleBuilder cb) + throws TaskException + { + switch (job.getMisfirePolicy()) + { + case ScheduleConstants.MISFIRE_DEFAULT: + return cb; + case ScheduleConstants.MISFIRE_IGNORE_MISFIRES: + return cb.withMisfireHandlingInstructionIgnoreMisfires(); + case ScheduleConstants.MISFIRE_FIRE_AND_PROCEED: + return cb.withMisfireHandlingInstructionFireAndProceed(); + case ScheduleConstants.MISFIRE_DO_NOTHING: + return cb.withMisfireHandlingInstructionDoNothing(); + default: + throw new TaskException("The task misfire policy '" + job.getMisfirePolicy() + + "' cannot be used in cron schedule tasks", Code.CONFIG_ERROR); + } + } + + /** + * 检查包名是否为白名单配置 + * + * @param invokeTarget 目标字符串 + * @return 结果 + */ + public static boolean whiteList(String invokeTarget) + { + String packageName = StringUtils.substringBefore(invokeTarget, "("); + int count = StringUtils.countMatches(packageName, "."); + if (count > 1) + { + return StringUtils.containsAnyIgnoreCase(invokeTarget, Constants.JOB_WHITELIST_STR); + } + Object obj = SpringUtils.getBean(StringUtils.split(invokeTarget, ".")[0]); + return StringUtils.containsAnyIgnoreCase(obj.getClass().getPackage().getName(), Constants.JOB_WHITELIST_STR); + } +} diff --git a/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobLogMapper.xml b/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobLogMapper.xml new file mode 100644 index 0000000..e608e42 --- /dev/null +++ b/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobLogMapper.xml @@ -0,0 +1,93 @@ + + + + + + + + + + + + + + + + + select job_log_id, job_name, job_group, invoke_target, job_message, status, exception_info, create_time + from sys_job_log + + + + + + + + + + delete from sys_job_log where job_log_id = #{jobLogId} + + + + delete from sys_job_log where job_log_id in + + #{jobLogId} + + + + + truncate table sys_job_log + + + + insert into sys_job_log( + job_log_id, + job_name, + job_group, + invoke_target, + job_message, + status, + exception_info, + create_time + )values( + #{jobLogId}, + #{jobName}, + #{jobGroup}, + #{invokeTarget}, + #{jobMessage}, + #{status}, + #{exceptionInfo}, + sysdate() + ) + + + \ No newline at end of file diff --git a/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobMapper.xml b/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobMapper.xml new file mode 100644 index 0000000..5605c44 --- /dev/null +++ b/ruoyi-quartz/src/main/resources/mapper/quartz/SysJobMapper.xml @@ -0,0 +1,111 @@ + + + + + + + + + + + + + + + + + + + + + + select job_id, job_name, job_group, invoke_target, cron_expression, misfire_policy, concurrent, status, create_by, create_time, remark + from sys_job + + + + + + + + + + delete from sys_job where job_id = #{jobId} + + + + delete from sys_job where job_id in + + #{jobId} + + + + + update sys_job + + job_name = #{jobName}, + job_group = #{jobGroup}, + invoke_target = #{invokeTarget}, + cron_expression = #{cronExpression}, + misfire_policy = #{misfirePolicy}, + concurrent = #{concurrent}, + status = #{status}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where job_id = #{jobId} + + + + insert into sys_job( + job_id, + job_name, + job_group, + invoke_target, + cron_expression, + misfire_policy, + concurrent, + status, + remark, + create_by, + create_time + )values( + #{jobId}, + #{jobName}, + #{jobGroup}, + #{invokeTarget}, + #{cronExpression}, + #{misfirePolicy}, + #{concurrent}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + \ No newline at end of file diff --git a/ruoyi-system/pom.xml b/ruoyi-system/pom.xml new file mode 100644 index 0000000..2f419a4 --- /dev/null +++ b/ruoyi-system/pom.xml @@ -0,0 +1,28 @@ + + + + ruoyi + com.ruoyi + 3.8.1 + + 4.0.0 + + ruoyi-system + + + system系统模块 + + + + + + + com.ruoyi + ruoyi-common + + + + + diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysConfig.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysConfig.java new file mode 100644 index 0000000..c54678c --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysConfig.java @@ -0,0 +1,111 @@ +package com.ruoyi.system.domain; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 参数配置表 sys_config + * + * @author ruoyi + */ +public class SysConfig extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 参数主键 */ + @Excel(name = "参数主键", cellType = ColumnType.NUMERIC) + private Long configId; + + /** 参数名称 */ + @Excel(name = "参数名称") + private String configName; + + /** 参数键名 */ + @Excel(name = "参数键名") + private String configKey; + + /** 参数键值 */ + @Excel(name = "参数键值") + private String configValue; + + /** 系统内置(Y是 N否) */ + @Excel(name = "系统内置", readConverterExp = "Y=是,N=否") + private String configType; + + public Long getConfigId() + { + return configId; + } + + public void setConfigId(Long configId) + { + this.configId = configId; + } + + @NotBlank(message = "参数名称不能为空") + @Size(min = 0, max = 100, message = "参数名称不能超过100个字符") + public String getConfigName() + { + return configName; + } + + public void setConfigName(String configName) + { + this.configName = configName; + } + + @NotBlank(message = "参数键名长度不能为空") + @Size(min = 0, max = 100, message = "参数键名长度不能超过100个字符") + public String getConfigKey() + { + return configKey; + } + + public void setConfigKey(String configKey) + { + this.configKey = configKey; + } + + @NotBlank(message = "参数键值不能为空") + @Size(min = 0, max = 500, message = "参数键值长度不能超过500个字符") + public String getConfigValue() + { + return configValue; + } + + public void setConfigValue(String configValue) + { + this.configValue = configValue; + } + + public String getConfigType() + { + return configType; + } + + public void setConfigType(String configType) + { + this.configType = configType; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("configId", getConfigId()) + .append("configName", getConfigName()) + .append("configKey", getConfigKey()) + .append("configValue", getConfigValue()) + .append("configType", getConfigType()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysLogininfor.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysLogininfor.java new file mode 100644 index 0000000..7fdea30 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysLogininfor.java @@ -0,0 +1,144 @@ +package com.ruoyi.system.domain; + +import java.util.Date; +import com.fasterxml.jackson.annotation.JsonFormat; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 系统访问记录表 sys_logininfor + * + * @author ruoyi + */ +public class SysLogininfor extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** ID */ + @Excel(name = "序号", cellType = ColumnType.NUMERIC) + private Long infoId; + + /** 用户账号 */ + @Excel(name = "用户账号") + private String userName; + + /** 登录状态 0成功 1失败 */ + @Excel(name = "登录状态", readConverterExp = "0=成功,1=失败") + private String status; + + /** 登录IP地址 */ + @Excel(name = "登录地址") + private String ipaddr; + + /** 登录地点 */ + @Excel(name = "登录地点") + private String loginLocation; + + /** 浏览器类型 */ + @Excel(name = "浏览器") + private String browser; + + /** 操作系统 */ + @Excel(name = "操作系统") + private String os; + + /** 提示消息 */ + @Excel(name = "提示消息") + private String msg; + + /** 访问时间 */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + @Excel(name = "访问时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") + private Date loginTime; + + public Long getInfoId() + { + return infoId; + } + + public void setInfoId(Long infoId) + { + this.infoId = infoId; + } + + public String getUserName() + { + return userName; + } + + public void setUserName(String userName) + { + this.userName = userName; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getIpaddr() + { + return ipaddr; + } + + public void setIpaddr(String ipaddr) + { + this.ipaddr = ipaddr; + } + + public String getLoginLocation() + { + return loginLocation; + } + + public void setLoginLocation(String loginLocation) + { + this.loginLocation = loginLocation; + } + + public String getBrowser() + { + return browser; + } + + public void setBrowser(String browser) + { + this.browser = browser; + } + + public String getOs() + { + return os; + } + + public void setOs(String os) + { + this.os = os; + } + + public String getMsg() + { + return msg; + } + + public void setMsg(String msg) + { + this.msg = msg; + } + + public Date getLoginTime() + { + return loginTime; + } + + public void setLoginTime(Date loginTime) + { + this.loginTime = loginTime; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysNotice.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysNotice.java new file mode 100644 index 0000000..8c07a54 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysNotice.java @@ -0,0 +1,102 @@ +package com.ruoyi.system.domain; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.core.domain.BaseEntity; +import com.ruoyi.common.xss.Xss; + +/** + * 通知公告表 sys_notice + * + * @author ruoyi + */ +public class SysNotice extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 公告ID */ + private Long noticeId; + + /** 公告标题 */ + private String noticeTitle; + + /** 公告类型(1通知 2公告) */ + private String noticeType; + + /** 公告内容 */ + private String noticeContent; + + /** 公告状态(0正常 1关闭) */ + private String status; + + public Long getNoticeId() + { + return noticeId; + } + + public void setNoticeId(Long noticeId) + { + this.noticeId = noticeId; + } + + public void setNoticeTitle(String noticeTitle) + { + this.noticeTitle = noticeTitle; + } + + @Xss(message = "公告标题不能包含脚本字符") + @NotBlank(message = "公告标题不能为空") + @Size(min = 0, max = 50, message = "公告标题不能超过50个字符") + public String getNoticeTitle() + { + return noticeTitle; + } + + public void setNoticeType(String noticeType) + { + this.noticeType = noticeType; + } + + public String getNoticeType() + { + return noticeType; + } + + public void setNoticeContent(String noticeContent) + { + this.noticeContent = noticeContent; + } + + public String getNoticeContent() + { + return noticeContent; + } + + public void setStatus(String status) + { + this.status = status; + } + + public String getStatus() + { + return status; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("noticeId", getNoticeId()) + .append("noticeTitle", getNoticeTitle()) + .append("noticeType", getNoticeType()) + .append("noticeContent", getNoticeContent()) + .append("status", getStatus()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysOperLog.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysOperLog.java new file mode 100644 index 0000000..175ee03 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysOperLog.java @@ -0,0 +1,255 @@ +package com.ruoyi.system.domain; + +import java.util.Date; +import com.fasterxml.jackson.annotation.JsonFormat; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 操作日志记录表 oper_log + * + * @author ruoyi + */ +public class SysOperLog extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 日志主键 */ + @Excel(name = "操作序号", cellType = ColumnType.NUMERIC) + private Long operId; + + /** 操作模块 */ + @Excel(name = "操作模块") + private String title; + + /** 业务类型(0其它 1新增 2修改 3删除) */ + @Excel(name = "业务类型", readConverterExp = "0=其它,1=新增,2=修改,3=删除,4=授权,5=导出,6=导入,7=强退,8=生成代码,9=清空数据") + private Integer businessType; + + /** 业务类型数组 */ + private Integer[] businessTypes; + + /** 请求方法 */ + @Excel(name = "请求方法") + private String method; + + /** 请求方式 */ + @Excel(name = "请求方式") + private String requestMethod; + + /** 操作类别(0其它 1后台用户 2手机端用户) */ + @Excel(name = "操作类别", readConverterExp = "0=其它,1=后台用户,2=手机端用户") + private Integer operatorType; + + /** 操作人员 */ + @Excel(name = "操作人员") + private String operName; + + /** 部门名称 */ + @Excel(name = "部门名称") + private String deptName; + + /** 请求url */ + @Excel(name = "请求地址") + private String operUrl; + + /** 操作地址 */ + @Excel(name = "操作地址") + private String operIp; + + /** 操作地点 */ + @Excel(name = "操作地点") + private String operLocation; + + /** 请求参数 */ + @Excel(name = "请求参数") + private String operParam; + + /** 返回参数 */ + @Excel(name = "返回参数") + private String jsonResult; + + /** 操作状态(0正常 1异常) */ + @Excel(name = "状态", readConverterExp = "0=正常,1=异常") + private Integer status; + + /** 错误消息 */ + @Excel(name = "错误消息") + private String errorMsg; + + /** 操作时间 */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + @Excel(name = "操作时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss") + private Date operTime; + + public Long getOperId() + { + return operId; + } + + public void setOperId(Long operId) + { + this.operId = operId; + } + + public String getTitle() + { + return title; + } + + public void setTitle(String title) + { + this.title = title; + } + + public Integer getBusinessType() + { + return businessType; + } + + public void setBusinessType(Integer businessType) + { + this.businessType = businessType; + } + + public Integer[] getBusinessTypes() + { + return businessTypes; + } + + public void setBusinessTypes(Integer[] businessTypes) + { + this.businessTypes = businessTypes; + } + + public String getMethod() + { + return method; + } + + public void setMethod(String method) + { + this.method = method; + } + + public String getRequestMethod() + { + return requestMethod; + } + + public void setRequestMethod(String requestMethod) + { + this.requestMethod = requestMethod; + } + + public Integer getOperatorType() + { + return operatorType; + } + + public void setOperatorType(Integer operatorType) + { + this.operatorType = operatorType; + } + + public String getOperName() + { + return operName; + } + + public void setOperName(String operName) + { + this.operName = operName; + } + + public String getDeptName() + { + return deptName; + } + + public void setDeptName(String deptName) + { + this.deptName = deptName; + } + + public String getOperUrl() + { + return operUrl; + } + + public void setOperUrl(String operUrl) + { + this.operUrl = operUrl; + } + + public String getOperIp() + { + return operIp; + } + + public void setOperIp(String operIp) + { + this.operIp = operIp; + } + + public String getOperLocation() + { + return operLocation; + } + + public void setOperLocation(String operLocation) + { + this.operLocation = operLocation; + } + + public String getOperParam() + { + return operParam; + } + + public void setOperParam(String operParam) + { + this.operParam = operParam; + } + + public String getJsonResult() + { + return jsonResult; + } + + public void setJsonResult(String jsonResult) + { + this.jsonResult = jsonResult; + } + + public Integer getStatus() + { + return status; + } + + public void setStatus(Integer status) + { + this.status = status; + } + + public String getErrorMsg() + { + return errorMsg; + } + + public void setErrorMsg(String errorMsg) + { + this.errorMsg = errorMsg; + } + + public Date getOperTime() + { + return operTime; + } + + public void setOperTime(Date operTime) + { + this.operTime = operTime; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysPost.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysPost.java new file mode 100644 index 0000000..1f1fcf4 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysPost.java @@ -0,0 +1,123 @@ +package com.ruoyi.system.domain; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.Size; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; +import com.ruoyi.common.annotation.Excel; +import com.ruoyi.common.annotation.Excel.ColumnType; +import com.ruoyi.common.core.domain.BaseEntity; + +/** + * 岗位表 sys_post + * + * @author ruoyi + */ +public class SysPost extends BaseEntity +{ + private static final long serialVersionUID = 1L; + + /** 岗位序号 */ + @Excel(name = "岗位序号", cellType = ColumnType.NUMERIC) + private Long postId; + + /** 岗位编码 */ + @Excel(name = "岗位编码") + private String postCode; + + /** 岗位名称 */ + @Excel(name = "岗位名称") + private String postName; + + /** 岗位排序 */ + @Excel(name = "岗位排序") + private String postSort; + + /** 状态(0正常 1停用) */ + @Excel(name = "状态", readConverterExp = "0=正常,1=停用") + private String status; + + /** 用户是否存在此岗位标识 默认不存在 */ + private boolean flag = false; + + public Long getPostId() + { + return postId; + } + + public void setPostId(Long postId) + { + this.postId = postId; + } + + @NotBlank(message = "岗位编码不能为空") + @Size(min = 0, max = 64, message = "岗位编码长度不能超过64个字符") + public String getPostCode() + { + return postCode; + } + + public void setPostCode(String postCode) + { + this.postCode = postCode; + } + + @NotBlank(message = "岗位名称不能为空") + @Size(min = 0, max = 50, message = "岗位名称长度不能超过50个字符") + public String getPostName() + { + return postName; + } + + public void setPostName(String postName) + { + this.postName = postName; + } + + @NotBlank(message = "显示顺序不能为空") + public String getPostSort() + { + return postSort; + } + + public void setPostSort(String postSort) + { + this.postSort = postSort; + } + + public String getStatus() + { + return status; + } + + public void setStatus(String status) + { + this.status = status; + } + + public boolean isFlag() + { + return flag; + } + + public void setFlag(boolean flag) + { + this.flag = flag; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("postId", getPostId()) + .append("postCode", getPostCode()) + .append("postName", getPostName()) + .append("postSort", getPostSort()) + .append("status", getStatus()) + .append("createBy", getCreateBy()) + .append("createTime", getCreateTime()) + .append("updateBy", getUpdateBy()) + .append("updateTime", getUpdateTime()) + .append("remark", getRemark()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleDept.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleDept.java new file mode 100644 index 0000000..47b21bf --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleDept.java @@ -0,0 +1,46 @@ +package com.ruoyi.system.domain; + +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +/** + * 角色和部门关联 sys_role_dept + * + * @author ruoyi + */ +public class SysRoleDept +{ + /** 角色ID */ + private Long roleId; + + /** 部门ID */ + private Long deptId; + + public Long getRoleId() + { + return roleId; + } + + public void setRoleId(Long roleId) + { + this.roleId = roleId; + } + + public Long getDeptId() + { + return deptId; + } + + public void setDeptId(Long deptId) + { + this.deptId = deptId; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("roleId", getRoleId()) + .append("deptId", getDeptId()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleMenu.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleMenu.java new file mode 100644 index 0000000..de10a74 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysRoleMenu.java @@ -0,0 +1,46 @@ +package com.ruoyi.system.domain; + +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +/** + * 角色和菜单关联 sys_role_menu + * + * @author ruoyi + */ +public class SysRoleMenu +{ + /** 角色ID */ + private Long roleId; + + /** 菜单ID */ + private Long menuId; + + public Long getRoleId() + { + return roleId; + } + + public void setRoleId(Long roleId) + { + this.roleId = roleId; + } + + public Long getMenuId() + { + return menuId; + } + + public void setMenuId(Long menuId) + { + this.menuId = menuId; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("roleId", getRoleId()) + .append("menuId", getMenuId()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserOnline.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserOnline.java new file mode 100644 index 0000000..2bbd318 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserOnline.java @@ -0,0 +1,113 @@ +package com.ruoyi.system.domain; + +/** + * 当前在线会话 + * + * @author ruoyi + */ +public class SysUserOnline +{ + /** 会话编号 */ + private String tokenId; + + /** 部门名称 */ + private String deptName; + + /** 用户名称 */ + private String userName; + + /** 登录IP地址 */ + private String ipaddr; + + /** 登录地址 */ + private String loginLocation; + + /** 浏览器类型 */ + private String browser; + + /** 操作系统 */ + private String os; + + /** 登录时间 */ + private Long loginTime; + + public String getTokenId() + { + return tokenId; + } + + public void setTokenId(String tokenId) + { + this.tokenId = tokenId; + } + + public String getDeptName() + { + return deptName; + } + + public void setDeptName(String deptName) + { + this.deptName = deptName; + } + + public String getUserName() + { + return userName; + } + + public void setUserName(String userName) + { + this.userName = userName; + } + + public String getIpaddr() + { + return ipaddr; + } + + public void setIpaddr(String ipaddr) + { + this.ipaddr = ipaddr; + } + + public String getLoginLocation() + { + return loginLocation; + } + + public void setLoginLocation(String loginLocation) + { + this.loginLocation = loginLocation; + } + + public String getBrowser() + { + return browser; + } + + public void setBrowser(String browser) + { + this.browser = browser; + } + + public String getOs() + { + return os; + } + + public void setOs(String os) + { + this.os = os; + } + + public Long getLoginTime() + { + return loginTime; + } + + public void setLoginTime(Long loginTime) + { + this.loginTime = loginTime; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserPost.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserPost.java new file mode 100644 index 0000000..6e8c416 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserPost.java @@ -0,0 +1,46 @@ +package com.ruoyi.system.domain; + +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +/** + * 用户和岗位关联 sys_user_post + * + * @author ruoyi + */ +public class SysUserPost +{ + /** 用户ID */ + private Long userId; + + /** 岗位ID */ + private Long postId; + + public Long getUserId() + { + return userId; + } + + public void setUserId(Long userId) + { + this.userId = userId; + } + + public Long getPostId() + { + return postId; + } + + public void setPostId(Long postId) + { + this.postId = postId; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("userId", getUserId()) + .append("postId", getPostId()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserRole.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserRole.java new file mode 100644 index 0000000..4d15810 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysUserRole.java @@ -0,0 +1,46 @@ +package com.ruoyi.system.domain; + +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +/** + * 用户和角色关联 sys_user_role + * + * @author ruoyi + */ +public class SysUserRole +{ + /** 用户ID */ + private Long userId; + + /** 角色ID */ + private Long roleId; + + public Long getUserId() + { + return userId; + } + + public void setUserId(Long userId) + { + this.userId = userId; + } + + public Long getRoleId() + { + return roleId; + } + + public void setRoleId(Long roleId) + { + this.roleId = roleId; + } + + @Override + public String toString() { + return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE) + .append("userId", getUserId()) + .append("roleId", getRoleId()) + .toString(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/MetaVo.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/MetaVo.java new file mode 100644 index 0000000..a5d5fdc --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/MetaVo.java @@ -0,0 +1,106 @@ +package com.ruoyi.system.domain.vo; + +import com.ruoyi.common.utils.StringUtils; + +/** + * 路由显示信息 + * + * @author ruoyi + */ +public class MetaVo +{ + /** + * 设置该路由在侧边栏和面包屑中展示的名字 + */ + private String title; + + /** + * 设置该路由的图标,对应路径src/assets/icons/svg + */ + private String icon; + + /** + * 设置为true,则不会被 缓存 + */ + private boolean noCache; + + /** + * 内链地址(http(s)://开头) + */ + private String link; + + public MetaVo() + { + } + + public MetaVo(String title, String icon) + { + this.title = title; + this.icon = icon; + } + + public MetaVo(String title, String icon, boolean noCache) + { + this.title = title; + this.icon = icon; + this.noCache = noCache; + } + + public MetaVo(String title, String icon, String link) + { + this.title = title; + this.icon = icon; + this.link = link; + } + + public MetaVo(String title, String icon, boolean noCache, String link) + { + this.title = title; + this.icon = icon; + this.noCache = noCache; + if (StringUtils.ishttp(link)) + { + this.link = link; + } + } + + public boolean isNoCache() + { + return noCache; + } + + public void setNoCache(boolean noCache) + { + this.noCache = noCache; + } + + public String getTitle() + { + return title; + } + + public void setTitle(String title) + { + this.title = title; + } + + public String getIcon() + { + return icon; + } + + public void setIcon(String icon) + { + this.icon = icon; + } + + public String getLink() + { + return link; + } + + public void setLink(String link) + { + this.link = link; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/RouterVo.java b/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/RouterVo.java new file mode 100644 index 0000000..afff8c9 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/domain/vo/RouterVo.java @@ -0,0 +1,148 @@ +package com.ruoyi.system.domain.vo; + +import com.fasterxml.jackson.annotation.JsonInclude; +import java.util.List; + +/** + * 路由配置信息 + * + * @author ruoyi + */ +@JsonInclude(JsonInclude.Include.NON_EMPTY) +public class RouterVo +{ + /** + * 路由名字 + */ + private String name; + + /** + * 路由地址 + */ + private String path; + + /** + * 是否隐藏路由,当设置 true 的时候该路由不会再侧边栏出现 + */ + private boolean hidden; + + /** + * 重定向地址,当设置 noRedirect 的时候该路由在面包屑导航中不可被点击 + */ + private String redirect; + + /** + * 组件地址 + */ + private String component; + + /** + * 路由参数:如 {"id": 1, "name": "ry"} + */ + private String query; + + /** + * 当你一个路由下面的 children 声明的路由大于1个时,自动会变成嵌套的模式--如组件页面 + */ + private Boolean alwaysShow; + + /** + * 其他元素 + */ + private MetaVo meta; + + /** + * 子路由 + */ + private List children; + + public String getName() + { + return name; + } + + public void setName(String name) + { + this.name = name; + } + + public String getPath() + { + return path; + } + + public void setPath(String path) + { + this.path = path; + } + + public boolean getHidden() + { + return hidden; + } + + public void setHidden(boolean hidden) + { + this.hidden = hidden; + } + + public String getRedirect() + { + return redirect; + } + + public void setRedirect(String redirect) + { + this.redirect = redirect; + } + + public String getComponent() + { + return component; + } + + public void setComponent(String component) + { + this.component = component; + } + + public String getQuery() + { + return query; + } + + public void setQuery(String query) + { + this.query = query; + } + + public Boolean getAlwaysShow() + { + return alwaysShow; + } + + public void setAlwaysShow(Boolean alwaysShow) + { + this.alwaysShow = alwaysShow; + } + + public MetaVo getMeta() + { + return meta; + } + + public void setMeta(MetaVo meta) + { + this.meta = meta; + } + + public List getChildren() + { + return children; + } + + public void setChildren(List children) + { + this.children = children; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysConfigMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysConfigMapper.java new file mode 100644 index 0000000..eb0c97c --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysConfigMapper.java @@ -0,0 +1,70 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysConfig; +import org.apache.ibatis.annotations.Mapper; + +/** + * 参数配置 数据层 + * + * @author ruoyi + */ +@Mapper +public interface SysConfigMapper +{ + /** + * 查询参数配置信息 + * + * @param config 参数配置信息 + * @return 参数配置信息 + */ + public SysConfig selectConfig(SysConfig config); + + /** + * 查询参数配置列表 + * + * @param config 参数配置信息 + * @return 参数配置集合 + */ + public List selectConfigList(SysConfig config); + + /** + * 根据键名查询参数配置信息 + * + * @param configKey 参数键名 + * @return 参数配置信息 + */ + public SysConfig checkConfigKeyUnique(String configKey); + + /** + * 新增参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + public int insertConfig(SysConfig config); + + /** + * 修改参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + public int updateConfig(SysConfig config); + + /** + * 删除参数配置 + * + * @param configId 参数ID + * @return 结果 + */ + public int deleteConfigById(Long configId); + + /** + * 批量删除参数信息 + * + * @param configIds 需要删除的参数ID + * @return 结果 + */ + public int deleteConfigByIds(Long[] configIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDeptMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDeptMapper.java new file mode 100644 index 0000000..384a9b6 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDeptMapper.java @@ -0,0 +1,118 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Param; +import com.ruoyi.common.core.domain.entity.SysDept; + +/** + * 部门管理 数据层 + * + * @author ruoyi + */ +public interface SysDeptMapper +{ + /** + * 查询部门管理数据 + * + * @param dept 部门信息 + * @return 部门信息集合 + */ + public List selectDeptList(SysDept dept); + + /** + * 根据角色ID查询部门树信息 + * + * @param roleId 角色ID + * @param deptCheckStrictly 部门树选择项是否关联显示 + * @return 选中部门列表 + */ + public List selectDeptListByRoleId(@Param("roleId") Long roleId, @Param("deptCheckStrictly") boolean deptCheckStrictly); + + /** + * 根据部门ID查询信息 + * + * @param deptId 部门ID + * @return 部门信息 + */ + public SysDept selectDeptById(Long deptId); + + /** + * 根据ID查询所有子部门 + * + * @param deptId 部门ID + * @return 部门列表 + */ + public List selectChildrenDeptById(Long deptId); + + /** + * 根据ID查询所有子部门(正常状态) + * + * @param deptId 部门ID + * @return 子部门数 + */ + public int selectNormalChildrenDeptById(Long deptId); + + /** + * 是否存在子节点 + * + * @param deptId 部门ID + * @return 结果 + */ + public int hasChildByDeptId(Long deptId); + + /** + * 查询部门是否存在用户 + * + * @param deptId 部门ID + * @return 结果 + */ + public int checkDeptExistUser(Long deptId); + + /** + * 校验部门名称是否唯一 + * + * @param deptName 部门名称 + * @param parentId 父部门ID + * @return 结果 + */ + public SysDept checkDeptNameUnique(@Param("deptName") String deptName, @Param("parentId") Long parentId); + + /** + * 新增部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + public int insertDept(SysDept dept); + + /** + * 修改部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + public int updateDept(SysDept dept); + + /** + * 修改所在部门正常状态 + * + * @param deptIds 部门ID组 + */ + public void updateDeptStatusNormal(Long[] deptIds); + + /** + * 修改子元素关系 + * + * @param depts 子元素 + * @return 结果 + */ + public int updateDeptChildren(@Param("depts") List depts); + + /** + * 删除部门管理信息 + * + * @param deptId 部门ID + * @return 结果 + */ + public int deleteDeptById(Long deptId); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictDataMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictDataMapper.java new file mode 100644 index 0000000..a341f1e --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictDataMapper.java @@ -0,0 +1,95 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Param; +import com.ruoyi.common.core.domain.entity.SysDictData; + +/** + * 字典表 数据层 + * + * @author ruoyi + */ +public interface SysDictDataMapper +{ + /** + * 根据条件分页查询字典数据 + * + * @param dictData 字典数据信息 + * @return 字典数据集合信息 + */ + public List selectDictDataList(SysDictData dictData); + + /** + * 根据字典类型查询字典数据 + * + * @param dictType 字典类型 + * @return 字典数据集合信息 + */ + public List selectDictDataByType(String dictType); + + /** + * 根据字典类型和字典键值查询字典数据信息 + * + * @param dictType 字典类型 + * @param dictValue 字典键值 + * @return 字典标签 + */ + public String selectDictLabel(@Param("dictType") String dictType, @Param("dictValue") String dictValue); + + /** + * 根据字典数据ID查询信息 + * + * @param dictCode 字典数据ID + * @return 字典数据 + */ + public SysDictData selectDictDataById(Long dictCode); + + /** + * 查询字典数据 + * + * @param dictType 字典类型 + * @return 字典数据 + */ + public int countDictDataByType(String dictType); + + /** + * 通过字典ID删除字典数据信息 + * + * @param dictCode 字典数据ID + * @return 结果 + */ + public int deleteDictDataById(Long dictCode); + + /** + * 批量删除字典数据信息 + * + * @param dictCodes 需要删除的字典数据ID + * @return 结果 + */ + public int deleteDictDataByIds(Long[] dictCodes); + + /** + * 新增字典数据信息 + * + * @param dictData 字典数据信息 + * @return 结果 + */ + public int insertDictData(SysDictData dictData); + + /** + * 修改字典数据信息 + * + * @param dictData 字典数据信息 + * @return 结果 + */ + public int updateDictData(SysDictData dictData); + + /** + * 同步修改字典类型 + * + * @param oldDictType 旧字典类型 + * @param newDictType 新旧字典类型 + * @return 结果 + */ + public int updateDictDataType(@Param("oldDictType") String oldDictType, @Param("newDictType") String newDictType); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictTypeMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictTypeMapper.java new file mode 100644 index 0000000..17545cd --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysDictTypeMapper.java @@ -0,0 +1,85 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Mapper; +import com.ruoyi.common.core.domain.entity.SysDictType; + +/** + * 字典表 数据层 + * + * @author ruoyi + */ +@Mapper +public interface SysDictTypeMapper +{ + /** + * 根据条件分页查询字典类型 + * + * @param dictType 字典类型信息 + * @return 字典类型集合信息 + */ + public List selectDictTypeList(SysDictType dictType); + + /** + * 根据所有字典类型 + * + * @return 字典类型集合信息 + */ + public List selectDictTypeAll(); + + /** + * 根据字典类型ID查询信息 + * + * @param dictId 字典类型ID + * @return 字典类型 + */ + public SysDictType selectDictTypeById(Long dictId); + + /** + * 根据字典类型查询信息 + * + * @param dictType 字典类型 + * @return 字典类型 + */ + public SysDictType selectDictTypeByType(String dictType); + + /** + * 通过字典ID删除字典信息 + * + * @param dictId 字典ID + * @return 结果 + */ + public int deleteDictTypeById(Long dictId); + + /** + * 批量删除字典类型信息 + * + * @param dictIds 需要删除的字典ID + * @return 结果 + */ + public int deleteDictTypeByIds(Long[] dictIds); + + /** + * 新增字典类型信息 + * + * @param dictType 字典类型信息 + * @return 结果 + */ + public int insertDictType(SysDictType dictType); + + /** + * 修改字典类型信息 + * + * @param dictType 字典类型信息 + * @return 结果 + */ + public int updateDictType(SysDictType dictType); + + /** + * 校验字典类型称是否唯一 + * + * @param dictType 字典类型 + * @return 结果 + */ + public SysDictType checkDictTypeUnique(String dictType); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysLogininforMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysLogininforMapper.java new file mode 100644 index 0000000..629866f --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysLogininforMapper.java @@ -0,0 +1,42 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysLogininfor; + +/** + * 系统访问日志情况信息 数据层 + * + * @author ruoyi + */ +public interface SysLogininforMapper +{ + /** + * 新增系统登录日志 + * + * @param logininfor 访问日志对象 + */ + public void insertLogininfor(SysLogininfor logininfor); + + /** + * 查询系统登录日志集合 + * + * @param logininfor 访问日志对象 + * @return 登录记录集合 + */ + public List selectLogininforList(SysLogininfor logininfor); + + /** + * 批量删除系统登录日志 + * + * @param infoIds 需要删除的登录日志ID + * @return 结果 + */ + public int deleteLogininforByIds(Long[] infoIds); + + /** + * 清空系统登录日志 + * + * @return 结果 + */ + public int cleanLogininfor(); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java new file mode 100644 index 0000000..123b47f --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java @@ -0,0 +1,117 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Param; +import com.ruoyi.common.core.domain.entity.SysMenu; + +/** + * 菜单表 数据层 + * + * @author ruoyi + */ +public interface SysMenuMapper +{ + /** + * 查询系统菜单列表 + * + * @param menu 菜单信息 + * @return 菜单列表 + */ + public List selectMenuList(SysMenu menu); + + /** + * 根据用户所有权限 + * + * @return 权限列表 + */ + public List selectMenuPerms(); + + /** + * 根据用户查询系统菜单列表 + * + * @param menu 菜单信息 + * @return 菜单列表 + */ + public List selectMenuListByUserId(SysMenu menu); + + /** + * 根据用户ID查询权限 + * + * @param userId 用户ID + * @return 权限列表 + */ + public List selectMenuPermsByUserId(Long userId); + + /** + * 根据用户ID查询菜单 + * + * @return 菜单列表 + */ + public List selectMenuTreeAll(); + + /** + * 根据用户ID查询菜单 + * + * @param userId 用户ID + * @return 菜单列表 + */ + public List selectMenuTreeByUserId(Long userId); + + /** + * 根据角色ID查询菜单树信息 + * + * @param roleId 角色ID + * @param menuCheckStrictly 菜单树选择项是否关联显示 + * @return 选中菜单列表 + */ + public List selectMenuListByRoleId(@Param("roleId") Long roleId, @Param("menuCheckStrictly") boolean menuCheckStrictly); + + /** + * 根据菜单ID查询信息 + * + * @param menuId 菜单ID + * @return 菜单信息 + */ + public SysMenu selectMenuById(Long menuId); + + /** + * 是否存在菜单子节点 + * + * @param menuId 菜单ID + * @return 结果 + */ + public int hasChildByMenuId(Long menuId); + + /** + * 新增菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + public int insertMenu(SysMenu menu); + + /** + * 修改菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + public int updateMenu(SysMenu menu); + + /** + * 删除菜单管理信息 + * + * @param menuId 菜单ID + * @return 结果 + */ + public int deleteMenuById(Long menuId); + + /** + * 校验菜单名称是否唯一 + * + * @param menuName 菜单名称 + * @param parentId 父菜单ID + * @return 结果 + */ + public SysMenu checkMenuNameUnique(@Param("menuName") String menuName, @Param("parentId") Long parentId); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysNoticeMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysNoticeMapper.java new file mode 100644 index 0000000..c34f0a2 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysNoticeMapper.java @@ -0,0 +1,60 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysNotice; + +/** + * 通知公告表 数据层 + * + * @author ruoyi + */ +public interface SysNoticeMapper +{ + /** + * 查询公告信息 + * + * @param noticeId 公告ID + * @return 公告信息 + */ + public SysNotice selectNoticeById(Long noticeId); + + /** + * 查询公告列表 + * + * @param notice 公告信息 + * @return 公告集合 + */ + public List selectNoticeList(SysNotice notice); + + /** + * 新增公告 + * + * @param notice 公告信息 + * @return 结果 + */ + public int insertNotice(SysNotice notice); + + /** + * 修改公告 + * + * @param notice 公告信息 + * @return 结果 + */ + public int updateNotice(SysNotice notice); + + /** + * 批量删除公告 + * + * @param noticeId 公告ID + * @return 结果 + */ + public int deleteNoticeById(Long noticeId); + + /** + * 批量删除公告信息 + * + * @param noticeIds 需要删除的公告ID + * @return 结果 + */ + public int deleteNoticeByIds(Long[] noticeIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysOperLogMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysOperLogMapper.java new file mode 100644 index 0000000..2ae6457 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysOperLogMapper.java @@ -0,0 +1,48 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysOperLog; + +/** + * 操作日志 数据层 + * + * @author ruoyi + */ +public interface SysOperLogMapper +{ + /** + * 新增操作日志 + * + * @param operLog 操作日志对象 + */ + public void insertOperlog(SysOperLog operLog); + + /** + * 查询系统操作日志集合 + * + * @param operLog 操作日志对象 + * @return 操作日志集合 + */ + public List selectOperLogList(SysOperLog operLog); + + /** + * 批量删除系统操作日志 + * + * @param operIds 需要删除的操作日志ID + * @return 结果 + */ + public int deleteOperLogByIds(Long[] operIds); + + /** + * 查询操作日志详细 + * + * @param operId 操作ID + * @return 操作日志对象 + */ + public SysOperLog selectOperLogById(Long operId); + + /** + * 清空操作日志 + */ + public void cleanOperLog(); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysPostMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysPostMapper.java new file mode 100644 index 0000000..19be227 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysPostMapper.java @@ -0,0 +1,99 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysPost; + +/** + * 岗位信息 数据层 + * + * @author ruoyi + */ +public interface SysPostMapper +{ + /** + * 查询岗位数据集合 + * + * @param post 岗位信息 + * @return 岗位数据集合 + */ + public List selectPostList(SysPost post); + + /** + * 查询所有岗位 + * + * @return 岗位列表 + */ + public List selectPostAll(); + + /** + * 通过岗位ID查询岗位信息 + * + * @param postId 岗位ID + * @return 角色对象信息 + */ + public SysPost selectPostById(Long postId); + + /** + * 根据用户ID获取岗位选择框列表 + * + * @param userId 用户ID + * @return 选中岗位ID列表 + */ + public List selectPostListByUserId(Long userId); + + /** + * 查询用户所属岗位组 + * + * @param userName 用户名 + * @return 结果 + */ + public List selectPostsByUserName(String userName); + + /** + * 删除岗位信息 + * + * @param postId 岗位ID + * @return 结果 + */ + public int deletePostById(Long postId); + + /** + * 批量删除岗位信息 + * + * @param postIds 需要删除的岗位ID + * @return 结果 + */ + public int deletePostByIds(Long[] postIds); + + /** + * 修改岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + public int updatePost(SysPost post); + + /** + * 新增岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + public int insertPost(SysPost post); + + /** + * 校验岗位名称 + * + * @param postName 岗位名称 + * @return 结果 + */ + public SysPost checkPostNameUnique(String postName); + + /** + * 校验岗位编码 + * + * @param postCode 岗位编码 + * @return 结果 + */ + public SysPost checkPostCodeUnique(String postCode); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleDeptMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleDeptMapper.java new file mode 100644 index 0000000..f9d3a2f --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleDeptMapper.java @@ -0,0 +1,44 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysRoleDept; + +/** + * 角色与部门关联表 数据层 + * + * @author ruoyi + */ +public interface SysRoleDeptMapper +{ + /** + * 通过角色ID删除角色和部门关联 + * + * @param roleId 角色ID + * @return 结果 + */ + public int deleteRoleDeptByRoleId(Long roleId); + + /** + * 批量删除角色部门关联信息 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteRoleDept(Long[] ids); + + /** + * 查询部门使用数量 + * + * @param deptId 部门ID + * @return 结果 + */ + public int selectCountRoleDeptByDeptId(Long deptId); + + /** + * 批量新增角色部门信息 + * + * @param roleDeptList 角色部门列表 + * @return 结果 + */ + public int batchRoleDept(List roleDeptList); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMapper.java new file mode 100644 index 0000000..cf2bd8c --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMapper.java @@ -0,0 +1,107 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.common.core.domain.entity.SysRole; + +/** + * 角色表 数据层 + * + * @author ruoyi + */ +public interface SysRoleMapper +{ + /** + * 根据条件分页查询角色数据 + * + * @param role 角色信息 + * @return 角色数据集合信息 + */ + public List selectRoleList(SysRole role); + + /** + * 根据用户ID查询角色 + * + * @param userId 用户ID + * @return 角色列表 + */ + public List selectRolePermissionByUserId(Long userId); + + /** + * 查询所有角色 + * + * @return 角色列表 + */ + public List selectRoleAll(); + + /** + * 根据用户ID获取角色选择框列表 + * + * @param userId 用户ID + * @return 选中角色ID列表 + */ + public List selectRoleListByUserId(Long userId); + + /** + * 通过角色ID查询角色 + * + * @param roleId 角色ID + * @return 角色对象信息 + */ + public SysRole selectRoleById(Long roleId); + + /** + * 根据用户ID查询角色 + * + * @param userName 用户名 + * @return 角色列表 + */ + public List selectRolesByUserName(String userName); + + /** + * 校验角色名称是否唯一 + * + * @param roleName 角色名称 + * @return 角色信息 + */ + public SysRole checkRoleNameUnique(String roleName); + + /** + * 校验角色权限是否唯一 + * + * @param roleKey 角色权限 + * @return 角色信息 + */ + public SysRole checkRoleKeyUnique(String roleKey); + + /** + * 修改角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + public int updateRole(SysRole role); + + /** + * 新增角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + public int insertRole(SysRole role); + + /** + * 通过角色ID删除角色 + * + * @param roleId 角色ID + * @return 结果 + */ + public int deleteRoleById(Long roleId); + + /** + * 批量删除角色信息 + * + * @param roleIds 需要删除的角色ID + * @return 结果 + */ + public int deleteRoleByIds(Long[] roleIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMenuMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMenuMapper.java new file mode 100644 index 0000000..6602bee --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysRoleMenuMapper.java @@ -0,0 +1,44 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysRoleMenu; + +/** + * 角色与菜单关联表 数据层 + * + * @author ruoyi + */ +public interface SysRoleMenuMapper +{ + /** + * 查询菜单使用数量 + * + * @param menuId 菜单ID + * @return 结果 + */ + public int checkMenuExistRole(Long menuId); + + /** + * 通过角色ID删除角色和菜单关联 + * + * @param roleId 角色ID + * @return 结果 + */ + public int deleteRoleMenuByRoleId(Long roleId); + + /** + * 批量删除角色菜单关联信息 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteRoleMenu(Long[] ids); + + /** + * 批量新增角色菜单信息 + * + * @param roleMenuList 角色菜单列表 + * @return 结果 + */ + public int batchRoleMenu(List roleMenuList); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserMapper.java new file mode 100644 index 0000000..b42ac52 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserMapper.java @@ -0,0 +1,127 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Param; +import com.ruoyi.common.core.domain.entity.SysUser; + +/** + * 用户表 数据层 + * + * @author ruoyi + */ +public interface SysUserMapper +{ + /** + * 根据条件分页查询用户列表 + * + * @param sysUser 用户信息 + * @return 用户信息集合信息 + */ + public List selectUserList(SysUser sysUser); + + /** + * 根据条件分页查询已配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + public List selectAllocatedList(SysUser user); + + /** + * 根据条件分页查询未分配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + public List selectUnallocatedList(SysUser user); + + /** + * 通过用户名查询用户 + * + * @param userName 用户名 + * @return 用户对象信息 + */ + public SysUser selectUserByUserName(String userName); + + /** + * 通过用户ID查询用户 + * + * @param userId 用户ID + * @return 用户对象信息 + */ + public SysUser selectUserById(Long userId); + + /** + * 新增用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + public int insertUser(SysUser user); + + /** + * 修改用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + public int updateUser(SysUser user); + + /** + * 修改用户头像 + * + * @param userName 用户名 + * @param avatar 头像地址 + * @return 结果 + */ + public int updateUserAvatar(@Param("userName") String userName, @Param("avatar") String avatar); + + /** + * 重置用户密码 + * + * @param userName 用户名 + * @param password 密码 + * @return 结果 + */ + public int resetUserPwd(@Param("userName") String userName, @Param("password") String password); + + /** + * 通过用户ID删除用户 + * + * @param userId 用户ID + * @return 结果 + */ + public int deleteUserById(Long userId); + + /** + * 批量删除用户信息 + * + * @param userIds 需要删除的用户ID + * @return 结果 + */ + public int deleteUserByIds(Long[] userIds); + + /** + * 校验用户名称是否唯一 + * + * @param userName 用户名称 + * @return 结果 + */ + public int checkUserNameUnique(String userName); + + /** + * 校验手机号码是否唯一 + * + * @param phonenumber 手机号码 + * @return 结果 + */ + public SysUser checkPhoneUnique(String phonenumber); + + /** + * 校验email是否唯一 + * + * @param email 用户邮箱 + * @return 结果 + */ + public SysUser checkEmailUnique(String email); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserPostMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserPostMapper.java new file mode 100644 index 0000000..e08991d --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserPostMapper.java @@ -0,0 +1,44 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import com.ruoyi.system.domain.SysUserPost; + +/** + * 用户与岗位关联表 数据层 + * + * @author ruoyi + */ +public interface SysUserPostMapper +{ + /** + * 通过用户ID删除用户和岗位关联 + * + * @param userId 用户ID + * @return 结果 + */ + public int deleteUserPostByUserId(Long userId); + + /** + * 通过岗位ID查询岗位使用数量 + * + * @param postId 岗位ID + * @return 结果 + */ + public int countUserPostById(Long postId); + + /** + * 批量删除用户和岗位关联 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteUserPost(Long[] ids); + + /** + * 批量新增用户岗位信息 + * + * @param userPostList 用户角色列表 + * @return 结果 + */ + public int batchUserPost(List userPostList); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserRoleMapper.java b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserRoleMapper.java new file mode 100644 index 0000000..3143ec8 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysUserRoleMapper.java @@ -0,0 +1,62 @@ +package com.ruoyi.system.mapper; + +import java.util.List; +import org.apache.ibatis.annotations.Param; +import com.ruoyi.system.domain.SysUserRole; + +/** + * 用户与角色关联表 数据层 + * + * @author ruoyi + */ +public interface SysUserRoleMapper +{ + /** + * 通过用户ID删除用户和角色关联 + * + * @param userId 用户ID + * @return 结果 + */ + public int deleteUserRoleByUserId(Long userId); + + /** + * 批量删除用户和角色关联 + * + * @param ids 需要删除的数据ID + * @return 结果 + */ + public int deleteUserRole(Long[] ids); + + /** + * 通过角色ID查询角色使用数量 + * + * @param roleId 角色ID + * @return 结果 + */ + public int countUserRoleByRoleId(Long roleId); + + /** + * 批量新增用户角色信息 + * + * @param userRoleList 用户角色列表 + * @return 结果 + */ + public int batchUserRole(List userRoleList); + + /** + * 删除用户和角色关联信息 + * + * @param userRole 用户和角色关联信息 + * @return 结果 + */ + public int deleteUserRoleInfo(SysUserRole userRole); + + /** + * 批量取消授权用户角色 + * + * @param roleId 角色ID + * @param userIds 需要删除的用户数据ID + * @return 结果 + */ + public int deleteUserRoleInfos(@Param("roleId") Long roleId, @Param("userIds") Long[] userIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysConfigService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysConfigService.java new file mode 100644 index 0000000..00b037a --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysConfigService.java @@ -0,0 +1,89 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.system.domain.SysConfig; + +/** + * 参数配置 服务层 + * + * @author ruoyi + */ +public interface ISysConfigService +{ + /** + * 查询参数配置信息 + * + * @param configId 参数配置ID + * @return 参数配置信息 + */ + public SysConfig selectConfigById(Long configId); + + /** + * 根据键名查询参数配置信息 + * + * @param configKey 参数键名 + * @return 参数键值 + */ + public String selectConfigByKey(String configKey); + + /** + * 获取验证码开关 + * + * @return true开启,false关闭 + */ + public boolean selectCaptchaOnOff(); + + /** + * 查询参数配置列表 + * + * @param config 参数配置信息 + * @return 参数配置集合 + */ + public List selectConfigList(SysConfig config); + + /** + * 新增参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + public int insertConfig(SysConfig config); + + /** + * 修改参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + public int updateConfig(SysConfig config); + + /** + * 批量删除参数信息 + * + * @param configIds 需要删除的参数ID + */ + public void deleteConfigByIds(Long[] configIds); + + /** + * 加载参数缓存数据 + */ + public void loadingConfigCache(); + + /** + * 清空参数缓存数据 + */ + public void clearConfigCache(); + + /** + * 重置参数缓存数据 + */ + public void resetConfigCache(); + + /** + * 校验参数键名是否唯一 + * + * @param config 参数信息 + * @return 结果 + */ + public String checkConfigKeyUnique(SysConfig config); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDeptService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDeptService.java new file mode 100644 index 0000000..72ca048 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDeptService.java @@ -0,0 +1,116 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.common.core.domain.TreeSelect; +import com.ruoyi.common.core.domain.entity.SysDept; + +/** + * 部门管理 服务层 + * + * @author ruoyi + */ +public interface ISysDeptService +{ + /** + * 查询部门管理数据 + * + * @param dept 部门信息 + * @return 部门信息集合 + */ + public List selectDeptList(SysDept dept); + + /** + * 构建前端所需要树结构 + * + * @param depts 部门列表 + * @return 树结构列表 + */ + public List buildDeptTree(List depts); + + /** + * 构建前端所需要下拉树结构 + * + * @param depts 部门列表 + * @return 下拉树结构列表 + */ + public List buildDeptTreeSelect(List depts); + + /** + * 根据角色ID查询部门树信息 + * + * @param roleId 角色ID + * @return 选中部门列表 + */ + public List selectDeptListByRoleId(Long roleId); + + /** + * 根据部门ID查询信息 + * + * @param deptId 部门ID + * @return 部门信息 + */ + public SysDept selectDeptById(Long deptId); + + /** + * 根据ID查询所有子部门(正常状态) + * + * @param deptId 部门ID + * @return 子部门数 + */ + public int selectNormalChildrenDeptById(Long deptId); + + /** + * 是否存在部门子节点 + * + * @param deptId 部门ID + * @return 结果 + */ + public boolean hasChildByDeptId(Long deptId); + + /** + * 查询部门是否存在用户 + * + * @param deptId 部门ID + * @return 结果 true 存在 false 不存在 + */ + public boolean checkDeptExistUser(Long deptId); + + /** + * 校验部门名称是否唯一 + * + * @param dept 部门信息 + * @return 结果 + */ + public String checkDeptNameUnique(SysDept dept); + + /** + * 校验部门是否有数据权限 + * + * @param deptId 部门id + */ + public void checkDeptDataScope(Long deptId); + + /** + * 新增保存部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + public int insertDept(SysDept dept); + + /** + * 修改保存部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + public int updateDept(SysDept dept); + + /** + * 删除部门管理信息 + * + * @param deptId 部门ID + * @return 结果 + */ + public int deleteDeptById(Long deptId); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictDataService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictDataService.java new file mode 100644 index 0000000..9bc4f13 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictDataService.java @@ -0,0 +1,60 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.common.core.domain.entity.SysDictData; + +/** + * 字典 业务层 + * + * @author ruoyi + */ +public interface ISysDictDataService +{ + /** + * 根据条件分页查询字典数据 + * + * @param dictData 字典数据信息 + * @return 字典数据集合信息 + */ + public List selectDictDataList(SysDictData dictData); + + /** + * 根据字典类型和字典键值查询字典数据信息 + * + * @param dictType 字典类型 + * @param dictValue 字典键值 + * @return 字典标签 + */ + public String selectDictLabel(String dictType, String dictValue); + + /** + * 根据字典数据ID查询信息 + * + * @param dictCode 字典数据ID + * @return 字典数据 + */ + public SysDictData selectDictDataById(Long dictCode); + + /** + * 批量删除字典数据信息 + * + * @param dictCodes 需要删除的字典数据ID + */ + public void deleteDictDataByIds(Long[] dictCodes); + + /** + * 新增保存字典数据信息 + * + * @param dictData 字典数据信息 + * @return 结果 + */ + public int insertDictData(SysDictData dictData); + + /** + * 修改保存字典数据信息 + * + * @param dictData 字典数据信息 + * @return 结果 + */ + public int updateDictData(SysDictData dictData); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictTypeService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictTypeService.java new file mode 100644 index 0000000..dbb2c16 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysDictTypeService.java @@ -0,0 +1,98 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.common.core.domain.entity.SysDictData; +import com.ruoyi.common.core.domain.entity.SysDictType; + +/** + * 字典 业务层 + * + * @author ruoyi + */ +public interface ISysDictTypeService +{ + /** + * 根据条件分页查询字典类型 + * + * @param dictType 字典类型信息 + * @return 字典类型集合信息 + */ + public List selectDictTypeList(SysDictType dictType); + + /** + * 根据所有字典类型 + * + * @return 字典类型集合信息 + */ + public List selectDictTypeAll(); + + /** + * 根据字典类型查询字典数据 + * + * @param dictType 字典类型 + * @return 字典数据集合信息 + */ + public List selectDictDataByType(String dictType); + + /** + * 根据字典类型ID查询信息 + * + * @param dictId 字典类型ID + * @return 字典类型 + */ + public SysDictType selectDictTypeById(Long dictId); + + /** + * 根据字典类型查询信息 + * + * @param dictType 字典类型 + * @return 字典类型 + */ + public SysDictType selectDictTypeByType(String dictType); + + /** + * 批量删除字典信息 + * + * @param dictIds 需要删除的字典ID + */ + public void deleteDictTypeByIds(Long[] dictIds); + + /** + * 加载字典缓存数据 + */ + public void loadingDictCache(); + + /** + * 清空字典缓存数据 + */ + public void clearDictCache(); + + /** + * 重置字典缓存数据 + */ + public void resetDictCache(); + + /** + * 新增保存字典类型信息 + * + * @param dictType 字典类型信息 + * @return 结果 + */ + public int insertDictType(SysDictType dictType); + + /** + * 修改保存字典类型信息 + * + * @param dictType 字典类型信息 + * @return 结果 + */ + public int updateDictType(SysDictType dictType); + + /** + * 校验字典类型称是否唯一 + * + * @param dictType 字典类型 + * @return 结果 + */ + public String checkDictTypeUnique(SysDictType dictType); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysLogininforService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysLogininforService.java new file mode 100644 index 0000000..ce3151d --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysLogininforService.java @@ -0,0 +1,40 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.system.domain.SysLogininfor; + +/** + * 系统访问日志情况信息 服务层 + * + * @author ruoyi + */ +public interface ISysLogininforService +{ + /** + * 新增系统登录日志 + * + * @param logininfor 访问日志对象 + */ + public void insertLogininfor(SysLogininfor logininfor); + + /** + * 查询系统登录日志集合 + * + * @param logininfor 访问日志对象 + * @return 登录记录集合 + */ + public List selectLogininforList(SysLogininfor logininfor); + + /** + * 批量删除系统登录日志 + * + * @param infoIds 需要删除的登录日志ID + * @return 结果 + */ + public int deleteLogininforByIds(Long[] infoIds); + + /** + * 清空系统登录日志 + */ + public void cleanLogininfor(); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysMenuService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysMenuService.java new file mode 100644 index 0000000..1b43174 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysMenuService.java @@ -0,0 +1,136 @@ +package com.ruoyi.system.service; + +import java.util.List; +import java.util.Set; +import com.ruoyi.common.core.domain.TreeSelect; +import com.ruoyi.common.core.domain.entity.SysMenu; +import com.ruoyi.system.domain.vo.RouterVo; + +/** + * 菜单 业务层 + * + * @author ruoyi + */ +public interface ISysMenuService +{ + /** + * 根据用户查询系统菜单列表 + * + * @param userId 用户ID + * @return 菜单列表 + */ + public List selectMenuList(Long userId); + + /** + * 根据用户查询系统菜单列表 + * + * @param menu 菜单信息 + * @param userId 用户ID + * @return 菜单列表 + */ + public List selectMenuList(SysMenu menu, Long userId); + + /** + * 根据用户ID查询权限 + * + * @param userId 用户ID + * @return 权限列表 + */ + public Set selectMenuPermsByUserId(Long userId); + + /** + * 根据用户ID查询菜单树信息 + * + * @param userId 用户ID + * @return 菜单列表 + */ + public List selectMenuTreeByUserId(Long userId); + + /** + * 根据角色ID查询菜单树信息 + * + * @param roleId 角色ID + * @return 选中菜单列表 + */ + public List selectMenuListByRoleId(Long roleId); + + /** + * 构建前端路由所需要的菜单 + * + * @param menus 菜单列表 + * @return 路由列表 + */ + public List buildMenus(List menus); + + /** + * 构建前端所需要树结构 + * + * @param menus 菜单列表 + * @return 树结构列表 + */ + public List buildMenuTree(List menus); + + /** + * 构建前端所需要下拉树结构 + * + * @param menus 菜单列表 + * @return 下拉树结构列表 + */ + public List buildMenuTreeSelect(List menus); + + /** + * 根据菜单ID查询信息 + * + * @param menuId 菜单ID + * @return 菜单信息 + */ + public SysMenu selectMenuById(Long menuId); + + /** + * 是否存在菜单子节点 + * + * @param menuId 菜单ID + * @return 结果 true 存在 false 不存在 + */ + public boolean hasChildByMenuId(Long menuId); + + /** + * 查询菜单是否存在角色 + * + * @param menuId 菜单ID + * @return 结果 true 存在 false 不存在 + */ + public boolean checkMenuExistRole(Long menuId); + + /** + * 新增保存菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + public int insertMenu(SysMenu menu); + + /** + * 修改保存菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + public int updateMenu(SysMenu menu); + + /** + * 删除菜单管理信息 + * + * @param menuId 菜单ID + * @return 结果 + */ + public int deleteMenuById(Long menuId); + + /** + * 校验菜单名称是否唯一 + * + * @param menu 菜单信息 + * @return 结果 + */ + public String checkMenuNameUnique(SysMenu menu); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysNoticeService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysNoticeService.java new file mode 100644 index 0000000..47ce1b7 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysNoticeService.java @@ -0,0 +1,60 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.system.domain.SysNotice; + +/** + * 公告 服务层 + * + * @author ruoyi + */ +public interface ISysNoticeService +{ + /** + * 查询公告信息 + * + * @param noticeId 公告ID + * @return 公告信息 + */ + public SysNotice selectNoticeById(Long noticeId); + + /** + * 查询公告列表 + * + * @param notice 公告信息 + * @return 公告集合 + */ + public List selectNoticeList(SysNotice notice); + + /** + * 新增公告 + * + * @param notice 公告信息 + * @return 结果 + */ + public int insertNotice(SysNotice notice); + + /** + * 修改公告 + * + * @param notice 公告信息 + * @return 结果 + */ + public int updateNotice(SysNotice notice); + + /** + * 删除公告信息 + * + * @param noticeId 公告ID + * @return 结果 + */ + public int deleteNoticeById(Long noticeId); + + /** + * 批量删除公告信息 + * + * @param noticeIds 需要删除的公告ID + * @return 结果 + */ + public int deleteNoticeByIds(Long[] noticeIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysOperLogService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysOperLogService.java new file mode 100644 index 0000000..4fd8e5a --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysOperLogService.java @@ -0,0 +1,48 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.system.domain.SysOperLog; + +/** + * 操作日志 服务层 + * + * @author ruoyi + */ +public interface ISysOperLogService +{ + /** + * 新增操作日志 + * + * @param operLog 操作日志对象 + */ + public void insertOperlog(SysOperLog operLog); + + /** + * 查询系统操作日志集合 + * + * @param operLog 操作日志对象 + * @return 操作日志集合 + */ + public List selectOperLogList(SysOperLog operLog); + + /** + * 批量删除系统操作日志 + * + * @param operIds 需要删除的操作日志ID + * @return 结果 + */ + public int deleteOperLogByIds(Long[] operIds); + + /** + * 查询操作日志详细 + * + * @param operId 操作ID + * @return 操作日志对象 + */ + public SysOperLog selectOperLogById(Long operId); + + /** + * 清空操作日志 + */ + public void cleanOperLog(); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysPostService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysPostService.java new file mode 100644 index 0000000..4ffee39 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysPostService.java @@ -0,0 +1,99 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.system.domain.SysPost; + +/** + * 岗位信息 服务层 + * + * @author ruoyi + */ +public interface ISysPostService +{ + /** + * 查询岗位信息集合 + * + * @param post 岗位信息 + * @return 岗位列表 + */ + public List selectPostList(SysPost post); + + /** + * 查询所有岗位 + * + * @return 岗位列表 + */ + public List selectPostAll(); + + /** + * 通过岗位ID查询岗位信息 + * + * @param postId 岗位ID + * @return 角色对象信息 + */ + public SysPost selectPostById(Long postId); + + /** + * 根据用户ID获取岗位选择框列表 + * + * @param userId 用户ID + * @return 选中岗位ID列表 + */ + public List selectPostListByUserId(Long userId); + + /** + * 校验岗位名称 + * + * @param post 岗位信息 + * @return 结果 + */ + public String checkPostNameUnique(SysPost post); + + /** + * 校验岗位编码 + * + * @param post 岗位信息 + * @return 结果 + */ + public String checkPostCodeUnique(SysPost post); + + /** + * 通过岗位ID查询岗位使用数量 + * + * @param postId 岗位ID + * @return 结果 + */ + public int countUserPostById(Long postId); + + /** + * 删除岗位信息 + * + * @param postId 岗位ID + * @return 结果 + */ + public int deletePostById(Long postId); + + /** + * 批量删除岗位信息 + * + * @param postIds 需要删除的岗位ID + * @return 结果 + */ + public int deletePostByIds(Long[] postIds); + + /** + * 新增保存岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + public int insertPost(SysPost post); + + /** + * 修改保存岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + public int updatePost(SysPost post); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysRoleService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysRoleService.java new file mode 100644 index 0000000..28c73cc --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysRoleService.java @@ -0,0 +1,173 @@ +package com.ruoyi.system.service; + +import java.util.List; +import java.util.Set; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.system.domain.SysUserRole; + +/** + * 角色业务层 + * + * @author ruoyi + */ +public interface ISysRoleService +{ + /** + * 根据条件分页查询角色数据 + * + * @param role 角色信息 + * @return 角色数据集合信息 + */ + public List selectRoleList(SysRole role); + + /** + * 根据用户ID查询角色列表 + * + * @param userId 用户ID + * @return 角色列表 + */ + public List selectRolesByUserId(Long userId); + + /** + * 根据用户ID查询角色权限 + * + * @param userId 用户ID + * @return 权限列表 + */ + public Set selectRolePermissionByUserId(Long userId); + + /** + * 查询所有角色 + * + * @return 角色列表 + */ + public List selectRoleAll(); + + /** + * 根据用户ID获取角色选择框列表 + * + * @param userId 用户ID + * @return 选中角色ID列表 + */ + public List selectRoleListByUserId(Long userId); + + /** + * 通过角色ID查询角色 + * + * @param roleId 角色ID + * @return 角色对象信息 + */ + public SysRole selectRoleById(Long roleId); + + /** + * 校验角色名称是否唯一 + * + * @param role 角色信息 + * @return 结果 + */ + public String checkRoleNameUnique(SysRole role); + + /** + * 校验角色权限是否唯一 + * + * @param role 角色信息 + * @return 结果 + */ + public String checkRoleKeyUnique(SysRole role); + + /** + * 校验角色是否允许操作 + * + * @param role 角色信息 + */ + public void checkRoleAllowed(SysRole role); + + /** + * 校验角色是否有数据权限 + * + * @param roleId 角色id + */ + public void checkRoleDataScope(Long roleId); + + /** + * 通过角色ID查询角色使用数量 + * + * @param roleId 角色ID + * @return 结果 + */ + public int countUserRoleByRoleId(Long roleId); + + /** + * 新增保存角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + public int insertRole(SysRole role); + + /** + * 修改保存角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + public int updateRole(SysRole role); + + /** + * 修改角色状态 + * + * @param role 角色信息 + * @return 结果 + */ + public int updateRoleStatus(SysRole role); + + /** + * 修改数据权限信息 + * + * @param role 角色信息 + * @return 结果 + */ + public int authDataScope(SysRole role); + + /** + * 通过角色ID删除角色 + * + * @param roleId 角色ID + * @return 结果 + */ + public int deleteRoleById(Long roleId); + + /** + * 批量删除角色信息 + * + * @param roleIds 需要删除的角色ID + * @return 结果 + */ + public int deleteRoleByIds(Long[] roleIds); + + /** + * 取消授权用户角色 + * + * @param userRole 用户和角色关联信息 + * @return 结果 + */ + public int deleteAuthUser(SysUserRole userRole); + + /** + * 批量取消授权用户角色 + * + * @param roleId 角色ID + * @param userIds 需要取消授权的用户数据ID + * @return 结果 + */ + public int deleteAuthUsers(Long roleId, Long[] userIds); + + /** + * 批量选择授权用户角色 + * + * @param roleId 角色ID + * @param userIds 需要删除的用户数据ID + * @return 结果 + */ + public int insertAuthUsers(Long roleId, Long[] userIds); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserOnlineService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserOnlineService.java new file mode 100644 index 0000000..8eb5448 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserOnlineService.java @@ -0,0 +1,48 @@ +package com.ruoyi.system.service; + +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.system.domain.SysUserOnline; + +/** + * 在线用户 服务层 + * + * @author ruoyi + */ +public interface ISysUserOnlineService +{ + /** + * 通过登录地址查询信息 + * + * @param ipaddr 登录地址 + * @param user 用户信息 + * @return 在线用户信息 + */ + public SysUserOnline selectOnlineByIpaddr(String ipaddr, LoginUser user); + + /** + * 通过用户名称查询信息 + * + * @param userName 用户名称 + * @param user 用户信息 + * @return 在线用户信息 + */ + public SysUserOnline selectOnlineByUserName(String userName, LoginUser user); + + /** + * 通过登录地址/用户名称查询信息 + * + * @param ipaddr 登录地址 + * @param userName 用户名称 + * @param user 用户信息 + * @return 在线用户信息 + */ + public SysUserOnline selectOnlineByInfo(String ipaddr, String userName, LoginUser user); + + /** + * 设置在线用户信息 + * + * @param user 用户信息 + * @return 在线用户 + */ + public SysUserOnline loginUserToUserOnline(LoginUser user); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserService.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserService.java new file mode 100644 index 0000000..59a41a1 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/ISysUserService.java @@ -0,0 +1,206 @@ +package com.ruoyi.system.service; + +import java.util.List; +import com.ruoyi.common.core.domain.entity.SysUser; + +/** + * 用户 业务层 + * + * @author ruoyi + */ +public interface ISysUserService +{ + /** + * 根据条件分页查询用户列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + public List selectUserList(SysUser user); + + /** + * 根据条件分页查询已分配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + public List selectAllocatedList(SysUser user); + + /** + * 根据条件分页查询未分配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + public List selectUnallocatedList(SysUser user); + + /** + * 通过用户名查询用户 + * + * @param userName 用户名 + * @return 用户对象信息 + */ + public SysUser selectUserByUserName(String userName); + + /** + * 通过用户ID查询用户 + * + * @param userId 用户ID + * @return 用户对象信息 + */ + public SysUser selectUserById(Long userId); + + /** + * 根据用户ID查询用户所属角色组 + * + * @param userName 用户名 + * @return 结果 + */ + public String selectUserRoleGroup(String userName); + + /** + * 根据用户ID查询用户所属岗位组 + * + * @param userName 用户名 + * @return 结果 + */ + public String selectUserPostGroup(String userName); + + /** + * 校验用户名称是否唯一 + * + * @param userName 用户名称 + * @return 结果 + */ + public String checkUserNameUnique(String userName); + + /** + * 校验手机号码是否唯一 + * + * @param user 用户信息 + * @return 结果 + */ + public String checkPhoneUnique(SysUser user); + + /** + * 校验email是否唯一 + * + * @param user 用户信息 + * @return 结果 + */ + public String checkEmailUnique(SysUser user); + + /** + * 校验用户是否允许操作 + * + * @param user 用户信息 + */ + public void checkUserAllowed(SysUser user); + + /** + * 校验用户是否有数据权限 + * + * @param userId 用户id + */ + public void checkUserDataScope(Long userId); + + /** + * 新增用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + public int insertUser(SysUser user); + + /** + * 注册用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + public boolean registerUser(SysUser user); + + /** + * 修改用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + public int updateUser(SysUser user); + + /** + * 用户授权角色 + * + * @param userId 用户ID + * @param roleIds 角色组 + */ + public void insertUserAuth(Long userId, Long[] roleIds); + + /** + * 修改用户状态 + * + * @param user 用户信息 + * @return 结果 + */ + public int updateUserStatus(SysUser user); + + /** + * 修改用户基本信息 + * + * @param user 用户信息 + * @return 结果 + */ + public int updateUserProfile(SysUser user); + + /** + * 修改用户头像 + * + * @param userName 用户名 + * @param avatar 头像地址 + * @return 结果 + */ + public boolean updateUserAvatar(String userName, String avatar); + + /** + * 重置用户密码 + * + * @param user 用户信息 + * @return 结果 + */ + public int resetPwd(SysUser user); + + /** + * 重置用户密码 + * + * @param userName 用户名 + * @param password 密码 + * @return 结果 + */ + public int resetUserPwd(String userName, String password); + + /** + * 通过用户ID删除用户 + * + * @param userId 用户ID + * @return 结果 + */ + public int deleteUserById(Long userId); + + /** + * 批量删除用户信息 + * + * @param userIds 需要删除的用户ID + * @return 结果 + */ + public int deleteUserByIds(Long[] userIds); + + /** + * 导入用户数据 + * + * @param userList 用户数据列表 + * @param isUpdateSupport 是否更新支持,如果已存在,则进行更新数据 + * @param operName 操作用户 + * @return 结果 + */ + public String importUser(List userList, Boolean isUpdateSupport, String operName); +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysConfigServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysConfigServiceImpl.java new file mode 100644 index 0000000..e723ce4 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysConfigServiceImpl.java @@ -0,0 +1,226 @@ +package com.ruoyi.system.service.impl; + +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.redis.RedisCache; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.domain.SysConfig; +import com.ruoyi.system.mapper.SysConfigMapper; +import com.ruoyi.system.service.ISysConfigService; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import javax.annotation.PostConstruct; +import java.util.Collection; +import java.util.List; + +/** + * 参数配置 服务层实现 + * + * @author ruoyi + */ +@Service +public class SysConfigServiceImpl implements ISysConfigService +{ + @Autowired + private SysConfigMapper configMapper; + + @Autowired + private RedisCache redisCache; + + /** + * 项目启动时,初始化参数到缓存 + */ + @PostConstruct + public void init() + { + loadingConfigCache(); + } + + /** + * 查询参数配置信息 + * + * @param configId 参数配置ID + * @return 参数配置信息 + */ + @Override + @DataSource(DataSourceType.MASTER) + public SysConfig selectConfigById(Long configId) + { + SysConfig config = new SysConfig(); + config.setConfigId(configId); + return configMapper.selectConfig(config); + } + + /** + * 根据键名查询参数配置信息 + * + * @param configKey 参数key + * @return 参数键值 + */ + @Override + public String selectConfigByKey(String configKey) + { + String configValue = Convert.toStr(redisCache.getCacheObject(getCacheKey(configKey))); + if (StringUtils.isNotEmpty(configValue)) + { + return configValue; + } + SysConfig config = new SysConfig(); + config.setConfigKey(configKey); + SysConfig retConfig = configMapper.selectConfig(config); + if (StringUtils.isNotNull(retConfig)) + { + redisCache.setCacheObject(getCacheKey(configKey), retConfig.getConfigValue()); + return retConfig.getConfigValue(); + } + return StringUtils.EMPTY; + } + + /** + * 获取验证码开关 + * + * @return true开启,false关闭 + */ + @Override + public boolean selectCaptchaOnOff() + { + String captchaOnOff = selectConfigByKey("sys.account.captchaOnOff"); + if (StringUtils.isEmpty(captchaOnOff)) + { + return true; + } + return Convert.toBool(captchaOnOff); + } + + /** + * 查询参数配置列表 + * + * @param config 参数配置信息 + * @return 参数配置集合 + */ + @Override + public List selectConfigList(SysConfig config) + { + return configMapper.selectConfigList(config); + } + + /** + * 新增参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + @Override + public int insertConfig(SysConfig config) + { + int row = configMapper.insertConfig(config); + if (row > 0) + { + redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue()); + } + return row; + } + + /** + * 修改参数配置 + * + * @param config 参数配置信息 + * @return 结果 + */ + @Override + public int updateConfig(SysConfig config) + { + int row = configMapper.updateConfig(config); + if (row > 0) + { + redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue()); + } + return row; + } + + /** + * 批量删除参数信息 + * + * @param configIds 需要删除的参数ID + */ + @Override + public void deleteConfigByIds(Long[] configIds) + { + for (Long configId : configIds) + { + SysConfig config = selectConfigById(configId); + if (StringUtils.equals(UserConstants.YES, config.getConfigType())) + { + throw new ServiceException(String.format("内置参数【%1$s】不能删除 ", config.getConfigKey())); + } + configMapper.deleteConfigById(configId); + redisCache.deleteObject(getCacheKey(config.getConfigKey())); + } + } + + /** + * 加载参数缓存数据 + */ + @Override + public void loadingConfigCache() + { + List configsList = configMapper.selectConfigList(new SysConfig()); + for (SysConfig config : configsList) + { + redisCache.setCacheObject(getCacheKey(config.getConfigKey()), config.getConfigValue()); + } + } + + /** + * 清空参数缓存数据 + */ + @Override + public void clearConfigCache() + { + Collection keys = redisCache.keys(Constants.SYS_CONFIG_KEY + "*"); + redisCache.deleteObject(keys); + } + + /** + * 重置参数缓存数据 + */ + @Override + public void resetConfigCache() + { + clearConfigCache(); + loadingConfigCache(); + } + + /** + * 校验参数键名是否唯一 + * + * @param config 参数配置信息 + * @return 结果 + */ + @Override + public String checkConfigKeyUnique(SysConfig config) + { + Long configId = StringUtils.isNull(config.getConfigId()) ? -1L : config.getConfigId(); + SysConfig info = configMapper.checkConfigKeyUnique(config.getConfigKey()); + if (StringUtils.isNotNull(info) && info.getConfigId().longValue() != configId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 设置cache key + * + * @param configKey 参数键 + * @return 缓存键key + */ + private String getCacheKey(String configKey) + { + return Constants.SYS_CONFIG_KEY + configKey; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDeptServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDeptServiceImpl.java new file mode 100644 index 0000000..96d215e --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDeptServiceImpl.java @@ -0,0 +1,329 @@ +package com.ruoyi.system.service.impl; + +import java.util.ArrayList; +import java.util.Iterator; +import java.util.List; +import java.util.stream.Collectors; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.common.annotation.DataScope; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.TreeSelect; +import com.ruoyi.common.core.domain.entity.SysDept; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.system.mapper.SysDeptMapper; +import com.ruoyi.system.mapper.SysRoleMapper; +import com.ruoyi.system.service.ISysDeptService; + +/** + * 部门管理 服务实现 + * + * @author ruoyi + */ +@Service +public class SysDeptServiceImpl implements ISysDeptService +{ + @Autowired + private SysDeptMapper deptMapper; + + @Autowired + private SysRoleMapper roleMapper; + + /** + * 查询部门管理数据 + * + * @param dept 部门信息 + * @return 部门信息集合 + */ + @Override + @DataScope(deptAlias = "d") + public List selectDeptList(SysDept dept) + { + return deptMapper.selectDeptList(dept); + } + + /** + * 构建前端所需要树结构 + * + * @param depts 部门列表 + * @return 树结构列表 + */ + @Override + public List buildDeptTree(List depts) + { + List returnList = new ArrayList(); + List tempList = new ArrayList(); + for (SysDept dept : depts) + { + tempList.add(dept.getDeptId()); + } + for (SysDept dept : depts) + { + // 如果是顶级节点, 遍历该父节点的所有子节点 + if (!tempList.contains(dept.getParentId())) + { + recursionFn(depts, dept); + returnList.add(dept); + } + } + if (returnList.isEmpty()) + { + returnList = depts; + } + return returnList; + } + + /** + * 构建前端所需要下拉树结构 + * + * @param depts 部门列表 + * @return 下拉树结构列表 + */ + @Override + public List buildDeptTreeSelect(List depts) + { + List deptTrees = buildDeptTree(depts); + return deptTrees.stream().map(TreeSelect::new).collect(Collectors.toList()); + } + + /** + * 根据角色ID查询部门树信息 + * + * @param roleId 角色ID + * @return 选中部门列表 + */ + @Override + public List selectDeptListByRoleId(Long roleId) + { + SysRole role = roleMapper.selectRoleById(roleId); + return deptMapper.selectDeptListByRoleId(roleId, role.isDeptCheckStrictly()); + } + + /** + * 根据部门ID查询信息 + * + * @param deptId 部门ID + * @return 部门信息 + */ + @Override + public SysDept selectDeptById(Long deptId) + { + return deptMapper.selectDeptById(deptId); + } + + /** + * 根据ID查询所有子部门(正常状态) + * + * @param deptId 部门ID + * @return 子部门数 + */ + @Override + public int selectNormalChildrenDeptById(Long deptId) + { + return deptMapper.selectNormalChildrenDeptById(deptId); + } + + /** + * 是否存在子节点 + * + * @param deptId 部门ID + * @return 结果 + */ + @Override + public boolean hasChildByDeptId(Long deptId) + { + int result = deptMapper.hasChildByDeptId(deptId); + return result > 0; + } + + /** + * 查询部门是否存在用户 + * + * @param deptId 部门ID + * @return 结果 true 存在 false 不存在 + */ + @Override + public boolean checkDeptExistUser(Long deptId) + { + int result = deptMapper.checkDeptExistUser(deptId); + return result > 0; + } + + /** + * 校验部门名称是否唯一 + * + * @param dept 部门信息 + * @return 结果 + */ + @Override + public String checkDeptNameUnique(SysDept dept) + { + Long deptId = StringUtils.isNull(dept.getDeptId()) ? -1L : dept.getDeptId(); + SysDept info = deptMapper.checkDeptNameUnique(dept.getDeptName(), dept.getParentId()); + if (StringUtils.isNotNull(info) && info.getDeptId().longValue() != deptId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验部门是否有数据权限 + * + * @param deptId 部门id + */ + @Override + public void checkDeptDataScope(Long deptId) + { + if (!SysUser.isAdmin(SecurityUtils.getUserId())) + { + SysDept dept = new SysDept(); + dept.setDeptId(deptId); + List depts = SpringUtils.getAopProxy(this).selectDeptList(dept); + if (StringUtils.isEmpty(depts)) + { + throw new ServiceException("没有权限访问部门数据!"); + } + } + } + + /** + * 新增保存部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + @Override + public int insertDept(SysDept dept) + { + SysDept info = deptMapper.selectDeptById(dept.getParentId()); + // 如果父节点不为正常状态,则不允许新增子节点 + if (!UserConstants.DEPT_NORMAL.equals(info.getStatus())) + { + throw new ServiceException("部门停用,不允许新增"); + } + dept.setAncestors(info.getAncestors() + "," + dept.getParentId()); + return deptMapper.insertDept(dept); + } + + /** + * 修改保存部门信息 + * + * @param dept 部门信息 + * @return 结果 + */ + @Override + public int updateDept(SysDept dept) + { + SysDept newParentDept = deptMapper.selectDeptById(dept.getParentId()); + SysDept oldDept = deptMapper.selectDeptById(dept.getDeptId()); + if (StringUtils.isNotNull(newParentDept) && StringUtils.isNotNull(oldDept)) + { + String newAncestors = newParentDept.getAncestors() + "," + newParentDept.getDeptId(); + String oldAncestors = oldDept.getAncestors(); + dept.setAncestors(newAncestors); + updateDeptChildren(dept.getDeptId(), newAncestors, oldAncestors); + } + int result = deptMapper.updateDept(dept); + if (UserConstants.DEPT_NORMAL.equals(dept.getStatus()) && StringUtils.isNotEmpty(dept.getAncestors()) + && !StringUtils.equals("0", dept.getAncestors())) + { + // 如果该部门是启用状态,则启用该部门的所有上级部门 + updateParentDeptStatusNormal(dept); + } + return result; + } + + /** + * 修改该部门的父级部门状态 + * + * @param dept 当前部门 + */ + private void updateParentDeptStatusNormal(SysDept dept) + { + String ancestors = dept.getAncestors(); + Long[] deptIds = Convert.toLongArray(ancestors); + deptMapper.updateDeptStatusNormal(deptIds); + } + + /** + * 修改子元素关系 + * + * @param deptId 被修改的部门ID + * @param newAncestors 新的父ID集合 + * @param oldAncestors 旧的父ID集合 + */ + public void updateDeptChildren(Long deptId, String newAncestors, String oldAncestors) + { + List children = deptMapper.selectChildrenDeptById(deptId); + for (SysDept child : children) + { + child.setAncestors(child.getAncestors().replaceFirst(oldAncestors, newAncestors)); + } + if (children.size() > 0) + { + deptMapper.updateDeptChildren(children); + } + } + + /** + * 删除部门管理信息 + * + * @param deptId 部门ID + * @return 结果 + */ + @Override + public int deleteDeptById(Long deptId) + { + return deptMapper.deleteDeptById(deptId); + } + + /** + * 递归列表 + */ + private void recursionFn(List list, SysDept t) + { + // 得到子节点列表 + List childList = getChildList(list, t); + t.setChildren(childList); + for (SysDept tChild : childList) + { + if (hasChild(list, tChild)) + { + recursionFn(list, tChild); + } + } + } + + /** + * 得到子节点列表 + */ + private List getChildList(List list, SysDept t) + { + List tlist = new ArrayList(); + Iterator it = list.iterator(); + while (it.hasNext()) + { + SysDept n = (SysDept) it.next(); + if (StringUtils.isNotNull(n.getParentId()) && n.getParentId().longValue() == t.getDeptId().longValue()) + { + tlist.add(n); + } + } + return tlist; + } + + /** + * 判断是否有子节点 + */ + private boolean hasChild(List list, SysDept t) + { + return getChildList(list, t).size() > 0; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictDataServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictDataServiceImpl.java new file mode 100644 index 0000000..fced569 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictDataServiceImpl.java @@ -0,0 +1,111 @@ +package com.ruoyi.system.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.common.core.domain.entity.SysDictData; +import com.ruoyi.common.utils.DictUtils; +import com.ruoyi.system.mapper.SysDictDataMapper; +import com.ruoyi.system.service.ISysDictDataService; + +/** + * 字典 业务层处理 + * + * @author ruoyi + */ +@Service +public class SysDictDataServiceImpl implements ISysDictDataService +{ + @Autowired + private SysDictDataMapper dictDataMapper; + + /** + * 根据条件分页查询字典数据 + * + * @param dictData 字典数据信息 + * @return 字典数据集合信息 + */ + @Override + public List selectDictDataList(SysDictData dictData) + { + return dictDataMapper.selectDictDataList(dictData); + } + + /** + * 根据字典类型和字典键值查询字典数据信息 + * + * @param dictType 字典类型 + * @param dictValue 字典键值 + * @return 字典标签 + */ + @Override + public String selectDictLabel(String dictType, String dictValue) + { + return dictDataMapper.selectDictLabel(dictType, dictValue); + } + + /** + * 根据字典数据ID查询信息 + * + * @param dictCode 字典数据ID + * @return 字典数据 + */ + @Override + public SysDictData selectDictDataById(Long dictCode) + { + return dictDataMapper.selectDictDataById(dictCode); + } + + /** + * 批量删除字典数据信息 + * + * @param dictCodes 需要删除的字典数据ID + */ + @Override + public void deleteDictDataByIds(Long[] dictCodes) + { + for (Long dictCode : dictCodes) + { + SysDictData data = selectDictDataById(dictCode); + dictDataMapper.deleteDictDataById(dictCode); + List dictDatas = dictDataMapper.selectDictDataByType(data.getDictType()); + DictUtils.setDictCache(data.getDictType(), dictDatas); + } + } + + /** + * 新增保存字典数据信息 + * + * @param data 字典数据信息 + * @return 结果 + */ + @Override + public int insertDictData(SysDictData data) + { + int row = dictDataMapper.insertDictData(data); + if (row > 0) + { + List dictDatas = dictDataMapper.selectDictDataByType(data.getDictType()); + DictUtils.setDictCache(data.getDictType(), dictDatas); + } + return row; + } + + /** + * 修改保存字典数据信息 + * + * @param data 字典数据信息 + * @return 结果 + */ + @Override + public int updateDictData(SysDictData data) + { + int row = dictDataMapper.updateDictData(data); + if (row > 0) + { + List dictDatas = dictDataMapper.selectDictDataByType(data.getDictType()); + DictUtils.setDictCache(data.getDictType(), dictDatas); + } + return row; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictTypeServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictTypeServiceImpl.java new file mode 100644 index 0000000..8da7983 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysDictTypeServiceImpl.java @@ -0,0 +1,223 @@ +package com.ruoyi.system.service.impl; + +import java.util.Comparator; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; +import javax.annotation.PostConstruct; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.entity.SysDictData; +import com.ruoyi.common.core.domain.entity.SysDictType; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.DictUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.mapper.SysDictDataMapper; +import com.ruoyi.system.mapper.SysDictTypeMapper; +import com.ruoyi.system.service.ISysDictTypeService; + +/** + * 字典 业务层处理 + * + * @author ruoyi + */ +@Service +public class SysDictTypeServiceImpl implements ISysDictTypeService +{ + @Autowired + private SysDictTypeMapper dictTypeMapper; + + @Autowired + private SysDictDataMapper dictDataMapper; + + /** + * 项目启动时,初始化字典到缓存 + */ + @PostConstruct + public void init() + { + loadingDictCache(); + } + + /** + * 根据条件分页查询字典类型 + * + * @param dictType 字典类型信息 + * @return 字典类型集合信息 + */ + @Override + public List selectDictTypeList(SysDictType dictType) + { + return dictTypeMapper.selectDictTypeList(dictType); + } + + /** + * 根据所有字典类型 + * + * @return 字典类型集合信息 + */ + @Override + public List selectDictTypeAll() + { + return dictTypeMapper.selectDictTypeAll(); + } + + /** + * 根据字典类型查询字典数据 + * + * @param dictType 字典类型 + * @return 字典数据集合信息 + */ + @Override + public List selectDictDataByType(String dictType) + { + List dictDatas = DictUtils.getDictCache(dictType); + if (StringUtils.isNotEmpty(dictDatas)) + { + return dictDatas; + } + dictDatas = dictDataMapper.selectDictDataByType(dictType); + if (StringUtils.isNotEmpty(dictDatas)) + { + DictUtils.setDictCache(dictType, dictDatas); + return dictDatas; + } + return null; + } + + /** + * 根据字典类型ID查询信息 + * + * @param dictId 字典类型ID + * @return 字典类型 + */ + @Override + public SysDictType selectDictTypeById(Long dictId) + { + return dictTypeMapper.selectDictTypeById(dictId); + } + + /** + * 根据字典类型查询信息 + * + * @param dictType 字典类型 + * @return 字典类型 + */ + @Override + public SysDictType selectDictTypeByType(String dictType) + { + return dictTypeMapper.selectDictTypeByType(dictType); + } + + /** + * 批量删除字典类型信息 + * + * @param dictIds 需要删除的字典ID + */ + @Override + public void deleteDictTypeByIds(Long[] dictIds) + { + for (Long dictId : dictIds) + { + SysDictType dictType = selectDictTypeById(dictId); + if (dictDataMapper.countDictDataByType(dictType.getDictType()) > 0) + { + throw new ServiceException(String.format("%1$s已分配,不能删除", dictType.getDictName())); + } + dictTypeMapper.deleteDictTypeById(dictId); + DictUtils.removeDictCache(dictType.getDictType()); + } + } + + /** + * 加载字典缓存数据 + */ + @Override + public void loadingDictCache() + { + SysDictData dictData = new SysDictData(); + dictData.setStatus("0"); + Map> dictDataMap = dictDataMapper.selectDictDataList(dictData).stream().collect(Collectors.groupingBy(SysDictData::getDictType)); + for (Map.Entry> entry : dictDataMap.entrySet()) + { + DictUtils.setDictCache(entry.getKey(), entry.getValue().stream().sorted(Comparator.comparing(SysDictData::getDictSort)).collect(Collectors.toList())); + } + } + + /** + * 清空字典缓存数据 + */ + @Override + public void clearDictCache() + { + DictUtils.clearDictCache(); + } + + /** + * 重置字典缓存数据 + */ + @Override + public void resetDictCache() + { + clearDictCache(); + loadingDictCache(); + } + + /** + * 新增保存字典类型信息 + * + * @param dict 字典类型信息 + * @return 结果 + */ + @Override + public int insertDictType(SysDictType dict) + { + int row = dictTypeMapper.insertDictType(dict); + if (row > 0) + { + DictUtils.setDictCache(dict.getDictType(), null); + } + return row; + } + + /** + * 修改保存字典类型信息 + * + * @param dict 字典类型信息 + * @return 结果 + */ + @Override + @Transactional + public int updateDictType(SysDictType dict) + { + SysDictType oldDict = dictTypeMapper.selectDictTypeById(dict.getDictId()); + dictDataMapper.updateDictDataType(oldDict.getDictType(), dict.getDictType()); + int row = dictTypeMapper.updateDictType(dict); + if (row > 0) + { + List dictDatas = dictDataMapper.selectDictDataByType(dict.getDictType()); + DictUtils.setDictCache(dict.getDictType(), dictDatas); + } + return row; + } + + /** + * 校验字典类型称是否唯一 + * + * @param dict 字典类型 + * @return 结果 + */ + @Override + public String checkDictTypeUnique(SysDictType dict) + { + Long dictId = StringUtils.isNull(dict.getDictId()) ? -1L : dict.getDictId(); + SysDictType dictType = dictTypeMapper.checkDictTypeUnique(dict.getDictType()); + if (StringUtils.isNotNull(dictType) && dictType.getDictId().longValue() != dictId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysLogininforServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysLogininforServiceImpl.java new file mode 100644 index 0000000..216aecb --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysLogininforServiceImpl.java @@ -0,0 +1,65 @@ +package com.ruoyi.system.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.system.domain.SysLogininfor; +import com.ruoyi.system.mapper.SysLogininforMapper; +import com.ruoyi.system.service.ISysLogininforService; + +/** + * 系统访问日志情况信息 服务层处理 + * + * @author ruoyi + */ +@Service +public class SysLogininforServiceImpl implements ISysLogininforService +{ + + @Autowired + private SysLogininforMapper logininforMapper; + + /** + * 新增系统登录日志 + * + * @param logininfor 访问日志对象 + */ + @Override + public void insertLogininfor(SysLogininfor logininfor) + { + logininforMapper.insertLogininfor(logininfor); + } + + /** + * 查询系统登录日志集合 + * + * @param logininfor 访问日志对象 + * @return 登录记录集合 + */ + @Override + public List selectLogininforList(SysLogininfor logininfor) + { + return logininforMapper.selectLogininforList(logininfor); + } + + /** + * 批量删除系统登录日志 + * + * @param infoIds 需要删除的登录日志ID + * @return 结果 + */ + @Override + public int deleteLogininforByIds(Long[] infoIds) + { + return logininforMapper.deleteLogininforByIds(infoIds); + } + + /** + * 清空系统登录日志 + */ + @Override + public void cleanLogininfor() + { + logininforMapper.cleanLogininfor(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java new file mode 100644 index 0000000..37dcc19 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java @@ -0,0 +1,514 @@ +package com.ruoyi.system.service.impl; + +import java.util.ArrayList; +import java.util.Arrays; +import java.util.HashSet; +import java.util.Iterator; +import java.util.LinkedList; +import java.util.List; +import java.util.Set; +import java.util.stream.Collectors; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.TreeSelect; +import com.ruoyi.common.core.domain.entity.SysMenu; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.domain.vo.MetaVo; +import com.ruoyi.system.domain.vo.RouterVo; +import com.ruoyi.system.mapper.SysMenuMapper; +import com.ruoyi.system.mapper.SysRoleMapper; +import com.ruoyi.system.mapper.SysRoleMenuMapper; +import com.ruoyi.system.service.ISysMenuService; + +/** + * 菜单 业务层处理 + * + * @author ruoyi + */ +@Service +public class SysMenuServiceImpl implements ISysMenuService +{ + public static final String PREMISSION_STRING = "perms[\"{0}\"]"; + + @Autowired + private SysMenuMapper menuMapper; + + @Autowired + private SysRoleMapper roleMapper; + + @Autowired + private SysRoleMenuMapper roleMenuMapper; + + /** + * 根据用户查询系统菜单列表 + * + * @param userId 用户ID + * @return 菜单列表 + */ + @Override + public List selectMenuList(Long userId) + { + return selectMenuList(new SysMenu(), userId); + } + + /** + * 查询系统菜单列表 + * + * @param menu 菜单信息 + * @return 菜单列表 + */ + @Override + public List selectMenuList(SysMenu menu, Long userId) + { + List menuList = null; + // 管理员显示所有菜单信息 + if (SysUser.isAdmin(userId)) + { + menuList = menuMapper.selectMenuList(menu); + } + else + { + menu.getParams().put("userId", userId); + menuList = menuMapper.selectMenuListByUserId(menu); + } + return menuList; + } + + /** + * 根据用户ID查询权限 + * + * @param userId 用户ID + * @return 权限列表 + */ + @Override + public Set selectMenuPermsByUserId(Long userId) + { + List perms = menuMapper.selectMenuPermsByUserId(userId); + Set permsSet = new HashSet<>(); + for (String perm : perms) + { + if (StringUtils.isNotEmpty(perm)) + { + permsSet.addAll(Arrays.asList(perm.trim().split(","))); + } + } + return permsSet; + } + + /** + * 根据用户ID查询菜单 + * + * @param userId 用户名称 + * @return 菜单列表 + */ + @Override + public List selectMenuTreeByUserId(Long userId) + { + List menus = null; + if (SecurityUtils.isAdmin(userId)) + { + menus = menuMapper.selectMenuTreeAll(); + } + else + { + menus = menuMapper.selectMenuTreeByUserId(userId); + } + return getChildPerms(menus, 0); + } + + /** + * 根据角色ID查询菜单树信息 + * + * @param roleId 角色ID + * @return 选中菜单列表 + */ + @Override + public List selectMenuListByRoleId(Long roleId) + { + SysRole role = roleMapper.selectRoleById(roleId); + return menuMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly()); + } + + /** + * 构建前端路由所需要的菜单 + * + * @param menus 菜单列表 + * @return 路由列表 + */ + @Override + public List buildMenus(List menus) + { + List routers = new LinkedList(); + for (SysMenu menu : menus) + { + RouterVo router = new RouterVo(); + router.setHidden("1".equals(menu.getVisible())); + router.setName(getRouteName(menu)); + router.setPath(getRouterPath(menu)); + router.setComponent(getComponent(menu)); + router.setQuery(menu.getQuery()); + router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath())); + List cMenus = menu.getChildren(); + if (!cMenus.isEmpty() && cMenus.size() > 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType())) + { + router.setAlwaysShow(true); + router.setRedirect("noRedirect"); + router.setChildren(buildMenus(cMenus)); + } + else if (isMenuFrame(menu)) + { + router.setMeta(null); + List childrenList = new ArrayList(); + RouterVo children = new RouterVo(); + children.setPath(menu.getPath()); + children.setComponent(menu.getComponent()); + children.setName(StringUtils.capitalize(menu.getPath())); + children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath())); + children.setQuery(menu.getQuery()); + childrenList.add(children); + router.setChildren(childrenList); + } + else if (menu.getParentId().intValue() == 0 && isInnerLink(menu)) + { + router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon())); + router.setPath("/"); + List childrenList = new ArrayList(); + RouterVo children = new RouterVo(); + String routerPath = innerLinkReplaceEach(menu.getPath()); + children.setPath(routerPath); + children.setComponent(UserConstants.INNER_LINK); + children.setName(StringUtils.capitalize(routerPath)); + children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), menu.getPath())); + childrenList.add(children); + router.setChildren(childrenList); + } + routers.add(router); + } + return routers; + } + + /** + * 构建前端所需要树结构 + * + * @param menus 菜单列表 + * @return 树结构列表 + */ + @Override + public List buildMenuTree(List menus) + { + List returnList = new ArrayList(); + List tempList = new ArrayList(); + for (SysMenu dept : menus) + { + tempList.add(dept.getMenuId()); + } + for (Iterator iterator = menus.iterator(); iterator.hasNext();) + { + SysMenu menu = (SysMenu) iterator.next(); + // 如果是顶级节点, 遍历该父节点的所有子节点 + if (!tempList.contains(menu.getParentId())) + { + recursionFn(menus, menu); + returnList.add(menu); + } + } + if (returnList.isEmpty()) + { + returnList = menus; + } + return returnList; + } + + /** + * 构建前端所需要下拉树结构 + * + * @param menus 菜单列表 + * @return 下拉树结构列表 + */ + @Override + public List buildMenuTreeSelect(List menus) + { + List menuTrees = buildMenuTree(menus); + return menuTrees.stream().map(TreeSelect::new).collect(Collectors.toList()); + } + + /** + * 根据菜单ID查询信息 + * + * @param menuId 菜单ID + * @return 菜单信息 + */ + @Override + public SysMenu selectMenuById(Long menuId) + { + return menuMapper.selectMenuById(menuId); + } + + /** + * 是否存在菜单子节点 + * + * @param menuId 菜单ID + * @return 结果 + */ + @Override + public boolean hasChildByMenuId(Long menuId) + { + int result = menuMapper.hasChildByMenuId(menuId); + return result > 0 ? true : false; + } + + /** + * 查询菜单使用数量 + * + * @param menuId 菜单ID + * @return 结果 + */ + @Override + public boolean checkMenuExistRole(Long menuId) + { + int result = roleMenuMapper.checkMenuExistRole(menuId); + return result > 0 ? true : false; + } + + /** + * 新增保存菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + @Override + public int insertMenu(SysMenu menu) + { + return menuMapper.insertMenu(menu); + } + + /** + * 修改保存菜单信息 + * + * @param menu 菜单信息 + * @return 结果 + */ + @Override + public int updateMenu(SysMenu menu) + { + return menuMapper.updateMenu(menu); + } + + /** + * 删除菜单管理信息 + * + * @param menuId 菜单ID + * @return 结果 + */ + @Override + public int deleteMenuById(Long menuId) + { + return menuMapper.deleteMenuById(menuId); + } + + /** + * 校验菜单名称是否唯一 + * + * @param menu 菜单信息 + * @return 结果 + */ + @Override + public String checkMenuNameUnique(SysMenu menu) + { + Long menuId = StringUtils.isNull(menu.getMenuId()) ? -1L : menu.getMenuId(); + SysMenu info = menuMapper.checkMenuNameUnique(menu.getMenuName(), menu.getParentId()); + if (StringUtils.isNotNull(info) && info.getMenuId().longValue() != menuId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 获取路由名称 + * + * @param menu 菜单信息 + * @return 路由名称 + */ + public String getRouteName(SysMenu menu) + { + String routerName = StringUtils.capitalize(menu.getPath()); + // 非外链并且是一级目录(类型为目录) + if (isMenuFrame(menu)) + { + routerName = StringUtils.EMPTY; + } + return routerName; + } + + /** + * 获取路由地址 + * + * @param menu 菜单信息 + * @return 路由地址 + */ + public String getRouterPath(SysMenu menu) + { + String routerPath = menu.getPath(); + // 内链打开外网方式 + if (menu.getParentId().intValue() != 0 && isInnerLink(menu)) + { + routerPath = innerLinkReplaceEach(routerPath); + } + // 非外链并且是一级目录(类型为目录) + if (0 == menu.getParentId().intValue() && UserConstants.TYPE_DIR.equals(menu.getMenuType()) + && UserConstants.NO_FRAME.equals(menu.getIsFrame())) + { + routerPath = "/" + menu.getPath(); + } + // 非外链并且是一级目录(类型为菜单) + else if (isMenuFrame(menu)) + { + routerPath = "/"; + } + return routerPath; + } + + /** + * 获取组件信息 + * + * @param menu 菜单信息 + * @return 组件信息 + */ + public String getComponent(SysMenu menu) + { + String component = UserConstants.LAYOUT; + if (StringUtils.isNotEmpty(menu.getComponent()) && !isMenuFrame(menu)) + { + component = menu.getComponent(); + } + else if (StringUtils.isEmpty(menu.getComponent()) && menu.getParentId().intValue() != 0 && isInnerLink(menu)) + { + component = UserConstants.INNER_LINK; + } + else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu)) + { + component = UserConstants.PARENT_VIEW; + } + return component; + } + + /** + * 是否为菜单内部跳转 + * + * @param menu 菜单信息 + * @return 结果 + */ + public boolean isMenuFrame(SysMenu menu) + { + return menu.getParentId().intValue() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType()) + && menu.getIsFrame().equals(UserConstants.NO_FRAME); + } + + /** + * 是否为内链组件 + * + * @param menu 菜单信息 + * @return 结果 + */ + public boolean isInnerLink(SysMenu menu) + { + return menu.getIsFrame().equals(UserConstants.NO_FRAME) && StringUtils.ishttp(menu.getPath()); + } + + /** + * 是否为parent_view组件 + * + * @param menu 菜单信息 + * @return 结果 + */ + public boolean isParentView(SysMenu menu) + { + return menu.getParentId().intValue() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType()); + } + + /** + * 根据父节点的ID获取所有子节点 + * + * @param list 分类表 + * @param parentId 传入的父节点ID + * @return String + */ + public List getChildPerms(List list, int parentId) + { + List returnList = new ArrayList(); + for (Iterator iterator = list.iterator(); iterator.hasNext();) + { + SysMenu t = (SysMenu) iterator.next(); + // 一、根据传入的某个父节点ID,遍历该父节点的所有子节点 + if (t.getParentId() == parentId) + { + recursionFn(list, t); + returnList.add(t); + } + } + return returnList; + } + + /** + * 递归列表 + * + * @param list + * @param t + */ + private void recursionFn(List list, SysMenu t) + { + // 得到子节点列表 + List childList = getChildList(list, t); + t.setChildren(childList); + for (SysMenu tChild : childList) + { + if (hasChild(list, tChild)) + { + recursionFn(list, tChild); + } + } + } + + /** + * 得到子节点列表 + */ + private List getChildList(List list, SysMenu t) + { + List tlist = new ArrayList(); + Iterator it = list.iterator(); + while (it.hasNext()) + { + SysMenu n = (SysMenu) it.next(); + if (n.getParentId().longValue() == t.getMenuId().longValue()) + { + tlist.add(n); + } + } + return tlist; + } + + /** + * 判断是否有子节点 + */ + private boolean hasChild(List list, SysMenu t) + { + return getChildList(list, t).size() > 0; + } + + /** + * 内链域名特殊字符替换 + * + * @return + */ + public String innerLinkReplaceEach(String path) + { + return StringUtils.replaceEach(path, new String[] { Constants.HTTP, Constants.HTTPS }, + new String[] { "", "" }); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysNoticeServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysNoticeServiceImpl.java new file mode 100644 index 0000000..765438b --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysNoticeServiceImpl.java @@ -0,0 +1,92 @@ +package com.ruoyi.system.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.system.domain.SysNotice; +import com.ruoyi.system.mapper.SysNoticeMapper; +import com.ruoyi.system.service.ISysNoticeService; + +/** + * 公告 服务层实现 + * + * @author ruoyi + */ +@Service +public class SysNoticeServiceImpl implements ISysNoticeService +{ + @Autowired + private SysNoticeMapper noticeMapper; + + /** + * 查询公告信息 + * + * @param noticeId 公告ID + * @return 公告信息 + */ + @Override + public SysNotice selectNoticeById(Long noticeId) + { + return noticeMapper.selectNoticeById(noticeId); + } + + /** + * 查询公告列表 + * + * @param notice 公告信息 + * @return 公告集合 + */ + @Override + public List selectNoticeList(SysNotice notice) + { + return noticeMapper.selectNoticeList(notice); + } + + /** + * 新增公告 + * + * @param notice 公告信息 + * @return 结果 + */ + @Override + public int insertNotice(SysNotice notice) + { + return noticeMapper.insertNotice(notice); + } + + /** + * 修改公告 + * + * @param notice 公告信息 + * @return 结果 + */ + @Override + public int updateNotice(SysNotice notice) + { + return noticeMapper.updateNotice(notice); + } + + /** + * 删除公告对象 + * + * @param noticeId 公告ID + * @return 结果 + */ + @Override + public int deleteNoticeById(Long noticeId) + { + return noticeMapper.deleteNoticeById(noticeId); + } + + /** + * 批量删除公告信息 + * + * @param noticeIds 需要删除的公告ID + * @return 结果 + */ + @Override + public int deleteNoticeByIds(Long[] noticeIds) + { + return noticeMapper.deleteNoticeByIds(noticeIds); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysOperLogServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysOperLogServiceImpl.java new file mode 100644 index 0000000..5489815 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysOperLogServiceImpl.java @@ -0,0 +1,76 @@ +package com.ruoyi.system.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.system.domain.SysOperLog; +import com.ruoyi.system.mapper.SysOperLogMapper; +import com.ruoyi.system.service.ISysOperLogService; + +/** + * 操作日志 服务层处理 + * + * @author ruoyi + */ +@Service +public class SysOperLogServiceImpl implements ISysOperLogService +{ + @Autowired + private SysOperLogMapper operLogMapper; + + /** + * 新增操作日志 + * + * @param operLog 操作日志对象 + */ + @Override + public void insertOperlog(SysOperLog operLog) + { + operLogMapper.insertOperlog(operLog); + } + + /** + * 查询系统操作日志集合 + * + * @param operLog 操作日志对象 + * @return 操作日志集合 + */ + @Override + public List selectOperLogList(SysOperLog operLog) + { + return operLogMapper.selectOperLogList(operLog); + } + + /** + * 批量删除系统操作日志 + * + * @param operIds 需要删除的操作日志ID + * @return 结果 + */ + @Override + public int deleteOperLogByIds(Long[] operIds) + { + return operLogMapper.deleteOperLogByIds(operIds); + } + + /** + * 查询操作日志详细 + * + * @param operId 操作ID + * @return 操作日志对象 + */ + @Override + public SysOperLog selectOperLogById(Long operId) + { + return operLogMapper.selectOperLogById(operId); + } + + /** + * 清空操作日志 + */ + @Override + public void cleanOperLog() + { + operLogMapper.cleanOperLog(); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysPostServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysPostServiceImpl.java new file mode 100644 index 0000000..78f1cb7 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysPostServiceImpl.java @@ -0,0 +1,178 @@ +package com.ruoyi.system.service.impl; + +import java.util.List; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.domain.SysPost; +import com.ruoyi.system.mapper.SysPostMapper; +import com.ruoyi.system.mapper.SysUserPostMapper; +import com.ruoyi.system.service.ISysPostService; + +/** + * 岗位信息 服务层处理 + * + * @author ruoyi + */ +@Service +public class SysPostServiceImpl implements ISysPostService +{ + @Autowired + private SysPostMapper postMapper; + + @Autowired + private SysUserPostMapper userPostMapper; + + /** + * 查询岗位信息集合 + * + * @param post 岗位信息 + * @return 岗位信息集合 + */ + @Override + public List selectPostList(SysPost post) + { + return postMapper.selectPostList(post); + } + + /** + * 查询所有岗位 + * + * @return 岗位列表 + */ + @Override + public List selectPostAll() + { + return postMapper.selectPostAll(); + } + + /** + * 通过岗位ID查询岗位信息 + * + * @param postId 岗位ID + * @return 角色对象信息 + */ + @Override + public SysPost selectPostById(Long postId) + { + return postMapper.selectPostById(postId); + } + + /** + * 根据用户ID获取岗位选择框列表 + * + * @param userId 用户ID + * @return 选中岗位ID列表 + */ + @Override + public List selectPostListByUserId(Long userId) + { + return postMapper.selectPostListByUserId(userId); + } + + /** + * 校验岗位名称是否唯一 + * + * @param post 岗位信息 + * @return 结果 + */ + @Override + public String checkPostNameUnique(SysPost post) + { + Long postId = StringUtils.isNull(post.getPostId()) ? -1L : post.getPostId(); + SysPost info = postMapper.checkPostNameUnique(post.getPostName()); + if (StringUtils.isNotNull(info) && info.getPostId().longValue() != postId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验岗位编码是否唯一 + * + * @param post 岗位信息 + * @return 结果 + */ + @Override + public String checkPostCodeUnique(SysPost post) + { + Long postId = StringUtils.isNull(post.getPostId()) ? -1L : post.getPostId(); + SysPost info = postMapper.checkPostCodeUnique(post.getPostCode()); + if (StringUtils.isNotNull(info) && info.getPostId().longValue() != postId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 通过岗位ID查询岗位使用数量 + * + * @param postId 岗位ID + * @return 结果 + */ + @Override + public int countUserPostById(Long postId) + { + return userPostMapper.countUserPostById(postId); + } + + /** + * 删除岗位信息 + * + * @param postId 岗位ID + * @return 结果 + */ + @Override + public int deletePostById(Long postId) + { + return postMapper.deletePostById(postId); + } + + /** + * 批量删除岗位信息 + * + * @param postIds 需要删除的岗位ID + * @return 结果 + */ + @Override + public int deletePostByIds(Long[] postIds) + { + for (Long postId : postIds) + { + SysPost post = selectPostById(postId); + if (countUserPostById(postId) > 0) + { + throw new ServiceException(String.format("%1$s已分配,不能删除", post.getPostName())); + } + } + return postMapper.deletePostByIds(postIds); + } + + /** + * 新增保存岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + @Override + public int insertPost(SysPost post) + { + return postMapper.insertPost(post); + } + + /** + * 修改保存岗位信息 + * + * @param post 岗位信息 + * @return 结果 + */ + @Override + public int updatePost(SysPost post) + { + return postMapper.updatePost(post); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java new file mode 100644 index 0000000..b9b8f32 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java @@ -0,0 +1,424 @@ +package com.ruoyi.system.service.impl; + +import java.util.ArrayList; +import java.util.Arrays; +import java.util.HashSet; +import java.util.List; +import java.util.Set; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import com.ruoyi.common.annotation.DataScope; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.system.domain.SysRoleDept; +import com.ruoyi.system.domain.SysRoleMenu; +import com.ruoyi.system.domain.SysUserRole; +import com.ruoyi.system.mapper.SysRoleDeptMapper; +import com.ruoyi.system.mapper.SysRoleMapper; +import com.ruoyi.system.mapper.SysRoleMenuMapper; +import com.ruoyi.system.mapper.SysUserRoleMapper; +import com.ruoyi.system.service.ISysRoleService; + +/** + * 角色 业务层处理 + * + * @author ruoyi + */ +@Service +public class SysRoleServiceImpl implements ISysRoleService +{ + @Autowired + private SysRoleMapper roleMapper; + + @Autowired + private SysRoleMenuMapper roleMenuMapper; + + @Autowired + private SysUserRoleMapper userRoleMapper; + + @Autowired + private SysRoleDeptMapper roleDeptMapper; + + /** + * 根据条件分页查询角色数据 + * + * @param role 角色信息 + * @return 角色数据集合信息 + */ + @Override + @DataScope(deptAlias = "d") + public List selectRoleList(SysRole role) + { + return roleMapper.selectRoleList(role); + } + + /** + * 根据用户ID查询角色 + * + * @param userId 用户ID + * @return 角色列表 + */ + @Override + public List selectRolesByUserId(Long userId) + { + List userRoles = roleMapper.selectRolePermissionByUserId(userId); + List roles = selectRoleAll(); + for (SysRole role : roles) + { + for (SysRole userRole : userRoles) + { + if (role.getRoleId().longValue() == userRole.getRoleId().longValue()) + { + role.setFlag(true); + break; + } + } + } + return roles; + } + + /** + * 根据用户ID查询权限 + * + * @param userId 用户ID + * @return 权限列表 + */ + @Override + public Set selectRolePermissionByUserId(Long userId) + { + List perms = roleMapper.selectRolePermissionByUserId(userId); + Set permsSet = new HashSet<>(); + for (SysRole perm : perms) + { + if (StringUtils.isNotNull(perm)) + { + permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(","))); + } + } + return permsSet; + } + + /** + * 查询所有角色 + * + * @return 角色列表 + */ + @Override + public List selectRoleAll() + { + return SpringUtils.getAopProxy(this).selectRoleList(new SysRole()); + } + + /** + * 根据用户ID获取角色选择框列表 + * + * @param userId 用户ID + * @return 选中角色ID列表 + */ + @Override + public List selectRoleListByUserId(Long userId) + { + return roleMapper.selectRoleListByUserId(userId); + } + + /** + * 通过角色ID查询角色 + * + * @param roleId 角色ID + * @return 角色对象信息 + */ + @Override + public SysRole selectRoleById(Long roleId) + { + return roleMapper.selectRoleById(roleId); + } + + /** + * 校验角色名称是否唯一 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + public String checkRoleNameUnique(SysRole role) + { + Long roleId = StringUtils.isNull(role.getRoleId()) ? -1L : role.getRoleId(); + SysRole info = roleMapper.checkRoleNameUnique(role.getRoleName()); + if (StringUtils.isNotNull(info) && info.getRoleId().longValue() != roleId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验角色权限是否唯一 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + public String checkRoleKeyUnique(SysRole role) + { + Long roleId = StringUtils.isNull(role.getRoleId()) ? -1L : role.getRoleId(); + SysRole info = roleMapper.checkRoleKeyUnique(role.getRoleKey()); + if (StringUtils.isNotNull(info) && info.getRoleId().longValue() != roleId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验角色是否允许操作 + * + * @param role 角色信息 + */ + @Override + public void checkRoleAllowed(SysRole role) + { + if (StringUtils.isNotNull(role.getRoleId()) && role.isAdmin()) + { + throw new ServiceException("不允许操作超级管理员角色"); + } + } + + /** + * 校验角色是否有数据权限 + * + * @param roleId 角色id + */ + @Override + public void checkRoleDataScope(Long roleId) + { + if (!SysUser.isAdmin(SecurityUtils.getUserId())) + { + SysRole role = new SysRole(); + role.setRoleId(roleId); + List roles = SpringUtils.getAopProxy(this).selectRoleList(role); + if (StringUtils.isEmpty(roles)) + { + throw new ServiceException("没有权限访问角色数据!"); + } + } + } + + /** + * 通过角色ID查询角色使用数量 + * + * @param roleId 角色ID + * @return 结果 + */ + @Override + public int countUserRoleByRoleId(Long roleId) + { + return userRoleMapper.countUserRoleByRoleId(roleId); + } + + /** + * 新增保存角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + @Transactional + public int insertRole(SysRole role) + { + // 新增角色信息 + roleMapper.insertRole(role); + return insertRoleMenu(role); + } + + /** + * 修改保存角色信息 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + @Transactional + public int updateRole(SysRole role) + { + // 修改角色信息 + roleMapper.updateRole(role); + // 删除角色与菜单关联 + roleMenuMapper.deleteRoleMenuByRoleId(role.getRoleId()); + return insertRoleMenu(role); + } + + /** + * 修改角色状态 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + public int updateRoleStatus(SysRole role) + { + return roleMapper.updateRole(role); + } + + /** + * 修改数据权限信息 + * + * @param role 角色信息 + * @return 结果 + */ + @Override + @Transactional + public int authDataScope(SysRole role) + { + // 修改角色信息 + roleMapper.updateRole(role); + // 删除角色与部门关联 + roleDeptMapper.deleteRoleDeptByRoleId(role.getRoleId()); + // 新增角色和部门信息(数据权限) + return insertRoleDept(role); + } + + /** + * 新增角色菜单信息 + * + * @param role 角色对象 + */ + public int insertRoleMenu(SysRole role) + { + int rows = 1; + // 新增用户与角色管理 + List list = new ArrayList(); + for (Long menuId : role.getMenuIds()) + { + SysRoleMenu rm = new SysRoleMenu(); + rm.setRoleId(role.getRoleId()); + rm.setMenuId(menuId); + list.add(rm); + } + if (list.size() > 0) + { + rows = roleMenuMapper.batchRoleMenu(list); + } + return rows; + } + + /** + * 新增角色部门信息(数据权限) + * + * @param role 角色对象 + */ + public int insertRoleDept(SysRole role) + { + int rows = 1; + // 新增角色与部门(数据权限)管理 + List list = new ArrayList(); + for (Long deptId : role.getDeptIds()) + { + SysRoleDept rd = new SysRoleDept(); + rd.setRoleId(role.getRoleId()); + rd.setDeptId(deptId); + list.add(rd); + } + if (list.size() > 0) + { + rows = roleDeptMapper.batchRoleDept(list); + } + return rows; + } + + /** + * 通过角色ID删除角色 + * + * @param roleId 角色ID + * @return 结果 + */ + @Override + @Transactional + public int deleteRoleById(Long roleId) + { + // 删除角色与菜单关联 + roleMenuMapper.deleteRoleMenuByRoleId(roleId); + // 删除角色与部门关联 + roleDeptMapper.deleteRoleDeptByRoleId(roleId); + return roleMapper.deleteRoleById(roleId); + } + + /** + * 批量删除角色信息 + * + * @param roleIds 需要删除的角色ID + * @return 结果 + */ + @Override + @Transactional + public int deleteRoleByIds(Long[] roleIds) + { + for (Long roleId : roleIds) + { + checkRoleAllowed(new SysRole(roleId)); + checkRoleDataScope(roleId); + SysRole role = selectRoleById(roleId); + if (countUserRoleByRoleId(roleId) > 0) + { + throw new ServiceException(String.format("%1$s已分配,不能删除", role.getRoleName())); + } + } + // 删除角色与菜单关联 + roleMenuMapper.deleteRoleMenu(roleIds); + // 删除角色与部门关联 + roleDeptMapper.deleteRoleDept(roleIds); + return roleMapper.deleteRoleByIds(roleIds); + } + + /** + * 取消授权用户角色 + * + * @param userRole 用户和角色关联信息 + * @return 结果 + */ + @Override + public int deleteAuthUser(SysUserRole userRole) + { + return userRoleMapper.deleteUserRoleInfo(userRole); + } + + /** + * 批量取消授权用户角色 + * + * @param roleId 角色ID + * @param userIds 需要取消授权的用户数据ID + * @return 结果 + */ + @Override + public int deleteAuthUsers(Long roleId, Long[] userIds) + { + return userRoleMapper.deleteUserRoleInfos(roleId, userIds); + } + + /** + * 批量选择授权用户角色 + * + * @param roleId 角色ID + * @param userIds 需要授权的用户数据ID + * @return 结果 + */ + @Override + public int insertAuthUsers(Long roleId, Long[] userIds) + { + // 新增用户与角色管理 + List list = new ArrayList(); + for (Long userId : userIds) + { + SysUserRole ur = new SysUserRole(); + ur.setUserId(userId); + ur.setRoleId(roleId); + list.add(ur); + } + return userRoleMapper.batchUserRole(list); + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserOnlineServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserOnlineServiceImpl.java new file mode 100644 index 0000000..f80a877 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserOnlineServiceImpl.java @@ -0,0 +1,96 @@ +package com.ruoyi.system.service.impl; + +import org.springframework.stereotype.Service; +import com.ruoyi.common.core.domain.model.LoginUser; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.system.domain.SysUserOnline; +import com.ruoyi.system.service.ISysUserOnlineService; + +/** + * 在线用户 服务层处理 + * + * @author ruoyi + */ +@Service +public class SysUserOnlineServiceImpl implements ISysUserOnlineService +{ + /** + * 通过登录地址查询信息 + * + * @param ipaddr 登录地址 + * @param user 用户信息 + * @return 在线用户信息 + */ + @Override + public SysUserOnline selectOnlineByIpaddr(String ipaddr, LoginUser user) + { + if (StringUtils.equals(ipaddr, user.getIpaddr())) + { + return loginUserToUserOnline(user); + } + return null; + } + + /** + * 通过用户名称查询信息 + * + * @param userName 用户名称 + * @param user 用户信息 + * @return 在线用户信息 + */ + @Override + public SysUserOnline selectOnlineByUserName(String userName, LoginUser user) + { + if (StringUtils.equals(userName, user.getUsername())) + { + return loginUserToUserOnline(user); + } + return null; + } + + /** + * 通过登录地址/用户名称查询信息 + * + * @param ipaddr 登录地址 + * @param userName 用户名称 + * @param user 用户信息 + * @return 在线用户信息 + */ + @Override + public SysUserOnline selectOnlineByInfo(String ipaddr, String userName, LoginUser user) + { + if (StringUtils.equals(ipaddr, user.getIpaddr()) && StringUtils.equals(userName, user.getUsername())) + { + return loginUserToUserOnline(user); + } + return null; + } + + /** + * 设置在线用户信息 + * + * @param user 用户信息 + * @return 在线用户 + */ + @Override + public SysUserOnline loginUserToUserOnline(LoginUser user) + { + if (StringUtils.isNull(user) || StringUtils.isNull(user.getUser())) + { + return null; + } + SysUserOnline sysUserOnline = new SysUserOnline(); + sysUserOnline.setTokenId(user.getToken()); + sysUserOnline.setUserName(user.getUsername()); + sysUserOnline.setIpaddr(user.getIpaddr()); + sysUserOnline.setLoginLocation(user.getLoginLocation()); + sysUserOnline.setBrowser(user.getBrowser()); + sysUserOnline.setOs(user.getOs()); + sysUserOnline.setLoginTime(user.getLoginTime()); + if (StringUtils.isNotNull(user.getUser().getDept())) + { + sysUserOnline.setDeptName(user.getUser().getDept().getDeptName()); + } + return sysUserOnline; + } +} diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java new file mode 100644 index 0000000..52f9cf9 --- /dev/null +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java @@ -0,0 +1,562 @@ +package com.ruoyi.system.service.impl; + +import java.util.ArrayList; +import java.util.List; +import java.util.stream.Collectors; +import javax.validation.Validator; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.util.CollectionUtils; +import com.ruoyi.common.annotation.DataScope; +import com.ruoyi.common.constant.UserConstants; +import com.ruoyi.common.core.domain.entity.SysRole; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.exception.ServiceException; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.StringUtils; +import com.ruoyi.common.utils.bean.BeanValidators; +import com.ruoyi.common.utils.spring.SpringUtils; +import com.ruoyi.system.domain.SysPost; +import com.ruoyi.system.domain.SysUserPost; +import com.ruoyi.system.domain.SysUserRole; +import com.ruoyi.system.mapper.SysPostMapper; +import com.ruoyi.system.mapper.SysRoleMapper; +import com.ruoyi.system.mapper.SysUserMapper; +import com.ruoyi.system.mapper.SysUserPostMapper; +import com.ruoyi.system.mapper.SysUserRoleMapper; +import com.ruoyi.system.service.ISysConfigService; +import com.ruoyi.system.service.ISysUserService; + +/** + * 用户 业务层处理 + * + * @author ruoyi + */ +@Service +public class SysUserServiceImpl implements ISysUserService +{ + private static final Logger log = LoggerFactory.getLogger(SysUserServiceImpl.class); + + @Autowired + private SysUserMapper userMapper; + + @Autowired + private SysRoleMapper roleMapper; + + @Autowired + private SysPostMapper postMapper; + + @Autowired + private SysUserRoleMapper userRoleMapper; + + @Autowired + private SysUserPostMapper userPostMapper; + + @Autowired + private ISysConfigService configService; + + @Autowired + protected Validator validator; + + /** + * 根据条件分页查询用户列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + @Override + @DataScope(deptAlias = "d", userAlias = "u") + public List selectUserList(SysUser user) + { + return userMapper.selectUserList(user); + } + + /** + * 根据条件分页查询已分配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + @Override + @DataScope(deptAlias = "d", userAlias = "u") + public List selectAllocatedList(SysUser user) + { + return userMapper.selectAllocatedList(user); + } + + /** + * 根据条件分页查询未分配用户角色列表 + * + * @param user 用户信息 + * @return 用户信息集合信息 + */ + @Override + @DataScope(deptAlias = "d", userAlias = "u") + public List selectUnallocatedList(SysUser user) + { + return userMapper.selectUnallocatedList(user); + } + + /** + * 通过用户名查询用户 + * + * @param userName 用户名 + * @return 用户对象信息 + */ + @Override + public SysUser selectUserByUserName(String userName) + { + return userMapper.selectUserByUserName(userName); + } + + /** + * 通过用户ID查询用户 + * + * @param userId 用户ID + * @return 用户对象信息 + */ + @Override + public SysUser selectUserById(Long userId) + { + return userMapper.selectUserById(userId); + } + + /** + * 查询用户所属角色组 + * + * @param userName 用户名 + * @return 结果 + */ + @Override + public String selectUserRoleGroup(String userName) + { + List list = roleMapper.selectRolesByUserName(userName); + if (CollectionUtils.isEmpty(list)) + { + return StringUtils.EMPTY; + } + return list.stream().map(SysRole::getRoleName).collect(Collectors.joining(",")); + } + + /** + * 查询用户所属岗位组 + * + * @param userName 用户名 + * @return 结果 + */ + @Override + public String selectUserPostGroup(String userName) + { + List list = postMapper.selectPostsByUserName(userName); + if (CollectionUtils.isEmpty(list)) + { + return StringUtils.EMPTY; + } + return list.stream().map(SysPost::getPostName).collect(Collectors.joining(",")); + } + + /** + * 校验用户名称是否唯一 + * + * @param userName 用户名称 + * @return 结果 + */ + @Override + public String checkUserNameUnique(String userName) + { + int count = userMapper.checkUserNameUnique(userName); + if (count > 0) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验手机号码是否唯一 + * + * @param user 用户信息 + * @return + */ + @Override + public String checkPhoneUnique(SysUser user) + { + Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId(); + SysUser info = userMapper.checkPhoneUnique(user.getPhonenumber()); + if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验email是否唯一 + * + * @param user 用户信息 + * @return + */ + @Override + public String checkEmailUnique(SysUser user) + { + Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId(); + SysUser info = userMapper.checkEmailUnique(user.getEmail()); + if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) + { + return UserConstants.NOT_UNIQUE; + } + return UserConstants.UNIQUE; + } + + /** + * 校验用户是否允许操作 + * + * @param user 用户信息 + */ + @Override + public void checkUserAllowed(SysUser user) + { + if (StringUtils.isNotNull(user.getUserId()) && user.isAdmin()) + { + throw new ServiceException("不允许操作超级管理员用户"); + } + } + + /** + * 校验用户是否有数据权限 + * + * @param userId 用户id + */ + @Override + public void checkUserDataScope(Long userId) + { + if (!SysUser.isAdmin(SecurityUtils.getUserId())) + { + SysUser user = new SysUser(); + user.setUserId(userId); + List users = SpringUtils.getAopProxy(this).selectUserList(user); + if (StringUtils.isEmpty(users)) + { + throw new ServiceException("没有权限访问用户数据!"); + } + } + } + + /** + * 新增保存用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + @Transactional + public int insertUser(SysUser user) + { + // 新增用户信息 + int rows = userMapper.insertUser(user); + // 新增用户岗位关联 + insertUserPost(user); + // 新增用户与角色管理 + insertUserRole(user); + return rows; + } + + /** + * 注册用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + public boolean registerUser(SysUser user) + { + return userMapper.insertUser(user) > 0; + } + + /** + * 修改保存用户信息 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + @Transactional + public int updateUser(SysUser user) + { + Long userId = user.getUserId(); + // 删除用户与角色关联 + userRoleMapper.deleteUserRoleByUserId(userId); + // 新增用户与角色管理 + insertUserRole(user); + // 删除用户与岗位关联 + userPostMapper.deleteUserPostByUserId(userId); + // 新增用户与岗位管理 + insertUserPost(user); + return userMapper.updateUser(user); + } + + /** + * 用户授权角色 + * + * @param userId 用户ID + * @param roleIds 角色组 + */ + @Override + @Transactional + public void insertUserAuth(Long userId, Long[] roleIds) + { + userRoleMapper.deleteUserRoleByUserId(userId); + insertUserRole(userId, roleIds); + } + + /** + * 修改用户状态 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + public int updateUserStatus(SysUser user) + { + return userMapper.updateUser(user); + } + + /** + * 修改用户基本信息 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + public int updateUserProfile(SysUser user) + { + return userMapper.updateUser(user); + } + + /** + * 修改用户头像 + * + * @param userName 用户名 + * @param avatar 头像地址 + * @return 结果 + */ + @Override + public boolean updateUserAvatar(String userName, String avatar) + { + return userMapper.updateUserAvatar(userName, avatar) > 0; + } + + /** + * 重置用户密码 + * + * @param user 用户信息 + * @return 结果 + */ + @Override + public int resetPwd(SysUser user) + { + return userMapper.updateUser(user); + } + + /** + * 重置用户密码 + * + * @param userName 用户名 + * @param password 密码 + * @return 结果 + */ + @Override + public int resetUserPwd(String userName, String password) + { + return userMapper.resetUserPwd(userName, password); + } + + /** + * 新增用户角色信息 + * + * @param user 用户对象 + */ + public void insertUserRole(SysUser user) + { + Long[] roles = user.getRoleIds(); + if (StringUtils.isNotNull(roles)) + { + // 新增用户与角色管理 + List list = new ArrayList(); + for (Long roleId : roles) + { + SysUserRole ur = new SysUserRole(); + ur.setUserId(user.getUserId()); + ur.setRoleId(roleId); + list.add(ur); + } + if (list.size() > 0) + { + userRoleMapper.batchUserRole(list); + } + } + } + + /** + * 新增用户岗位信息 + * + * @param user 用户对象 + */ + public void insertUserPost(SysUser user) + { + Long[] posts = user.getPostIds(); + if (StringUtils.isNotNull(posts)) + { + // 新增用户与岗位管理 + List list = new ArrayList(); + for (Long postId : posts) + { + SysUserPost up = new SysUserPost(); + up.setUserId(user.getUserId()); + up.setPostId(postId); + list.add(up); + } + if (list.size() > 0) + { + userPostMapper.batchUserPost(list); + } + } + } + + /** + * 新增用户角色信息 + * + * @param userId 用户ID + * @param roleIds 角色组 + */ + public void insertUserRole(Long userId, Long[] roleIds) + { + if (StringUtils.isNotNull(roleIds)) + { + // 新增用户与角色管理 + List list = new ArrayList(); + for (Long roleId : roleIds) + { + SysUserRole ur = new SysUserRole(); + ur.setUserId(userId); + ur.setRoleId(roleId); + list.add(ur); + } + if (list.size() > 0) + { + userRoleMapper.batchUserRole(list); + } + } + } + + /** + * 通过用户ID删除用户 + * + * @param userId 用户ID + * @return 结果 + */ + @Override + @Transactional + public int deleteUserById(Long userId) + { + // 删除用户与角色关联 + userRoleMapper.deleteUserRoleByUserId(userId); + // 删除用户与岗位表 + userPostMapper.deleteUserPostByUserId(userId); + return userMapper.deleteUserById(userId); + } + + /** + * 批量删除用户信息 + * + * @param userIds 需要删除的用户ID + * @return 结果 + */ + @Override + @Transactional + public int deleteUserByIds(Long[] userIds) + { + for (Long userId : userIds) + { + checkUserAllowed(new SysUser(userId)); + checkUserDataScope(userId); + } + // 删除用户与角色关联 + userRoleMapper.deleteUserRole(userIds); + // 删除用户与岗位关联 + userPostMapper.deleteUserPost(userIds); + return userMapper.deleteUserByIds(userIds); + } + + /** + * 导入用户数据 + * + * @param userList 用户数据列表 + * @param isUpdateSupport 是否更新支持,如果已存在,则进行更新数据 + * @param operName 操作用户 + * @return 结果 + */ + @Override + public String importUser(List userList, Boolean isUpdateSupport, String operName) + { + if (StringUtils.isNull(userList) || userList.size() == 0) + { + throw new ServiceException("导入用户数据不能为空!"); + } + int successNum = 0; + int failureNum = 0; + StringBuilder successMsg = new StringBuilder(); + StringBuilder failureMsg = new StringBuilder(); + String password = configService.selectConfigByKey("sys.user.initPassword"); + for (SysUser user : userList) + { + try + { + // 验证是否存在这个用户 + SysUser u = userMapper.selectUserByUserName(user.getUserName()); + if (StringUtils.isNull(u)) + { + BeanValidators.validateWithException(validator, user); + user.setPassword(SecurityUtils.encryptPassword(password)); + user.setCreateBy(operName); + this.insertUser(user); + successNum++; + successMsg.append("
" + successNum + "、账号 " + user.getUserName() + " 导入成功"); + } + else if (isUpdateSupport) + { + BeanValidators.validateWithException(validator, user); + user.setUpdateBy(operName); + this.updateUser(user); + successNum++; + successMsg.append("
" + successNum + "、账号 " + user.getUserName() + " 更新成功"); + } + else + { + failureNum++; + failureMsg.append("
" + failureNum + "、账号 " + user.getUserName() + " 已存在"); + } + } + catch (Exception e) + { + failureNum++; + String msg = "
" + failureNum + "、账号 " + user.getUserName() + " 导入失败:"; + failureMsg.append(msg + e.getMessage()); + log.error(msg, e); + } + } + if (failureNum > 0) + { + failureMsg.insert(0, "很抱歉,导入失败!共 " + failureNum + " 条数据格式不正确,错误如下:"); + throw new ServiceException(failureMsg.toString()); + } + else + { + successMsg.insert(0, "恭喜您,数据已全部导入成功!共 " + successNum + " 条,数据如下:"); + } + return successMsg.toString(); + } +} diff --git a/ruoyi-system/src/main/resources/mapper/system/SysConfigMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysConfigMapper.xml new file mode 100644 index 0000000..8b97906 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysConfigMapper.xml @@ -0,0 +1,112 @@ + + + + + + + + + + + + + + + + + + select config_id, config_name, config_key, config_value, config_type, create_by, create_time, update_by, update_time, remark + from sys_config + + + + + + + and config_id = #{configId} + + + and config_key = #{configKey} + + + + + + + + + + + + insert into sys_config ( + config_name, + config_key, + config_value, + config_type, + create_by, + remark, + create_time + )values( + #{configName}, + #{configKey}, + #{configValue}, + #{configType}, + #{createBy}, + #{remark}, + sysdate() + ) + + + + update sys_config + + config_name = #{configName}, + config_key = #{configKey}, + config_value = #{configValue}, + config_type = #{configType}, + update_by = #{updateBy}, + remark = #{remark}, + update_time = sysdate() + + where config_id = #{configId} + + + + delete from sys_config where config_id = #{configId} + + + + delete from sys_config where config_id in + + #{configId} + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysDeptMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysDeptMapper.xml new file mode 100644 index 0000000..b01bb32 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysDeptMapper.xml @@ -0,0 +1,157 @@ + + + + + + + + + + + + + + + + + + + + + + + + select d.dept_id, d.parent_id, d.ancestors, d.dept_name, d.order_num, d.leader, d.phone, d.email, d.status, d.del_flag, d.create_by, d.create_time + from sys_dept d + + + + + + + + + + + + + + + + + + + + insert into sys_dept( + dept_id, + parent_id, + dept_name, + ancestors, + order_num, + leader, + phone, + email, + status, + create_by, + create_time + )values( + #{deptId}, + #{parentId}, + #{deptName}, + #{ancestors}, + #{orderNum}, + #{leader}, + #{phone}, + #{email}, + #{status}, + #{createBy}, + sysdate() + ) + + + + update sys_dept + + parent_id = #{parentId}, + dept_name = #{deptName}, + ancestors = #{ancestors}, + order_num = #{orderNum}, + leader = #{leader}, + phone = #{phone}, + email = #{email}, + status = #{status}, + update_by = #{updateBy}, + update_time = sysdate() + + where dept_id = #{deptId} + + + + update sys_dept set ancestors = + + when #{item.deptId} then #{item.ancestors} + + where dept_id in + + #{item.deptId} + + + + + update sys_dept set status = '0' where dept_id in + + #{deptId} + + + + + update sys_dept set del_flag = '2' where dept_id = #{deptId} + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysDictDataMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysDictDataMapper.xml new file mode 100644 index 0000000..8da9030 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysDictDataMapper.xml @@ -0,0 +1,124 @@ + + + + + + + + + + + + + + + + + + + + + + select dict_code, dict_sort, dict_label, dict_value, dict_type, css_class, list_class, is_default, status, create_by, create_time, remark + from sys_dict_data + + + + + + + + + + + + + + delete from sys_dict_data where dict_code = #{dictCode} + + + + delete from sys_dict_data where dict_code in + + #{dictCode} + + + + + update sys_dict_data + + dict_sort = #{dictSort}, + dict_label = #{dictLabel}, + dict_value = #{dictValue}, + dict_type = #{dictType}, + css_class = #{cssClass}, + list_class = #{listClass}, + is_default = #{isDefault}, + status = #{status}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where dict_code = #{dictCode} + + + + update sys_dict_data set dict_type = #{newDictType} where dict_type = #{oldDictType} + + + + insert into sys_dict_data( + dict_sort, + dict_label, + dict_value, + dict_type, + css_class, + list_class, + is_default, + status, + remark, + create_by, + create_time + )values( + #{dictSort}, + #{dictLabel}, + #{dictValue}, + #{dictType}, + #{cssClass}, + #{listClass}, + #{isDefault}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysDictTypeMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysDictTypeMapper.xml new file mode 100644 index 0000000..55b4075 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysDictTypeMapper.xml @@ -0,0 +1,105 @@ + + + + + + + + + + + + + + + + + select dict_id, dict_name, dict_type, status, create_by, create_time, remark + from sys_dict_type + + + + + + + + + + + + + + delete from sys_dict_type where dict_id = #{dictId} + + + + delete from sys_dict_type where dict_id in + + #{dictId} + + + + + update sys_dict_type + + dict_name = #{dictName}, + dict_type = #{dictType}, + status = #{status}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where dict_id = #{dictId} + + + + insert into sys_dict_type( + dict_name, + dict_type, + status, + remark, + create_by, + create_time + )values( + #{dictName}, + #{dictType}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysLogininforMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysLogininforMapper.xml new file mode 100644 index 0000000..b8178fa --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysLogininforMapper.xml @@ -0,0 +1,57 @@ + + + + + + + + + + + + + + + + + + insert into sys_logininfor (user_name, status, ipaddr, login_location, browser, os, msg, login_time) + values (#{userName}, #{status}, #{ipaddr}, #{loginLocation}, #{browser}, #{os}, #{msg}, sysdate()) + + + + + + delete from sys_logininfor where info_id in + + #{infoId} + + + + + truncate table sys_logininfor + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysMenuMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysMenuMapper.xml new file mode 100644 index 0000000..2ad833c --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysMenuMapper.xml @@ -0,0 +1,195 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + select menu_id, menu_name, parent_id, order_num, path, component, `query`, is_frame, is_cache, menu_type, visible, status, ifnull(perms,'') as perms, icon, create_time + from sys_menu + + + + + + + + + + + + + + + + + + + + + + + + update sys_menu + + menu_name = #{menuName}, + parent_id = #{parentId}, + order_num = #{orderNum}, + path = #{path}, + component = #{component}, + `query` = #{query}, + is_frame = #{isFrame}, + is_cache = #{isCache}, + menu_type = #{menuType}, + visible = #{visible}, + status = #{status}, + perms = #{perms}, + icon = #{icon}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where menu_id = #{menuId} + + + + insert into sys_menu( + menu_id, + parent_id, + menu_name, + order_num, + path, + component, + `query`, + is_frame, + is_cache, + menu_type, + visible, + status, + perms, + icon, + remark, + create_by, + create_time + )values( + #{menuId}, + #{parentId}, + #{menuName}, + #{orderNum}, + #{path}, + #{component}, + #{query}, + #{isFrame}, + #{isCache}, + #{menuType}, + #{visible}, + #{status}, + #{perms}, + #{icon}, + #{remark}, + #{createBy}, + sysdate() + ) + + + + delete from sys_menu where menu_id = #{menuId} + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysNoticeMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysNoticeMapper.xml new file mode 100644 index 0000000..65d3079 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysNoticeMapper.xml @@ -0,0 +1,89 @@ + + + + + + + + + + + + + + + + + + + select notice_id, notice_title, notice_type, cast(notice_content as char) as notice_content, status, create_by, create_time, update_by, update_time, remark + from sys_notice + + + + + + + + insert into sys_notice ( + notice_title, + notice_type, + notice_content, + status, + remark, + create_by, + create_time + )values( + #{noticeTitle}, + #{noticeType}, + #{noticeContent}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + + update sys_notice + + notice_title = #{noticeTitle}, + notice_type = #{noticeType}, + notice_content = #{noticeContent}, + status = #{status}, + update_by = #{updateBy}, + update_time = sysdate() + + where notice_id = #{noticeId} + + + + delete from sys_notice where notice_id = #{noticeId} + + + + delete from sys_notice where notice_id in + + #{noticeId} + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysOperLogMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysOperLogMapper.xml new file mode 100644 index 0000000..f017ef8 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysOperLogMapper.xml @@ -0,0 +1,83 @@ + + + + + + + + + + + + + + + + + + + + + + + + + select oper_id, title, business_type, method, request_method, operator_type, oper_name, dept_name, oper_url, oper_ip, oper_location, oper_param, json_result, status, error_msg, oper_time + from sys_oper_log + + + + insert into sys_oper_log(title, business_type, method, request_method, operator_type, oper_name, dept_name, oper_url, oper_ip, oper_location, oper_param, json_result, status, error_msg, oper_time) + values (#{title}, #{businessType}, #{method}, #{requestMethod}, #{operatorType}, #{operName}, #{deptName}, #{operUrl}, #{operIp}, #{operLocation}, #{operParam}, #{jsonResult}, #{status}, #{errorMsg}, sysdate()) + + + + + + delete from sys_oper_log where oper_id in + + #{operId} + + + + + + + truncate table sys_oper_log + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysPostMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysPostMapper.xml new file mode 100644 index 0000000..fc37f49 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysPostMapper.xml @@ -0,0 +1,122 @@ + + + + + + + + + + + + + + + + + + + select post_id, post_code, post_name, post_sort, status, create_by, create_time, remark + from sys_post + + + + + + + + + + + + + + + + + + update sys_post + + post_code = #{postCode}, + post_name = #{postName}, + post_sort = #{postSort}, + status = #{status}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where post_id = #{postId} + + + + insert into sys_post( + post_id, + post_code, + post_name, + post_sort, + status, + remark, + create_by, + create_time + )values( + #{postId}, + #{postCode}, + #{postName}, + #{postSort}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + + delete from sys_post where post_id = #{postId} + + + + delete from sys_post where post_id in + + #{postId} + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysRoleDeptMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysRoleDeptMapper.xml new file mode 100644 index 0000000..7c4139b --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysRoleDeptMapper.xml @@ -0,0 +1,34 @@ + + + + + + + + + + + delete from sys_role_dept where role_id=#{roleId} + + + + + + delete from sys_role_dept where role_id in + + #{roleId} + + + + + insert into sys_role_dept(role_id, dept_id) values + + (#{item.roleId},#{item.deptId}) + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysRoleMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysRoleMapper.xml new file mode 100644 index 0000000..96e8615 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysRoleMapper.xml @@ -0,0 +1,152 @@ + + + + + + + + + + + + + + + + + + + + + + + select distinct r.role_id, r.role_name, r.role_key, r.role_sort, r.data_scope, r.menu_check_strictly, r.dept_check_strictly, + r.status, r.del_flag, r.create_time, r.remark + from sys_role r + left join sys_user_role ur on ur.role_id = r.role_id + left join sys_user u on u.user_id = ur.user_id + left join sys_dept d on u.dept_id = d.dept_id + + + + + + + + + + + + + + + + + + + + insert into sys_role( + role_id, + role_name, + role_key, + role_sort, + data_scope, + menu_check_strictly, + dept_check_strictly, + status, + remark, + create_by, + create_time + )values( + #{roleId}, + #{roleName}, + #{roleKey}, + #{roleSort}, + #{dataScope}, + #{menuCheckStrictly}, + #{deptCheckStrictly}, + #{status}, + #{remark}, + #{createBy}, + sysdate() + ) + + + + update sys_role + + role_name = #{roleName}, + role_key = #{roleKey}, + role_sort = #{roleSort}, + data_scope = #{dataScope}, + menu_check_strictly = #{menuCheckStrictly}, + dept_check_strictly = #{deptCheckStrictly}, + status = #{status}, + remark = #{remark}, + update_by = #{updateBy}, + update_time = sysdate() + + where role_id = #{roleId} + + + + update sys_role set del_flag = '2' where role_id = #{roleId} + + + + update sys_role set del_flag = '2' where role_id in + + #{roleId} + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysRoleMenuMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysRoleMenuMapper.xml new file mode 100644 index 0000000..cb60a85 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysRoleMenuMapper.xml @@ -0,0 +1,34 @@ + + + + + + + + + + + + + delete from sys_role_menu where role_id=#{roleId} + + + + delete from sys_role_menu where role_id in + + #{roleId} + + + + + insert into sys_role_menu(role_id, menu_id) values + + (#{item.roleId},#{item.menuId}) + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysUserMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysUserMapper.xml new file mode 100644 index 0000000..02f3a7b --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysUserMapper.xml @@ -0,0 +1,221 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + select u.user_id, u.dept_id, u.user_name, u.nick_name, u.email, u.avatar, u.phonenumber, u.password, u.sex, u.status, u.del_flag, u.login_ip, u.login_date, u.create_by, u.create_time, u.remark, + d.dept_id, d.parent_id, d.ancestors, d.dept_name, d.order_num, d.leader, d.status as dept_status, + r.role_id, r.role_name, r.role_key, r.role_sort, r.data_scope, r.status as role_status + from sys_user u + left join sys_dept d on u.dept_id = d.dept_id + left join sys_user_role ur on u.user_id = ur.user_id + left join sys_role r on r.role_id = ur.role_id + + + + + + + + + + + + + + + + + + + + insert into sys_user( + user_id, + dept_id, + user_name, + nick_name, + email, + avatar, + phonenumber, + sex, + password, + status, + create_by, + remark, + create_time + )values( + #{userId}, + #{deptId}, + #{userName}, + #{nickName}, + #{email}, + #{avatar}, + #{phonenumber}, + #{sex}, + #{password}, + #{status}, + #{createBy}, + #{remark}, + sysdate() + ) + + + + update sys_user + + dept_id = #{deptId}, + user_name = #{userName}, + nick_name = #{nickName}, + email = #{email}, + phonenumber = #{phonenumber}, + sex = #{sex}, + avatar = #{avatar}, + password = #{password}, + status = #{status}, + login_ip = #{loginIp}, + login_date = #{loginDate}, + update_by = #{updateBy}, + remark = #{remark}, + update_time = sysdate() + + where user_id = #{userId} + + + + update sys_user set status = #{status} where user_id = #{userId} + + + + update sys_user set avatar = #{avatar} where user_name = #{userName} + + + + update sys_user set password = #{password} where user_name = #{userName} + + + + update sys_user set del_flag = '2' where user_id = #{userId} + + + + update sys_user set del_flag = '2' where user_id in + + #{userId} + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysUserPostMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysUserPostMapper.xml new file mode 100644 index 0000000..2b90bc4 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysUserPostMapper.xml @@ -0,0 +1,34 @@ + + + + + + + + + + + delete from sys_user_post where user_id=#{userId} + + + + + + delete from sys_user_post where user_id in + + #{userId} + + + + + insert into sys_user_post(user_id, post_id) values + + (#{item.userId},#{item.postId}) + + + + \ No newline at end of file diff --git a/ruoyi-system/src/main/resources/mapper/system/SysUserRoleMapper.xml b/ruoyi-system/src/main/resources/mapper/system/SysUserRoleMapper.xml new file mode 100644 index 0000000..dd72689 --- /dev/null +++ b/ruoyi-system/src/main/resources/mapper/system/SysUserRoleMapper.xml @@ -0,0 +1,44 @@ + + + + + + + + + + + delete from sys_user_role where user_id=#{userId} + + + + + + delete from sys_user_role where user_id in + + #{userId} + + + + + insert into sys_user_role(user_id, role_id) values + + (#{item.userId},#{item.roleId}) + + + + + delete from sys_user_role where user_id=#{userId} and role_id=#{roleId} + + + + delete from sys_user_role where role_id=#{roleId} and user_id in + + #{userId} + + + \ No newline at end of file diff --git a/ry.bat b/ry.bat new file mode 100644 index 0000000..ac1e437 --- /dev/null +++ b/ry.bat @@ -0,0 +1,67 @@ +@echo off + +rem jarƽĿ¼ +set AppName=ruoyi-admin.jar + +rem JVM +set JVM_OPTS="-Dname=%AppName% -Duser.timezone=Asia/Shanghai -Xms512m -Xmx1024m -XX:MetaspaceSize=128m -XX:MaxMetaspaceSize=512m -XX:+HeapDumpOnOutOfMemoryError -XX:+PrintGCDateStamps -XX:+PrintGCDetails -XX:NewRatio=1 -XX:SurvivorRatio=30 -XX:+UseParallelGC -XX:+UseParallelOldGC" + + +ECHO. + ECHO. [1] %AppName% + ECHO. [2] ر%AppName% + ECHO. [3] %AppName% + ECHO. [4] ״̬ %AppName% + ECHO. [5] +ECHO. + +ECHO.ѡĿ: +set /p ID= + IF "%id%"=="1" GOTO start + IF "%id%"=="2" GOTO stop + IF "%id%"=="3" GOTO restart + IF "%id%"=="4" GOTO status + IF "%id%"=="5" EXIT +PAUSE +:start + for /f "usebackq tokens=1-2" %%a in (`jps -l ^| findstr %AppName%`) do ( + set pid=%%a + set image_name=%%b + ) + if defined pid ( + echo %%is running + PAUSE + ) + +start javaw %JVM_OPTS% -jar %AppName% + +echo starting +echo Start %AppName% success... +goto:eof + +rem stopͨjpspid +:stop + for /f "usebackq tokens=1-2" %%a in (`jps -l ^| findstr %AppName%`) do ( + set pid=%%a + set image_name=%%b + ) + if not defined pid (echo process %AppName% does not exists) else ( + echo prepare to kill %image_name% + echo start kill %pid% ... + rem ݽIDkill + taskkill /f /pid %pid% + ) +goto:eof +:restart + call :stop + call :start +goto:eof +:status + for /f "usebackq tokens=1-2" %%a in (`jps -l ^| findstr %AppName%`) do ( + set pid=%%a + set image_name=%%b + ) + if not defined pid (echo process %AppName% is dead ) else ( + echo %image_name% is running + ) +goto:eof diff --git a/ry.sh b/ry.sh new file mode 100644 index 0000000..d6a9cf3 --- /dev/null +++ b/ry.sh @@ -0,0 +1,86 @@ +#!/bin/sh +# ./ry.sh start 启动 stop 停止 restart 重启 status 状态 +AppName=ruoyi-admin.jar + +# JVM参数 +JVM_OPTS="-Dname=$AppName -Duser.timezone=Asia/Shanghai -Xms512m -Xmx1024m -XX:MetaspaceSize=128m -XX:MaxMetaspaceSize=512m -XX:+HeapDumpOnOutOfMemoryError -XX:+PrintGCDateStamps -XX:+PrintGCDetails -XX:NewRatio=1 -XX:SurvivorRatio=30 -XX:+UseParallelGC -XX:+UseParallelOldGC" +APP_HOME=`pwd` +LOG_PATH=$APP_HOME/logs/$AppName.log + +if [ "$1" = "" ]; +then + echo -e "\033[0;31m 未输入操作名 \033[0m \033[0;34m {start|stop|restart|status} \033[0m" + exit 1 +fi + +if [ "$AppName" = "" ]; +then + echo -e "\033[0;31m 未输入应用名 \033[0m" + exit 1 +fi + +function start() +{ + PID=`ps -ef |grep java|grep $AppName|grep -v grep|awk '{print $2}'` + + if [ x"$PID" != x"" ]; then + echo "$AppName is running..." + else + nohup java $JVM_OPTS -jar $AppName > /dev/null 2>&1 & + echo "Start $AppName success..." + fi +} + +function stop() +{ + echo "Stop $AppName" + + PID="" + query(){ + PID=`ps -ef |grep java|grep $AppName|grep -v grep|awk '{print $2}'` + } + + query + if [ x"$PID" != x"" ]; then + kill -TERM $PID + echo "$AppName (pid:$PID) exiting..." + while [ x"$PID" != x"" ] + do + sleep 1 + query + done + echo "$AppName exited." + else + echo "$AppName already stopped." + fi +} + +function restart() +{ + stop + sleep 2 + start +} + +function status() +{ + PID=`ps -ef |grep java|grep $AppName|grep -v grep|wc -l` + if [ $PID != 0 ];then + echo "$AppName is running..." + else + echo "$AppName is not running..." + fi +} + +case $1 in + start) + start;; + stop) + stop;; + restart) + restart;; + status) + status;; + *) + +esac diff --git a/sql/quartz.sql b/sql/quartz.sql new file mode 100644 index 0000000..cee613b --- /dev/null +++ b/sql/quartz.sql @@ -0,0 +1,174 @@ +DROP TABLE IF EXISTS QRTZ_FIRED_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_PAUSED_TRIGGER_GRPS; +DROP TABLE IF EXISTS QRTZ_SCHEDULER_STATE; +DROP TABLE IF EXISTS QRTZ_LOCKS; +DROP TABLE IF EXISTS QRTZ_SIMPLE_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_SIMPROP_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_CRON_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_BLOB_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_TRIGGERS; +DROP TABLE IF EXISTS QRTZ_JOB_DETAILS; +DROP TABLE IF EXISTS QRTZ_CALENDARS; + +-- ---------------------------- +-- 1、存储每一个已配置的 jobDetail 的详细信息 +-- ---------------------------- +create table QRTZ_JOB_DETAILS ( + sched_name varchar(120) not null comment '调度名称', + job_name varchar(200) not null comment '任务名称', + job_group varchar(200) not null comment '任务组名', + description varchar(250) null comment '相关介绍', + job_class_name varchar(250) not null comment '执行任务类名称', + is_durable varchar(1) not null comment '是否持久化', + is_nonconcurrent varchar(1) not null comment '是否并发', + is_update_data varchar(1) not null comment '是否更新数据', + requests_recovery varchar(1) not null comment '是否接受恢复执行', + job_data blob null comment '存放持久化job对象', + primary key (sched_name, job_name, job_group) +) engine=innodb comment = '任务详细信息表'; + +-- ---------------------------- +-- 2、 存储已配置的 Trigger 的信息 +-- ---------------------------- +create table QRTZ_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + trigger_name varchar(200) not null comment '触发器的名字', + trigger_group varchar(200) not null comment '触发器所属组的名字', + job_name varchar(200) not null comment 'qrtz_job_details表job_name的外键', + job_group varchar(200) not null comment 'qrtz_job_details表job_group的外键', + description varchar(250) null comment '相关介绍', + next_fire_time bigint(13) null comment '上一次触发时间(毫秒)', + prev_fire_time bigint(13) null comment '下一次触发时间(默认为-1表示不触发)', + priority integer null comment '优先级', + trigger_state varchar(16) not null comment '触发器状态', + trigger_type varchar(8) not null comment '触发器的类型', + start_time bigint(13) not null comment '开始时间', + end_time bigint(13) null comment '结束时间', + calendar_name varchar(200) null comment '日程表名称', + misfire_instr smallint(2) null comment '补偿执行的策略', + job_data blob null comment '存放持久化job对象', + primary key (sched_name, trigger_name, trigger_group), + foreign key (sched_name, job_name, job_group) references QRTZ_JOB_DETAILS(sched_name, job_name, job_group) +) engine=innodb comment = '触发器详细信息表'; + +-- ---------------------------- +-- 3、 存储简单的 Trigger,包括重复次数,间隔,以及已触发的次数 +-- ---------------------------- +create table QRTZ_SIMPLE_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + trigger_name varchar(200) not null comment 'qrtz_triggers表trigger_name的外键', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + repeat_count bigint(7) not null comment '重复的次数统计', + repeat_interval bigint(12) not null comment '重复的间隔时间', + times_triggered bigint(10) not null comment '已经触发的次数', + primary key (sched_name, trigger_name, trigger_group), + foreign key (sched_name, trigger_name, trigger_group) references QRTZ_TRIGGERS(sched_name, trigger_name, trigger_group) +) engine=innodb comment = '简单触发器的信息表'; + +-- ---------------------------- +-- 4、 存储 Cron Trigger,包括 Cron 表达式和时区信息 +-- ---------------------------- +create table QRTZ_CRON_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + trigger_name varchar(200) not null comment 'qrtz_triggers表trigger_name的外键', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + cron_expression varchar(200) not null comment 'cron表达式', + time_zone_id varchar(80) comment '时区', + primary key (sched_name, trigger_name, trigger_group), + foreign key (sched_name, trigger_name, trigger_group) references QRTZ_TRIGGERS(sched_name, trigger_name, trigger_group) +) engine=innodb comment = 'Cron类型的触发器表'; + +-- ---------------------------- +-- 5、 Trigger 作为 Blob 类型存储(用于 Quartz 用户用 JDBC 创建他们自己定制的 Trigger 类型,JobStore 并不知道如何存储实例的时候) +-- ---------------------------- +create table QRTZ_BLOB_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + trigger_name varchar(200) not null comment 'qrtz_triggers表trigger_name的外键', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + blob_data blob null comment '存放持久化Trigger对象', + primary key (sched_name, trigger_name, trigger_group), + foreign key (sched_name, trigger_name, trigger_group) references QRTZ_TRIGGERS(sched_name, trigger_name, trigger_group) +) engine=innodb comment = 'Blob类型的触发器表'; + +-- ---------------------------- +-- 6、 以 Blob 类型存储存放日历信息, quartz可配置一个日历来指定一个时间范围 +-- ---------------------------- +create table QRTZ_CALENDARS ( + sched_name varchar(120) not null comment '调度名称', + calendar_name varchar(200) not null comment '日历名称', + calendar blob not null comment '存放持久化calendar对象', + primary key (sched_name, calendar_name) +) engine=innodb comment = '日历信息表'; + +-- ---------------------------- +-- 7、 存储已暂停的 Trigger 组的信息 +-- ---------------------------- +create table QRTZ_PAUSED_TRIGGER_GRPS ( + sched_name varchar(120) not null comment '调度名称', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + primary key (sched_name, trigger_group) +) engine=innodb comment = '暂停的触发器表'; + +-- ---------------------------- +-- 8、 存储与已触发的 Trigger 相关的状态信息,以及相联 Job 的执行信息 +-- ---------------------------- +create table QRTZ_FIRED_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + entry_id varchar(95) not null comment '调度器实例id', + trigger_name varchar(200) not null comment 'qrtz_triggers表trigger_name的外键', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + instance_name varchar(200) not null comment '调度器实例名', + fired_time bigint(13) not null comment '触发的时间', + sched_time bigint(13) not null comment '定时器制定的时间', + priority integer not null comment '优先级', + state varchar(16) not null comment '状态', + job_name varchar(200) null comment '任务名称', + job_group varchar(200) null comment '任务组名', + is_nonconcurrent varchar(1) null comment '是否并发', + requests_recovery varchar(1) null comment '是否接受恢复执行', + primary key (sched_name, entry_id) +) engine=innodb comment = '已触发的触发器表'; + +-- ---------------------------- +-- 9、 存储少量的有关 Scheduler 的状态信息,假如是用于集群中,可以看到其他的 Scheduler 实例 +-- ---------------------------- +create table QRTZ_SCHEDULER_STATE ( + sched_name varchar(120) not null comment '调度名称', + instance_name varchar(200) not null comment '实例名称', + last_checkin_time bigint(13) not null comment '上次检查时间', + checkin_interval bigint(13) not null comment '检查间隔时间', + primary key (sched_name, instance_name) +) engine=innodb comment = '调度器状态表'; + +-- ---------------------------- +-- 10、 存储程序的悲观锁的信息(假如使用了悲观锁) +-- ---------------------------- +create table QRTZ_LOCKS ( + sched_name varchar(120) not null comment '调度名称', + lock_name varchar(40) not null comment '悲观锁名称', + primary key (sched_name, lock_name) +) engine=innodb comment = '存储的悲观锁信息表'; + +-- ---------------------------- +-- 11、 Quartz集群实现同步机制的行锁表 +-- ---------------------------- +create table QRTZ_SIMPROP_TRIGGERS ( + sched_name varchar(120) not null comment '调度名称', + trigger_name varchar(200) not null comment 'qrtz_triggers表trigger_name的外键', + trigger_group varchar(200) not null comment 'qrtz_triggers表trigger_group的外键', + str_prop_1 varchar(512) null comment 'String类型的trigger的第一个参数', + str_prop_2 varchar(512) null comment 'String类型的trigger的第二个参数', + str_prop_3 varchar(512) null comment 'String类型的trigger的第三个参数', + int_prop_1 int null comment 'int类型的trigger的第一个参数', + int_prop_2 int null comment 'int类型的trigger的第二个参数', + long_prop_1 bigint null comment 'long类型的trigger的第一个参数', + long_prop_2 bigint null comment 'long类型的trigger的第二个参数', + dec_prop_1 numeric(13,4) null comment 'decimal类型的trigger的第一个参数', + dec_prop_2 numeric(13,4) null comment 'decimal类型的trigger的第二个参数', + bool_prop_1 varchar(1) null comment 'Boolean类型的trigger的第一个参数', + bool_prop_2 varchar(1) null comment 'Boolean类型的trigger的第二个参数', + primary key (sched_name, trigger_name, trigger_group), + foreign key (sched_name, trigger_name, trigger_group) references QRTZ_TRIGGERS(sched_name, trigger_name, trigger_group) +) engine=innodb comment = '同步机制的行锁表'; + +commit; \ No newline at end of file diff --git a/sql/ry_20210908.sql b/sql/ry_20210908.sql new file mode 100644 index 0000000..650238a --- /dev/null +++ b/sql/ry_20210908.sql @@ -0,0 +1,688 @@ +-- ---------------------------- +-- 1、部门表 +-- ---------------------------- +drop table if exists sys_dept; +create table sys_dept ( + dept_id bigint(20) not null auto_increment comment '部门id', + parent_id bigint(20) default 0 comment '父部门id', + ancestors varchar(50) default '' comment '祖级列表', + dept_name varchar(30) default '' comment '部门名称', + order_num int(4) default 0 comment '显示顺序', + leader varchar(20) default null comment '负责人', + phone varchar(11) default null comment '联系电话', + email varchar(50) default null comment '邮箱', + status char(1) default '0' comment '部门状态(0正常 1停用)', + del_flag char(1) default '0' comment '删除标志(0代表存在 2代表删除)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + primary key (dept_id) +) engine=innodb auto_increment=200 comment = '部门表'; + +-- ---------------------------- +-- 初始化-部门表数据 +-- ---------------------------- +insert into sys_dept values(100, 0, '0', '若依科技', 0, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(101, 100, '0,100', '深圳总公司', 1, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(102, 100, '0,100', '长沙分公司', 2, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(103, 101, '0,100,101', '研发部门', 1, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(104, 101, '0,100,101', '市场部门', 2, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(105, 101, '0,100,101', '测试部门', 3, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(106, 101, '0,100,101', '财务部门', 4, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(107, 101, '0,100,101', '运维部门', 5, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(108, 102, '0,100,102', '市场部门', 1, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); +insert into sys_dept values(109, 102, '0,100,102', '财务部门', 2, '若依', '15888888888', 'ry@qq.com', '0', '0', 'admin', sysdate(), '', null); + + +-- ---------------------------- +-- 2、用户信息表 +-- ---------------------------- +drop table if exists sys_user; +create table sys_user ( + user_id bigint(20) not null auto_increment comment '用户ID', + dept_id bigint(20) default null comment '部门ID', + user_name varchar(30) not null comment '用户账号', + nick_name varchar(30) not null comment '用户昵称', + user_type varchar(2) default '00' comment '用户类型(00系统用户)', + email varchar(50) default '' comment '用户邮箱', + phonenumber varchar(11) default '' comment '手机号码', + sex char(1) default '0' comment '用户性别(0男 1女 2未知)', + avatar varchar(100) default '' comment '头像地址', + password varchar(100) default '' comment '密码', + status char(1) default '0' comment '帐号状态(0正常 1停用)', + del_flag char(1) default '0' comment '删除标志(0代表存在 2代表删除)', + login_ip varchar(128) default '' comment '最后登录IP', + login_date datetime comment '最后登录时间', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (user_id) +) engine=innodb auto_increment=100 comment = '用户信息表'; + +-- ---------------------------- +-- 初始化-用户信息表数据 +-- ---------------------------- +insert into sys_user values(1, 103, 'admin', '若依', '00', 'ry@163.com', '15888888888', '1', '', '$2a$10$7JB720yubVSZvUI0rEqK/.VqGOZTH.ulu33dHOiBE8ByOhJIrdAu2', '0', '0', '127.0.0.1', sysdate(), 'admin', sysdate(), '', null, '管理员'); +insert into sys_user values(2, 105, 'ry', '若依', '00', 'ry@qq.com', '15666666666', '1', '', '$2a$10$7JB720yubVSZvUI0rEqK/.VqGOZTH.ulu33dHOiBE8ByOhJIrdAu2', '0', '0', '127.0.0.1', sysdate(), 'admin', sysdate(), '', null, '测试员'); + + +-- ---------------------------- +-- 3、岗位信息表 +-- ---------------------------- +drop table if exists sys_post; +create table sys_post +( + post_id bigint(20) not null auto_increment comment '岗位ID', + post_code varchar(64) not null comment '岗位编码', + post_name varchar(50) not null comment '岗位名称', + post_sort int(4) not null comment '显示顺序', + status char(1) not null comment '状态(0正常 1停用)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (post_id) +) engine=innodb comment = '岗位信息表'; + +-- ---------------------------- +-- 初始化-岗位信息表数据 +-- ---------------------------- +insert into sys_post values(1, 'ceo', '董事长', 1, '0', 'admin', sysdate(), '', null, ''); +insert into sys_post values(2, 'se', '项目经理', 2, '0', 'admin', sysdate(), '', null, ''); +insert into sys_post values(3, 'hr', '人力资源', 3, '0', 'admin', sysdate(), '', null, ''); +insert into sys_post values(4, 'user', '普通员工', 4, '0', 'admin', sysdate(), '', null, ''); + + +-- ---------------------------- +-- 4、角色信息表 +-- ---------------------------- +drop table if exists sys_role; +create table sys_role ( + role_id bigint(20) not null auto_increment comment '角色ID', + role_name varchar(30) not null comment '角色名称', + role_key varchar(100) not null comment '角色权限字符串', + role_sort int(4) not null comment '显示顺序', + data_scope char(1) default '1' comment '数据范围(1:全部数据权限 2:自定数据权限 3:本部门数据权限 4:本部门及以下数据权限)', + menu_check_strictly tinyint(1) default 1 comment '菜单树选择项是否关联显示', + dept_check_strictly tinyint(1) default 1 comment '部门树选择项是否关联显示', + status char(1) not null comment '角色状态(0正常 1停用)', + del_flag char(1) default '0' comment '删除标志(0代表存在 2代表删除)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (role_id) +) engine=innodb auto_increment=100 comment = '角色信息表'; + +-- ---------------------------- +-- 初始化-角色信息表数据 +-- ---------------------------- +insert into sys_role values('1', '超级管理员', 'admin', 1, 1, 1, 1, '0', '0', 'admin', sysdate(), '', null, '超级管理员'); +insert into sys_role values('2', '普通角色', 'common', 2, 2, 1, 1, '0', '0', 'admin', sysdate(), '', null, '普通角色'); + + +-- ---------------------------- +-- 5、菜单权限表 +-- ---------------------------- +drop table if exists sys_menu; +create table sys_menu ( + menu_id bigint(20) not null auto_increment comment '菜单ID', + menu_name varchar(50) not null comment '菜单名称', + parent_id bigint(20) default 0 comment '父菜单ID', + order_num int(4) default 0 comment '显示顺序', + path varchar(200) default '' comment '路由地址', + component varchar(255) default null comment '组件路径', + query varchar(255) default null comment '路由参数', + is_frame int(1) default 1 comment '是否为外链(0是 1否)', + is_cache int(1) default 0 comment '是否缓存(0缓存 1不缓存)', + menu_type char(1) default '' comment '菜单类型(M目录 C菜单 F按钮)', + visible char(1) default 0 comment '菜单状态(0显示 1隐藏)', + status char(1) default 0 comment '菜单状态(0正常 1停用)', + perms varchar(100) default null comment '权限标识', + icon varchar(100) default '#' comment '菜单图标', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default '' comment '备注', + primary key (menu_id) +) engine=innodb auto_increment=2000 comment = '菜单权限表'; + +-- ---------------------------- +-- 初始化-菜单信息表数据 +-- ---------------------------- +-- 一级菜单 +insert into sys_menu values('1', '系统管理', '0', '1', 'system', null, '', 1, 0, 'M', '0', '0', '', 'system', 'admin', sysdate(), '', null, '系统管理目录'); +insert into sys_menu values('2', '系统监控', '0', '2', 'monitor', null, '', 1, 0, 'M', '0', '0', '', 'monitor', 'admin', sysdate(), '', null, '系统监控目录'); +insert into sys_menu values('3', '系统工具', '0', '3', 'tool', null, '', 1, 0, 'M', '0', '0', '', 'tool', 'admin', sysdate(), '', null, '系统工具目录'); +insert into sys_menu values('4', '若依官网', '0', '4', 'http://ruoyi.vip', null, '', 0, 0, 'M', '0', '0', '', 'guide', 'admin', sysdate(), '', null, '若依官网地址'); +-- 二级菜单 +insert into sys_menu values('100', '用户管理', '1', '1', 'user', 'system/user/index', '', 1, 0, 'C', '0', '0', 'system:user:list', 'user', 'admin', sysdate(), '', null, '用户管理菜单'); +insert into sys_menu values('101', '角色管理', '1', '2', 'role', 'system/role/index', '', 1, 0, 'C', '0', '0', 'system:role:list', 'peoples', 'admin', sysdate(), '', null, '角色管理菜单'); +insert into sys_menu values('102', '菜单管理', '1', '3', 'menu', 'system/menu/index', '', 1, 0, 'C', '0', '0', 'system:menu:list', 'tree-table', 'admin', sysdate(), '', null, '菜单管理菜单'); +insert into sys_menu values('103', '部门管理', '1', '4', 'dept', 'system/dept/index', '', 1, 0, 'C', '0', '0', 'system:dept:list', 'tree', 'admin', sysdate(), '', null, '部门管理菜单'); +insert into sys_menu values('104', '岗位管理', '1', '5', 'post', 'system/post/index', '', 1, 0, 'C', '0', '0', 'system:post:list', 'post', 'admin', sysdate(), '', null, '岗位管理菜单'); +insert into sys_menu values('105', '字典管理', '1', '6', 'dict', 'system/dict/index', '', 1, 0, 'C', '0', '0', 'system:dict:list', 'dict', 'admin', sysdate(), '', null, '字典管理菜单'); +insert into sys_menu values('106', '参数设置', '1', '7', 'config', 'system/config/index', '', 1, 0, 'C', '0', '0', 'system:config:list', 'edit', 'admin', sysdate(), '', null, '参数设置菜单'); +insert into sys_menu values('107', '通知公告', '1', '8', 'notice', 'system/notice/index', '', 1, 0, 'C', '0', '0', 'system:notice:list', 'message', 'admin', sysdate(), '', null, '通知公告菜单'); +insert into sys_menu values('108', '日志管理', '1', '9', 'log', '', '', 1, 0, 'M', '0', '0', '', 'log', 'admin', sysdate(), '', null, '日志管理菜单'); +insert into sys_menu values('109', '在线用户', '2', '1', 'online', 'monitor/online/index', '', 1, 0, 'C', '0', '0', 'monitor:online:list', 'online', 'admin', sysdate(), '', null, '在线用户菜单'); +insert into sys_menu values('110', '定时任务', '2', '2', 'job', 'monitor/job/index', '', 1, 0, 'C', '0', '0', 'monitor:job:list', 'job', 'admin', sysdate(), '', null, '定时任务菜单'); +insert into sys_menu values('111', '数据监控', '2', '3', 'druid', 'monitor/druid/index', '', 1, 0, 'C', '0', '0', 'monitor:druid:list', 'druid', 'admin', sysdate(), '', null, '数据监控菜单'); +insert into sys_menu values('112', '服务监控', '2', '4', 'server', 'monitor/server/index', '', 1, 0, 'C', '0', '0', 'monitor:server:list', 'server', 'admin', sysdate(), '', null, '服务监控菜单'); +insert into sys_menu values('113', '缓存监控', '2', '5', 'cache', 'monitor/cache/index', '', 1, 0, 'C', '0', '0', 'monitor:cache:list', 'redis', 'admin', sysdate(), '', null, '缓存监控菜单'); +insert into sys_menu values('114', '表单构建', '3', '1', 'build', 'tool/build/index', '', 1, 0, 'C', '0', '0', 'tool:build:list', 'build', 'admin', sysdate(), '', null, '表单构建菜单'); +insert into sys_menu values('115', '代码生成', '3', '2', 'gen', 'tool/gen/index', '', 1, 0, 'C', '0', '0', 'tool:gen:list', 'code', 'admin', sysdate(), '', null, '代码生成菜单'); +insert into sys_menu values('116', '系统接口', '3', '3', 'swagger', 'tool/swagger/index', '', 1, 0, 'C', '0', '0', 'tool:swagger:list', 'swagger', 'admin', sysdate(), '', null, '系统接口菜单'); +-- 三级菜单 +insert into sys_menu values('500', '操作日志', '108', '1', 'operlog', 'monitor/operlog/index', '', 1, 0, 'C', '0', '0', 'monitor:operlog:list', 'form', 'admin', sysdate(), '', null, '操作日志菜单'); +insert into sys_menu values('501', '登录日志', '108', '2', 'logininfor', 'monitor/logininfor/index', '', 1, 0, 'C', '0', '0', 'monitor:logininfor:list', 'logininfor', 'admin', sysdate(), '', null, '登录日志菜单'); +-- 用户管理按钮 +insert into sys_menu values('1001', '用户查询', '100', '1', '', '', '', 1, 0, 'F', '0', '0', 'system:user:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1002', '用户新增', '100', '2', '', '', '', 1, 0, 'F', '0', '0', 'system:user:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1003', '用户修改', '100', '3', '', '', '', 1, 0, 'F', '0', '0', 'system:user:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1004', '用户删除', '100', '4', '', '', '', 1, 0, 'F', '0', '0', 'system:user:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1005', '用户导出', '100', '5', '', '', '', 1, 0, 'F', '0', '0', 'system:user:export', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1006', '用户导入', '100', '6', '', '', '', 1, 0, 'F', '0', '0', 'system:user:import', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1007', '重置密码', '100', '7', '', '', '', 1, 0, 'F', '0', '0', 'system:user:resetPwd', '#', 'admin', sysdate(), '', null, ''); +-- 角色管理按钮 +insert into sys_menu values('1008', '角色查询', '101', '1', '', '', '', 1, 0, 'F', '0', '0', 'system:role:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1009', '角色新增', '101', '2', '', '', '', 1, 0, 'F', '0', '0', 'system:role:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1010', '角色修改', '101', '3', '', '', '', 1, 0, 'F', '0', '0', 'system:role:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1011', '角色删除', '101', '4', '', '', '', 1, 0, 'F', '0', '0', 'system:role:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1012', '角色导出', '101', '5', '', '', '', 1, 0, 'F', '0', '0', 'system:role:export', '#', 'admin', sysdate(), '', null, ''); +-- 菜单管理按钮 +insert into sys_menu values('1013', '菜单查询', '102', '1', '', '', '', 1, 0, 'F', '0', '0', 'system:menu:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1014', '菜单新增', '102', '2', '', '', '', 1, 0, 'F', '0', '0', 'system:menu:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1015', '菜单修改', '102', '3', '', '', '', 1, 0, 'F', '0', '0', 'system:menu:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1016', '菜单删除', '102', '4', '', '', '', 1, 0, 'F', '0', '0', 'system:menu:remove', '#', 'admin', sysdate(), '', null, ''); +-- 部门管理按钮 +insert into sys_menu values('1017', '部门查询', '103', '1', '', '', '', 1, 0, 'F', '0', '0', 'system:dept:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1018', '部门新增', '103', '2', '', '', '', 1, 0, 'F', '0', '0', 'system:dept:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1019', '部门修改', '103', '3', '', '', '', 1, 0, 'F', '0', '0', 'system:dept:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1020', '部门删除', '103', '4', '', '', '', 1, 0, 'F', '0', '0', 'system:dept:remove', '#', 'admin', sysdate(), '', null, ''); +-- 岗位管理按钮 +insert into sys_menu values('1021', '岗位查询', '104', '1', '', '', '', 1, 0, 'F', '0', '0', 'system:post:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1022', '岗位新增', '104', '2', '', '', '', 1, 0, 'F', '0', '0', 'system:post:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1023', '岗位修改', '104', '3', '', '', '', 1, 0, 'F', '0', '0', 'system:post:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1024', '岗位删除', '104', '4', '', '', '', 1, 0, 'F', '0', '0', 'system:post:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1025', '岗位导出', '104', '5', '', '', '', 1, 0, 'F', '0', '0', 'system:post:export', '#', 'admin', sysdate(), '', null, ''); +-- 字典管理按钮 +insert into sys_menu values('1026', '字典查询', '105', '1', '#', '', '', 1, 0, 'F', '0', '0', 'system:dict:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1027', '字典新增', '105', '2', '#', '', '', 1, 0, 'F', '0', '0', 'system:dict:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1028', '字典修改', '105', '3', '#', '', '', 1, 0, 'F', '0', '0', 'system:dict:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1029', '字典删除', '105', '4', '#', '', '', 1, 0, 'F', '0', '0', 'system:dict:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1030', '字典导出', '105', '5', '#', '', '', 1, 0, 'F', '0', '0', 'system:dict:export', '#', 'admin', sysdate(), '', null, ''); +-- 参数设置按钮 +insert into sys_menu values('1031', '参数查询', '106', '1', '#', '', '', 1, 0, 'F', '0', '0', 'system:config:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1032', '参数新增', '106', '2', '#', '', '', 1, 0, 'F', '0', '0', 'system:config:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1033', '参数修改', '106', '3', '#', '', '', 1, 0, 'F', '0', '0', 'system:config:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1034', '参数删除', '106', '4', '#', '', '', 1, 0, 'F', '0', '0', 'system:config:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1035', '参数导出', '106', '5', '#', '', '', 1, 0, 'F', '0', '0', 'system:config:export', '#', 'admin', sysdate(), '', null, ''); +-- 通知公告按钮 +insert into sys_menu values('1036', '公告查询', '107', '1', '#', '', '', 1, 0, 'F', '0', '0', 'system:notice:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1037', '公告新增', '107', '2', '#', '', '', 1, 0, 'F', '0', '0', 'system:notice:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1038', '公告修改', '107', '3', '#', '', '', 1, 0, 'F', '0', '0', 'system:notice:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1039', '公告删除', '107', '4', '#', '', '', 1, 0, 'F', '0', '0', 'system:notice:remove', '#', 'admin', sysdate(), '', null, ''); +-- 操作日志按钮 +insert into sys_menu values('1040', '操作查询', '500', '1', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:operlog:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1041', '操作删除', '500', '2', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:operlog:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1042', '日志导出', '500', '4', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:operlog:export', '#', 'admin', sysdate(), '', null, ''); +-- 登录日志按钮 +insert into sys_menu values('1043', '登录查询', '501', '1', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:logininfor:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1044', '登录删除', '501', '2', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:logininfor:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1045', '日志导出', '501', '3', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:logininfor:export', '#', 'admin', sysdate(), '', null, ''); +-- 在线用户按钮 +insert into sys_menu values('1046', '在线查询', '109', '1', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:online:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1047', '批量强退', '109', '2', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:online:batchLogout', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1048', '单条强退', '109', '3', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:online:forceLogout', '#', 'admin', sysdate(), '', null, ''); +-- 定时任务按钮 +insert into sys_menu values('1049', '任务查询', '110', '1', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1050', '任务新增', '110', '2', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:add', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1051', '任务修改', '110', '3', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1052', '任务删除', '110', '4', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1053', '状态修改', '110', '5', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:changeStatus', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1054', '任务导出', '110', '7', '#', '', '', 1, 0, 'F', '0', '0', 'monitor:job:export', '#', 'admin', sysdate(), '', null, ''); +-- 代码生成按钮 +insert into sys_menu values('1055', '生成查询', '115', '1', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:query', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1056', '生成修改', '115', '2', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:edit', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1057', '生成删除', '115', '3', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:remove', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1058', '导入代码', '115', '2', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:import', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1059', '预览代码', '115', '4', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:preview', '#', 'admin', sysdate(), '', null, ''); +insert into sys_menu values('1060', '生成代码', '115', '5', '#', '', '', 1, 0, 'F', '0', '0', 'tool:gen:code', '#', 'admin', sysdate(), '', null, ''); + + +-- ---------------------------- +-- 6、用户和角色关联表 用户N-1角色 +-- ---------------------------- +drop table if exists sys_user_role; +create table sys_user_role ( + user_id bigint(20) not null comment '用户ID', + role_id bigint(20) not null comment '角色ID', + primary key(user_id, role_id) +) engine=innodb comment = '用户和角色关联表'; + +-- ---------------------------- +-- 初始化-用户和角色关联表数据 +-- ---------------------------- +insert into sys_user_role values ('1', '1'); +insert into sys_user_role values ('2', '2'); + + +-- ---------------------------- +-- 7、角色和菜单关联表 角色1-N菜单 +-- ---------------------------- +drop table if exists sys_role_menu; +create table sys_role_menu ( + role_id bigint(20) not null comment '角色ID', + menu_id bigint(20) not null comment '菜单ID', + primary key(role_id, menu_id) +) engine=innodb comment = '角色和菜单关联表'; + +-- ---------------------------- +-- 初始化-角色和菜单关联表数据 +-- ---------------------------- +insert into sys_role_menu values ('2', '1'); +insert into sys_role_menu values ('2', '2'); +insert into sys_role_menu values ('2', '3'); +insert into sys_role_menu values ('2', '4'); +insert into sys_role_menu values ('2', '100'); +insert into sys_role_menu values ('2', '101'); +insert into sys_role_menu values ('2', '102'); +insert into sys_role_menu values ('2', '103'); +insert into sys_role_menu values ('2', '104'); +insert into sys_role_menu values ('2', '105'); +insert into sys_role_menu values ('2', '106'); +insert into sys_role_menu values ('2', '107'); +insert into sys_role_menu values ('2', '108'); +insert into sys_role_menu values ('2', '109'); +insert into sys_role_menu values ('2', '110'); +insert into sys_role_menu values ('2', '111'); +insert into sys_role_menu values ('2', '112'); +insert into sys_role_menu values ('2', '113'); +insert into sys_role_menu values ('2', '114'); +insert into sys_role_menu values ('2', '115'); +insert into sys_role_menu values ('2', '116'); +insert into sys_role_menu values ('2', '500'); +insert into sys_role_menu values ('2', '501'); +insert into sys_role_menu values ('2', '1000'); +insert into sys_role_menu values ('2', '1001'); +insert into sys_role_menu values ('2', '1002'); +insert into sys_role_menu values ('2', '1003'); +insert into sys_role_menu values ('2', '1004'); +insert into sys_role_menu values ('2', '1005'); +insert into sys_role_menu values ('2', '1006'); +insert into sys_role_menu values ('2', '1007'); +insert into sys_role_menu values ('2', '1008'); +insert into sys_role_menu values ('2', '1009'); +insert into sys_role_menu values ('2', '1010'); +insert into sys_role_menu values ('2', '1011'); +insert into sys_role_menu values ('2', '1012'); +insert into sys_role_menu values ('2', '1013'); +insert into sys_role_menu values ('2', '1014'); +insert into sys_role_menu values ('2', '1015'); +insert into sys_role_menu values ('2', '1016'); +insert into sys_role_menu values ('2', '1017'); +insert into sys_role_menu values ('2', '1018'); +insert into sys_role_menu values ('2', '1019'); +insert into sys_role_menu values ('2', '1020'); +insert into sys_role_menu values ('2', '1021'); +insert into sys_role_menu values ('2', '1022'); +insert into sys_role_menu values ('2', '1023'); +insert into sys_role_menu values ('2', '1024'); +insert into sys_role_menu values ('2', '1025'); +insert into sys_role_menu values ('2', '1026'); +insert into sys_role_menu values ('2', '1027'); +insert into sys_role_menu values ('2', '1028'); +insert into sys_role_menu values ('2', '1029'); +insert into sys_role_menu values ('2', '1030'); +insert into sys_role_menu values ('2', '1031'); +insert into sys_role_menu values ('2', '1032'); +insert into sys_role_menu values ('2', '1033'); +insert into sys_role_menu values ('2', '1034'); +insert into sys_role_menu values ('2', '1035'); +insert into sys_role_menu values ('2', '1036'); +insert into sys_role_menu values ('2', '1037'); +insert into sys_role_menu values ('2', '1038'); +insert into sys_role_menu values ('2', '1039'); +insert into sys_role_menu values ('2', '1040'); +insert into sys_role_menu values ('2', '1041'); +insert into sys_role_menu values ('2', '1042'); +insert into sys_role_menu values ('2', '1043'); +insert into sys_role_menu values ('2', '1044'); +insert into sys_role_menu values ('2', '1045'); +insert into sys_role_menu values ('2', '1046'); +insert into sys_role_menu values ('2', '1047'); +insert into sys_role_menu values ('2', '1048'); +insert into sys_role_menu values ('2', '1049'); +insert into sys_role_menu values ('2', '1050'); +insert into sys_role_menu values ('2', '1051'); +insert into sys_role_menu values ('2', '1052'); +insert into sys_role_menu values ('2', '1053'); +insert into sys_role_menu values ('2', '1054'); +insert into sys_role_menu values ('2', '1055'); +insert into sys_role_menu values ('2', '1056'); +insert into sys_role_menu values ('2', '1057'); +insert into sys_role_menu values ('2', '1058'); +insert into sys_role_menu values ('2', '1059'); +insert into sys_role_menu values ('2', '1060'); + +-- ---------------------------- +-- 8、角色和部门关联表 角色1-N部门 +-- ---------------------------- +drop table if exists sys_role_dept; +create table sys_role_dept ( + role_id bigint(20) not null comment '角色ID', + dept_id bigint(20) not null comment '部门ID', + primary key(role_id, dept_id) +) engine=innodb comment = '角色和部门关联表'; + +-- ---------------------------- +-- 初始化-角色和部门关联表数据 +-- ---------------------------- +insert into sys_role_dept values ('2', '100'); +insert into sys_role_dept values ('2', '101'); +insert into sys_role_dept values ('2', '105'); + + +-- ---------------------------- +-- 9、用户与岗位关联表 用户1-N岗位 +-- ---------------------------- +drop table if exists sys_user_post; +create table sys_user_post +( + user_id bigint(20) not null comment '用户ID', + post_id bigint(20) not null comment '岗位ID', + primary key (user_id, post_id) +) engine=innodb comment = '用户与岗位关联表'; + +-- ---------------------------- +-- 初始化-用户与岗位关联表数据 +-- ---------------------------- +insert into sys_user_post values ('1', '1'); +insert into sys_user_post values ('2', '2'); + + +-- ---------------------------- +-- 10、操作日志记录 +-- ---------------------------- +drop table if exists sys_oper_log; +create table sys_oper_log ( + oper_id bigint(20) not null auto_increment comment '日志主键', + title varchar(50) default '' comment '模块标题', + business_type int(2) default 0 comment '业务类型(0其它 1新增 2修改 3删除)', + method varchar(100) default '' comment '方法名称', + request_method varchar(10) default '' comment '请求方式', + operator_type int(1) default 0 comment '操作类别(0其它 1后台用户 2手机端用户)', + oper_name varchar(50) default '' comment '操作人员', + dept_name varchar(50) default '' comment '部门名称', + oper_url varchar(255) default '' comment '请求URL', + oper_ip varchar(128) default '' comment '主机地址', + oper_location varchar(255) default '' comment '操作地点', + oper_param varchar(2000) default '' comment '请求参数', + json_result varchar(2000) default '' comment '返回参数', + status int(1) default 0 comment '操作状态(0正常 1异常)', + error_msg varchar(2000) default '' comment '错误消息', + oper_time datetime comment '操作时间', + primary key (oper_id) +) engine=innodb auto_increment=100 comment = '操作日志记录'; + + +-- ---------------------------- +-- 11、字典类型表 +-- ---------------------------- +drop table if exists sys_dict_type; +create table sys_dict_type +( + dict_id bigint(20) not null auto_increment comment '字典主键', + dict_name varchar(100) default '' comment '字典名称', + dict_type varchar(100) default '' comment '字典类型', + status char(1) default '0' comment '状态(0正常 1停用)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (dict_id), + unique (dict_type) +) engine=innodb auto_increment=100 comment = '字典类型表'; + +insert into sys_dict_type values(1, '用户性别', 'sys_user_sex', '0', 'admin', sysdate(), '', null, '用户性别列表'); +insert into sys_dict_type values(2, '菜单状态', 'sys_show_hide', '0', 'admin', sysdate(), '', null, '菜单状态列表'); +insert into sys_dict_type values(3, '系统开关', 'sys_normal_disable', '0', 'admin', sysdate(), '', null, '系统开关列表'); +insert into sys_dict_type values(4, '任务状态', 'sys_job_status', '0', 'admin', sysdate(), '', null, '任务状态列表'); +insert into sys_dict_type values(5, '任务分组', 'sys_job_group', '0', 'admin', sysdate(), '', null, '任务分组列表'); +insert into sys_dict_type values(6, '系统是否', 'sys_yes_no', '0', 'admin', sysdate(), '', null, '系统是否列表'); +insert into sys_dict_type values(7, '通知类型', 'sys_notice_type', '0', 'admin', sysdate(), '', null, '通知类型列表'); +insert into sys_dict_type values(8, '通知状态', 'sys_notice_status', '0', 'admin', sysdate(), '', null, '通知状态列表'); +insert into sys_dict_type values(9, '操作类型', 'sys_oper_type', '0', 'admin', sysdate(), '', null, '操作类型列表'); +insert into sys_dict_type values(10, '系统状态', 'sys_common_status', '0', 'admin', sysdate(), '', null, '登录状态列表'); + + +-- ---------------------------- +-- 12、字典数据表 +-- ---------------------------- +drop table if exists sys_dict_data; +create table sys_dict_data +( + dict_code bigint(20) not null auto_increment comment '字典编码', + dict_sort int(4) default 0 comment '字典排序', + dict_label varchar(100) default '' comment '字典标签', + dict_value varchar(100) default '' comment '字典键值', + dict_type varchar(100) default '' comment '字典类型', + css_class varchar(100) default null comment '样式属性(其他样式扩展)', + list_class varchar(100) default null comment '表格回显样式', + is_default char(1) default 'N' comment '是否默认(Y是 N否)', + status char(1) default '0' comment '状态(0正常 1停用)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (dict_code) +) engine=innodb auto_increment=100 comment = '字典数据表'; + +insert into sys_dict_data values(1, 1, '男', '0', 'sys_user_sex', '', '', 'Y', '0', 'admin', sysdate(), '', null, '性别男'); +insert into sys_dict_data values(2, 2, '女', '1', 'sys_user_sex', '', '', 'N', '0', 'admin', sysdate(), '', null, '性别女'); +insert into sys_dict_data values(3, 3, '未知', '2', 'sys_user_sex', '', '', 'N', '0', 'admin', sysdate(), '', null, '性别未知'); +insert into sys_dict_data values(4, 1, '显示', '0', 'sys_show_hide', '', 'primary', 'Y', '0', 'admin', sysdate(), '', null, '显示菜单'); +insert into sys_dict_data values(5, 2, '隐藏', '1', 'sys_show_hide', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '隐藏菜单'); +insert into sys_dict_data values(6, 1, '正常', '0', 'sys_normal_disable', '', 'primary', 'Y', '0', 'admin', sysdate(), '', null, '正常状态'); +insert into sys_dict_data values(7, 2, '停用', '1', 'sys_normal_disable', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '停用状态'); +insert into sys_dict_data values(8, 1, '正常', '0', 'sys_job_status', '', 'primary', 'Y', '0', 'admin', sysdate(), '', null, '正常状态'); +insert into sys_dict_data values(9, 2, '暂停', '1', 'sys_job_status', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '停用状态'); +insert into sys_dict_data values(10, 1, '默认', 'DEFAULT', 'sys_job_group', '', '', 'Y', '0', 'admin', sysdate(), '', null, '默认分组'); +insert into sys_dict_data values(11, 2, '系统', 'SYSTEM', 'sys_job_group', '', '', 'N', '0', 'admin', sysdate(), '', null, '系统分组'); +insert into sys_dict_data values(12, 1, '是', 'Y', 'sys_yes_no', '', 'primary', 'Y', '0', 'admin', sysdate(), '', null, '系统默认是'); +insert into sys_dict_data values(13, 2, '否', 'N', 'sys_yes_no', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '系统默认否'); +insert into sys_dict_data values(14, 1, '通知', '1', 'sys_notice_type', '', 'warning', 'Y', '0', 'admin', sysdate(), '', null, '通知'); +insert into sys_dict_data values(15, 2, '公告', '2', 'sys_notice_type', '', 'success', 'N', '0', 'admin', sysdate(), '', null, '公告'); +insert into sys_dict_data values(16, 1, '正常', '0', 'sys_notice_status', '', 'primary', 'Y', '0', 'admin', sysdate(), '', null, '正常状态'); +insert into sys_dict_data values(17, 2, '关闭', '1', 'sys_notice_status', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '关闭状态'); +insert into sys_dict_data values(18, 1, '新增', '1', 'sys_oper_type', '', 'info', 'N', '0', 'admin', sysdate(), '', null, '新增操作'); +insert into sys_dict_data values(19, 2, '修改', '2', 'sys_oper_type', '', 'info', 'N', '0', 'admin', sysdate(), '', null, '修改操作'); +insert into sys_dict_data values(20, 3, '删除', '3', 'sys_oper_type', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '删除操作'); +insert into sys_dict_data values(21, 4, '授权', '4', 'sys_oper_type', '', 'primary', 'N', '0', 'admin', sysdate(), '', null, '授权操作'); +insert into sys_dict_data values(22, 5, '导出', '5', 'sys_oper_type', '', 'warning', 'N', '0', 'admin', sysdate(), '', null, '导出操作'); +insert into sys_dict_data values(23, 6, '导入', '6', 'sys_oper_type', '', 'warning', 'N', '0', 'admin', sysdate(), '', null, '导入操作'); +insert into sys_dict_data values(24, 7, '强退', '7', 'sys_oper_type', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '强退操作'); +insert into sys_dict_data values(25, 8, '生成代码', '8', 'sys_oper_type', '', 'warning', 'N', '0', 'admin', sysdate(), '', null, '生成操作'); +insert into sys_dict_data values(26, 9, '清空数据', '9', 'sys_oper_type', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '清空操作'); +insert into sys_dict_data values(27, 1, '成功', '0', 'sys_common_status', '', 'primary', 'N', '0', 'admin', sysdate(), '', null, '正常状态'); +insert into sys_dict_data values(28, 2, '失败', '1', 'sys_common_status', '', 'danger', 'N', '0', 'admin', sysdate(), '', null, '停用状态'); + + +-- ---------------------------- +-- 13、参数配置表 +-- ---------------------------- +drop table if exists sys_config; +create table sys_config ( + config_id int(5) not null auto_increment comment '参数主键', + config_name varchar(100) default '' comment '参数名称', + config_key varchar(100) default '' comment '参数键名', + config_value varchar(500) default '' comment '参数键值', + config_type char(1) default 'N' comment '系统内置(Y是 N否)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (config_id) +) engine=innodb auto_increment=100 comment = '参数配置表'; + +insert into sys_config values(1, '主框架页-默认皮肤样式名称', 'sys.index.skinName', 'skin-blue', 'Y', 'admin', sysdate(), '', null, '蓝色 skin-blue、绿色 skin-green、紫色 skin-purple、红色 skin-red、黄色 skin-yellow' ); +insert into sys_config values(2, '用户管理-账号初始密码', 'sys.user.initPassword', '123456', 'Y', 'admin', sysdate(), '', null, '初始化密码 123456' ); +insert into sys_config values(3, '主框架页-侧边栏主题', 'sys.index.sideTheme', 'theme-dark', 'Y', 'admin', sysdate(), '', null, '深色主题theme-dark,浅色主题theme-light' ); +insert into sys_config values(4, '账号自助-验证码开关', 'sys.account.captchaOnOff', 'true', 'Y', 'admin', sysdate(), '', null, '是否开启验证码功能(true开启,false关闭)'); +insert into sys_config values(5, '账号自助-是否开启用户注册功能', 'sys.account.registerUser', 'false', 'Y', 'admin', sysdate(), '', null, '是否开启注册用户功能(true开启,false关闭)'); + + +-- ---------------------------- +-- 14、系统访问记录 +-- ---------------------------- +drop table if exists sys_logininfor; +create table sys_logininfor ( + info_id bigint(20) not null auto_increment comment '访问ID', + user_name varchar(50) default '' comment '用户账号', + ipaddr varchar(128) default '' comment '登录IP地址', + login_location varchar(255) default '' comment '登录地点', + browser varchar(50) default '' comment '浏览器类型', + os varchar(50) default '' comment '操作系统', + status char(1) default '0' comment '登录状态(0成功 1失败)', + msg varchar(255) default '' comment '提示消息', + login_time datetime comment '访问时间', + primary key (info_id) +) engine=innodb auto_increment=100 comment = '系统访问记录'; + + +-- ---------------------------- +-- 15、定时任务调度表 +-- ---------------------------- +drop table if exists sys_job; +create table sys_job ( + job_id bigint(20) not null auto_increment comment '任务ID', + job_name varchar(64) default '' comment '任务名称', + job_group varchar(64) default 'DEFAULT' comment '任务组名', + invoke_target varchar(500) not null comment '调用目标字符串', + cron_expression varchar(255) default '' comment 'cron执行表达式', + misfire_policy varchar(20) default '3' comment '计划执行错误策略(1立即执行 2执行一次 3放弃执行)', + concurrent char(1) default '1' comment '是否并发执行(0允许 1禁止)', + status char(1) default '0' comment '状态(0正常 1暂停)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default '' comment '备注信息', + primary key (job_id, job_name, job_group) +) engine=innodb auto_increment=100 comment = '定时任务调度表'; + +insert into sys_job values(1, '系统默认(无参)', 'DEFAULT', 'ryTask.ryNoParams', '0/10 * * * * ?', '3', '1', '1', 'admin', sysdate(), '', null, ''); +insert into sys_job values(2, '系统默认(有参)', 'DEFAULT', 'ryTask.ryParams(\'ry\')', '0/15 * * * * ?', '3', '1', '1', 'admin', sysdate(), '', null, ''); +insert into sys_job values(3, '系统默认(多参)', 'DEFAULT', 'ryTask.ryMultipleParams(\'ry\', true, 2000L, 316.50D, 100)', '0/20 * * * * ?', '3', '1', '1', 'admin', sysdate(), '', null, ''); + + +-- ---------------------------- +-- 16、定时任务调度日志表 +-- ---------------------------- +drop table if exists sys_job_log; +create table sys_job_log ( + job_log_id bigint(20) not null auto_increment comment '任务日志ID', + job_name varchar(64) not null comment '任务名称', + job_group varchar(64) not null comment '任务组名', + invoke_target varchar(500) not null comment '调用目标字符串', + job_message varchar(500) comment '日志信息', + status char(1) default '0' comment '执行状态(0正常 1失败)', + exception_info varchar(2000) default '' comment '异常信息', + create_time datetime comment '创建时间', + primary key (job_log_id) +) engine=innodb comment = '定时任务调度日志表'; + + +-- ---------------------------- +-- 17、通知公告表 +-- ---------------------------- +drop table if exists sys_notice; +create table sys_notice ( + notice_id int(4) not null auto_increment comment '公告ID', + notice_title varchar(50) not null comment '公告标题', + notice_type char(1) not null comment '公告类型(1通知 2公告)', + notice_content longblob default null comment '公告内容', + status char(1) default '0' comment '公告状态(0正常 1关闭)', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(255) default null comment '备注', + primary key (notice_id) +) engine=innodb auto_increment=10 comment = '通知公告表'; + +-- ---------------------------- +-- 初始化-公告信息表数据 +-- ---------------------------- +insert into sys_notice values('1', '温馨提醒:2018-07-01 若依新版本发布啦', '2', '新版本内容', '0', 'admin', sysdate(), '', null, '管理员'); +insert into sys_notice values('2', '维护通知:2018-07-01 若依系统凌晨维护', '1', '维护内容', '0', 'admin', sysdate(), '', null, '管理员'); + + +-- ---------------------------- +-- 18、代码生成业务表 +-- ---------------------------- +drop table if exists gen_table; +create table gen_table ( + table_id bigint(20) not null auto_increment comment '编号', + table_name varchar(200) default '' comment '表名称', + table_comment varchar(500) default '' comment '表描述', + sub_table_name varchar(64) default null comment '关联子表的表名', + sub_table_fk_name varchar(64) default null comment '子表关联的外键名', + class_name varchar(100) default '' comment '实体类名称', + tpl_category varchar(200) default 'crud' comment '使用的模板(crud单表操作 tree树表操作)', + package_name varchar(100) comment '生成包路径', + module_name varchar(30) comment '生成模块名', + business_name varchar(30) comment '生成业务名', + function_name varchar(50) comment '生成功能名', + function_author varchar(50) comment '生成功能作者', + gen_type char(1) default '0' comment '生成代码方式(0zip压缩包 1自定义路径)', + gen_path varchar(200) default '/' comment '生成路径(不填默认项目路径)', + options varchar(1000) comment '其它生成选项', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + remark varchar(500) default null comment '备注', + primary key (table_id) +) engine=innodb auto_increment=1 comment = '代码生成业务表'; + + +-- ---------------------------- +-- 19、代码生成业务表字段 +-- ---------------------------- +drop table if exists gen_table_column; +create table gen_table_column ( + column_id bigint(20) not null auto_increment comment '编号', + table_id varchar(64) comment '归属表编号', + column_name varchar(200) comment '列名称', + column_comment varchar(500) comment '列描述', + column_type varchar(100) comment '列类型', + java_type varchar(500) comment 'JAVA类型', + java_field varchar(200) comment 'JAVA字段名', + is_pk char(1) comment '是否主键(1是)', + is_increment char(1) comment '是否自增(1是)', + is_required char(1) comment '是否必填(1是)', + is_insert char(1) comment '是否为插入字段(1是)', + is_edit char(1) comment '是否编辑字段(1是)', + is_list char(1) comment '是否列表字段(1是)', + is_query char(1) comment '是否查询字段(1是)', + query_type varchar(200) default 'EQ' comment '查询方式(等于、不等于、大于、小于、范围)', + html_type varchar(200) comment '显示类型(文本框、文本域、下拉框、复选框、单选框、日期控件)', + dict_type varchar(200) default '' comment '字典类型', + sort int comment '排序', + create_by varchar(64) default '' comment '创建者', + create_time datetime comment '创建时间', + update_by varchar(64) default '' comment '更新者', + update_time datetime comment '更新时间', + primary key (column_id) +) engine=innodb auto_increment=1 comment = '代码生成业务表字段'; \ No newline at end of file diff --git a/swlscx/pom.xml b/swlscx/pom.xml new file mode 100644 index 0000000..3e61d3a --- /dev/null +++ b/swlscx/pom.xml @@ -0,0 +1,33 @@ + + + 4.0.0 + + com.ruoyi + ruoyi + 3.8.1 + + + swlscx + + + 11 + 11 + UTF-8 + + + + + + com.ruoyi + ruoyi-common + + + + org.springframework.boot + spring-boot-test + + org.apache.maven.pluginsmaven-compiler-plugin88 + + \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseExportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseExportController.java new file mode 100644 index 0000000..ad11c6b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseExportController.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.basic.controller; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.service.HyStscAService; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/9/27 15:16 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/report") +@RequiredArgsConstructor +public class BaseExportController { + + private final HyStscAService hyStscAService; + + + /** + * 导出测站一览表 + */ + @RequestMapping("/station") + public R exportStation(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyStscAService.exportStation(response,startTime,endTime,stcd,stnm); + } +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseInfoController.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseInfoController.java new file mode 100644 index 0000000..06ab7aa --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/controller/BaseInfoController.java @@ -0,0 +1,68 @@ +package com.ruoyi.swlscx.basic.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.basic.service.HyStscAService; +import lombok.RequiredArgsConstructor; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletResponse; +import java.util.Map; + +/** + * @Author al + * @Date 2024/8/4 14:36 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/basic") +@RequiredArgsConstructor +public class BaseInfoController { + + private final HyStscAService hyStscAService; + + private final YcExportTaskService ycExportTaskService; + + /** + * 查询站点信息 + */ + @GetMapping("/getStationPageByInfo") + public R getStationPageByInfo(Integer pageNum,Integer pageSize, String stnm,String stcd, String startTime, String endTime) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage stationPageByInfo = hyStscAService.getStationPageByInfo(pageParams, stnm, stcd, startTime, endTime); + return R.ok().put("data", stationPageByInfo.getRecords()).put("count", stationPageByInfo.getTotal()); + } + + /** + * 查询列表 + */ + @RequestMapping("/cgtask/list") + public R list(@RequestParam Map params){ + return ycExportTaskService.queryPage(params); + } + + @RequestMapping("/downloadExcel/{id}") + public void batchDownload(HttpServletResponse response, @PathVariable("id") Long id){ + YcExportTask t = ycExportTaskService.getById(id); + ycExportTaskService.downloadExcel(response,t); + } + + @DeleteMapping("/cgtask/delete/{id}") + public R delete(@PathVariable("id")Integer id){ + return ycExportTaskService.removeTask(id); + } + + + + @PostMapping("/importDateToSoft") + public R importDateToSoft(String startTime, String endTime){ + return hyStscAService.importDateTosoft(startTime,endTime); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/bo/HyStscABo.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/bo/HyStscABo.java new file mode 100644 index 0000000..aabc759 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/bo/HyStscABo.java @@ -0,0 +1,107 @@ +package com.ruoyi.swlscx.basic.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; + +/** + * 测站一览表 + * @Author al + * @Date 2024/7/30 13:52 + * @Description: TODO + * @Version + */ + +@Data +public class HyStscABo implements Serializable { + + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + @ExcelProperty(value = "站名",index = 2) + private String stnm; + + + private String stct; + + @ExcelProperty(value = "流域名称",index = 3) + private String bshncd; + + @ExcelProperty(value = "水系名称",index = 4) + private String hnnm; + + + @ExcelProperty(value = "河流名称",index = 5) + private String rvnm; + + @ExcelProperty(value = "施测项目码",index = 6) + private String obitmcd; + + @ExcelProperty(value = "行政区划码",index = 7) + private String addvcd; + + @ExcelProperty(value = "水资源分区码",index = 8) + private String wrrgcd; + + @ExcelProperty(value = "设站年份",index = 9) + private String esstyr; + + + @ExcelProperty(value = "设站月份",index = 10) + private String esstmth; + + @ExcelProperty(value = "撤站年份",index = 11) + private String wdstyr; + + @ExcelProperty(value = "撤站月份",index = 12) + private String wdstmth; + + @ExcelProperty(value = "集水面积",index = 13) + private String drar; + + + @ExcelProperty(value = "流入何处",index = 14) + private String flto; + + + @ExcelProperty(value = "至河口距离",index = 15) + private String dstrvm; + + + @ExcelProperty(value = "基准基面名称",index = 16) + private String fdtmnm; + + + @ExcelProperty(value = "领导机关",index = 17) + private String admag; + + + @ExcelProperty(value = "管理单位",index = 18) + private String admnst; + + + @ExcelProperty(value = "站址",index = 19) + private String stlc; + + + @ExcelProperty(value = "东经",index = 20) + private String lgtd; + + + @ExcelProperty(value = "北纬",index = 21) + private String lttd; + + @ExcelProperty(value = "测站等级",index = 22) + private String stgrd; + + @ExcelProperty(value = "报汛等级",index = 23) + private String frgrd; + + @ExcelProperty(value = "备注",index = 24) + private String nt; + + private static final long serialVersionUID = 1L; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/HyStscA.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/HyStscA.java new file mode 100644 index 0000000..3cd620f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/HyStscA.java @@ -0,0 +1,68 @@ +package com.ruoyi.swlscx.basic.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import lombok.Data; + +/** + * @TableName 测站一览表 + */ +@TableName(value ="hy_stsc_a") +@Data +public class HyStscA implements Serializable { + private String stcd; + + private String stnm; + + private String stct; + + private String bshncd; + + private String hnnm; + + private String rvnm; + + private String obitmcd; + + private String addvcd; + + private String wrrgcd; + + private Integer esstyr; + + private Integer esstmth; + + private Integer wdstyr; + + private Integer wdstmth; + + private BigDecimal drar; + + private String flto; + + private BigDecimal dstrvm; + + private String fdtmnm; + + private String admag; + + private String admnst; + + private String stlc; + + private BigDecimal lgtd; + + private BigDecimal lttd; + + private String stgrd; + + private String frgrd; + + private String nt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/YcExportTask.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/YcExportTask.java new file mode 100644 index 0000000..2825490 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/po/YcExportTask.java @@ -0,0 +1,54 @@ +package com.ruoyi.swlscx.basic.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.util.Date; +import lombok.Data; + +/** + * 导出任务 + * @TableName yc_export_task + */ +@TableName(value ="yc_export_task") +@Data +public class YcExportTask implements Serializable { + /** + * + */ + @TableId(value = "id", type = IdType.AUTO) + private Long id; + + /** + * 开始时间 + */ + private Date startTime; + + /** + * 结束时间 + */ + private Date endTime; + + /** + * 状态 0 进行中 1 完成 + */ + private Integer status; + + + /** + * 文件名称 + */ + private String filename; + + /** + * + */ + private Long userId; + + private String filepath; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/vo/HyStscAVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/vo/HyStscAVo.java new file mode 100644 index 0000000..808b014 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/domain/vo/HyStscAVo.java @@ -0,0 +1,105 @@ +package com.ruoyi.swlscx.basic.domain.vo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; + +/** + * 测站一览表 + * @Author al + * @Date 2024/7/30 13:52 + * @Description: TODO + * @Version + */ + +@Data +public class HyStscAVo implements Serializable { + + /** 站码 */ + private String stcd; + + /**站名**/ + private String stnm; + + + /**流域名称**/ + private String bshncd; + + /**水系名称**/ + private String hnnm; + + + /**河流名称**/ + private String rvnm; + + /**施测项目码**/ + private String obitmcd; + + /**行政区划码**/ + private String addvcd; + + /**水资源分区码 **/ + private String wrrgcd; + + /**设站年份**/ + private String esstyr; + + + /**设站月份**/ + private String esstmth; + + /**撤站年份**/ + private String wdstyr; + + /**撤站月份**/ + private String wdstmth; + + /**集水面积**/ + private String drar; + + + /**流入何处**/ + private String flto; + + + /**至河口距离**/ + private String dstrvm; + + + /**基准基面名称**/ + private String fdtmnm; + + + /**领导机关**/ + private String admag; + + + /**管理单位**/ + private String admnst; + + + /**站址**/ + private String stlc; + + + /**东经**/ + private BigDecimal lgtd; + + + /**北纬**/ + private BigDecimal lttd; + + /**测站等级**/ + private String stgrd; + + /**报汛等级**/ + private String frgrd; + + /**备注**/ + private String nt; + + private static final long serialVersionUID = 1L; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/HyStscAMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/HyStscAMapper.java new file mode 100644 index 0000000..b656999 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/HyStscAMapper.java @@ -0,0 +1,40 @@ +package com.ruoyi.swlscx.basic.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_stsc_a】的数据库操作Mapper +* @createDate 2024-07-30 13:37:07 +* @Entity com.ruoyi.swlscx.domain.po.po.HyStscA +*/ + +@Mapper +public interface HyStscAMapper extends BaseMapper { + + /** + * 查询站点信息 + */ + IPage selectHyStscAPageByInfo(Page page, @Param("map") Map map); + + + + void deleteTempData(); + + void mergeToData(); + + void insertTempData(@Param("list") List records); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/YcExportTaskMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/YcExportTaskMapper.java new file mode 100644 index 0000000..4769bac --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/mapper/YcExportTaskMapper.java @@ -0,0 +1,27 @@ +package com.ruoyi.swlscx.basic.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【yc_export_task(导出任务)】的数据库操作Mapper +* @createDate 2024-10-23 13:57:23 +* @Entity com.ruoyi.swlscx.basic.domain.po.YcExportTask +*/ +public interface YcExportTaskMapper extends BaseMapper { + + IPage queryPage(Page page,@Param("map") Map params); + + List> test(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/HyStscAService.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/HyStscAService.java new file mode 100644 index 0000000..05f11bd --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/HyStscAService.java @@ -0,0 +1,42 @@ +package com.ruoyi.swlscx.basic.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_stsc_a】的数据库操作Service +* @createDate 2024-07-30 13:37:07 +*/ +public interface HyStscAService extends IService { + + /** + * 导入站点信息数据 + */ + void importHyStsc(MultipartFile file); + + /** + * 查询站点信息 + */ + IPage getStationPageByInfo(PageParams pageParams, String stnm, String stcd, String startTime, String endTime); + + /** + * 导出测站一览表 + */ + R exportStation(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); + + + /** + * 导出测站一览表 + */ + R testSqlServer(); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/YcExportTaskService.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/YcExportTaskService.java new file mode 100644 index 0000000..79318ac --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/YcExportTaskService.java @@ -0,0 +1,26 @@ +package com.ruoyi.swlscx.basic.service; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.baomidou.mybatisplus.extension.service.IService; + +import javax.servlet.http.HttpServletResponse; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【yc_export_task(导出任务)】的数据库操作Service +* @createDate 2024-10-23 13:57:23 +*/ +public interface YcExportTaskService extends IService { + + void addTask(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R queryPage(Map params); + + void downloadExcel(HttpServletResponse response, YcExportTask t); + + R removeTask(Integer id); + + R getTest(); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/HyStscAServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/HyStscAServiceImpl.java new file mode 100644 index 0000000..1b984f7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/HyStscAServiceImpl.java @@ -0,0 +1,349 @@ +package com.ruoyi.swlscx.basic.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.basic.domain.bo.HyStscABo; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import com.ruoyi.swlscx.basic.mapper.HyStscAMapper; +import com.ruoyi.swlscx.basic.service.HyStscAService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_stsc_a】的数据库操作Service实现 +* @createDate 2024-07-30 13:37:07 +*/ +@Service +public class HyStscAServiceImpl extends ServiceImpl + implements HyStscAService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + + + + + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyStsc(MultipartFile file) { + try { + List hyStationBoList = ExcelUtils.readExcel(file.getInputStream(), HyStscABo.class, 2); + + List hyStationList = hyStationBoList.stream().map(i -> { + HyStscA hyStscA = new HyStscA(); + BeanUtils.copyProperties(i, hyStscA); + hyStscA.setDstrvm(Convert.toBigDecimal(i.getDstrvm())); + hyStscA.setDrar(Convert.toBigDecimal(i.getDrar())); + hyStscA.setLgtd(Convert.toBigDecimal(i.getLgtd())); + hyStscA.setLttd(Convert.toBigDecimal(i.getLttd())); + hyStscA.setEsstmth(Convert.toInt(i.getEsstmth())); + hyStscA.setEsstyr(Convert.toInt(i.getEsstyr())); + hyStscA.setWdstmth(Convert.toInt(i.getWdstmth())); + hyStscA.setEsstmth(Convert.toInt(i.getEsstmth())); + return hyStscA; + }).collect(Collectors.toList()); + this.saveBatch(hyStationList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询站点信息 + */ + @Override + @DataSource(DataSourceType.MASTER) + public IPage getStationPageByInfo(PageParams pageParams, String stnm,String stcd , String startTime, String endTime) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyStscAPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出测站一览表 + */ + @Override + public R exportStation(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "测站一览表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyStscAVoList) { + List records = hyStscAVoList.stream().map(i->{ + HyStscA hyStscA = new HyStscA(); + BeanUtils.copyProperties(i, hyStscA); + hyStscA.setDstrvm(Convert.toBigDecimal(i.getDstrvm())); + hyStscA.setDrar(Convert.toBigDecimal(i.getDrar())); + hyStscA.setLgtd(Convert.toBigDecimal(i.getLgtd())); + hyStscA.setLttd(Convert.toBigDecimal(i.getLttd())); + hyStscA.setEsstmth(Convert.toInt(i.getEsstmth())); + hyStscA.setEsstyr(Convert.toInt(i.getEsstyr())); + hyStscA.setWdstmth(Convert.toInt(i.getWdstmth())); + hyStscA.setEsstmth(Convert.toInt(i.getEsstmth())); + return hyStscA; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "测站一览表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "测站一览表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage stationPageByInfo = this.getStationPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyStscAServiceImpl hyStscAServiceImpl = context.getBean(HyStscAServiceImpl.class); + hyStscAServiceImpl.importMysqlToSqlserver(stationPageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + @Override + @DataSource(DataSourceType.SHARDING) + public R testSqlServer() { + List list = this.list().subList(0,100); + return R.ok().put("data",list); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("station.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyStscAVoIPage = this.baseMapper.selectHyStscAPageByInfo(new Query().getPage(map), map); + List records = hyStscAVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyStscAVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("测站一览表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getBshncd() != null ? vo.getBshncd() : "", style); + ExcelUtils.formatCell(row, 4, vo.getHnnm() != null ? vo.getHnnm() : "", style); + ExcelUtils.formatCell(row, 5, vo.getRvnm() != null ? vo.getRvnm() : "", style); + ExcelUtils.formatCell(row, 6, vo.getObitmcd() != null ? vo.getObitmcd() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getWrrgcd() != null ? vo.getWrrgcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getEsstyr() != null ? vo.getEsstyr() : "", style); + ExcelUtils.formatCell(row, 10, vo.getEsstmth() != null ? vo.getEsstmth() : "", style); + ExcelUtils.formatCell(row, 11, vo.getWdstyr() != null ? vo.getWdstyr() : "", style); + ExcelUtils.formatCell(row, 12, vo.getWdstmth() != null ? vo.getWdstmth() : "", style); + ExcelUtils.formatCell(row, 13, vo.getDrar() != null ? vo.getDrar() : "", style); + ExcelUtils.formatCell(row, 14, vo.getFlto() != null ? vo.getFlto() : "", style); + ExcelUtils.formatCell(row, 15, vo.getDstrvm() != null ? vo.getDstrvm() : "", style); + ExcelUtils.formatCell(row, 16, vo.getAdmag() != null ? vo.getAdmag() : "", style); + ExcelUtils.formatCell(row, 17, vo.getAdmnst() != null ? vo.getAdmnst() : "", style); + ExcelUtils.formatCell(row, 18, vo.getStlc() != null ? vo.getStlc() : "", style); + ExcelUtils.formatCell(row, 19, vo.getLgtd() != null ? CommonUtils.formatNum(vo.getLgtd()) : "", style); + ExcelUtils.formatCell(row, 20, vo.getLttd() != null ? CommonUtils.formatNum(vo.getLttd()) : "", style); + ExcelUtils.formatCell(row, 21, vo.getStgrd() != null ? vo.getStgrd() : "", style); + ExcelUtils.formatCell(row, 22, vo.getFrgrd() != null ? vo.getFrgrd() : "", style); + ExcelUtils.formatCell(row, 23, vo.getNt() != null ? vo.getNt() : "", style); + ExcelUtils.formatCell(row, 24, "", style); + ExcelUtils.formatCell(row, 25, "", style); + ExcelUtils.formatCell(row, 26, "", style); + ExcelUtils.formatCell(row, 27, "", style); + ExcelUtils.formatCell(row, 28, "", style); + ExcelUtils.formatCell(row, 29, "", style); + ExcelUtils.formatCell(row, 30, "", style); + ExcelUtils.formatCell(row, 31, "", style); + ExcelUtils.formatCell(row, 32, "", style); + ExcelUtils.formatCell(row, 33, "", style); + ExcelUtils.formatCell(row, 34, "", style); + ExcelUtils.formatCell(row, 35, "", style); + ExcelUtils.formatCell(row, 36, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + + + + + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/YcExportTaskServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/YcExportTaskServiceImpl.java new file mode 100644 index 0000000..c6ba524 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/basic/service/impl/YcExportTaskServiceImpl.java @@ -0,0 +1,99 @@ +package com.ruoyi.swlscx.basic.service.impl; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.Query; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.mapper.YcExportTaskMapper; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; + +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.List; +import java.util.Map; + +/** + * @author 12974 + * @description 针对表【yc_export_task(导出任务)】的数据库操作Service实现 + * @createDate 2024-10-23 13:57:23 + */ +@Service +@Slf4j +public class YcExportTaskServiceImpl extends ServiceImpl + implements YcExportTaskService { + + + @Override + public void addTask(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + } + + @Override + public R queryPage(Map params) { + Long userId = SecurityUtils.getUserId(); + params.put("userId", userId); + IPage page = this.baseMapper.queryPage( + new Query().getPage(params), + params + ); + + return R.ok().put("count", page.getTotal()).put("data", page.getRecords()); + } + + @Override + public void downloadExcel(HttpServletResponse response, YcExportTask t) { + ExcelUtils.downLoadExcel(response, t); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public R removeTask(Integer id) { + + //先删除路径下的文件 + YcExportTask task = this.getById(id); + String fullPath = task.getFilepath()+task.getFilename() + ".xls"; + log.info("fullPath{}",fullPath); + File file = new File(fullPath); + if (!file.exists()) { + throw new RuntimeException("该文件不存在"); + } + if (file.delete()){ + this.removeById(id); + return R.ok(); + } + return R.error("删除失败"); + } + + @Override + @DataSource(DataSourceType.SHARDING) + public R getTest() { + List> list=this.baseMapper.test(); + return R.ok().put("data",list); + } + + + // 删除文件的方法 + private boolean deleteFile(String fullPath) { + File file = new File(fullPath); + if (file.exists() && file.isFile()) { + boolean isDeleted = file.delete(); + log.info("文件删除状态: {}", isDeleted ? "成功" : "失败"); + return isDeleted; + } + log.warn("文件不存在或不是文件: {}", fullPath); + return false; + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/PageParams.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/PageParams.java new file mode 100644 index 0000000..7f6db86 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/PageParams.java @@ -0,0 +1,29 @@ +package com.ruoyi.swlscx.common; + +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; + +/** + * @Author al + * @Date 2023/12/18 9:55 + * @Description: TODO + * @Version + */ +@Data +@AllArgsConstructor +@NoArgsConstructor +public class PageParams { + + /** + * 当前页码 + */ + private Long pageNum; + + /** + * 每页记录数默认值 + */ + private Long pageSize; + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/RequestVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/RequestVo.java new file mode 100644 index 0000000..7efccb2 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/RequestVo.java @@ -0,0 +1,27 @@ +package com.ruoyi.swlscx.common; + +import lombok.Data; + +/** + * @Author al + * @Date 2024/9/19 13:40 + * @Description: TODO + * @Version + */ + +@Data +public class RequestVo { + + private Long pageNum; + + private Long pageSize; + + private String stcd; + + private String stnm; + + private String startTime; + + private String endTime; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/constants/SystemConstants.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/constants/SystemConstants.java new file mode 100644 index 0000000..0287402 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/constants/SystemConstants.java @@ -0,0 +1,12 @@ +package com.ruoyi.swlscx.common.constants; + +/** + * @Author al + * @Date 2024/10/18 14:14 + * @Description: TODO + * @Version + */ +public class SystemConstants { + + public static final String TIME_FORMAT="yyyy-MM-dd HH:mm:ss"; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator.java new file mode 100644 index 0000000..7eea9fb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator.java @@ -0,0 +1,16 @@ +package com.ruoyi.swlscx.common.functions; + +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Sheet; + +import java.io.IOException; +import java.util.concurrent.ExecutionException; + +/*@Author: tongw + *@CreateDate: 2019/7/26 16:01 + *@Description: + **/ +@FunctionalInterface +public interface ExcelGenerator { + void doSheet(Sheet sheet, CellStyle style) throws IOException, ExecutionException, InterruptedException; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_1.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_1.java new file mode 100644 index 0000000..936d4f1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_1.java @@ -0,0 +1,13 @@ +package com.ruoyi.swlscx.common.functions; + +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Sheet; + +/*@Author: tongw + *@CreateDate: 2019/7/26 16:01 + *@Description: + **/ +@FunctionalInterface +public interface ExcelGenerator2_1 { + void doSheet(Sheet sheet, CellStyle style, CellStyle style1, CellStyle style2, CellStyle style3); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_2.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_2.java new file mode 100644 index 0000000..5bcc83b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator2_2.java @@ -0,0 +1,13 @@ +package com.ruoyi.swlscx.common.functions; + +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Sheet; + +/*@Author: tongw + *@CreateDate: 2019/7/26 16:01 + *@Description: + **/ +@FunctionalInterface +public interface ExcelGenerator2_2 { + int doSheet(Sheet sheet, CellStyle style, CellStyle style1, CellStyle style2, CellStyle style3); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator3.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator3.java new file mode 100644 index 0000000..199cab4 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/functions/ExcelGenerator3.java @@ -0,0 +1,15 @@ +package com.ruoyi.swlscx.common.functions; + +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Sheet; + +import java.io.IOException; + +/*@Author: tongw + *@CreateDate: 2019/7/26 16:01 + *@Description: + **/ +@FunctionalInterface +public interface ExcelGenerator3 { + int doSheet(Sheet sheet, CellStyle style) throws IOException; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/listener/ExcelListener.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/listener/ExcelListener.java new file mode 100644 index 0000000..1a51153 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/listener/ExcelListener.java @@ -0,0 +1,52 @@ +package com.ruoyi.swlscx.common.listener; + +import com.alibaba.excel.context.AnalysisContext; +import com.alibaba.excel.event.AnalysisEventListener; +import lombok.extern.slf4j.Slf4j; + +import java.util.ArrayList; +import java.util.List; + +/** + * @Author al + * @Date 2024/7/23 14:49 + * @Description: TODO + * @Version + */ +@Slf4j +public class ExcelListener extends AnalysisEventListener { + + // 可以通过实例获取该值 + private List data = new ArrayList<>(); + + /** + * 读取excel内容 + * 从excel中的第二行开始读取,把每行数据数据都读出来然后封装到T对象中 + * 每解析一行数据就会调用一次该方法 + * + * @param t + * @param analysisContext + */ + @Override + public void invoke(T t, AnalysisContext analysisContext) { +// log.info("data:{}", t); + // 数据存储到list,供批量处理,或后续自己业务逻辑处理。 + data.add(t); + } + + /** + * 所有操作都完成之后才会执行本方法 + * + * @param analysisContext + */ + @Override + public void doAfterAllAnalysed(AnalysisContext analysisContext) { + // excel解析完毕以后需要执行的代码 +// log.info("This is doAfterAllAnalysed!!!!"); + } + + + public List getData() { + return data; + } +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/CustomNameThreadFactory.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/CustomNameThreadFactory.java new file mode 100644 index 0000000..3224348 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/CustomNameThreadFactory.java @@ -0,0 +1,47 @@ +package com.ruoyi.swlscx.common.thread; + +import org.apache.commons.lang3.StringUtils; + +import java.util.concurrent.ThreadFactory; +import java.util.concurrent.atomic.AtomicInteger; + +/** + * 自定义名称的线程工厂 + * + * @author: ChickenWing + * @date: 2023/11/26 + */ +public class CustomNameThreadFactory implements ThreadFactory { + + private static final AtomicInteger poolNumber = new AtomicInteger(1); + private final ThreadGroup group; + private final AtomicInteger threadNumber = new AtomicInteger(1); + private final String namePrefix; + + CustomNameThreadFactory(String name) { + SecurityManager s = System.getSecurityManager(); + group = (s != null) ? s.getThreadGroup() : + Thread.currentThread().getThreadGroup(); + if (StringUtils.isBlank(name)) { + name = "pool"; + } + namePrefix = name + "-" + + poolNumber.getAndIncrement() + + "-thread-"; + } + + @Override + public Thread newThread(Runnable r) { + Thread t = new Thread(group, r, + namePrefix + threadNumber.getAndIncrement(), + 0); + if (t.isDaemon()){ + t.setDaemon(false); + } + if (t.getPriority() != Thread.NORM_PRIORITY){ + t.setPriority(Thread.NORM_PRIORITY); + } + return t; + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThread.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThread.java new file mode 100644 index 0000000..79a291d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThread.java @@ -0,0 +1,40 @@ +package com.ruoyi.swlscx.common.thread; + +import com.baomidou.mybatisplus.core.toolkit.Wrappers; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; + +/** + * @Author al + * @Date 2024/10/23 13:54 + * @Description: TODO + * @Version + */ +public class ExportThread implements Runnable { + + private final YcExportTask params; + + private final YcExportTaskService taskService; + + public ExportThread(YcExportTask params, YcExportTaskService taskService) { + this.params = params; + this.taskService = taskService; + } + + @Override + public void run() { + + + + + + // 执行成功后 + taskService.update(Wrappers.lambdaUpdate(YcExportTask.class) + .set(YcExportTask::getStatus, 1) + .eq(YcExportTask::getId, params.getId())); + + + + } +} + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThreadPoolConfig.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThreadPoolConfig.java new file mode 100644 index 0000000..116ae6e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ExportThreadPoolConfig.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.common.thread; + +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +import java.util.concurrent.LinkedBlockingDeque; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.concurrent.TimeUnit; + +/** + * 线程池的config管理 + * + * @author: ChickenWing + * @date: 2023/11/26 + */ +@Configuration +public class ExportThreadPoolConfig { + + @Bean(name = "labelThreadPool") + public ThreadPoolExecutor getLabelThreadPool() { + return new ThreadPoolExecutor(20, 100, 5, + TimeUnit.SECONDS, new LinkedBlockingDeque<>(40), + new CustomNameThreadFactory("label"), + new ThreadPoolExecutor.CallerRunsPolicy()); + } + + @Bean(name = "importThreadPool") + public ThreadPoolExecutor getImportThreadPool() { + return new ThreadPoolExecutor(20, 100, 5, + TimeUnit.SECONDS, new LinkedBlockingDeque<>(40), + new CustomNameThreadFactory("import"), + new ThreadPoolExecutor.CallerRunsPolicy()); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ThreadConfig.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ThreadConfig.java new file mode 100644 index 0000000..8da84f3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/thread/ThreadConfig.java @@ -0,0 +1,29 @@ +package com.ruoyi.swlscx.common.thread; + +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.scheduling.annotation.EnableAsync; +import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; + +import java.util.concurrent.Executor; + +/** + * @Author al + * @Date 2024/10/23 14:09 + * @Description: TODO + * @Version + */ +@Configuration +@EnableAsync // 启用异步执行支持 +public class ThreadConfig { + + @Bean(name = "taskExecutor") + public Executor taskExecutor() { + ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); + executor.setCorePoolSize(5); // 核心线程数 + executor.setMaxPoolSize(10); // 最大线程数 + executor.setQueueCapacity(25); // 队列容量 + executor.initialize(); + return executor; + } +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/CommonUtils.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/CommonUtils.java new file mode 100644 index 0000000..7a4100e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/CommonUtils.java @@ -0,0 +1,67 @@ +package com.ruoyi.swlscx.common.utils; + +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.swlscx.common.PageParams; +import org.apache.commons.lang3.StringUtils; + +import java.math.BigDecimal; +import java.util.HashMap; +import java.util.Map; + +/** + * @Author al + * @Date 2024/9/21 14:54 + * @Description: TODO + * @Version + */ +public class CommonUtils { + + public static Map getYearAndDayDataMap(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = new HashMap<>(); + map.put("startTime", startTime); + map.put("endTime", endTime); + map.put("stnm", stnm); + map.put("stcd", stcd); + map.put("page", Convert.toStr(pageParams.getPageNum())); + map.put("limit", Convert.toStr(pageParams.getPageSize())); + return map; + } + + public static Map getMonthDataMap(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + String startYear = ""; + String endYear = ""; + if (StringUtils.isNotBlank(startTime)) { + startYear = startTime.length() > 4 ? startTime.substring(0, 4) : startTime; + } + if (StringUtils.isNotBlank(endTime)) { + endYear = endTime.length() > 4 ? endTime.substring(0, 4) : endTime; + } + Map map = new HashMap<>(); + map.put("startYear", Convert.toInt(startYear)); + map.put("startTime", startTime); + map.put("endYear", Convert.toInt(endYear)); + map.put("endTme", endTime); + map.put("stnm", stnm); + map.put("stcd", stcd); + map.put("page", Convert.toStr(pageParams.getPageNum())); + map.put("limit", Convert.toStr(pageParams.getPageSize())); + return map; + } + + + + public static String formatNum(BigDecimal number) { + if (number == null) { + return ""; + } + return String.format("%.2f", number).replaceAll("0*$", "").replaceAll("\\.$", ""); + } + + public static String formatNum(Double number) { + if (number == null) { + return ""; + } + return String.format("%.2f", number).replaceAll("0*$", "").replaceAll("\\.$", ""); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ExcelUtils.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ExcelUtils.java new file mode 100644 index 0000000..8433c94 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ExcelUtils.java @@ -0,0 +1,412 @@ +package com.ruoyi.swlscx.common.utils; + +import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.ExcelReader; +import com.alibaba.excel.analysis.ExcelReadExecutor; +import com.alibaba.excel.read.metadata.ReadSheet; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.common.functions.ExcelGenerator; +import com.ruoyi.swlscx.common.functions.ExcelGenerator2_1; +import com.ruoyi.swlscx.common.functions.ExcelGenerator2_2; +import com.ruoyi.swlscx.common.functions.ExcelGenerator3; +import com.ruoyi.swlscx.common.listener.ExcelListener; +import lombok.extern.slf4j.Slf4j; +import org.apache.poi.hssf.usermodel.HSSFWorkbook; +import org.apache.poi.ss.usermodel.*; +import org.apache.poi.ss.util.CellRangeAddress; +import org.apache.poi.xssf.usermodel.XSSFWorkbook; +import org.springframework.core.io.ClassPathResource; + +import javax.servlet.http.HttpServletResponse; +import java.io.*; +import java.net.URLEncoder; +import java.nio.file.Files; +import java.util.ArrayList; +import java.util.List; +import java.util.Map; + +@Slf4j +public class ExcelUtils { + + /** + * 从指定行开始读取Excel数据 + * + * @param inputStream 文件输入流 + * @param clazz 映射的实体类 + * @param headRowNumber 从第几行开始读取(从0开始计数) + * @param 实体类类型 + * @return 实体类对象的列表 + */ + public static List readExcel(InputStream inputStream, Class clazz, int headRowNumber) { + // 读取Excel文件 + ExcelReader excelReader = EasyExcel.read(inputStream).build(); + ExcelReadExecutor excelReadExecutor = excelReader.excelExecutor(); + List sheets = excelReadExecutor.sheetList(); + List sheetData = new ArrayList<>(); + //读取到的数据 + for (ReadSheet sheet : sheets) { + ExcelListener listener = new ExcelListener<>(); + ReadSheet readSheet1 = EasyExcel.readSheet(sheet.getSheetNo()).head(clazz).headRowNumber(headRowNumber).registerReadListener(listener).build(); + excelReader.read(readSheet1); + // 得到sheet名称和其对应的数据 + sheetData.addAll(listener.getData()); + } + return sheetData; + } + + + + /** + * 导出 + * + * @param fileName + * @param response + * @param workbook + */ + public static void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) { + try { + response.setCharacterEncoding("UTF-8"); + response.setHeader("content-Type", "application/vnd.ms-excel"); + response.setHeader("Content-Disposition", + "attachment;filename=\"" + URLEncoder.encode(fileName, "UTF-8") + "\""); + workbook.write(response.getOutputStream()); + } catch (IOException e) { + e.printStackTrace(); + } + } + + public static void downLoadTxt(String fileName, HttpServletResponse response) throws IOException { + response.setCharacterEncoding("UTF-8"); + response.setHeader("content-Type", "text/plain"); + response.setHeader("Content-Disposition", + "attachment;filename=\"" + URLEncoder.encode(fileName, "UTF-8") + "\""); + OutputStream outputStream = response.getOutputStream(); + byte[] buff = new byte[1024]; + BufferedInputStream bis ; + ClassPathResource cpr = new ClassPathResource("/excel/" + fileName); + bis = new BufferedInputStream(cpr.getInputStream()); + int i = bis.read(buff); + while (i != -1) { + outputStream.write(buff, 0, buff.length); + outputStream.flush(); + i = bis.read(buff); + } + } + + public static void excelExport(HttpServletResponse response, String templateName, String excelName, ExcelGenerator excelGenerator) { + try { + ClassPathResource cpr = new ClassPathResource("/excel/" + templateName); + InputStream is = cpr.getInputStream(); + Workbook workbook = null; + if ("xlsx".equals(templateName.split("\\.")[1])) { + workbook = new XSSFWorkbook(is); + } else { + workbook = new HSSFWorkbook(is); + } + + CellStyle style = workbook.createCellStyle(); + Font font = workbook.createFont(); + font.setFontHeightInPoints((short)9); + font.setFontName("宋体"); + style.setFont(font); + + Sheet sheet0 = workbook.getSheetAt(0); + excelGenerator.doSheet(sheet0, style); + + ExcelUtils.downLoadExcel(excelName + "." + templateName.split("\\.")[1], response, workbook); + }catch (Exception e) { + e.printStackTrace(); + } + } + + public static void excelExport2(HttpServletResponse response, String templateName, String excelName, ExcelGenerator2_1 excelGenerator) { + try { + ClassPathResource cpr = new ClassPathResource("/excel/" + templateName); + InputStream is = cpr.getInputStream(); + Workbook workbook = null; + if ("xlsx".equals(templateName.split("\\.")[1])) { + workbook = new XSSFWorkbook(is); + } else { + workbook = new HSSFWorkbook(is); + } + + CellStyle style = workbook.createCellStyle(); + Font font = workbook.createFont(); + font.setFontHeightInPoints((short)10); + font.setFontName("宋体"); + style.setFont(font); + + Font font2 = workbook.createFont(); + font2.setFontHeightInPoints((short)12); + font2.setFontName("宋体"); + + CellStyle style1 = workbook.createCellStyle(); + style1.setFont(font2); + + CellStyle style2 = workbook.createCellStyle(); + style2.setAlignment(HorizontalAlignment.LEFT); + style2.setFont(font); + + CellStyle style3 = workbook.createCellStyle(); + style3.setAlignment(HorizontalAlignment.RIGHT); + style3.setFont(font); + + + Sheet sheet0 = workbook.getSheetAt(0); + excelGenerator.doSheet(sheet0, style,style1,style2,style3); + + ExcelUtils.downLoadExcel(excelName + "." + templateName.split("\\.")[1], response, workbook); + }catch (IOException e) { + e.printStackTrace(); + } + } + + public static void listToExport(HttpServletResponse response, String excelName, List> dataMap, Integer columnNum){ + try { + ClassPathResource cpr = new ClassPathResource("/excel/" + "list2excel.xls"); + InputStream is = cpr.getInputStream(); + Workbook workbook = new HSSFWorkbook(is); + Sheet sheet0 = workbook.getSheetAt(0); + Row row; + for(int i=0; i< dataMap.size(); i++){ + row=sheet0.createRow(i); + int j=0; + for (Map.Entry entry : dataMap.get(i).entrySet()) { + Cell cell = row.createCell(j); + cell.setCellValue(entry.getValue() == null ? "" : entry.getValue().toString()); + cell.getCellStyle().setAlignment(HorizontalAlignment.CENTER); + j++; + } + } + for(int i=0; i" + res); + if (res == 0){ + ExcelUtils.downLoadBackExcel(excelName + "." + templateName.split("\\.")[1], workbook); + } + }catch (IOException e) { + e.printStackTrace(); + } + } + + /** + * 逐潮高低潮表专用 + */ + public static void excelExportTide1Back(String templateName, String excelName, ExcelGenerator2_2 excelGenerator) { + try { + ClassPathResource cpr = new ClassPathResource("/excel/" + templateName); + InputStream is = cpr.getInputStream(); + Workbook workbook = null; + if ("xlsx".equals(templateName.split("\\.")[1])) { + workbook = new XSSFWorkbook(is); + } else { + workbook = new HSSFWorkbook(is); + } + + CellStyle style = workbook.createCellStyle(); + Font font = workbook.createFont(); + font.setFontHeightInPoints((short)10); + font.setFontName("宋体"); + style.setFont(font); + + Font font2 = workbook.createFont(); + font2.setFontHeightInPoints((short)12); + font2.setFontName("宋体"); + + CellStyle style1 = workbook.createCellStyle(); + style1.setFont(font2); + + CellStyle style2 = workbook.createCellStyle(); + style2.setAlignment(HorizontalAlignment.LEFT); + style2.setFont(font); + + CellStyle style3 = workbook.createCellStyle(); + style3.setAlignment(HorizontalAlignment.RIGHT); + style3.setFont(font); + + + Sheet sheet0 = workbook.getSheetAt(0); + int res = excelGenerator.doSheet(sheet0, style,style1,style2,style3); + if (res == 0) { + ExcelUtils.downLoadBackExcel(excelName + "." + templateName.split("\\.")[1], workbook); + } + }catch (IOException e) { + e.printStackTrace(); + } + } + + public static void downLoadBackExcel(String fileName, Workbook workbook) { + try { + FileOutputStream fileOut = new FileOutputStream(fileName); + workbook.write(fileOut); + fileOut.close(); + workbook.close(); + } catch (IOException e) { + e.printStackTrace(); + } + } + + public static void formatCell(Row row, int index, String value, CellStyle style) { + Cell cell = row.createCell(index); + cell.setCellValue(value); + // 对齐方式 + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + // 设置上下左右边框 + style.setBorderLeft(BorderStyle.THIN); + style.setBorderRight(BorderStyle.THIN); + style.setBorderTop(BorderStyle.THIN); // 添加顶部边框 + style.setBorderBottom(BorderStyle.THIN); // 添加底部边框 + // 应用样式 + cell.setCellStyle(style); + } + + + public static Row createRow(Sheet sheet, int index) { + if (sheet.getRow(index) == null) { + return sheet.createRow(index); + } else { + return sheet.getRow(index); + } + } + + + public static void downLoadExcel(HttpServletResponse response, YcExportTask t) { + Long userId = SecurityUtils.getUserId(); + String fullPath = t.getFilepath()+t.getFilename() + ".xls"; + log.info("fullPath{}",fullPath); + File file = new File(fullPath); + if (!file.exists()) { + throw new RuntimeException("该文件不存在"); + } + + try (InputStream in = Files.newInputStream(file.toPath()); + OutputStream out = response.getOutputStream()) { + // 设置响应头,通知浏览器以附件形式下载 + response.setCharacterEncoding("UTF-8"); + response.setHeader("content-Type", "application/vnd.ms-excel"); + response.setHeader("Content-Disposition", + "attachment;filename=\"" + URLEncoder.encode(t.getFilename(), "UTF-8") + "\""); + byte[] buffer = new byte[1024]; + int len; + while ((len = in.read(buffer)) != -1) { + out.write(buffer, 0, len); + } + out.flush(); // 确保所有数据写入响应 + } catch (IOException e) { + throw new RuntimeException("文件下载失败", e); + } + } + + + + public static void copyHeader(Sheet targetSheet, CellStyle style) { + // 获取第一个 sheet + Sheet sourceSheet = targetSheet.getWorkbook().getSheetAt(0); // 假设源 sheet 是第一个 + + // 复制第一行(表头) + Row headerRow1 = sourceSheet.getRow(0); // 第1行 + Row newHeaderRow1 = targetSheet.createRow(0); // 在新sheet中创建第一行 + copyRow(headerRow1, newHeaderRow1); + + // 复制第二行 + Row headerRow2 = sourceSheet.getRow(1); // 第2行 + Row newHeaderRow2 = targetSheet.createRow(1); // 在新sheet中创建第二行 + copyRow(headerRow2, newHeaderRow2); + + // 复制合并单元格 + int mergedRegionsCount = sourceSheet.getNumMergedRegions(); + for (int i = 0; i < mergedRegionsCount; i++) { + CellRangeAddress mergedRegion = sourceSheet.getMergedRegion(i); + // 检查合并区域是否在需要复制的行范围内 + if (mergedRegion.getFirstRow() <= 1 && mergedRegion.getLastRow() >= 0) { + // 检查目标 sheet 中是否已存在相同的合并区域 + boolean exists = false; + for (int j = 0; j < targetSheet.getNumMergedRegions(); j++) { + CellRangeAddress existingRegion = targetSheet.getMergedRegion(j); + if (existingRegion.equals(mergedRegion)) { + exists = true; + break; + } + } + + // 如果不存在,则添加合并区域 + if (!exists) { + targetSheet.addMergedRegion(new CellRangeAddress( + mergedRegion.getFirstRow(), mergedRegion.getLastRow(), + mergedRegion.getFirstColumn(), mergedRegion.getLastColumn() + )); + } + } + } + } + + // 假设 copyRow 方法是已定义的方法,用于复制行的内容和样式 + public static void copyRow(Row sourceRow, Row targetRow) { + for (int i = 0; i < sourceRow.getPhysicalNumberOfCells(); i++) { + Cell sourceCell = sourceRow.getCell(i); + Cell targetCell = targetRow.createCell(i); + if (sourceCell != null) { + // 复制单元格内容和样式 + targetCell.setCellValue(sourceCell.getStringCellValue()); + targetCell.setCellStyle(sourceCell.getCellStyle()); + } + } + } + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/FileUtil.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/FileUtil.java new file mode 100644 index 0000000..e676188 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/FileUtil.java @@ -0,0 +1,80 @@ +package com.ruoyi.swlscx.common.utils; + +import org.springframework.util.CollectionUtils; + +import java.io.File; +import java.io.InputStream; +import java.util.ArrayList; +import java.util.List; + +/** + * @Author al + * @Date 2024/8/7 15:16 + * @Description: TODO + * @Version + */ +public class FileUtil { + public static InputStream getResourcesFileInputStream(String fileName) { + return Thread.currentThread().getContextClassLoader().getResourceAsStream("" + fileName); + } + + public static String getPath() { + return FileUtil.class.getResource("/").getPath(); + } + + public static PathBuild pathBuild() { + return new PathBuild(); + } + + public static File createNewFile(String pathName) { + File file = new File(getPath() + pathName); + if (file.exists()) { + file.delete(); + } else { + if (!file.getParentFile().exists()) { + file.getParentFile().mkdirs(); + } + } + return file; + } + + public static File readFile(String pathName) { + return new File(getPath() + pathName); + } + + public static File readUserHomeFile(String pathName) { + return new File(System.getProperty("user.home") + File.separator + pathName); + } + + /** + * build to test file path + **/ + public static class PathBuild { + private PathBuild() { + subPath = new ArrayList<>(); + } + + private final List subPath; + + public PathBuild sub(String dirOrFile) { + subPath.add(dirOrFile); + return this; + } + + public String getPath() { + if (CollectionUtils.isEmpty(subPath)) { + return FileUtil.class.getResource("/").getPath(); + } + if (subPath.size() == 1) { + return FileUtil.class.getResource("/").getPath() + subPath.get(0); + } + StringBuilder path = new StringBuilder(FileUtil.class.getResource("/").getPath()); + path.append(subPath.get(0)); + for (int i = 1; i < subPath.size(); i++) { + path.append(File.separator).append(subPath.get(i)); + } + return path.toString(); + } + + } +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ImportDateToSqlserverDto.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ImportDateToSqlserverDto.java new file mode 100644 index 0000000..18d91cd --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/ImportDateToSqlserverDto.java @@ -0,0 +1,30 @@ +package com.ruoyi.swlscx.common.utils; + +import lombok.Data; +import org.apache.poi.ss.formula.functions.T; + +import java.util.List; + +/** + * @Author al + * @Date 2025/1/9 16:04 + * @Description: TODO + * @Version + */ +@Data +public class ImportDateToSqlserverDto { + + private String startTime; + + private String endTime; + + private String filename; + + private List dataList; + + private Class classBean; + + + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/Query.java b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/Query.java new file mode 100644 index 0000000..56cfd44 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/common/utils/Query.java @@ -0,0 +1,76 @@ +/** + * Copyright (c) 2016-2019 人人开源 All rights reserved. + * + * https://www.renren.io + * + * 版权所有,侵权必究! + */ + +package com.ruoyi.swlscx.common.utils; + +import com.baomidou.mybatisplus.core.metadata.OrderItem; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.common.constant.Constants; +import com.ruoyi.common.xss.SQLFilter; +import org.apache.commons.lang3.StringUtils; + +import java.util.Map; + +/** + * 查询参数 + * + * @author Mark sunlightcs@gmail.com + */ +public class Query { + + public Page getPage(Map params) { + return this.getPage(params, null, false); + } + + public Page getPage(Map params, String defaultOrderField, boolean isAsc) { + //分页参数 + long curPage = 1; + long limit = 10; + + if(params.get(Constants.PAGE) != null){ + curPage = Long.parseLong((String)params.get(Constants.PAGE)); + } + if(params.get(Constants.LIMIT) != null){ + limit = Long.parseLong((String)params.get(Constants.LIMIT)); + } + + //分页对象 + Page page = new Page<>(curPage, limit); + + //分页参数 + params.put(Constants.PAGE, page); + + //排序字段 + //防止SQL注入(因为sidx、order是通过拼接SQL实现排序的,会有SQL注入风险) + String orderField = SQLFilter.sqlInject((String)params.get(Constants.ORDER_FIELD)); + String order = (String)params.get(Constants.ORDER); + + //前端字段排序 + if(StringUtils.isNotEmpty(orderField) && StringUtils.isNotEmpty(order)){ + if(Constants.ASC.equalsIgnoreCase(order)) { + return page.addOrder(OrderItem.asc(orderField)); + }else { + return page.addOrder(OrderItem.desc(orderField)); + } + } + + //没有排序字段,则不排序 + if(StringUtils.isBlank(defaultOrderField)){ + return page; + } + + //默认排序 + if(isAsc) { + page.addOrder(OrderItem.asc(defaultOrderField)); + }else { + page.addOrder(OrderItem.desc(defaultOrderField)); + } + + return page; + } +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayExportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayExportController.java new file mode 100644 index 0000000..fdc3471 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayExportController.java @@ -0,0 +1,112 @@ +package com.ruoyi.swlscx.day.controller; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.day.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/10/18 13:47 + * @Description: TODO + * @Version + */ +@RequestMapping("/report") +@RequiredArgsConstructor +@RestController +public class DayExportController { + + private final HyDpCService hyDpCService; + + private final HyDweCService hyDweCService; + + private final HyDqCService hyDqCService; + + private final HyDzCService hyDzCService; + + private final HyDcsFService hyDcsFService; + + private final HyDwtCService hyDwtCService; + + + + /** + * 导出日降雨量表数据 + */ + @PostMapping("/dayRain") + public R exportRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDpCService.exportHyDpCData(response, startTime, endTime, stcd, stnm); + } + + + /** + * 导出日水面蒸发表数据 + */ + @PostMapping("/evaporationWater") + public R exportEvaporationWater(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDweCService.exportHyDweCData(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出日流量表数据 + */ + @PostMapping("/dayFlow") + public R exportDayFlow(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDqCService.exportDayFlow(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出日平均水位表数据 + */ + @PostMapping("/dayWaterLever") + public R exportDayWaterLever(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDzCService.exportDayWaterLever(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出日平均含沙量表表数据 + */ + @PostMapping("/daySedimentConcentration") + public R exportDaySedimentConcentration(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDcsFService.exportDaySedimentConcentration(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出日水温表 + */ + @PostMapping("/dayWaterTemperature") + public R exportDayWaterTemperature(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDwtCService.exportDayWaterTemperature(response, startTime, endTime, stcd, stnm); + } + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayImportToSoftController.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayImportToSoftController.java new file mode 100644 index 0000000..defb10b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayImportToSoftController.java @@ -0,0 +1,93 @@ +package com.ruoyi.swlscx.day.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.day.domain.vo.HyDpCVo; +import com.ruoyi.swlscx.day.domain.vo.HyDqCVo; +import com.ruoyi.swlscx.day.domain.vo.HyDzCVo; +import com.ruoyi.swlscx.day.service.*; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.*; + +/** + * @Author al + * @Date 2025/1/5 16:11 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/day/import") +@RequiredArgsConstructor +@Slf4j +public class DayImportToSoftController { + + + + private final HyDpCService hyDpCService; + + private final HyDweCService hyDweCService; + + private final HyDqCService hyDqCService; + + private final HyDzCService hyDzCService; + + private final HyDcsFService hyDcsFService; + + private final HyDwtCService hyDwtCService; + + + /** + * 导入日水面蒸发表数据 + */ + @PostMapping("/importHyDweCDateToSoft") + public R importHyDweCDateToSoft(@RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime){ + return hyDweCService.importDateTosoft(startTime,endTime); + } + + + /** + * 导入日水温表数据 + */ + @PostMapping("/importHyDwtCDateToSoft") + public R importHyDwtCDateToSoft(@RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime){ + return hyDwtCService.importDateTosoft(startTime,endTime); + } + + + /** + * 导入日降雨量表数据 + */ + @PostMapping("/importRainListDateToSoft") + public R getRainList(@RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime) { + return hyDpCService.importDateTosoft(startTime, endTime); + } + + /** + * 导入日流量表数据 + */ + @PostMapping("/importDayFlowDateToSoft") + public R importDayFlowDateToSoft(@RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime) { + return hyDqCService.importDateTosoft(startTime, endTime); + } + + /**导入日水位表数据**/ + @PostMapping("/importWaterLeverDateToSoft") + public R importWaterLeverDateToSoft( String startTime, String endTime) { + return hyDzCService.importDateTosoft(startTime, endTime); + } + + + /**导入日平均含沙量表表数据**/ + @PostMapping("/importDaySedimentConcentrationDateToSoft") + public R importDaySedimentConcentrationDateToSoft( String startTime, String endTime) { + return hyDcsFService.importDateTosoft(startTime, endTime); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayTableController.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayTableController.java new file mode 100644 index 0000000..ff693ac --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/controller/DayTableController.java @@ -0,0 +1,127 @@ +package com.ruoyi.swlscx.day.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.vo.*; +import com.ruoyi.swlscx.day.service.*; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/10/17 15:16 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/day") +@RequiredArgsConstructor +@Slf4j +public class DayTableController { + + private final HyDpCService hyDpCService; + + private final HyDweCService hyDweCService; + + private final HyDqCService hyDqCService; + + private final HyDzCService hyDzCService; + + private final HyDcsFService hyDcsFService; + + private final HyDwtCService hyDwtCService; + + /** + * 获取日降雨量表数据 + */ + @GetMapping("/getDayRainList") + public R getRainList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDpCVoIPage = hyDpCService.selectHyDpCDataByPageAndInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyDpCVoIPage.getRecords()).put("count", hyDpCVoIPage.getTotal()); + } + + /** + * 查询日水面蒸发表数据 + */ + @GetMapping("/getEvaporationWaterList") + public R getEvaporationWaterList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDweCVoIPage = hyDweCService.selectHyDweCDataByPageAndInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyDweCVoIPage.getRecords()).put("count", hyDweCVoIPage.getTotal()); + } + + /** + * 查询日流量表数据 + */ + @GetMapping("/getDayFlowList") + public R getDayFlowList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDqCVoIPage = hyDqCService.selectHyDqCDataByPageAndInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyDqCVoIPage.getRecords()).put("count", hyDqCVoIPage.getTotal()); + } + + /**查询日水位表数据**/ + @GetMapping("/getDayWaterLeverList") + public R getWaterLeverList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDzCVoIPage = hyDzCService.selectHyDzCPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyDzCVoIPage.getRecords()).put("count", hyDzCVoIPage.getTotal()); + } + + + /**查询日平均含沙量表表数据**/ + @GetMapping("/getDaySedimentConcentrationList") + public R getDaySedimentConcentrationList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDcsFVoIPage = hyDcsFService.selectHyDcsFPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyDcsFVoIPage.getRecords()).put("count", hyDcsFVoIPage.getTotal()); + } + + + /**查询日水温表数据**/ + @GetMapping("/getDayWaterTemperatureList") + public R getDayWaterTemperatureList(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDwtCVoIPage = hyDwtCService.selectHyDwtCPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyDwtCVoIPage.getRecords()).put("count", hyDwtCVoIPage.getTotal()); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDcsFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDcsFBo.java new file mode 100644 index 0000000..ffa9a53 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDcsFBo.java @@ -0,0 +1,43 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均含沙量表 + * @TableName hy_dcs_f + */ +@Data +public class HyDcsFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期",index = 3) + private String dt; + + /** + * 平均含沙量(千克每立方米) + */ + @ExcelProperty(value = "平均含沙量",index = 4) + private String avcs; + + /** + * 平均含沙量注解码 + */ + @ExcelProperty(value = "平均含沙量注解码",index = 5) + private String avcsrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDpCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDpCBo.java new file mode 100644 index 0000000..ddc17da --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDpCBo.java @@ -0,0 +1,45 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日降水表 + * @TableName hy_dp_c + */ +@Data +public class HyDpCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期",index = 3) + private String dt; + + /** + * 降水量(毫米) + */ + @ExcelProperty(value = "降水量",index = 4) + private String p; + + /** + * 降水量注解码 + */ + @ExcelProperty(value = "站码",index = 5) + private String prcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqCBo.java new file mode 100644 index 0000000..fc74d98 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqCBo.java @@ -0,0 +1,44 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均流量表 + * @TableName hy_dq_c + */ +@Data +public class HyDqCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期",index = 3) + private String dt; + + /** + * 平均流量 + */ + @ExcelProperty(value = "平均流量",index = 4) + private String avq; + + /** + * 平均流量注解码 + */ + @ExcelProperty(value = "平均流量注解码",index = 5) + private String avqrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqsCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqsCBo.java new file mode 100644 index 0000000..921fb79 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDqsCBo.java @@ -0,0 +1,46 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均输沙率表 + * @TableName hy_dqs_c + */ +@Data +public class HyDqsCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 泥沙类型 + */@ExcelProperty(value = "泥沙类型",index = 3) + private String sdtp; + + /** + * 日期 + */@ExcelProperty(value = "日期",index = 4) + private String dt; + + /** + * 平均输沙率(千克每秒) + */@ExcelProperty(value = "平均输沙率",index = 5) + private String avqs; + + /** + * 平均输沙率注解码 + */@ExcelProperty(value = "平均输沙率注解码",index = 6) + private String avqsrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDweCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDweCBo.java new file mode 100644 index 0000000..72c967e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDweCBo.java @@ -0,0 +1,46 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 日水面蒸发量表 + * + * @TableName hy_dwe_c + */ +@Data +public class HyDweCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期", index = 3) + private String dt; + + /** + * 蒸发器型式 + */ + @ExcelProperty(value = "蒸发器型式", index = 4) + private String eetp; + + /** + * 水面蒸发量(毫米) + */ + @ExcelProperty(value = "水面蒸发量", index = 5) + private String wsfe; + + /** + * 水面蒸发量注解码 + */ + @ExcelProperty(value = "水面蒸发量注解码", index = 6) + private String wsfercd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDwtCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDwtCBo.java new file mode 100644 index 0000000..9dbfb57 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDwtCBo.java @@ -0,0 +1,42 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import lombok.Data; + +import java.io.Serializable; + +/** + * 日水温表 + * + * @TableName hy_dwt_c + */ +@Data +public class HyDwtCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期", index = 3) + private String dt; + + /** + * 水温 + */ + @ExcelProperty(value = "水温", index = 4) + private String wtmp; + + /** + * 水温注解码 + */ + @ExcelProperty(value = "水温注解码", index = 5) + private String wtmprcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDzCBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDzCBo.java new file mode 100644 index 0000000..ea476df --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/bo/HyDzCBo.java @@ -0,0 +1,44 @@ +package com.ruoyi.swlscx.day.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均水位表 + * @TableName hy_dz_c + */ +@Data +public class HyDzCBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 日期 + */ + @ExcelProperty(value = "日期",index = 3) + private String dt; + + /** + * 平均水位(米) + */ + @ExcelProperty(value = "平均水位",index = 4) + private String avz; + + /** + * 平均水位注解码 + */ + @ExcelProperty(value = "平均水位注解码",index = 5) + private String avzrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDcsF.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDcsF.java new file mode 100644 index 0000000..802811a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDcsF.java @@ -0,0 +1,41 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日平均含沙量表 + * @TableName hy_dcs_f + */ +@TableName(value ="hy_dcs_f") +@Data +public class HyDcsF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDpC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDpC.java new file mode 100644 index 0000000..d6108fc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDpC.java @@ -0,0 +1,42 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日降水表 + * @TableName hy_dp_c + */ +@TableName(value ="hy_dp_c") +@Data +public class HyDpC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqC.java new file mode 100644 index 0000000..5058aeb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqC.java @@ -0,0 +1,41 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日平均流量表 + * @TableName hy_dq_c + */ +@TableName(value ="hy_dq_c") +@Data +public class HyDqC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均流量 + */ + private BigDecimal avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqsC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqsC.java new file mode 100644 index 0000000..56f18c7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDqsC.java @@ -0,0 +1,46 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日平均输沙率表 + * @TableName hy_dqs_c + */ +@TableName(value ="hy_dqs_c") +@Data +public class HyDqsC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均输沙率(千克每秒) + */ + private BigDecimal avqs; + + /** + * 平均输沙率注解码 + */ + private String avqsrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDweC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDweC.java new file mode 100644 index 0000000..402248d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDweC.java @@ -0,0 +1,46 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日水面蒸发量表 + * @TableName hy_dwe_c + */ +@TableName(value ="hy_dwe_c") +@Data +public class HyDweC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 水面蒸发量(毫米) + */ + private BigDecimal wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDwtC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDwtC.java new file mode 100644 index 0000000..fa66fd4 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDwtC.java @@ -0,0 +1,41 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日水温表 + * @TableName hy_dwt_c + */ +@TableName(value ="hy_dwt_c") +@Data +public class HyDwtC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 水温 + */ + private BigDecimal wtmp; + + /** + * 水温注解码 + */ + private String wtmprcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDzC.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDzC.java new file mode 100644 index 0000000..c7a7236 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/po/HyDzC.java @@ -0,0 +1,41 @@ +package com.ruoyi.swlscx.day.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日平均水位表 + * @TableName hy_dz_c + */ +@TableName(value ="hy_dz_c") +@Data +public class HyDzC implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均水位(米) + */ + private BigDecimal avz; + + /** + * 平均水位注解码 + */ + private String avzrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDcsFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDcsFVo.java new file mode 100644 index 0000000..eb643dc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDcsFVo.java @@ -0,0 +1,43 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均含沙量表 + * @TableName hy_dcs_f + */ +@Data +public class HyDcsFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDpCVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDpCVo.java new file mode 100644 index 0000000..b5026c5 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDpCVo.java @@ -0,0 +1,45 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import com.fasterxml.jackson.annotation.JsonFormat; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日降水表 + * @TableName hy_dp_c + */ +@Data +public class HyDpCVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDqCVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDqCVo.java new file mode 100644 index 0000000..91f94cb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDqCVo.java @@ -0,0 +1,43 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均流量表 + * @TableName hy_dq_c + */ +@Data +public class HyDqCVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均流量 + */ + private BigDecimal avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDweCVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDweCVo.java new file mode 100644 index 0000000..6d3be6f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDweCVo.java @@ -0,0 +1,49 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日水面蒸发量表 + * @TableName hy_dwe_c + */ +@Data +public class HyDweCVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 水面蒸发量(毫米) + */ + private BigDecimal wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDwtCVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDwtCVo.java new file mode 100644 index 0000000..e238dfa --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDwtCVo.java @@ -0,0 +1,43 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日水温表 + * @TableName hy_dwt_c + */ +@Data +public class HyDwtCVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 水温 + */ + private BigDecimal wtmp; + + /** + * 水温注解码 + */ + private String wtmprcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDzCVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDzCVo.java new file mode 100644 index 0000000..c2e0a82 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/domain/vo/HyDzCVo.java @@ -0,0 +1,43 @@ +package com.ruoyi.swlscx.day.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日平均水位表 + * @TableName hy_dz_c + */ +@Data +public class HyDzCVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 站名 + */ + private String stnm; + + /** + * 日期 + */ + private Date dt; + + /** + * 平均水位(米) + */ + private BigDecimal avz; + + /** + * 平均水位注解码 + */ + private String avzrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDcsFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDcsFMapper.java new file mode 100644 index 0000000..b1edd63 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDcsFMapper.java @@ -0,0 +1,38 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDcsF; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dcs_f(日平均含沙量表)】的数据库操作Mapper +* @createDate 2024-10-16 16:12:22 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDcsF +*/ +public interface HyDcsFMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + IPage selectHyDcsFPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDpCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDpCMapper.java new file mode 100644 index 0000000..246bda7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDpCMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDpC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDpCVo; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dp_c(日降水表)】的数据库操作Mapper +* @createDate 2024-10-15 17:40:46 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDpC +*/ +@Mapper +public interface HyDpCMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List batchList); + + void batchInsertOrUpdate(); + + IPage selectHyDpCDataByPageAndInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqCMapper.java new file mode 100644 index 0000000..29273a9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqCMapper.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDqC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDqCVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dq_c(日平均流量表)】的数据库操作Mapper +* @createDate 2024-10-16 15:40:17 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDqC +*/ +@Mapper +public interface HyDqCMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + IPage selectHyMtqEPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqsCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqsCMapper.java new file mode 100644 index 0000000..4710c62 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDqsCMapper.java @@ -0,0 +1,18 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.ruoyi.swlscx.day.domain.po.HyDqsC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; + +/** +* @author 12974 +* @description 针对表【hy_dqs_c(日平均输沙率表)】的数据库操作Mapper +* @createDate 2024-10-16 16:29:38 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDqsC +*/ +public interface HyDqsCMapper extends BaseMapper { + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDweCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDweCMapper.java new file mode 100644 index 0000000..8418b22 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDweCMapper.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDweC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDweCVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrweFVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dwe_c(日水面蒸发量表)】的数据库操作Mapper +* @createDate 2024-10-16 17:24:19 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDweC +*/ +public interface HyDweCMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + /** + * 查询日水面蒸发表数据 + */ + IPage selectHyDweCDataByPageAndInfo(Page page,@Param("map") Map map); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDwtCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDwtCMapper.java new file mode 100644 index 0000000..652f542 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDwtCMapper.java @@ -0,0 +1,26 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDwtC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDweCVo; +import com.ruoyi.swlscx.day.domain.vo.HyDwtCVo; +import org.apache.ibatis.annotations.Param; + +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dwt_c(日水温表)】的数据库操作Mapper +* @createDate 2024-10-17 15:07:12 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDwtC +*/ +public interface HyDwtCMapper extends BaseMapper { + + IPage selectHyDwtCPageByInfo(Page page, @Param("map") Map map); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDzCMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDzCMapper.java new file mode 100644 index 0000000..0c1dd7c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/mapper/HyDzCMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.day.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDzC; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.day.domain.vo.HyDzCVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dz_c(日平均水位表)】的数据库操作Mapper +* @createDate 2024-10-16 16:46:04 +* @Entity com.ruoyi.swlscx.day.domain.po.HyDzC +*/ +public interface HyDzCMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + IPage selectHyDzCPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDcsFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDcsFService.java new file mode 100644 index 0000000..d954f91 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDcsFService.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDcsF; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dcs_f(日平均含沙量表)】的数据库操作Service +* @createDate 2024-10-16 16:12:22 +*/ +public interface HyDcsFService extends IService { + + /** + * 导入日平均含沙量表数据 + */ + void importHyDcsFData(MultipartFile file); + + /** + * 查询日平均含沙量表表数据 + */ + IPage selectHyDcsFPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出日平均含沙量表表数据 + */ + R exportDaySedimentConcentration(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDpCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDpCService.java new file mode 100644 index 0000000..a3384ed --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDpCService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDpC; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDpCVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dp_c(日降水表)】的数据库操作Service +* @createDate 2024-10-15 17:40:46 +*/ +public interface HyDpCService extends IService { + + + /** + * 导入日降水表数据 + */ + void importHyDpCData(MultipartFile file); + + /** + * 查询日降水表数据 + */ + IPage selectHyDpCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出日降雨量表数据 + */ + R exportHyDpCData(HttpServletResponse response,String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqCService.java new file mode 100644 index 0000000..3563203 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqCService.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDqC; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDqCVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dq_c(日平均流量表)】的数据库操作Service +* @createDate 2024-10-16 15:40:17 +*/ +public interface HyDqCService extends IService { + /** + * 导入日平均流量表数据 + */ + void importHyDqCData(MultipartFile file); + + /** + * 查询日流量表数据 + */ + IPage selectHyDqCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出查询日流量表数据 + */ + R exportDayFlow(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqsCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqsCService.java new file mode 100644 index 0000000..28abad9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDqsCService.java @@ -0,0 +1,19 @@ +package com.ruoyi.swlscx.day.service; + +import com.ruoyi.swlscx.day.domain.po.HyDqsC; +import com.baomidou.mybatisplus.extension.service.IService; +import org.springframework.web.multipart.MultipartFile; + +/** +* @author 12974 +* @description 针对表【hy_dqs_c(日平均输沙率表)】的数据库操作Service +* @createDate 2024-10-16 16:29:38 +*/ +public interface HyDqsCService extends IService { + + /** + * 导入日平均输沙率表数据 + */ + void importHyDqsCData(MultipartFile file); + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDweCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDweCService.java new file mode 100644 index 0000000..e8ee184 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDweCService.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDweC; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDweCVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dwe_c(日水面蒸发量表)】的数据库操作Service +* @createDate 2024-10-16 17:24:19 +*/ +public interface HyDweCService extends IService { + /** + * 导入日水面蒸发量表数据 + */ + void importHyDweCData(MultipartFile file); + + /** + * 查询日水面蒸发表数据 + */ + IPage selectHyDweCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出日水面蒸发表数据 + */ + R exportHyDweCData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDwtCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDwtCService.java new file mode 100644 index 0000000..e0f7a97 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDwtCService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDwtC; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDwtCVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dwt_c(日水温表)】的数据库操作Service +* @createDate 2024-10-17 15:07:12 +*/ +public interface HyDwtCService extends IService { + + + /** + * 导入日水温表数据 + */ + void importHyDwtCData(MultipartFile file); + + /** + * 查询日水温表数据 + */ + IPage selectHyDwtCPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出日平均水位表数据 + */ + R exportDayWaterTemperature(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDzCService.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDzCService.java new file mode 100644 index 0000000..436d907 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/HyDzCService.java @@ -0,0 +1,38 @@ +package com.ruoyi.swlscx.day.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.domain.po.HyDzC; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.day.domain.vo.HyDzCVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dz_c(日平均水位表)】的数据库操作Service +* @createDate 2024-10-16 16:46:04 +*/ +public interface HyDzCService extends IService { + + + /** + * 导入日平均水位表数据 + */ + void importHyDzCData(MultipartFile file); + + + /** + * 分页查询日平均水位表数据 + */ + IPage selectHyDzCPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出日平均水位表数据 + */ + R exportDayWaterLever(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDcsFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDcsFServiceImpl.java new file mode 100644 index 0000000..34fcd1b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDcsFServiceImpl.java @@ -0,0 +1,345 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDcsFBo; +import com.ruoyi.swlscx.day.domain.po.HyDcsF; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.day.service.HyDcsFService; +import com.ruoyi.swlscx.day.mapper.HyDcsFMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dcs_f(日平均含沙量表)】的数据库操作Service实现 +* @createDate 2024-10-16 16:12:22 +*/ +@Service +public class HyDcsFServiceImpl extends ServiceImpl + implements HyDcsFService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入日平均含沙量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDcsFData(MultipartFile file) { + try { + List hyDcsFBos = ExcelUtils.readExcel(file.getInputStream(), + HyDcsFBo.class, 2); + List hyDcsFS = hyDcsFBos.stream().map(i -> { + HyDcsF hyDcsF = new HyDcsF(); + BeanUtils.copyProperties(i, hyDcsF); + hyDcsF.setDt(DateUtils.parseDate(i.getDt())); + hyDcsF.setAvcs(Convert.toBigDecimal(i.getAvcs())); + return hyDcsF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyDcsFS.size(); i += 1000) { + int end = Math.min(i + 1000, hyDcsFS.size()); + List batchList = hyDcsFS.subList(i, end); + List collect = batchList.stream().distinct().collect(Collectors.toList()); + this.baseMapper.batchInsert(collect); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询日平均含沙量表表数据 + */ + @Override + public IPage selectHyDcsFPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + + } + + @Override + public R exportDaySedimentConcentration(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "日平均含沙量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + // Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "daySedimentConcentration.xls", "日平均含沙量表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDcsFVoList) { + List records = hyDcsFVoList.stream().map(i->{ + HyDcsF hyDcsF = new HyDcsF(); + BeanUtils.copyProperties(i, hyDcsF); + hyDcsF.setDt(i.getDt()); + hyDcsF.setAvcs(Convert.toBigDecimal(i.getAvcs())); + return hyDcsF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日平均含沙量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "日平均含沙量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyDcsFPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyDcsFServiceImpl hyDcsFService = context.getBean(HyDcsFServiceImpl.class); + hyDcsFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("daySedimentConcentration.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDcsFVoIPage = this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + List records = hyDcsFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDcsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均含沙量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvcsrcd() != null ? vo.getAvcsrcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDcsFVoIPage = this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + List records = hyDcsFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDcsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均含沙量表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvcsrcd() != null ? vo.getAvcsrcd() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImpl.java new file mode 100644 index 0000000..57dff22 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDpCServiceImpl.java @@ -0,0 +1,359 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.HyStscAService; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDpCBo; +import com.ruoyi.swlscx.day.domain.po.HyDpC; +import com.ruoyi.swlscx.day.domain.vo.HyDpCVo; +import com.ruoyi.swlscx.day.service.HyDpCService; +import com.ruoyi.swlscx.day.mapper.HyDpCMapper; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.month.service.impl.HyMtweEServiceImpl; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrpFVo; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dp_c(日降水表)】的数据库操作Service实现 +* @createDate 2024-10-15 17:40:46 +*/ +@Service +public class HyDpCServiceImpl extends ServiceImpl + implements HyDpCService{ + + @Autowired + private HyStscAService hyStscAService; + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入日降水表数据 + * @param file + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDpCData(MultipartFile file) { + try { + List hyDpCBos = ExcelUtils.readExcel(file.getInputStream(), HyDpCBo.class, 2); + List hyDpCList = hyDpCBos.stream().map(i -> { + HyDpC hyDpC = new HyDpC(); + BeanUtils.copyProperties(i, hyDpC); + hyDpC.setDt(DateUtils.parseDate(i.getDt())); + hyDpC.setP(Convert.toBigDecimal(i.getP())); + return hyDpC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyDpCList.size(); i += 1000) { + int end = Math.min(i + 1000, hyDpCList.size()); + List batchList = hyDpCList.subList(i, end); + List collect = batchList.stream().distinct().collect(Collectors.toList()); + this.baseMapper.batchInsert(collect); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询日降水表数据 + */ + @Override + public IPage selectHyDpCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDpCDataByPageAndInfo(new Query().getPage(map), map); + } + + @Override + public R exportHyDpCData(HttpServletResponse response,String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "日降水表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDpCVoList) { + List records = hyDpCVoList.stream().map(i->{ + HyDpC hyDpC = new HyDpC(); + BeanUtils.copyProperties(i, hyDpC); + hyDpC.setDt((i.getDt())); + hyDpC.setP(Convert.toBigDecimal(i.getP())); + return hyDpC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日降雨量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "日降雨量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyDpCDataByPageAndInfo(new PageParams(0L, 100L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyDpCServiceImpl hyDpCService = context.getBean(HyDpCServiceImpl.class); + hyDpCService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDpCVoIPage = this.baseMapper.selectHyDpCDataByPageAndInfo(new Query().getPage(map), map); + List list = hyStscAService.list(); + List records = hyDpCVoIPage.getRecords().stream().map(i -> { + list.stream().filter(j -> j.getStcd().equals(i.getStcd())).findFirst().ifPresent(j -> { + i.setStnm(j.getStnm()); + }); + return i; + }).collect(Collectors.toList()); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDpCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日降水表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getP() != null ? vo.getP().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getPrcd() != null ? vo.getPrcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDpCVoIPage = this.baseMapper.selectHyDpCDataByPageAndInfo(new Query().getPage(map), map); + List records = hyDpCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDpCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日降水表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getP() != null ? vo.getP().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getPrcd() != null ? vo.getPrcd() : "", style); + rowIndex++; + } + } + } + + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqCServiceImpl.java new file mode 100644 index 0000000..7e3cb7a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqCServiceImpl.java @@ -0,0 +1,343 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDqCBo; +import com.ruoyi.swlscx.day.domain.po.HyDqC; +import com.ruoyi.swlscx.day.domain.vo.HyDqCVo; +import com.ruoyi.swlscx.day.service.HyDqCService; +import com.ruoyi.swlscx.day.mapper.HyDqCMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dq_c(日平均流量表)】的数据库操作Service实现 +* @createDate 2024-10-16 15:40:17 +*/ +@Service +public class HyDqCServiceImpl extends ServiceImpl + implements HyDqCService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入日平均流量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDqCData(MultipartFile file) { + try { + List hyDqCBos = ExcelUtils.readExcel(file.getInputStream(), + HyDqCBo.class, 2); + List hyDqCS = hyDqCBos.stream().map(i -> { + HyDqC hyDqC = new HyDqC(); + BeanUtils.copyProperties(i, hyDqC); + hyDqC.setDt(DateUtils.parseDate(i.getDt())); + hyDqC.setAvq(Convert.toBigDecimal(i.getAvq())); + return hyDqC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyDqCS.size(); i += 1000) { + int end = Math.min(i + 1000, hyDqCS.size()); + List batchList = hyDqCS.subList(i, end); + List collect = batchList.stream().distinct().collect(Collectors.toList()); + this.baseMapper.batchInsert(collect); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + @Override + public IPage selectHyDqCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportDayFlow(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "日平均流量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "dayFlow.xls", "日平均流量表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDqCVoList) { + List records = hyDqCVoList.stream().map(i->{ + HyDqC hyDqC = new HyDqC(); + BeanUtils.copyProperties(i, hyDqC); + hyDqC.setDt(i.getDt()); + hyDqC.setAvq(Convert.toBigDecimal(i.getAvq())); + return hyDqC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日平均流量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "日平均流量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyDqCDataByPageAndInfo(new PageParams(0L, 100000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyDqCServiceImpl hyDqCService = context.getBean(HyDqCServiceImpl.class); + hyDqCService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayFlow.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDqCVoIPage = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyDqCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDqCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均流量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvq() != null ? vo.getAvq().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDqCVoIPage = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyDqCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDqCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均流量表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvq() != null ? vo.getAvq().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqsCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqsCServiceImpl.java new file mode 100644 index 0000000..35c7343 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDqsCServiceImpl.java @@ -0,0 +1,52 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.day.domain.bo.HyDqsCBo; +import com.ruoyi.swlscx.day.domain.po.HyDqsC; +import com.ruoyi.swlscx.day.service.HyDqsCService; +import com.ruoyi.swlscx.day.mapper.HyDqsCMapper; +import org.springframework.beans.BeanUtils; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import java.util.List; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dqs_c(日平均输沙率表)】的数据库操作Service实现 +* @createDate 2024-10-16 16:29:38 +*/ +@Service +public class HyDqsCServiceImpl extends ServiceImpl + implements HyDqsCService{ + + /** + * 导入日平均输沙率表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDqsCData(MultipartFile file) { + try { + List hyDqsCBos = ExcelUtils.readExcel(file.getInputStream(), HyDqsCBo.class, 2); + List hyMtpEList = hyDqsCBos.stream().map(i -> { + HyDqsC hyDqsC = new HyDqsC(); + BeanUtils.copyProperties(i, hyDqsC); + hyDqsC.setDt(DateUtils.parseDate(i.getDt())); + hyDqsC.setAvqs(Convert.toBigDecimal(i.getAvqs())); + return hyDqsC; + }).collect(Collectors.toList()); + this.saveBatch(hyMtpEList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDweCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDweCServiceImpl.java new file mode 100644 index 0000000..bbb4bf9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDweCServiceImpl.java @@ -0,0 +1,330 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.domain.vo.HyStscAVo; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDweCBo; +import com.ruoyi.swlscx.day.domain.po.HyDweC; +import com.ruoyi.swlscx.day.domain.vo.HyDweCVo; +import com.ruoyi.swlscx.day.service.HyDweCService; +import com.ruoyi.swlscx.day.mapper.HyDweCMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dwe_c(日水面蒸发量表)】的数据库操作Service实现 +* @createDate 2024-10-16 17:24:19 +*/ +@Service +public class HyDweCServiceImpl extends ServiceImpl + implements HyDweCService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Autowired + private ApplicationContext context; + + /** + * 导入日水面蒸发量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDweCData(MultipartFile file) { + try { + List hyDweCBos = ExcelUtils.readExcel(file.getInputStream(), + HyDweCBo.class, 2); + List hyDweCList = hyDweCBos.stream().map(i -> { + HyDweC hyDweC = new HyDweC(); + BeanUtils.copyProperties(i, hyDweC); + hyDweC.setDt(DateUtils.parseDate(i.getDt())); + hyDweC.setWsfe(Convert.toBigDecimal(i.getWsfe())); + return hyDweC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyDweCList.size(); i += 1000) { + int end = Math.min(i + 1000, hyDweCList.size()); + List batchList = hyDweCList.subList(i, end); + List collect = batchList.stream().distinct().collect(Collectors.toList()); + this.baseMapper.batchInsert(collect); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询日水面蒸发表数据 + */ + @Override + @DataSource(DataSourceType.MASTER) + public IPage selectHyDweCDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDweCDataByPageAndInfo(new Query().getPage(map), map); + } + + /** + * 导出日水面蒸发表数据 + */ + @Override + public R exportHyDweCData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "日水面蒸发量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "dayEvaporationWater.xls", "日水面蒸发量表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日水面蒸发量表" + startTime + "-" + endTime+"导入到南方片"; + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage stationPageByInfo = this.selectHyDweCDataByPageAndInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + HyDweCServiceImpl hyDweCService = context.getBean(HyDweCServiceImpl.class); + hyDweCService.importMysqlToSqlserver(stationPageByInfo.getRecords()); + ycExportTask.setStatus(2); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + ycExportTask.setStatus(4); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDweCVoList) { + List records = hyDweCVoList.stream().map(i->{ + HyDweC hyDweC = new HyDweC(); + BeanUtils.copyProperties(i, hyDweC); + hyDweC.setWsfe(Convert.toBigDecimal(i.getWsfe())); + return hyDweC; + }).collect(Collectors.toList()); + this.saveBatch(records,10000); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayEvaporationWater.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDweCVoIPage = this.baseMapper.selectHyDweCDataByPageAndInfo(new Query().getPage(map), map); + List records = hyDweCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDweCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日降水表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 5, vo.getWsfe() != null ? vo.getWsfe().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDweCVoIPage = this.baseMapper.selectHyDweCDataByPageAndInfo(new Query().getPage(map), map); + List records = hyDweCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDweCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日水面蒸发量表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 5, vo.getWsfe() != null ? vo.getWsfe().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDwtCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDwtCServiceImpl.java new file mode 100644 index 0000000..13152fd --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDwtCServiceImpl.java @@ -0,0 +1,311 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDwtCBo; +import com.ruoyi.swlscx.day.domain.po.HyDwtC; +import com.ruoyi.swlscx.day.domain.vo.HyDweCVo; +import com.ruoyi.swlscx.day.domain.vo.HyDwtCVo; +import com.ruoyi.swlscx.day.service.HyDwtCService; +import com.ruoyi.swlscx.day.mapper.HyDwtCMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_dwt_c(日水温表)】的数据库操作Service实现 +* @createDate 2024-10-17 15:07:12 +*/ +@Service +public class HyDwtCServiceImpl extends ServiceImpl + implements HyDwtCService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Autowired + private ApplicationContext context; + + @Resource + private ThreadPoolExecutor importThreadPool; + + /** + * 导入日水温表数据 + */ + @Transactional(rollbackFor = Exception.class) + @Override + public void importHyDwtCData(MultipartFile file) { + try { + List hyDwtCBos = ExcelUtils.readExcel(file.getInputStream(), + HyDwtCBo.class, 2); + List hyDwtCList = hyDwtCBos.stream().map(i -> { + HyDwtC hyDwtC = new HyDwtC(); + BeanUtils.copyProperties(i, hyDwtC); + hyDwtC.setDt(DateUtils.parseDate(i.getDt())); + hyDwtC.setWtmp(Convert.toBigDecimal(i.getWtmp())); + return hyDwtC; + }).collect(Collectors.toList()); + this.saveBatch(hyDwtCList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询日水温表数据 + */ + @Override + public IPage selectHyDwtCPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDwtCPageByInfo(new Query().getPage(map), map); + + } + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDwtCVoList) { + List records = hyDwtCVoList.stream().map(i->{ + HyDwtC hyDwtC = new HyDwtC(); + com.ruoyi.common.utils.bean.BeanUtils.copyProperties(i, hyDwtC); + return hyDwtC; + }).collect(Collectors.toList()); + this.saveBatch(records,10000); + } + + + @Override + public R exportDayWaterTemperature(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "日水温表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "dayWaterTemperature.xls", "日水温表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日水温表" + startTime + "-" + endTime+"导入到南方片"; + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage stationPageByInfo = this.selectHyDwtCPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + HyDwtCServiceImpl hyDwtCService = context.getBean(HyDwtCServiceImpl.class); + hyDwtCService.importMysqlToSqlserver(stationPageByInfo.getRecords()); + ycExportTask.setStatus(2); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + ycExportTask.setStatus(4); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayWaterTemperature.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDweCVoIPage = this.baseMapper.selectHyDwtCPageByInfo(new Query().getPage(map), map); + List records = hyDweCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDwtCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日水温表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getWtmp() != null ? CommonUtils.formatNum(vo.getWtmp()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getWtmprcd() != null ? vo.getWtmprcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDweCVoIPage = this.baseMapper.selectHyDwtCPageByInfo(new Query().getPage(map), map); + List records = hyDweCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDwtCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日水温表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getWtmp() != null ? CommonUtils.formatNum(vo.getWtmp()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getWtmprcd() != null ? vo.getWtmprcd() : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDzCServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDzCServiceImpl.java new file mode 100644 index 0000000..b9540ef --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/day/service/impl/HyDzCServiceImpl.java @@ -0,0 +1,354 @@ +package com.ruoyi.swlscx.day.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.bo.HyDzCBo; +import com.ruoyi.swlscx.day.domain.po.HyDzC; +import com.ruoyi.swlscx.day.domain.vo.HyDpCVo; +import com.ruoyi.swlscx.day.domain.vo.HyDzCVo; +import com.ruoyi.swlscx.day.mapper.HyDzCMapper; +import com.ruoyi.swlscx.day.service.HyDzCService; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrzFVo; +import com.ruoyi.swlscx.year.service.impl.HyHmxpFServiceImpl; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_dz_c(日平均水位表)】的数据库操作Service实现 + * @createDate 2024-10-16 16:46:04 + */ +@Service +public class HyDzCServiceImpl extends ServiceImpl + implements HyDzCService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入日平均水位表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDzCData(MultipartFile file) { + try { + List hyDzCBos = ExcelUtils.readExcel(file.getInputStream(), + HyDzCBo.class, 2); + List hyDpCList = hyDzCBos.stream().map(i -> { + HyDzC hyDzC = new HyDzC(); + BeanUtils.copyProperties(i, hyDzC); + hyDzC.setDt(DateUtils.parseDate(i.getDt())); + hyDzC.setAvz(Convert.toBigDecimal(i.getAvz())); + return hyDzC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyDpCList.size(); i += 1000) { + int end = Math.min(i + 1000, hyDpCList.size()); + List batchList = hyDpCList.subList(i, end); + List collect = batchList.stream().distinct().collect(Collectors.toList()); + this.baseMapper.batchInsert(collect); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 分页查询日平均水位表数据 + */ + @Override + public IPage selectHyDzCPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDzCPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出日平均水位表数据 + */ + @Override + public R exportDayWaterLever(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "日平均水位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "dayWaterLever.xls", "日平均水位表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDzCVoList) { + List records = hyDzCVoList.stream().map(i->{ + HyDzC hyDzC = new HyDzC(); + BeanUtils.copyProperties(i, hyDzC); + hyDzC.setDt(i.getDt()); + hyDzC.setAvz(Convert.toBigDecimal(i.getAvz())); + return hyDzC; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日平均水位表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "日平均水位表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyDzCPageByInfo(new PageParams(0L, 100L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyDzCServiceImpl hyDzCService = context.getBean(HyDzCServiceImpl.class); + hyDzCService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayWaterLever.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDzCVoIPage = this.baseMapper.selectHyDzCPageByInfo(new Query().getPage(map), map); + List records = hyDzCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDzCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均水位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvz() != null ? vo.getAvz().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvzrcd() != null ? vo.getAvzrcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDzCVoIPage = this.baseMapper.selectHyDzCPageByInfo(new Query().getPage(map), map); + List records = hyDzCVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDzCVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日平均水位表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getDt() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getDt()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getAvz() != null ? vo.getAvz().toString() : "", style); + ExcelUtils.formatCell(row, 5, vo.getAvzrcd() != null ? vo.getAvzrcd() : "", style); + rowIndex++; + } + } + } + + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptExportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptExportController.java new file mode 100644 index 0000000..bbcaed7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptExportController.java @@ -0,0 +1,96 @@ +package com.ruoyi.swlscx.excerpt.controller; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.excerpt.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/10/18 9:19 + * @Description: TODO + * @Version + */ +@RequestMapping("/report") +@RequiredArgsConstructor +@RestController +public class ExcerptExportController { + + private final HyFdheexBService hyFdheexBService; + + private final HyPrexBService hyPrexBService; + + private final HyRvfhexBService hyRvfhexBService; + + private final HyWsfhexBService hyWsfhexBService; + + private final HyHltdzBService hyHltdzBService; + + /** + * 导出洪水水文要素摘录表 + */ + @PostMapping("/floodExcerpt") + public R exportFloodExcerpt(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyFdheexBService.exportFloodExcerpt(response, startTime, endTime, stnm, stcd); + } + + + /** + * 导出降雨量摘录表 + */ + @PostMapping("/rainExcerpt") + public R exportRainExcerpt(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyPrexBService.exportRainExcerpt(response, startTime, endTime, stnm, stcd); + } + + /** + * 导出水库洪水水文要素摘录表 + */ + @PostMapping("/rsverFloodExcerpt") + public R exportRsverFloodExcerpt(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyRvfhexBService.exportRsverFloodExcerpt(response, startTime, endTime, stnm, stcd); + } + + /** + * 导出水闸洪水水文要素摘录表 + */ + @PostMapping("/szFloodExcerpt") + public R exportSzFloodExcerpt(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyWsfhexBService.exportSzFloodExcerpt(response, startTime, endTime, stnm, stcd); + } + + + /** + * 导出逐潮高低潮位表 + */ + @PostMapping("/followingTheTideExcerpt") + public R exportFollowingTheTideExcerpt(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyHltdzBService.exportFollowingTheTideExcerpt(response, startTime, endTime, stnm, stcd); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptImportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptImportController.java new file mode 100644 index 0000000..583fb45 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptImportController.java @@ -0,0 +1,79 @@ +package com.ruoyi.swlscx.excerpt.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.vo.HyFdheexBVo; +import com.ruoyi.swlscx.excerpt.domain.vo.HyHltdzBVo; +import com.ruoyi.swlscx.excerpt.domain.vo.HyRvfhexBVo; +import com.ruoyi.swlscx.excerpt.domain.vo.HyWsfhexBVo; +import com.ruoyi.swlscx.excerpt.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/10/18 9:19 + * @Description: TODO + * @Version + */ +@RequestMapping("/excerpt") +@RequiredArgsConstructor +@RestController +public class ExcerptImportController { + + private final HyFdheexBService hyFdheexBService; + + private final HyPrexBService hyPrexBService; + + private final HyRvfhexBService hyRvfhexBService; + + private final HyWsfhexBService hyWsfhexBService; + + private final HyHltdzBService hyHltdzBService; + + + /** + *导入洪水水文要素摘录表 + */ + @PostMapping("/importFloodExcerpt") + public R importFloodExcerpt(String startTime, String endTime){ + return hyFdheexBService.importDateTosoft(startTime, endTime); + } + + /** + *导入降雨量摘录表 + */ + @PostMapping("/importRainExcerpt") + public R importRainExcerpt(String startTime, String endTime){ + return hyPrexBService.importDateTosoft(startTime, endTime); + } + + + + /** + *导入水库洪水水文要素摘录表 + */ + @PostMapping("/importRsverFloodExcerpt") + public R importRsverFloodExcerpt(String startTime, String endTime){ + return hyRvfhexBService.importDateTosoft(startTime, endTime); + } + + /** + *导入水闸洪水水文要素摘录表 + */ + @PostMapping("/importSzFloodExcerpt") + public R importSzFloodExcerpt(String startTime, String endTime){ + return hyWsfhexBService.importDateTosoft(startTime, endTime); + } + + /** + *查询逐潮高低潮位表 + */ + @PostMapping("/importFollowingTheTideExcerpt") + public R importFollowingTheTideExcerpt(String startTime, String endTime){ + return hyHltdzBService.importDateTosoft(startTime, endTime); + } +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptTableController.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptTableController.java new file mode 100644 index 0000000..67bb6b3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/controller/ExcerptTableController.java @@ -0,0 +1,111 @@ +package com.ruoyi.swlscx.excerpt.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.vo.*; +import com.ruoyi.swlscx.excerpt.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +/** + * @Author al + * @Date 2024/10/18 9:19 + * @Description: TODO + * @Version + */ +@RequestMapping("/excerpt") +@RequiredArgsConstructor +@RestController +public class ExcerptTableController { + + private final HyFdheexBService hyFdheexBService; + + private final HyPrexBService hyPrexBService; + + private final HyRvfhexBService hyRvfhexBService; + + private final HyWsfhexBService hyWsfhexBService; + + private final HyHltdzBService hyHltdzBService; + + /** + *查询洪水水文要素摘录表 + */ + @GetMapping("/getFloodExcerpt") + public R getFloodExcerpt(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyFdheexBVoIPage = hyFdheexBService.selectHyFdheexBPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyFdheexBVoIPage.getRecords()).put("count", hyFdheexBVoIPage.getTotal()); + } + + /** + *查询降雨量摘录表 + */ + @GetMapping("/getRainExcerpt") + public R getRainExcerpt(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyPrexBVoIPage = hyPrexBService.selectHyPrexBPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyPrexBVoIPage.getRecords()).put("count", hyPrexBVoIPage.getTotal()); + } + + /** + *查询水库洪水水文要素摘录表 + */ + @GetMapping("/getRsverFloodExcerpt") + public R getRsverFloodExcerpt(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyRvfhexBVoIPage = hyRvfhexBService.selectHyRvfhexBPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyRvfhexBVoIPage.getRecords()).put("count", hyRvfhexBVoIPage.getTotal()); + } + + /** + *查询水闸洪水水文要素摘录表 + */ + @GetMapping("/getSzFloodExcerpt") + public R getSzFloodExcerpt(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyWsfhexBVoIPage = hyWsfhexBService.selectHyWsfhexBPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyWsfhexBVoIPage.getRecords()).put("count", hyWsfhexBVoIPage.getTotal()); + } + + /** + *查询逐潮高低潮位表 + */ + @GetMapping("/getFollowingTheTideExcerpt") + public R getFollowingTheTideExcerpt(@RequestParam("pageNum") Integer pageNum, + @RequestParam("pageSize") Integer pageSize, + @RequestParam("startTime") String startTime, + @RequestParam("endTime") String endTime, + @RequestParam("stcd") String stcd, + @RequestParam("stnm") String stnm){ + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyHltdzBVoIPage = hyHltdzBService.selectHyHltdzBPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyHltdzBVoIPage.getRecords()).put("count", hyHltdzBVoIPage.getTotal()); + } + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyFdheexBBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyFdheexBBo.java new file mode 100644 index 0000000..85ce4a9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyFdheexBBo.java @@ -0,0 +1,52 @@ +package com.ruoyi.swlscx.excerpt.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 洪水水文要素摘录表 + * + * @TableName hy_fdheex_b + */ +@Data +public class HyFdheexBBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 时间 + */ + @ExcelProperty(value = "时间", index = 3) + private String tm; + + /** + * 水位 + */ + @ExcelProperty(value = "水位", index = 4) + private String z; + + /** + * 水位注解码 + */ + @ExcelProperty(value = "水位注解码", index = 5) + private String zrcd; + + /** + * 流量 + */ + @ExcelProperty(value = "流量", index = 6) + private String q; + + /** + * 含沙量 + */ + @ExcelProperty(value = "含沙量", index = 7) + private String s; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyHltdzBBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyHltdzBBo.java new file mode 100644 index 0000000..9cab5c6 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyHltdzBBo.java @@ -0,0 +1,64 @@ +package com.ruoyi.swlscx.excerpt.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 逐潮高低潮位表 + * + * @TableName hy_hltdz_b + */ +@Data +public class HyHltdzBBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 时间 + */ + @ExcelProperty(value = "时间", index = 3) + private String tm; + + /** + * 潮别 + */ + @ExcelProperty(value = "潮别", index = 4) + private String tdtp; + + /** + * 潮位 + */ + @ExcelProperty(value = "潮位", index = 5) + private String tdz; + + /** + * 潮位注解码 + */ + @ExcelProperty(value = "潮位注解码", index = 6) + private String tdzrcd; + + /** + * 潮差 + */ + @ExcelProperty(value = "潮差", index = 7) + private String tdrng; + + /** + * 历时 + */ + @ExcelProperty(value = "历时", index = 8) + private String dr; + + /** + * 备注 + */ + @ExcelProperty(value = "备注", index = 9) + private String nt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyPrexBBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyPrexBBo.java new file mode 100644 index 0000000..3bd1288 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyPrexBBo.java @@ -0,0 +1,46 @@ +package com.ruoyi.swlscx.excerpt.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 降水量摘录表 + * + * @TableName hy_prex_b + */ +@Data +public class HyPrexBBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 起时间 + */ + @ExcelProperty(value = "起时间", index = 3) + private String bgtm; + + /** + * 止时间 + */ + @ExcelProperty(value = "止时间", index = 4) + private String endtm; + + /** + * 降水量(毫米) + */ + @ExcelProperty(value = "降水量", index = 5) + private String p; + + /** + * 降水量注解码 + */ + @ExcelProperty(value = "降水量注解码", index = 6) + private String prcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyRvfhexBBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyRvfhexBBo.java new file mode 100644 index 0000000..05298d6 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyRvfhexBBo.java @@ -0,0 +1,64 @@ +package com.ruoyi.swlscx.excerpt.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 水库洪水水文要素摘录表 + * + * @TableName hy_rvfhex_b + */ +@Data +public class HyRvfhexBBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 时间 + */ + @ExcelProperty(value = "时间", index = 3) + private String tm; + + /** + * 坝上水位 + */ + @ExcelProperty(value = "坝上水位", index = 4) + private String dambhdz; + + /** + * 坝上水位注解码 + */ + @ExcelProperty(value = "坝上水位注解码", index = 5) + private String dambhdzrcd; + + /** + * 流量 + */ + @ExcelProperty(value = "流量", index = 6) + private String q; + + /** + * 含沙量 + */ + @ExcelProperty(value = "含沙量", index = 7) + private String s; + + /** + * 蓄水量 + */ + @ExcelProperty(value = "蓄水量", index = 8) + private String w; + + /** + * 备注 + */ + @ExcelProperty(value = "备注", index = 9) + private String nt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyWsfhexBBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyWsfhexBBo.java new file mode 100644 index 0000000..3dc08cc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/bo/HyWsfhexBBo.java @@ -0,0 +1,67 @@ +package com.ruoyi.swlscx.excerpt.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 水闸洪水水文要素摘录表 + * + * @TableName hy_wsfhex_b + */ +@Data +public class HyWsfhexBBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 时间 + */ + @ExcelProperty(value = "时间", index = 3) + private String tm; + + /** + * 闸上水位 + */ + @ExcelProperty(value = "闸上水位", index = 4) + private String upz; + + /** + * 闸上水位注解码 + */ + @ExcelProperty(value = "闸上水位注解码", index = 5) + private String upzrcd; + + /** + * 闸下水位 + */ + @ExcelProperty(value = "闸下水位", index = 6) + private String dwz; + + /** + * 闸下水位注解码 + */ + @ExcelProperty(value = "闸下水位注解码", index = 7) + private String dwzrcd; + + /** + * 流量 + */ + @ExcelProperty(value = "流量", index = 8) + private String q; + + /** + * 含沙量 + */ + @ExcelProperty(value = "含沙量", index = 9) + private String s; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyFdheexB.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyFdheexB.java new file mode 100644 index 0000000..0ad77fa --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyFdheexB.java @@ -0,0 +1,52 @@ +package com.ruoyi.swlscx.excerpt.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 洪水水文要素摘录表 + * @TableName hy_fdheex_b + */ +@TableName(value ="hy_fdheex_b") +@Data +public class HyFdheexB implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 时间 + */ + private Date tm; + + /** + * 水位 + */ + private BigDecimal z; + + /** + * 水位注解码 + + */ + private String zrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyHltdzB.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyHltdzB.java new file mode 100644 index 0000000..51ac30c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyHltdzB.java @@ -0,0 +1,61 @@ +package com.ruoyi.swlscx.excerpt.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 逐潮高低潮位表 + * @TableName hy_hltdz_b + */ +@TableName(value ="hy_hltdz_b") +@Data +public class HyHltdzB implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 时间 + */ + private Date tm; + + /** + * 潮别 + */ + private String tdtp; + + /** + * 潮位 + */ + private BigDecimal tdz; + + /** + * 潮位注解码 + */ + private String tdzrcd; + + /** + * 潮差 + */ + private BigDecimal tdrng; + + /** + * 历时 + */ + private String dr; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyPrexB.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyPrexB.java new file mode 100644 index 0000000..3600e9b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyPrexB.java @@ -0,0 +1,48 @@ +package com.ruoyi.swlscx.excerpt.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; +import lombok.EqualsAndHashCode; + +/** + * 降水量摘录表 + * @TableName hy_prex_b + */ +@TableName(value ="hy_prex_b") +@Data +@EqualsAndHashCode +public class HyPrexB implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 起时间 + */ + private Date bgtm; + + /** + * 止时间 + */ + private Date endtm; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyRvfhexB.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyRvfhexB.java new file mode 100644 index 0000000..d663907 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyRvfhexB.java @@ -0,0 +1,61 @@ +package com.ruoyi.swlscx.excerpt.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 水库洪水水文要素摘录表 + * @TableName hy_rvfhex_b + */ +@TableName(value ="hy_rvfhex_b") +@Data +public class HyRvfhexB implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 时间 + */ + private Date tm; + + /** + * 坝上水位 + */ + private BigDecimal dambhdz; + + /** + * 坝上水位注解码 + */ + private String dambhdzrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + /** + * 蓄水量 + */ + private BigDecimal w; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyWsfhexB.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyWsfhexB.java new file mode 100644 index 0000000..84a14cd --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/po/HyWsfhexB.java @@ -0,0 +1,61 @@ +package com.ruoyi.swlscx.excerpt.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 水闸洪水水文要素摘录表 + * @TableName hy_wsfhex_b + */ +@TableName(value ="hy_wsfhex_b") +@Data +public class HyWsfhexB implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 时间 + */ + private Date tm; + + /** + * 闸上水位 + */ + private BigDecimal upz; + + /** + * 闸上水位注解码 + */ + private String upzrcd; + + /** + * 闸下水位 + */ + private BigDecimal dwz; + + /** + * 闸下水位注解码 + */ + private String dwzrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyFdheexBVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyFdheexBVo.java new file mode 100644 index 0000000..62a4466 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyFdheexBVo.java @@ -0,0 +1,57 @@ + +package com.ruoyi.swlscx.excerpt.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 洪水水文要素摘录表 + * @TableName hy_fdheex_b + */ +@Data +public class HyFdheexBVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 时间 + */ + private Date tm; + + /** + * 水位 + */ + private BigDecimal z; + + /** + * 水位注解码 + + */ + private String zrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyHltdzBVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyHltdzBVo.java new file mode 100644 index 0000000..26f4161 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyHltdzBVo.java @@ -0,0 +1,64 @@ +package com.ruoyi.swlscx.excerpt.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 逐潮高低潮位表 + * @TableName hy_hltdz_b + */ +@Data +public class HyHltdzBVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 时间 + */ + private Date tm; + + /** + * 潮别 + */ + private String tdtp; + + /** + * 潮位 + */ + private BigDecimal tdz; + + /** + * 潮位注解码 + */ + private String tdzrcd; + + /** + * 潮差 + */ + private BigDecimal tdrng; + + /** + * 历时 + */ + private String dr; + + /** + * 备注 + */ + private String nt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyPrexBVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyPrexBVo.java new file mode 100644 index 0000000..b778d29 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyPrexBVo.java @@ -0,0 +1,51 @@ +package com.ruoyi.swlscx.excerpt.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; +import lombok.EqualsAndHashCode; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 降水量摘录表 + * @TableName hy_prex_b + */ +@Data +@EqualsAndHashCode +public class HyPrexBVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 起时间 + */ + private Date bgtm; + + /** + * 止时间 + */ + private Date endtm; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyRvfhexBVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyRvfhexBVo.java new file mode 100644 index 0000000..8464228 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyRvfhexBVo.java @@ -0,0 +1,65 @@ +package com.ruoyi.swlscx.excerpt.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 水库洪水水文要素摘录表 + * @TableName hy_rvfhex_b + */ +@Data +public class HyRvfhexBVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 站名 + */ + private String stnm; + + + /** + * 时间 + */ + private Date tm; + + /** + * 坝上水位 + */ + private BigDecimal dambhdz; + + /** + * 坝上水位注解码 + */ + private String dambhdzrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + /** + * 蓄水量 + */ + private BigDecimal w; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyWsfhexBVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyWsfhexBVo.java new file mode 100644 index 0000000..499388e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/domain/vo/HyWsfhexBVo.java @@ -0,0 +1,64 @@ +package com.ruoyi.swlscx.excerpt.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 水闸洪水水文要素摘录表 + * @TableName hy_wsfhex_b + */ +@Data +public class HyWsfhexBVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + + /** + * 站名 + */ + private String stnm; + + /** + * 时间 + */ + private Date tm; + + /** + * 闸上水位 + */ + private BigDecimal upz; + + /** + * 闸上水位注解码 + */ + private String upzrcd; + + /** + * 闸下水位 + */ + private BigDecimal dwz; + + /** + * 闸下水位注解码 + */ + private String dwzrcd; + + /** + * 流量 + */ + private BigDecimal q; + + /** + * 含沙量 + */ + private BigDecimal s; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyFdheexBMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyFdheexBMapper.java new file mode 100644 index 0000000..a857a01 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyFdheexBMapper.java @@ -0,0 +1,39 @@ +package com.ruoyi.swlscx.excerpt.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.excerpt.domain.po.HyFdheexB; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.excerpt.domain.po.HyPrexB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyFdheexBVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_fdheex_b(洪水水文要素摘录表)】的数据库操作Mapper +* @createDate 2024-10-17 13:15:04 +* @Entity com.ruoyi.swlscx.excerpt.domain.po.HyFdheexB +*/ +@Mapper +public interface HyFdheexBMapper extends BaseMapper { + + void deleteAll(); + + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + IPage selectHyFdheexBPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyHltdzBMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyHltdzBMapper.java new file mode 100644 index 0000000..e50ebf8 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyHltdzBMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.excerpt.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.excerpt.domain.po.HyHltdzB; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.excerpt.domain.vo.HyHltdzBVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_hltdz_b(逐潮高低潮位表)】的数据库操作Mapper +* @createDate 2024-10-17 14:28:45 +* @Entity com.ruoyi.swlscx.excerpt.domain.po.HyHltdzB +*/ +public interface HyHltdzBMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List batchList); + + void batchInsertOrUpdate(); + + IPage selectHyHltdzBPageByInfo(Page page, @Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyPrexBMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyPrexBMapper.java new file mode 100644 index 0000000..96f8e46 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyPrexBMapper.java @@ -0,0 +1,40 @@ +package com.ruoyi.swlscx.excerpt.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.day.domain.po.HyDweC; +import com.ruoyi.swlscx.excerpt.domain.po.HyPrexB; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.excerpt.domain.vo.HyPrexBVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_prex_b(降水量摘录表)】的数据库操作Mapper +* @createDate 2024-10-17 09:17:42 +* @Entity com.ruoyi.swlscx.excerpt.domain.po.HyPrexB +*/ +@Mapper +public interface HyPrexBMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List collect); + + void batchInsertOrUpdate(); + + /** + * 查询降雨量摘录表 + */ + IPage selectHyDcsFPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyRvfhexBMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyRvfhexBMapper.java new file mode 100644 index 0000000..78c4b7e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyRvfhexBMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.excerpt.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.excerpt.domain.po.HyRvfhexB; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.excerpt.domain.vo.HyRvfhexBVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_rvfhex_b(水库洪水水文要素摘录表)】的数据库操作Mapper +* @createDate 2024-10-17 13:54:18 +* @Entity com.ruoyi.swlscx.excerpt.domain.po.HyRvfhexB +*/ +public interface HyRvfhexBMapper extends BaseMapper { + + void deleteAll(); + + void batchInsert(@Param("list") List batchList); + + void batchInsertOrUpdate(); + + IPage selectHyRvfhexBPageByInfo(Page page,@Param("map") Map map); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyWsfhexBMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyWsfhexBMapper.java new file mode 100644 index 0000000..5f7287c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/mapper/HyWsfhexBMapper.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.excerpt.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.excerpt.domain.po.HyWsfhexB; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.excerpt.domain.vo.HyWsfhexBVo; +import org.apache.ibatis.annotations.Param; +import org.springframework.security.core.parameters.P; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_wsfhex_b(水闸洪水水文要素摘录表)】的数据库操作Mapper +* @createDate 2024-10-17 14:14:32 +* @Entity com.ruoyi.swlscx.excerpt.domain.po.HyWsfhexB +*/ +public interface HyWsfhexBMapper extends BaseMapper { + + /** + * 查询水闸洪水水文要素摘录表 + */ + IPage selectHyWsfhexBPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyFdheexBService.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyFdheexBService.java new file mode 100644 index 0000000..ac8eecd --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyFdheexBService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.excerpt.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.po.HyFdheexB; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.excerpt.domain.vo.HyFdheexBVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_fdheex_b(洪水水文要素摘录表)】的数据库操作Service +* @createDate 2024-10-17 13:15:04 +*/ +public interface HyFdheexBService extends IService { + + /** + * 导入洪水水文要素摘录表 + */ + void imporHyFdheexBData(MultipartFile file); + + + /** + * 查询洪水水文要素摘录表 + */ + IPage selectHyFdheexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出洪水水文要素摘录表 + */ + R exportFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyHltdzBService.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyHltdzBService.java new file mode 100644 index 0000000..20e1992 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyHltdzBService.java @@ -0,0 +1,38 @@ +package com.ruoyi.swlscx.excerpt.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.po.HyHltdzB; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.excerpt.domain.vo.HyHltdzBVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_hltdz_b(逐潮高低潮位表)】的数据库操作Service +* @createDate 2024-10-17 14:28:45 +*/ +public interface HyHltdzBService extends IService { + + + + /** + * 导入逐潮高低潮位表数据 + */ + void importHyHltdzBData(MultipartFile file); + + /** + * 查询逐潮高低潮位表 + */ + IPage selectHyHltdzBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出逐潮高低潮位表 + */ + R exportFollowingTheTideExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyPrexBService.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyPrexBService.java new file mode 100644 index 0000000..4cb2baa --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyPrexBService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.excerpt.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.po.HyPrexB; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.excerpt.domain.vo.HyPrexBVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_prex_b(降水量摘录表)】的数据库操作Service +* @createDate 2024-10-17 09:17:42 +*/ +public interface HyPrexBService extends IService { + + + /** + * 导入降水量摘录表数据 + */ + void importHyPrexBData(MultipartFile file); + + /** + * 查询降雨量摘录表 + */ + IPage selectHyPrexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出降雨量摘录表 + */ + R exportRainExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyRvfhexBService.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyRvfhexBService.java new file mode 100644 index 0000000..209f092 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyRvfhexBService.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.excerpt.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.po.HyRvfhexB; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.excerpt.domain.vo.HyRvfhexBVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_rvfhex_b(水库洪水水文要素摘录表)】的数据库操作Service +* @createDate 2024-10-17 13:54:18 +*/ +public interface HyRvfhexBService extends IService { + + /** + * 导入水库洪水水文要素摘录表 + */ + void importHyRvfhexBData(MultipartFile file); + + /** + * 查询水库洪水水文要素摘录表 + */ + IPage selectHyRvfhexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出水库洪水水文要素摘录表 + */ + R exportRsverFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyWsfhexBService.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyWsfhexBService.java new file mode 100644 index 0000000..df25646 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/HyWsfhexBService.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.excerpt.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.excerpt.domain.po.HyWsfhexB; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.excerpt.domain.vo.HyWsfhexBVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_wsfhex_b(水闸洪水水文要素摘录表)】的数据库操作Service +* @createDate 2024-10-17 14:14:32 +*/ +public interface HyWsfhexBService extends IService { + + /** + * 导入水闸洪水水文要素摘录表 + */ + void importHyWsfhexBData(MultipartFile file); + + /** + * 查询水闸洪水水文要素摘录表 + */ + IPage selectHyWsfhexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出水库洪水水文要素摘录表 + */ + R exportSzFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyFdheexBServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyFdheexBServiceImpl.java new file mode 100644 index 0000000..05df0ab --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyFdheexBServiceImpl.java @@ -0,0 +1,359 @@ +package com.ruoyi.swlscx.excerpt.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.excerpt.domain.bo.HyFdheexBBo; +import com.ruoyi.swlscx.excerpt.domain.po.HyFdheexB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyFdheexBVo; +import com.ruoyi.swlscx.excerpt.service.HyFdheexBService; +import com.ruoyi.swlscx.excerpt.mapper.HyFdheexBMapper; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.service.impl.HyHmxpFServiceImpl; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_fdheex_b(洪水水文要素摘录表)】的数据库操作Service实现 +* @createDate 2024-10-17 13:15:04 +*/ +@Service +public class HyFdheexBServiceImpl extends ServiceImpl + implements HyFdheexBService{ + + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入洪水水文要素摘录表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void imporHyFdheexBData(MultipartFile file) { + try { + List hyFdheexBBos = ExcelUtils.readExcel(file.getInputStream(), + HyFdheexBBo.class, 2); + List hyFdheexBList = hyFdheexBBos.stream().map(i -> { + HyFdheexB hyFdheexB = new HyFdheexB(); + BeanUtils.copyProperties(i, hyFdheexB); + hyFdheexB.setTm(DateUtils.parseDate(i.getTm())); + hyFdheexB.setZ(Convert.toBigDecimal(i.getZ())); + hyFdheexB.setQ(Convert.toBigDecimal(i.getQ())); + hyFdheexB.setS(Convert.toBigDecimal(i.getS())); + return hyFdheexB; + }).collect(Collectors.toList()); + List collect = hyFdheexBList.stream().distinct(). + collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < collect.size(); i += 1000) { + int end = Math.min(i + 1000, collect.size()); + List batchList = collect.subList(i, end); + this.baseMapper.batchInsert(batchList); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询洪水水文要素摘录表 + */ + @Override + public IPage selectHyFdheexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyFdheexBPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd) { + Long userId = SecurityUtils.getUserId(); + String tableName = "洪水水文要素摘录表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "floodExcerpt.xls", "洪水水文要素摘录表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyFdheexBVoList) { + List records = hyFdheexBVoList.stream().map(i->{ + HyFdheexB hyFdheexB = new HyFdheexB(); + BeanUtils.copyProperties(i, hyFdheexB); + hyFdheexB.setZ(Convert.toBigDecimal(i.getZ())); + hyFdheexB.setQ(Convert.toBigDecimal(i.getQ())); + hyFdheexB.setS(Convert.toBigDecimal(i.getS())); + return hyFdheexB; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "洪水水文要素摘录表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "洪水水文要素摘录表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyFdheexBPageByInfo(new PageParams(0L, 1000000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyFdheexBServiceImpl hyFdheexBService = context.getBean(HyFdheexBServiceImpl.class); + hyFdheexBService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("floodExcerpt.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyFdheexBVoIPage = this.baseMapper.selectHyFdheexBPageByInfo(new Query().getPage(map), map); + List records = hyFdheexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyFdheexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("洪水水文要素摘录表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getZ() != null ? CommonUtils.formatNum(vo.getZ()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getZrcd() != null ? vo.getZrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyFdheexBVoIPage = this.baseMapper.selectHyFdheexBPageByInfo(new Query().getPage(map), map); + List records = hyFdheexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyFdheexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("洪水水文要素摘录表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getZ() != null ? CommonUtils.formatNum(vo.getZ()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getZrcd() != null ? vo.getZrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyHltdzBServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyHltdzBServiceImpl.java new file mode 100644 index 0000000..70d52cb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyHltdzBServiceImpl.java @@ -0,0 +1,363 @@ +package com.ruoyi.swlscx.excerpt.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.excerpt.domain.bo.HyHltdzBBo; +import com.ruoyi.swlscx.excerpt.domain.po.HyHltdzB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyHltdzBVo; +import com.ruoyi.swlscx.excerpt.mapper.HyHltdzBMapper; +import com.ruoyi.swlscx.excerpt.service.HyHltdzBService; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.service.impl.HyHmxpFServiceImpl; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_hltdz_b(逐潮高低潮位表)】的数据库操作Service实现 + * @createDate 2024-10-17 14:28:45 + */ +@Service +public class HyHltdzBServiceImpl extends ServiceImpl + implements HyHltdzBService { + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + + /** + * 导入逐潮高低潮位表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyHltdzBData(MultipartFile file) { + try { + List hyHltdzBBos = ExcelUtils.readExcel(file.getInputStream(), + HyHltdzBBo.class, 2); + List hyHltdzBList = hyHltdzBBos.stream().map(i -> { + HyHltdzB hyHltdzB = new HyHltdzB(); + BeanUtils.copyProperties(i, hyHltdzB); + hyHltdzB.setTm(DateUtils.parseDate(i.getTm())); + hyHltdzB.setTdz(Convert.toBigDecimal(i.getTdz())); + hyHltdzB.setTdrng(Convert.toBigDecimal(i.getTdrng())); + return hyHltdzB; + }).collect(Collectors.toList()); + List collect = hyHltdzBList.stream().distinct().collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < collect.size(); i += 1000) { + int end = Math.min(i + 1000, collect.size()); + List batchList = collect.subList(i, end); + this.baseMapper.batchInsert(batchList); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询逐潮高低潮位表 + */ + @Override + public IPage selectHyHltdzBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyHltdzBPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出逐潮高低潮位表 + */ + @Override + public R exportFollowingTheTideExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd) { + Long userId = SecurityUtils.getUserId(); + String tableName = "逐潮高低潮位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + + // Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "followingTheTideExcerpt.xls", "逐潮高低潮位表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyHltdzBVoList) { + List records = hyHltdzBVoList.stream().map(i->{ + HyHltdzB hyHltdzB = new HyHltdzB(); + BeanUtils.copyProperties(i, hyHltdzB); + hyHltdzB.setTdz(Convert.toBigDecimal(i.getTdz())); + hyHltdzB.setTdrng(Convert.toBigDecimal(i.getTdrng())); + return hyHltdzB; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "逐潮高低潮位表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "逐潮高低潮位表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyHltdzBPageByInfo(new PageParams(0L, 100000000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyHltdzBServiceImpl hyHltdzBService = context.getBean(HyHltdzBServiceImpl.class); + hyHltdzBService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("followingTheTideExcerpt.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyHltdzBVoIPage = this.baseMapper.selectHyHltdzBPageByInfo(new Query().getPage(map), map); + List records = hyHltdzBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyHltdzBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("逐潮高低潮位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT, vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getTdtp() != null ? vo.getTdtp() : "", style); + ExcelUtils.formatCell(row, 5, vo.getTdz() != null ? CommonUtils.formatNum(vo.getTdz()) : "", style); + ExcelUtils.formatCell(row, 6, vo.getTdzrcd() != null ? vo.getTdzrcd() : "", style); + ExcelUtils.formatCell(row, 7, vo.getTdrng() != null ? CommonUtils.formatNum(vo.getTdrng()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getDr() != null ? vo.getDr() : "", style); + ExcelUtils.formatCell(row, 9, vo.getNt() != null ? vo.getNt() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyHltdzBVoIPage = this.baseMapper.selectHyHltdzBPageByInfo(new Query().getPage(map), map); + List records = hyHltdzBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyHltdzBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("逐潮高低潮位表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT, vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getTdtp() != null ? vo.getTdtp() : "", style); + ExcelUtils.formatCell(row, 5, vo.getTdz() != null ? CommonUtils.formatNum(vo.getTdz()) : "", style); + ExcelUtils.formatCell(row, 6, vo.getTdzrcd() != null ? vo.getTdzrcd() : "", style); + ExcelUtils.formatCell(row, 7, vo.getTdrng() != null ? CommonUtils.formatNum(vo.getTdrng()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getDr() != null ? vo.getDr() : "", style); + ExcelUtils.formatCell(row, 9, vo.getNt() != null ? vo.getNt() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyPrexBServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyPrexBServiceImpl.java new file mode 100644 index 0000000..826cea3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyPrexBServiceImpl.java @@ -0,0 +1,355 @@ +package com.ruoyi.swlscx.excerpt.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.day.domain.vo.HyDcsFVo; +import com.ruoyi.swlscx.excerpt.domain.bo.HyPrexBBo; +import com.ruoyi.swlscx.excerpt.domain.po.HyPrexB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyFdheexBVo; +import com.ruoyi.swlscx.excerpt.domain.vo.HyPrexBVo; +import com.ruoyi.swlscx.excerpt.service.HyPrexBService; +import com.ruoyi.swlscx.excerpt.mapper.HyPrexBMapper; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.service.impl.HyHmxpFServiceImpl; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_prex_b(降水量摘录表)】的数据库操作Service实现 +* @createDate 2024-10-17 09:17:42 +*/ +@Service +public class HyPrexBServiceImpl extends ServiceImpl + implements HyPrexBService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入降水量摘录表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyPrexBData(MultipartFile file) { + try { + List hyPrexBBos = ExcelUtils.readExcel(file.getInputStream(), + HyPrexBBo.class, 2); + List hyPrexBList = hyPrexBBos.stream().map(i -> { + HyPrexB hyPrexB = new HyPrexB(); + BeanUtils.copyProperties(i, hyPrexB); + hyPrexB.setBgtm(DateUtils.parseDate(i.getBgtm())); + hyPrexB.setEndtm(DateUtils.parseDate(i.getEndtm())); + hyPrexB.setP(Convert.toBigDecimal(i.getP())); + return hyPrexB; + }).collect(Collectors.toList()); + List collect = hyPrexBList.stream().distinct().collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < collect.size(); i += 1000) { + int end = Math.min(i + 1000, collect.size()); + List batchList = collect.subList(i, end); + this.baseMapper.batchInsert(batchList); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询降雨量摘录表 + */ + @Override + public IPage selectHyPrexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出降雨量摘录表 + */ + @Override + public R exportRainExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd) { + Long userId = SecurityUtils.getUserId(); + String tableName = "降水量摘录表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "rainExcerpt.xls", "降水量摘录表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyPrexBVoList) { + List records = hyPrexBVoList.stream().map(i->{ + HyPrexB hyPrexB = new HyPrexB(); + BeanUtils.copyProperties(i, hyPrexB); + hyPrexB.setP(Convert.toBigDecimal(i.getP())); + return hyPrexB; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "降水量摘录表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "降水量摘录表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyPrexBPageByInfo(new PageParams(0L, 100000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyPrexBServiceImpl hyPrexBService = context.getBean(HyPrexBServiceImpl.class); + hyPrexBService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("rainExcerpt.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyPrexBVoIPage = this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + List records = hyPrexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyPrexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("降水量摘录表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getBgtm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getBgtm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getEndtm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getEndtm()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getP() != null ? CommonUtils.formatNum(vo.getP()) : "", style); + ExcelUtils.formatCell(row, 6, vo.getPrcd() != null ? vo.getPrcd() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyPrexBVoIPage = this.baseMapper.selectHyDcsFPageByInfo(new Query().getPage(map), map); + List records = hyPrexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyPrexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("降水量摘录表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getBgtm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getBgtm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getEndtm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getEndtm()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getP() != null ? CommonUtils.formatNum(vo.getP()) : "", style); + ExcelUtils.formatCell(row, 6, vo.getPrcd() != null ? vo.getPrcd() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyRvfhexBServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyRvfhexBServiceImpl.java new file mode 100644 index 0000000..032d027 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyRvfhexBServiceImpl.java @@ -0,0 +1,359 @@ +package com.ruoyi.swlscx.excerpt.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.excerpt.domain.bo.HyRvfhexBBo; +import com.ruoyi.swlscx.excerpt.domain.po.HyRvfhexB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyRvfhexBVo; +import com.ruoyi.swlscx.excerpt.service.HyRvfhexBService; +import com.ruoyi.swlscx.excerpt.mapper.HyRvfhexBMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_rvfhex_b(水库洪水水文要素摘录表)】的数据库操作Service实现 +* @createDate 2024-10-17 13:54:18 +*/ +@Service +public class HyRvfhexBServiceImpl extends ServiceImpl + implements HyRvfhexBService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入水库洪水水文要素摘录表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyRvfhexBData(MultipartFile file) { + try { + List hyRvfhexBBos = ExcelUtils.readExcel(file.getInputStream(), + HyRvfhexBBo.class, 2); + List hyRvfhexBS = hyRvfhexBBos.stream().map(i -> { + HyRvfhexB hyRvfhexB = new HyRvfhexB(); + BeanUtils.copyProperties(i, hyRvfhexB); + hyRvfhexB.setTm(DateUtils.parseDate(i.getTm())); + hyRvfhexB.setDambhdz(Convert.toBigDecimal(i.getDambhdz())); + hyRvfhexB.setQ(Convert.toBigDecimal(i.getQ())); + hyRvfhexB.setS(Convert.toBigDecimal(i.getS())); + hyRvfhexB.setW(Convert.toBigDecimal(i.getW())); + return hyRvfhexB; + }).collect(Collectors.toList()); + List collect = hyRvfhexBS.stream().distinct().collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < collect.size(); i += 1000) { + int end = Math.min(i + 1000, collect.size()); + List batchList = collect.subList(i, end); + this.baseMapper.batchInsert(batchList); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询水库洪水水文要素摘录表 + */ + @Override + public IPage selectHyRvfhexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyRvfhexBPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportRsverFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "水库洪水水文要素摘录表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "rsverFloodExcerpt.xls", "水库洪水水文要素摘录表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyRvfhexBVoList) { + List records = hyRvfhexBVoList.stream().map(i->{ + HyRvfhexB hyRvfhexB = new HyRvfhexB(); + BeanUtils.copyProperties(i, hyRvfhexB); + hyRvfhexB.setDambhdz(Convert.toBigDecimal(i.getDambhdz())); + hyRvfhexB.setQ(Convert.toBigDecimal(i.getQ())); + hyRvfhexB.setS(Convert.toBigDecimal(i.getS())); + hyRvfhexB.setW(Convert.toBigDecimal(i.getW())); + return hyRvfhexB; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "水库洪水水文要素摘录表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "水库洪水水文要素摘录表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyRvfhexBPageByInfo(new PageParams(0L, 10000000000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyRvfhexBServiceImpl hyRvfhexBService = context.getBean(HyRvfhexBServiceImpl.class); + hyRvfhexBService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("rsverFloodExcerpt.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyRvfhexBVoIPage = this.baseMapper.selectHyRvfhexBPageByInfo(new Query().getPage(map), map); + List records = hyRvfhexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyRvfhexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("水库洪水水文要素摘录表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getDambhdz() != null ? CommonUtils.formatNum(vo.getDambhdz()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getDambhdzrcd() != null ? vo.getDambhdzrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getW() != null ? CommonUtils.formatNum(vo.getW()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getNt() != null ? vo.getNt() : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyRvfhexBVoIPage = this.baseMapper.selectHyRvfhexBPageByInfo(new Query().getPage(map), map); + List records = hyRvfhexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyRvfhexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("水库洪水水文要素摘录表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getDambhdz() != null ? CommonUtils.formatNum(vo.getDambhdz()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getDambhdzrcd() != null ? vo.getDambhdzrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getW() != null ? CommonUtils.formatNum(vo.getW()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getNt() != null ? vo.getNt() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyWsfhexBServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyWsfhexBServiceImpl.java new file mode 100644 index 0000000..9ebfc82 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/excerpt/service/impl/HyWsfhexBServiceImpl.java @@ -0,0 +1,347 @@ +package com.ruoyi.swlscx.excerpt.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.constants.SystemConstants; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.excerpt.domain.bo.HyWsfhexBBo; +import com.ruoyi.swlscx.excerpt.domain.po.HyWsfhexB; +import com.ruoyi.swlscx.excerpt.domain.vo.HyWsfhexBVo; +import com.ruoyi.swlscx.excerpt.service.HyWsfhexBService; +import com.ruoyi.swlscx.excerpt.mapper.HyWsfhexBMapper; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_wsfhex_b(水闸洪水水文要素摘录表)】的数据库操作Service实现 +* @createDate 2024-10-17 14:14:32 +*/ +@Service +public class HyWsfhexBServiceImpl extends ServiceImpl + implements HyWsfhexBService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入水闸洪水水文要素摘录表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyWsfhexBData(MultipartFile file) { + try { + List hyWsfhexBBos = ExcelUtils.readExcel(file.getInputStream(), + HyWsfhexBBo.class, 2); + List hyWsfhexBList = hyWsfhexBBos.stream().map(i -> { + HyWsfhexB hyWsfhexB = new HyWsfhexB(); + BeanUtils.copyProperties(i, hyWsfhexB); + hyWsfhexB.setTm(DateUtils.parseDate(i.getTm())); + hyWsfhexB.setUpz(Convert.toBigDecimal(i.getUpz())); + hyWsfhexB.setDwz(Convert.toBigDecimal(i.getDwz())); + hyWsfhexB.setQ(Convert.toBigDecimal(i.getQ())); + hyWsfhexB.setS(Convert.toBigDecimal(i.getS())); + return hyWsfhexB; + }).collect(Collectors.toList()); + this.saveBatch(hyWsfhexBList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询水闸洪水水文要素摘录表 + */ + @Override + public IPage selectHyWsfhexBPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyWsfhexBPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportSzFloodExcerpt(HttpServletResponse response, String startTime, String endTime, String stnm, String stcd) { + Long userId = SecurityUtils.getUserId(); + String tableName = "水闸洪水水文要素摘录表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + +// Map dataMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "szFloodExcerpt.xls", "水闸洪水水文要素摘录表", (Sheet sheet, CellStyle style) -> { +// dataModel(dataMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyWsfhexBVoList) { + List records = hyWsfhexBVoList.stream().map(i->{ + HyWsfhexB hyWsfhexB = new HyWsfhexB(); + BeanUtils.copyProperties(i, hyWsfhexB); + hyWsfhexB.setUpz(Convert.toBigDecimal(i.getUpz())); + hyWsfhexB.setDwz(Convert.toBigDecimal(i.getDwz())); + hyWsfhexB.setQ(Convert.toBigDecimal(i.getQ())); + hyWsfhexB.setS(Convert.toBigDecimal(i.getS())); + return hyWsfhexB; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "水闸洪水水文要素摘录表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "水闸洪水水文要素摘录表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyWsfhexBPageByInfo(new PageParams(0L, 100000000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyWsfhexBServiceImpl hyWsfhexBService = context.getBean(HyWsfhexBServiceImpl.class); + hyWsfhexBService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,10); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,10); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("szFloodExcerpt.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyWsfhexBVoIPage = this.baseMapper.selectHyWsfhexBPageByInfo(new Query().getPage(map), map); + List records = hyWsfhexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyWsfhexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("水闸洪水水文要素摘录表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getUpz() != null ? CommonUtils.formatNum(vo.getUpz()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getUpzrcd() != null ? vo.getUpzrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getDwz() != null ? CommonUtils.formatNum(vo.getDwz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getDwzrcd() != null ? vo.getDwzrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyWsfhexBVoIPage = this.baseMapper.selectHyWsfhexBPageByInfo(new Query().getPage(map), map); + List records = hyWsfhexBVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyWsfhexBVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("水闸洪水水文要素摘录表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getTm() != null ? DateUtils.parseDateToStr(SystemConstants.TIME_FORMAT,vo.getTm()) : "", style); + ExcelUtils.formatCell(row, 4, vo.getUpz() != null ? CommonUtils.formatNum(vo.getUpz()) : "", style); + ExcelUtils.formatCell(row, 5, vo.getUpzrcd() != null ? vo.getUpzrcd() : "", style); + ExcelUtils.formatCell(row, 6, vo.getDwz() != null ? CommonUtils.formatNum(vo.getDwz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getDwzrcd() != null ? vo.getDwzrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getQ() != null ? CommonUtils.formatNum(vo.getQ()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getS() != null ? CommonUtils.formatNum(vo.getS()) : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthExportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthExportController.java new file mode 100644 index 0000000..578605f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthExportController.java @@ -0,0 +1,136 @@ +package com.ruoyi.swlscx.month.controller; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.month.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/8/7 14:10 + * @Description: TODO + * @Version + */ + +@RequestMapping("/report") +@RequiredArgsConstructor +@RestController +public class MonthExportController { + + private final HyMtpEService hyMtpEService; + + private final HyMtweEService hyMtweEService; + + private final HyMtwtEService hyMtwtEService; + + private final HyMtqsEService hyMtqsEService; + + private final HyMtqEService hyMtqEService; + + private final HyMtzEService hyMtzEService; + + private final HyMtcsEService hyMtcsEService; + + private final HyMttdzEService hyMttdzEService; + + /** + * 导出月降水表数据 + */ + @RequestMapping("/monthRain") + public R exportMonthRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtpEService.exportHyMtpData(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月水面蒸发量表 + */ + @RequestMapping("/monthEvaporationWater") + public R exportMonthEvaporationWaterList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtweEService.exportMonthEvaporationWaterList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月水温表 + */ + @RequestMapping("/monthWaterTemperature") + public R exportMonthWaterTemperatureList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtwtEService.exportMonthWaterTemperatureList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月输沙率表 + */ + @RequestMapping("/monthSedimentDischarge") + public R exportMonthSedimentDischargeList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtqsEService.exportMonthSedimentDischargeList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月流量表 + */ + @RequestMapping("/monthFlow") + public R exportMonthFlowList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtqEService.exportMonthFlowList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月水位表 + */ + @RequestMapping("/monthWaterLever") + public R exportMonthWaterLeverList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtzEService.exportMonthWaterLeverList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月含沙量表 + */ + @RequestMapping("/monthSedimentConcentration") + public R exportMonthSedimentContentList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMtcsEService.exportMonthSedimentContentList(response, startTime, endTime, stcd, stnm); + } + + /** + * 导出月潮位表 + */ + @RequestMapping("/monthTideLever") + public R exportMonthTideLeverList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMttdzEService.exportMonthTideLeverList(response, startTime, endTime, stcd, stnm); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthTableController.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthTableController.java new file mode 100644 index 0000000..2bf9f4d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthTableController.java @@ -0,0 +1,110 @@ +package com.ruoyi.swlscx.month.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.vo.*; +import com.ruoyi.swlscx.month.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +/** + * @Author al + * @Date 2024/7/31 13:34 + * @Description: TODO + * @Version + */ + +@RestController +@RequestMapping("/month") +@RequiredArgsConstructor +public class MonthTableController { + + private final com.ruoyi.swlscx.month.service.HyMtpEService HyMtpEService; + + private final HyMtzEService hyMtzEService; + + private final HyMtqEService hyMtqEService; + + private final HyMttdzEService hyMttdzEService; + + private final HyMtweEService hyMtweEService; + + private final HyMtcsEService hyMtcsEService; + + private final HyMtqsEService hyMtqsEService; + + private final HyMtwtEService hyMtwtEService; + + /** + * 查询月降水表数据 + */ + @GetMapping("/getMonthRainList") + public R getMonthRainList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMptEVoIPage = HyMtpEService.selectHyMtpDataByPageAndInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyMptEVoIPage.getRecords()).put("count", hyMptEVoIPage.getTotal()); + } + + + /**查询月水位表数据**/ + @GetMapping("/getWaterLeverList") + public R getWaterLeverList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stnm,String stcd) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtzEVoIPage = hyMtzEService.selectHyMtzEPageByInfo(pageParams, stcdIds, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyMtzEVoIPage.getRecords()).put("count", hyMtzEVoIPage.getTotal()); + } + + /**查询月流量表数据**/ + @GetMapping("/getFlowList") + public R getFlowList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtqEVoIPage = hyMtqEService.selectHyMtqEListPageByInfo(pageParams, stcdIds, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyMtqEVoIPage.getRecords()).put("count", hyMtqEVoIPage.getTotal()); + } + + /**查询月潮位表数据**/ + @GetMapping("/getTideLeverList") + public R getTideLeverList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + return hyMttdzEService.selectHyMttdzEListPageByInfo(pageParams, stcdIds, startTime, endTime,stcd,stnm); + } + + /**查询月水面蒸发表数据**/ + @RequestMapping("/getEvaporationWaterList") + public R getEvaporationWaterList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtweEVoIPage = hyMtweEService.selectHyMtweEListPageByInfo(pageParams, stcdIds, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyMtweEVoIPage.getRecords()).put("count", hyMtweEVoIPage.getTotal()); + } + + /**查询月含沙量数据**/ + @RequestMapping("/getSedimentConcentrationList") + public R getSedimentConcentrationList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtcsEVoIPage = hyMtcsEService.selectHyMtcsEByPageAndInfo(pageParams, stcdIds, startTime, endTime, stcd, stnm); + return R.ok().put("data",hyMtcsEVoIPage.getRecords()).put("count",hyMtcsEVoIPage.getTotal()); + } + + + /**查询月输沙率表数据**/ + @RequestMapping("/getSedimentTransportRateList") + public R getSedimentTransportRateList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtqsEVoIPage = hyMtqsEService.selectHyMtqsEDataByPageAndInfo(pageParams, stcdIds, startTime, endTime, stcd, stnm); + return R.ok().put("data",hyMtqsEVoIPage.getRecords()).put("count",hyMtqsEVoIPage.getTotal()); + } + + /** + * 查询月水温表数据 + */ + @RequestMapping("/getWaterTemperatureList") + public R getWaterTemperatureList(Integer pageNum,Integer pageSize, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMtwtEVoIPage = hyMtwtEService.selectHyMtweEDataByPageAndInfo(pageParams, stcdIds, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyMtwtEVoIPage.getRecords()).put("count", hyMtwtEVoIPage.getTotal()); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthToSoftController.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthToSoftController.java new file mode 100644 index 0000000..0a70408 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/controller/MonthToSoftController.java @@ -0,0 +1,93 @@ +package com.ruoyi.swlscx.month.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.vo.*; +import com.ruoyi.swlscx.month.service.*; +import com.ruoyi.swlscx.year.service.*; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.security.core.parameters.P; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +/** + * @Author al + * @Date 2025/1/5 16:11 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/month/import") +@RequiredArgsConstructor +@Slf4j +public class MonthToSoftController { + + private final com.ruoyi.swlscx.month.service.HyMtpEService HyMtpEService; + + private final HyMtzEService hyMtzEService; + + private final HyMtqEService hyMtqEService; + + private final HyMttdzEService hyMttdzEService; + + private final HyMtweEService hyMtweEService; + + private final HyMtcsEService hyMtcsEService; + + private final HyMtqsEService hyMtqsEService; + + private final HyMtwtEService hyMtwtEService; + + /** + * 导入月降水表数据 + */ + @PostMapping("/importMonthRainList") + public R importMonthRainList(String startTime, String endTime) { + return HyMtpEService.importDateTosoft(startTime, endTime); + } + + /**导入月水面蒸发表数据**/ + @RequestMapping("/importEvaporationWaterList") + public R importEvaporationWaterList(String startTime, String endTime) { + return hyMtweEService.importDateTosoft(startTime, endTime); + } + + /**导入月水位表数据**/ + @PostMapping("/importWaterLeverList") + public R getWaterLeverList(String startTime, String endTime) { + return hyMtzEService.importDateTosoft(startTime, endTime); + } + /**导入月含沙量数据**/ + @RequestMapping("/importSedimentConcentrationList") + public R importSedimentConcentrationList(String startTime, String endTime) { + return hyMtcsEService.importDateTosoft(startTime, endTime); + } + + + /**导入月流量表数据**/ + @RequestMapping("/importFlowList") + public R getFlowList(String startTime, String endTime) { + return hyMtqEService.importDateTosoft(startTime, endTime); + } + + /** + * 导入月水温表数据 + */ + @RequestMapping("/importWaterTemperatureList") + public R importWaterTemperatureList(String startTime, String endTime) { + return hyMtwtEService.importDateTosoft(startTime, endTime); + + } + + + /**查询月输沙率表数据**/ + @RequestMapping("/importSedimentTransportRateList") + public R importSedimentTransportRateList(String startTime, String endTime) { + return hyMtqsEService.importDateTosoft(startTime, endTime); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyHmxpFExcelBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyHmxpFExcelBo.java new file mode 100644 index 0000000..322b7fb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyHmxpFExcelBo.java @@ -0,0 +1,62 @@ +package com.ruoyi.swlscx.month.domain.bo; + + +import com.alibaba.excel.annotation.ExcelProperty; +import com.alibaba.excel.annotation.format.DateTimeFormat; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @author 12974 + * @TableName hy_hmxp_f + */ +@TableName(value ="hy_hmxp_f") +@Data +public class HyHmxpFExcelBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 最大降水量时段长(小时) + */ + @ExcelProperty(value = "最大降水量时段长",index = 7) + private Integer mxpdr; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private Integer yr; + + /** + * 起时间 + */ + @ExcelProperty(value = "起时间",index = 6) + private String bgtm; + + /** + * 最大降水量(毫米) + */ + @ExcelProperty(value = "最大降水量",index = 8) + private String mxp; + + /** + * 最大降水量注解码 + */ + @ExcelProperty(value = "最大降水量注解码",index = 9) + private String mxprc; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; + + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtcsEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtcsEBo.java new file mode 100644 index 0000000..0c3ba53 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtcsEBo.java @@ -0,0 +1,85 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月含沙量表 + * @TableName hy_mtcs_e + */ +@Data +public class HyMtcsEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均含沙量(千克每立方米) + */ + @ExcelProperty(value = "平均含沙量",index = 7) + private String avcs; + + /** + * 平均含沙量注解码 + */ + @ExcelProperty(value = "平均含沙量注解码",index = 8) + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + @ExcelProperty(value = "最大含沙量",index = 9) + private String mxs; + + /** + * 最大含沙量注解码 + */ + @ExcelProperty(value = "最大含沙量注解码",index = 10) + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + @ExcelProperty(value = "最大含沙量日期",index = 11) + private String mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + @ExcelProperty(value = "最小含沙量",index = 12) + private String mns; + + /** + * 最小含沙量注解码 + */ + @ExcelProperty(value = "最小含沙量注解码",index = 13) + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + @ExcelProperty(value = "最小含沙量日期",index = 14) + private String mnsdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpEBo.java new file mode 100644 index 0000000..8f12f92 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpEBo.java @@ -0,0 +1,83 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * @Author al + * @Date 2024/7/30 15:52 + * @Description: TODO + * @Version + */ +@Data +public class HyMtpEBo implements Serializable { + + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 4) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 5) + private String mth; + + /** + * 降水量(毫米) + */ + @ExcelProperty(value = "降水量",index = 6) + private String p; + + /** + * 降水量注解码 + */ + @ExcelProperty(value = "降水量注解码",index = 7) + private String prcd; + + /** + * 降水日数 + */ + @ExcelProperty(value = "降水日数",index = 8) + private String pdynum; + + /** + * 降水日数注解码 + */ + @ExcelProperty(value = "降水日数注解码",index = 9) + private String pdynumrcd; + + /** + * 最大降水量(毫米) + */ + @ExcelProperty(value = "最大日降水量",index = 10) + private String mxdyp; + + /** + * 最大降水量注解码 + */ + @ExcelProperty(value = "最大日降水量注解码",index = 11) + private String mxdyprcd; + + /** + * 最大日降水量出现日期 + */ + @ExcelProperty(value = "最大日降水量出现日期",index = 12) + private String mxdypodt; + + private static final long serialVersionUID = 1L; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpddbEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpddbEBo.java new file mode 100644 index 0000000..4b42624 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtpddbEBo.java @@ -0,0 +1,47 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; + +/** + * 月平均泥沙颗粒级配表 + * @TableName hy_mtpddb_e + */ +@Data +public class HyMtpddbEBo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private String yr; + + /** + * 月 + */ + private String mth; + + /** + * 上限粒径(毫米) + */ + private String ltpd; + + /** + * 平均沙重百分数 + */ + private String avswpct; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqEBo.java new file mode 100644 index 0000000..b440d98 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqEBo.java @@ -0,0 +1,86 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月流量表 + * @TableName hy_mtq_e + */ +@Data +public class HyMtqEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均流量 + */ + @ExcelProperty(value = "平均流量",index = 7) + private String avq; + + /** + * 平均流量注解码 + */ + @ExcelProperty(value = "平均流量注解码",index = 8) + private String avqrcd; + + /** + * 最高流量 + */ + @ExcelProperty(value = "最高流量",index = 9) + private String mxq; + + /** + * 最高流量注解码 + */ + @ExcelProperty(value = "最高流量注解码",index = 10) + private String mxqrcd; + + /** + * 最高流量日期 + */ + @ExcelProperty(value = "最高流量日期",index = 11) + private String mxqdt; + + /** + * 最小流量 + */ + @ExcelProperty(value = "最小流量",index = 12) + private String mnq; + + /** + * 最小流量注解码 + */ + @ExcelProperty(value = "最小流量注解码",index = 13) + private String mnqrcd; + + /** + * 最小流量日期 + */ + @ExcelProperty(value = "最小流量日期",index = 14) + private String mnqdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqsEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqsEBo.java new file mode 100644 index 0000000..3652f11 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtqsEBo.java @@ -0,0 +1,75 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月输沙率表 + * @TableName hy_mtqs_e + */ +@TableName(value ="hy_mtqs_e") +@Data +public class HyMtqsEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 泥沙类型 + */ + @ExcelProperty(value = "泥沙类型",index = 7) + private String sdtp; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均输沙率(千克每秒) + */ + @ExcelProperty(value = "平均输沙率",index = 8) + private String avqs; + + /** + * 平均输沙率注解码 + */ + @ExcelProperty(value = "平均输沙率注解码",index = 9) + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + @ExcelProperty(value = "最大日平均沙率",index = 10) + private String mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + @ExcelProperty(value = "最大日平均沙率注解码",index = 11) + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + @ExcelProperty(value = "最大日平均沙率出现日期",index = 12) + private String mxdyqsodt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMttdzEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMttdzEBo.java new file mode 100644 index 0000000..ad571be --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMttdzEBo.java @@ -0,0 +1,361 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName hy_mttdz_e + */ +@Data +public class HyMttdzEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均高潮潮位 + */ + @ExcelProperty(value = "平均高潮潮位",index = 7) + private String avhtdz; + + /** + * 平均高潮潮位注解码 + */ + @ExcelProperty(value = "平均高潮潮位注解码",index = 8) + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + @ExcelProperty(value = "最高高潮位",index = 9) + private String hthtdz; + + /** + * 最高高潮位注解码 + */ + @ExcelProperty(value = "最高高潮位注解码",index = 10) + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + @ExcelProperty(value = "最高高潮位出现时间",index = 11) + private String hthtdzotm; + + /** + * 最低高潮位 + */ + @ExcelProperty(value = "最低高潮位",index = 12) + private String lthtdz; + + /** + * 最低高潮位注解码 + */ + @ExcelProperty(value = "最低高潮位注解码",index = 13) + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + @ExcelProperty(value = "最低高潮位出现时间",index = 14) + private String lthtdzotm; + + /** + * 平均低潮潮位 + */ + @ExcelProperty(value = "平均低潮潮位",index = 15) + private String avltdz; + + /** + * 平均低潮潮位注解码 + */ + @ExcelProperty(value = "平均低潮潮位注解码",index = 16) + private String avltdzrcd; + + /** + * 最高低潮位 + */ + @ExcelProperty(value = "最高低潮位",index = 17) + private String htltdz; + + /** + * 最高低潮位注解码 + */ + @ExcelProperty(value = "最高低潮位注解码",index = 18) + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + @ExcelProperty(value = "最高低潮潮位出现时间",index = 19) + private String htltdzotm; + + /** + * 最低低潮位 + */ + @ExcelProperty(value = "最低低潮位",index = 20) + private String ltltdz; + + /** + * 最低低潮潮位注解码 + */ + @ExcelProperty(value = "最低低潮潮位注解码",index = 21) + private String ltltdzrcd; + + /** + * 最低地潮位出现时间 + */ + @ExcelProperty(value = "最低地潮位出现时间",index = 22) + private String ltltdzotm; + + /** + * 平均涨潮潮差 + */ + @ExcelProperty(value = "平均涨潮潮差",index = 23) + private String avftdr; + + /** + * 平均涨潮潮差注解码 + */ + @ExcelProperty(value = "平均涨潮潮差注解码",index = 24) + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + @ExcelProperty(value = "最大涨潮潮差",index = 25) + private String mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + @ExcelProperty(value = "最大涨潮潮差注解码",index = 26) + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最大涨潮潮差(高潮)时间",index = 27) + private String mxfltdrhtm; + + /** + * 最小潮潮差 + */ + @ExcelProperty(value = "最小潮潮差",index = 28) + private String mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + @ExcelProperty(value = "最小涨潮潮差注解码",index = 29) + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最小涨潮潮差(高潮)时间",index = 30) + private String mnfltdrhtm; + + /** + * 平均涨潮潮差 + */ + @ExcelProperty(value = "平均涨潮潮差",index = 31) + private String aver; + + /** + * 平均涨潮潮差注解码 + */ + @ExcelProperty(value = "平均涨潮潮差注解码",index = 32) + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + @ExcelProperty(value = "最大落潮潮差",index = 33) + private String mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + @ExcelProperty(value = "最大落潮潮差注解码",index = 34) + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最大落潮潮差(高潮)时间",index = 35) + private String mxebtdrht; + + /** + * 最小落潮潮差 + */ + @ExcelProperty(value = "最小落潮潮差",index = 36) + private String mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + @ExcelProperty(value = "最小落潮潮差注解码",index = 37) + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最小落潮潮差(高潮)时间",index = 38) + private String mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + @ExcelProperty(value = "平均涨潮历时",index = 39) + private String avftd; + + /** + * 平均涨潮历时注解码 + */ + @ExcelProperty(value = "平均涨潮历时注解码",index = 40) + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + @ExcelProperty(value = "最长涨潮历时",index = 41) + private String mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + @ExcelProperty(value = "最长涨潮历时注解码",index = 42) + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + @ExcelProperty(value = "最长涨潮历时(高潮)时间",index = 43) + private String mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + @ExcelProperty(value = "最短涨潮历时",index = 44) + private String mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + @ExcelProperty(value = "最短涨潮历时注解码",index = 45) + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + @ExcelProperty(value = "最短涨潮历时(高潮)时间",index = 46) + private String mnfltddrhtm; + + /** + * 平均落潮历时 + */ + @ExcelProperty(value = "平均落潮历时",index = 47) + private String avebbdr; + + /** + * 平均落潮历时注解码 + */ + @ExcelProperty(value = "平均落潮历时注解码",index = 48) + private String avedrc; + + /** + * 最长落潮历时 + */ + @ExcelProperty(value = "最长落潮历时",index = 49) + private String mxebtddr; + + /** + * 最长落潮历时注解码 + */ + @ExcelProperty(value = "最长落潮历时注解码",index = 50) + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + @ExcelProperty(value = "最长落潮历时(高潮)时间",index = 51) + private String mxebtddrhtm; + + /** + * 最短落潮历时 + */ + @ExcelProperty(value = "最短落潮历时",index = 52) + private String mnbtddr; + + /** + * 最短落潮历时注解码 + */ + @ExcelProperty(value = "最短落潮历时注解码",index = 53) + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + @ExcelProperty(value = "最短落潮历时(高潮)时间",index = 54) + private String mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + @ExcelProperty(value = "逐时平均潮位",index = 55) + private String hravtdz; + + /** + * 逐时平均潮位注解码 + */ + @ExcelProperty(value = "逐时平均潮位注解码",index = 56) + private String hravtdzrcd; + + /** + * 平均潮差 + */ + @ExcelProperty(value = "平均潮差",index = 57) + private String avtdr; + + /** + * 平均潮差注解码 + */ + @ExcelProperty(value = "平均潮差注解码",index = 58) + private String avtdrrcd; + + /** + * 平均历时 + */ + @ExcelProperty(value = "平均历时",index = 59) + private String avtddr; + + /** + * 平均历时注解码 + */ + @ExcelProperty(value = "平均历时注解码",index = 60) + private String avtddrrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtweEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtweEBo.java new file mode 100644 index 0000000..7719a55 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtweEBo.java @@ -0,0 +1,79 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; + +/** + * 月水面蒸发量表 + * @TableName hy_mtwe_e + */ +@Data +public class HyMtweEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 蒸发器型式 + */ + @ExcelProperty(value = "蒸发器型式",index = 5) + private String eetp; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 6) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 7) + private String mth; + + /** + * 水面蒸发量(毫米) + */ + @ExcelProperty(value = "水面蒸发量",index = 8) + private String wsfe; + + /** + * 水面蒸发量注解码 + */ + @ExcelProperty(value = "水面蒸发量注解码",index = 9) + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + @ExcelProperty(value = "最大日水面蒸发量",index = 10) + private String mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + @ExcelProperty(value = "最大日水面蒸发量注解码",index = 11) + private String mxdyercd; + + /** + * 最小日水面蒸发量 + */ + @ExcelProperty(value = "最小日水面蒸发量",index = 12) + private String mndye; + + /** + * 最小日水面蒸发量注解码 + */ + @ExcelProperty(value = "最小日水面蒸发量注解码",index = 13) + private String mndyercd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtwtEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtwtEBo.java new file mode 100644 index 0000000..14774bc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtwtEBo.java @@ -0,0 +1,85 @@ +package com.ruoyi.swlscx.month.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月水温表字段定义 + * @TableName hy_mtwt_e + */ +@Data +public class HyMtwtEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均水温(摄氏度) + */ + @ExcelProperty(value = "平均水温",index = 7) + private String avwtmp; + + /** + * 平均水温注解码 + */ + @ExcelProperty(value = "平均水温注解码",index = 8) + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + @ExcelProperty(value = "最高水温",index = 9) + private String mxwtmp; + + /** + * 最高水温注解码 + */ + @ExcelProperty(value = "最高水温注解码",index = 10) + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + @ExcelProperty(value = "最高水温日期",index = 11) + private String mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + @ExcelProperty(value = "最低水温",index = 12) + private String mnwtmp; + + /** + * 最低水温注解码 + */ + @ExcelProperty(value = "最低水温注解码",index = 13) + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + @ExcelProperty(value = "最低水温日期",index = 14) + private String mnwtmpdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtzEBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtzEBo.java new file mode 100644 index 0000000..b42deeb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/bo/HyMtzEBo.java @@ -0,0 +1,87 @@ +package com.ruoyi.swlscx.month.domain.bo; + + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName 月水位表 + */ +@Data +public class HyMtzEBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 月 + */ + @ExcelProperty(value = "月",index = 6) + private String mth; + + /** + * 平均水位 + */ + @ExcelProperty(value = "平均水位",index = 7) + private String avz; + + /** + * 平均水位注解码 + */ + @ExcelProperty(value = "平均水位注解码",index = 8) + private String avzrcd; + + /** + * 最高水位 + */ + @ExcelProperty(value = "最高水位",index = 9) + private String htz; + + /** + * 最高水位注解码 + */ + @ExcelProperty(value = "最高水位注解码",index = 10) + private String htzrcd; + + /** + * 最高水位日期 + */ + @ExcelProperty(value = "最高水位日期",index = 11) + private String htzdt; + + /** + * 最低水位 + */ + @ExcelProperty(value = "最低水位",index = 12) + private String mnz; + + /** + * 最低水位注解码 + */ + @ExcelProperty(value = "最低水位注解码",index = 13) + private String mnzrcd; + + /** + * 最低水位日期 + */ + @ExcelProperty(value = "最低水位日期",index = 14) + private String mnzdt; + + private static final long serialVersionUID = 1L; + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtchpdE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtchpdE.java new file mode 100644 index 0000000..1e6b8ba --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtchpdE.java @@ -0,0 +1,60 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import lombok.Data; + +/** + * 月泥沙特征粒径表字段定义 + * @TableName hy_mtchpd_e + */ +@TableName(value ="hy_mtchpd_e") +@Data +public class HyMtchpdE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 中数粒径(毫米) + */ + private BigDecimal mdpd; + + /** + * 平均粒径(毫米) + */ + private BigDecimal avpd; + + /** + * 最大粒径(毫米) + */ + private BigDecimal mxpd; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtcsE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtcsE.java new file mode 100644 index 0000000..9920613 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtcsE.java @@ -0,0 +1,76 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 月含沙量表 + * @TableName hy_mtcs_e + */ +@TableName(value ="hy_mtcs_e") +@Data +public class HyMtcsE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + private BigDecimal mxs; + + /** + * 最大含沙量注解码 + */ + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + private Date mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + private BigDecimal mns; + + /** + * 最小含沙量注解码 + */ + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + private Date mnsdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtiqE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtiqE.java new file mode 100644 index 0000000..678566a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtiqE.java @@ -0,0 +1,61 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 月冰流量表 + * @TableName hy_mtiq_e + */ +@TableName(value ="hy_mtiq_e") +@Data +public class HyMtiqE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 总冰流量(万立方米) + */ + private BigDecimal tiq; + + /** + * 总冰流量注解码 + */ + private String tiqrcd; + + /** + * 最大冰流量(万立方米) + */ + private BigDecimal mxiq; + + /** + * 最大冰流量注解码 + */ + private String mxiqrcd; + + /** + * 最大冰流量日期 + */ + private Date mxiqdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpE.java new file mode 100644 index 0000000..76eb858 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpE.java @@ -0,0 +1,270 @@ +package com.ruoyi.swlscx.month.domain.po; + + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月降水表 + * @TableName hy_mtp_e + */ +@TableName(value ="hy_mtp_e") +public class HyMtpE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + /** + * 降水日数 + */ + private String pdynum; + + /** + * 降水日数注解码 + */ + private String pdynumrcd; + + /** + * 最大降水量(毫米) + */ + private BigDecimal mxdyp; + + /** + * 最大降水量注解码 + */ + private String mxdyprcd; + + /** + * 最大日降水量出现日期 + */ + private Date mxdypodt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; + + /** + * 站码 + */ + public String getStcd() { + return stcd; + } + + /** + * 站码 + */ + public void setStcd(String stcd) { + this.stcd = stcd; + } + + /** + * 年 + */ + public Integer getYr() { + return yr; + } + + /** + * 年 + */ + public void setYr(Integer yr) { + this.yr = yr; + } + + /** + * 月 + */ + public Integer getMth() { + return mth; + } + + /** + * 月 + */ + public void setMth(Integer mth) { + this.mth = mth; + } + + /** + * 降水量(毫米) + */ + public BigDecimal getP() { + return p; + } + + /** + * 降水量(毫米) + */ + public void setP(BigDecimal p) { + this.p = p; + } + + /** + * 降水量注解码 + */ + public String getPrcd() { + return prcd; + } + + /** + * 降水量注解码 + */ + public void setPrcd(String prcd) { + this.prcd = prcd; + } + + /** + * 降水日数 + */ + public String getPdynum() { + return pdynum; + } + + /** + * 降水日数 + */ + public void setPdynum(String pdynum) { + this.pdynum = pdynum; + } + + /** + * 降水日数注解码 + */ + public String getPdynumrcd() { + return pdynumrcd; + } + + /** + * 降水日数注解码 + */ + public void setPdynumrcd(String pdynumrcd) { + this.pdynumrcd = pdynumrcd; + } + + /** + * 最大降水量(毫米) + */ + public BigDecimal getMxdyp() { + return mxdyp; + } + + /** + * 最大降水量(毫米) + */ + public void setMxdyp(BigDecimal mxdyp) { + this.mxdyp = mxdyp; + } + + /** + * 最大降水量注解码 + */ + public String getMxdyprcd() { + return mxdyprcd; + } + + /** + * 最大降水量注解码 + */ + public void setMxdyprcd(String mxdyprcd) { + this.mxdyprcd = mxdyprcd; + } + + /** + * 最大日降水量出现日期 + */ + public Date getMxdypodt() { + return mxdypodt; + } + + /** + * 最大日降水量出现日期 + */ + public void setMxdypodt(Date mxdypodt) { + this.mxdypodt = mxdypodt; + } + + @Override + public boolean equals(Object that) { + if (this == that) { + return true; + } + if (that == null) { + return false; + } + if (getClass() != that.getClass()) { + return false; + } + HyMtpE other = (HyMtpE) that; + return (this.getStcd() == null ? other.getStcd() == null : this.getStcd().equals(other.getStcd())) + && (this.getYr() == null ? other.getYr() == null : this.getYr().equals(other.getYr())) + && (this.getMth() == null ? other.getMth() == null : this.getMth().equals(other.getMth())) + && (this.getP() == null ? other.getP() == null : this.getP().equals(other.getP())) + && (this.getPrcd() == null ? other.getPrcd() == null : this.getPrcd().equals(other.getPrcd())) + && (this.getPdynum() == null ? other.getPdynum() == null : this.getPdynum().equals(other.getPdynum())) + && (this.getPdynumrcd() == null ? other.getPdynumrcd() == null : this.getPdynumrcd().equals(other.getPdynumrcd())) + && (this.getMxdyp() == null ? other.getMxdyp() == null : this.getMxdyp().equals(other.getMxdyp())) + && (this.getMxdyprcd() == null ? other.getMxdyprcd() == null : this.getMxdyprcd().equals(other.getMxdyprcd())) + && (this.getMxdypodt() == null ? other.getMxdypodt() == null : this.getMxdypodt().equals(other.getMxdypodt())); + } + + @Override + public int hashCode() { + final int prime = 31; + int result = 1; + result = prime * result + ((getStcd() == null) ? 0 : getStcd().hashCode()); + result = prime * result + ((getYr() == null) ? 0 : getYr().hashCode()); + result = prime * result + ((getMth() == null) ? 0 : getMth().hashCode()); + result = prime * result + ((getP() == null) ? 0 : getP().hashCode()); + result = prime * result + ((getPrcd() == null) ? 0 : getPrcd().hashCode()); + result = prime * result + ((getPdynum() == null) ? 0 : getPdynum().hashCode()); + result = prime * result + ((getPdynumrcd() == null) ? 0 : getPdynumrcd().hashCode()); + result = prime * result + ((getMxdyp() == null) ? 0 : getMxdyp().hashCode()); + result = prime * result + ((getMxdyprcd() == null) ? 0 : getMxdyprcd().hashCode()); + result = prime * result + ((getMxdypodt() == null) ? 0 : getMxdypodt().hashCode()); + return result; + } + + @Override + public String toString() { + StringBuilder sb = new StringBuilder(); + sb.append(getClass().getSimpleName()); + sb.append(" ["); + sb.append("Hash = ").append(hashCode()); + sb.append(", stcd=").append(stcd); + sb.append(", yr=").append(yr); + sb.append(", mth=").append(mth); + sb.append(", p=").append(p); + sb.append(", prcd=").append(prcd); + sb.append(", pdynum=").append(pdynum); + sb.append(", pdynumrcd=").append(pdynumrcd); + sb.append(", mxdyp=").append(mxdyp); + sb.append(", mxdyprcd=").append(mxdyprcd); + sb.append(", mxdypodt=").append(mxdypodt); + sb.append(", serialVersionUID=").append(serialVersionUID); + sb.append("]"); + return sb.toString(); + } +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpddbE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpddbE.java new file mode 100644 index 0000000..3a73edb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtpddbE.java @@ -0,0 +1,49 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import lombok.Data; + +/** + * 月平均泥沙颗粒级配表 + * @TableName hy_mtpddb_e + */ +@TableName(value ="hy_mtpddb_e") +@Data +public class HyMtpddbE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 上限粒径(毫米) + */ + private BigDecimal ltpd; + + /** + * 平均沙重百分数 + */ + private BigDecimal avswpct; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqE.java new file mode 100644 index 0000000..92009fc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqE.java @@ -0,0 +1,76 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 月流量表 + * @TableName hy_mtq_e + */ +@TableName(value ="hy_mtq_e") +@Data +public class HyMtqE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均流量 + */ + private BigDecimal avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + /** + * 最高流量 + */ + private BigDecimal mxq; + + /** + * 最高流量注解码 + */ + private String mxqrcd; + + /** + * 最高流量日期 + */ + private Date mxqdt; + + /** + * 最小流量 + */ + private BigDecimal mnq; + + /** + * 最小流量注解码 + */ + private String mnqrcd; + + /** + * 最小流量日期 + */ + private Date mnqdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqsE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqsE.java new file mode 100644 index 0000000..0bfde30 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtqsE.java @@ -0,0 +1,66 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 月输沙率表 + * @TableName hy_mtqs_e + */ +@TableName(value ="hy_mtqs_e") +@Data +public class HyMtqsE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均输沙率(千克每秒) + */ + private BigDecimal avqs; + + /** + * 平均输沙率注解码 + */ + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + private BigDecimal mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + private Date mxdyqsodt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMttdzE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMttdzE.java new file mode 100644 index 0000000..6c79dea --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMttdzE.java @@ -0,0 +1,306 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * + * @TableName hy_mttdz_e + */ +@TableName(value ="hy_mttdz_e") +@Data +public class HyMttdzE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均高潮潮位 + */ + private BigDecimal avhtdz; + + /** + * 平均高潮潮位注解码 + */ + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + private BigDecimal hthtdz; + + /** + * 最高高潮位注解码 + */ + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + private Date hthtdzotm; + + /** + * 最低高潮位 + */ + private BigDecimal lthtdz; + + /** + * 最低高潮位注解码 + */ + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + private Date lthtdzotm; + + /** + * 平均低潮潮位 + */ + private BigDecimal avltdz; + + /** + * 平均低潮潮位注解码 + */ + private String avltdzrcd; + + /** + * 最高低潮位 + */ + private BigDecimal htltdz; + + /** + * 最高低潮位注解码 + */ + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + private Date htltdzotm; + + /** + * 最低低潮位 + */ + private BigDecimal ltltdz; + + /** + * 最低低潮潮位注解码 + */ + private String ltltdzrcd; + + /** + * 最低地潮位出现时间 + */ + private Date ltltdzotm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal avftdr; + + /** + * 平均涨潮潮差注解码 + */ + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + private BigDecimal mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + private Date mxfltdrhtm; + + /** + * 最小潮潮差 + */ + private BigDecimal mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + private Date mnfltdrhtm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal aver; + + /** + * 平均涨潮潮差注解码 + */ + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + private BigDecimal mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + private Date mxebtdrht; + + /** + * 最小落潮潮差 + */ + private BigDecimal mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + private Date mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + private Integer avftd; + + /** + * 平均涨潮历时注解码 + */ + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + private Integer mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + private Date mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + private Integer mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + private Date mnfltddrhtm; + + /** + * 平均落潮历时 + */ + private Integer avebbdr; + + /** + * 平均落潮历时注解码 + */ + private String avedrc; + + /** + * 最长落潮历时 + */ + private Integer mxebtddr; + + /** + * 最长落潮历时注解码 + */ + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + private Date mxebtddrhtm; + + /** + * 最短落潮历时 + */ + private Integer mnbtddr; + + /** + * 最短落潮历时注解码 + */ + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + private Date mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + private BigDecimal hravtdz; + + /** + * 逐时平均潮位注解码 + */ + private String hravtdzrcd; + + /** + * 平均潮差 + */ + private BigDecimal avtdr; + + /** + * 平均潮差注解码 + */ + private String avtdrrcd; + + /** + * 平均历时 + */ + private Integer avtddr; + + /** + * 平均历时注解码 + */ + private String avtddrrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtweE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtweE.java new file mode 100644 index 0000000..b02d188 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtweE.java @@ -0,0 +1,70 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import lombok.Data; + +/** + * 月水面蒸发量表 + * @TableName hy_mtwe_e + */ +@TableName(value ="hy_mtwe_e") +@Data +public class HyMtweE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 水面蒸发量(毫米) + */ + private BigDecimal wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + private BigDecimal mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + private String mxdyercd; + + /** + * 最小日水面蒸发量 + */ + private BigDecimal mndye; + + /** + * 最小日水面蒸发量注解码 + */ + private String mndyercd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtwtE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtwtE.java new file mode 100644 index 0000000..8bbfec1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtwtE.java @@ -0,0 +1,76 @@ +package com.ruoyi.swlscx.month.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 月水温表字段定义 + * @TableName hy_mtwt_e + */ +@TableName(value ="hy_mtwt_e") +@Data +public class HyMtwtE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均水温(摄氏度) + */ + private BigDecimal avwtmp; + + /** + * 平均水温注解码 + */ + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + private BigDecimal mxwtmp; + + /** + * 最高水温注解码 + */ + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + private Date mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + private BigDecimal mnwtmp; + + /** + * 最低水温注解码 + */ + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + private Date mnwtmpdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtzE.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtzE.java new file mode 100644 index 0000000..3f008b2 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/po/HyMtzE.java @@ -0,0 +1,292 @@ +package com.ruoyi.swlscx.month.domain.po; + + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName 月水位表 + */ +@TableName(value ="hy_mtz_e") +public class HyMtzE implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均水位 + */ + private BigDecimal avz; + + /** + * 平均水位注解码 + */ + private String avzrcd; + + /** + * 最高水位 + */ + private BigDecimal htz; + + /** + * 最高水位注解码 + */ + private String htzrcd; + + /** + * 最高水位日期 + */ + private Date htzdt; + + /** + * 最低水位 + */ + private BigDecimal mnz; + + /** + * 最低水位注解码 + */ + private String mnzrcd; + + /** + * 最低水位日期 + */ + private Date mnzdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; + + /** + * 站码 + */ + public String getStcd() { + return stcd; + } + + /** + * 站码 + */ + public void setStcd(String stcd) { + this.stcd = stcd; + } + + /** + * 年 + */ + public Integer getYr() { + return yr; + } + + /** + * 年 + */ + public void setYr(Integer yr) { + this.yr = yr; + } + + /** + * 月 + */ + public Integer getMth() { + return mth; + } + + /** + * 月 + */ + public void setMth(Integer mth) { + this.mth = mth; + } + + /** + * 平均水位 + */ + public BigDecimal getAvz() { + return avz; + } + + /** + * 平均水位 + */ + public void setAvz(BigDecimal avz) { + this.avz = avz; + } + + /** + * 平均水位注解码 + */ + public String getAvzrcd() { + return avzrcd; + } + + /** + * 平均水位注解码 + */ + public void setAvzrcd(String avzrcd) { + this.avzrcd = avzrcd; + } + + /** + * 最高水位 + */ + public BigDecimal getHtz() { + return htz; + } + + /** + * 最高水位 + */ + public void setHtz(BigDecimal htz) { + this.htz = htz; + } + + /** + * 最高水位注解码 + */ + public String getHtzrcd() { + return htzrcd; + } + + /** + * 最高水位注解码 + */ + public void setHtzrcd(String htzrcd) { + this.htzrcd = htzrcd; + } + + /** + * 最高水位日期 + */ + public Date getHtzdt() { + return htzdt; + } + + /** + * 最高水位日期 + */ + public void setHtzdt(Date htzdt) { + this.htzdt = htzdt; + } + + /** + * 最低水位 + */ + public BigDecimal getMnz() { + return mnz; + } + + /** + * 最低水位 + */ + public void setMnz(BigDecimal mnz) { + this.mnz = mnz; + } + + /** + * 最低水位注解码 + */ + public String getMnzrcd() { + return mnzrcd; + } + + /** + * 最低水位注解码 + */ + public void setMnzrcd(String mnzrcd) { + this.mnzrcd = mnzrcd; + } + + /** + * 最低水位日期 + */ + public Date getMnzdt() { + return mnzdt; + } + + /** + * 最低水位日期 + */ + public void setMnzdt(Date mnzdt) { + this.mnzdt = mnzdt; + } + + @Override + public boolean equals(Object that) { + if (this == that) { + return true; + } + if (that == null) { + return false; + } + if (getClass() != that.getClass()) { + return false; + } + HyMtzE other = (HyMtzE) that; + return (this.getStcd() == null ? other.getStcd() == null : this.getStcd().equals(other.getStcd())) + && (this.getYr() == null ? other.getYr() == null : this.getYr().equals(other.getYr())) + && (this.getMth() == null ? other.getMth() == null : this.getMth().equals(other.getMth())) + && (this.getAvz() == null ? other.getAvz() == null : this.getAvz().equals(other.getAvz())) + && (this.getAvzrcd() == null ? other.getAvzrcd() == null : this.getAvzrcd().equals(other.getAvzrcd())) + && (this.getHtz() == null ? other.getHtz() == null : this.getHtz().equals(other.getHtz())) + && (this.getHtzrcd() == null ? other.getHtzrcd() == null : this.getHtzrcd().equals(other.getHtzrcd())) + && (this.getHtzdt() == null ? other.getHtzdt() == null : this.getHtzdt().equals(other.getHtzdt())) + && (this.getMnz() == null ? other.getMnz() == null : this.getMnz().equals(other.getMnz())) + && (this.getMnzrcd() == null ? other.getMnzrcd() == null : this.getMnzrcd().equals(other.getMnzrcd())) + && (this.getMnzdt() == null ? other.getMnzdt() == null : this.getMnzdt().equals(other.getMnzdt())); + } + + @Override + public int hashCode() { + final int prime = 31; + int result = 1; + result = prime * result + ((getStcd() == null) ? 0 : getStcd().hashCode()); + result = prime * result + ((getYr() == null) ? 0 : getYr().hashCode()); + result = prime * result + ((getMth() == null) ? 0 : getMth().hashCode()); + result = prime * result + ((getAvz() == null) ? 0 : getAvz().hashCode()); + result = prime * result + ((getAvzrcd() == null) ? 0 : getAvzrcd().hashCode()); + result = prime * result + ((getHtz() == null) ? 0 : getHtz().hashCode()); + result = prime * result + ((getHtzrcd() == null) ? 0 : getHtzrcd().hashCode()); + result = prime * result + ((getHtzdt() == null) ? 0 : getHtzdt().hashCode()); + result = prime * result + ((getMnz() == null) ? 0 : getMnz().hashCode()); + result = prime * result + ((getMnzrcd() == null) ? 0 : getMnzrcd().hashCode()); + result = prime * result + ((getMnzdt() == null) ? 0 : getMnzdt().hashCode()); + return result; + } + + @Override + public String toString() { + StringBuilder sb = new StringBuilder(); + sb.append(getClass().getSimpleName()); + sb.append(" ["); + sb.append("Hash = ").append(hashCode()); + sb.append(", stcd=").append(stcd); + sb.append(", yr=").append(yr); + sb.append(", mth=").append(mth); + sb.append(", avz=").append(avz); + sb.append(", avzrcd=").append(avzrcd); + sb.append(", htz=").append(htz); + sb.append(", htzrcd=").append(htzrcd); + sb.append(", htzdt=").append(htzdt); + sb.append(", mnz=").append(mnz); + sb.append(", mnzrcd=").append(mnzrcd); + sb.append(", mnzdt=").append(mnzdt); + sb.append(", serialVersionUID=").append(serialVersionUID); + sb.append("]"); + return sb.toString(); + } +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMptEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMptEVo.java new file mode 100644 index 0000000..15dc337 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMptEVo.java @@ -0,0 +1,39 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import lombok.Data; + +/** + * @Author al + * @Date 2024/7/31 14:50 + * @Description: TODO + * @Version + */ + +@Data +public class HyMptEVo { + + private String stcd; + + private String stnm; + + private String addvcd; + + private String yr; + + private String mth; + + private String p; + + private String prcd; + + private String pdynum; + + private String pdynumrcd; + + private String mxdyp; + + private String mxdyprcd; + + private String mxdypodt; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtcsEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtcsEVo.java new file mode 100644 index 0000000..b63f3f8 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtcsEVo.java @@ -0,0 +1,87 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月含沙量表 + * @TableName hy_mtcs_e + */ +@Data +public class HyMtcsEVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + private BigDecimal mxs; + + /** + * 最大含沙量注解码 + */ + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + private String mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + private BigDecimal mns; + + /** + * 最小含沙量注解码 + */ + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + private String mnsdt; + + /** + * 流域水系码 + */ + private String lysxm; + + /**站名**/ + private String stnm; + + /** + * 行政区划码 + */ + private String addvcd; + + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtpEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtpEVo.java new file mode 100644 index 0000000..3f0c35c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtpEVo.java @@ -0,0 +1,85 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.alibaba.excel.annotation.ExcelProperty; +import com.baomidou.mybatisplus.annotation.TableField; +import lombok.Data; + +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @Author al + * @Date 2024/8/1 10:24 + * @Description: TODO + * @Version + */ + +@Data +public class HyMtpEVo { + + /** + * 站码 + */ + private String stcd; + + + /**站名**/ + private String stnm; + + /** + * 年 + */ + private Integer yr; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 月 + */ + private Integer mth; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + /** + * 降水日数 + */ + private String pdynum; + + /** + * 降水日数注解码 + */ + private String pdynumrcd; + + /** + * 最大降水量(毫米) + */ + private BigDecimal mxdyp; + + /** + * 最大降水量注解码 + */ + private String mxdyprcd; + + /** + * 最大日降水量出现日期 + */ + private Date mxdypodt; + + /** + * 流域水系码 + */ + private String lysxm; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqEVo.java new file mode 100644 index 0000000..0f94d65 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqEVo.java @@ -0,0 +1,86 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 月流量表 + * @TableName hy_mtq_e + */ +@Data +public class HyMtqEVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private String yr; + + /** + * 月 + */ + private String mth; + + /** + * 平均流量 + */ + private String avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + /** + * 最高流量 + */ + private String mxq; + + /** + * 最高流量注解码 + */ + private String mxqrcd; + + /** + * 最高流量日期 + */ + private String mxqdt; + + /** + * 最小流量 + */ + private String mnq; + + /** + * 最小流量注解码 + */ + private String mnqrcd; + + /** + * 最小流量日期 + */ + private String mnqdt; + + + /** + * 站名 + */ + private String stnm; + + + private String addvcd; + + + /** + * 流域水系码 + */ + private String lysxm; + + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqsEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqsEVo.java new file mode 100644 index 0000000..7a4ce72 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtqsEVo.java @@ -0,0 +1,79 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月输沙率表 + * @TableName hy_mtqs_e + */ +@TableName(value ="hy_mtqs_e") +@Data +public class HyMtqsEVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均输沙率(千克每秒) + */ + private BigDecimal avqs; + + /** + * 平均输沙率注解码 + */ + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + private BigDecimal mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + private String mxdyqsodt; + + + /** + * 站名 + */ + private String stnm; + + + private String addvcd; + + + /** + * 流域水系码 + */ + private String lysxm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMttdzEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMttdzEVo.java new file mode 100644 index 0000000..823130b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMttdzEVo.java @@ -0,0 +1,317 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName hy_mttdz_e + */ +@Data +public class HyMttdzEVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均高潮潮位 + */ + private BigDecimal avhtdz; + + /** + * 平均高潮潮位注解码 + */ + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + private BigDecimal hthtdz; + + /** + * 最高高潮位注解码 + */ + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + private String hthtdzotm; + + /** + * 最低高潮位 + */ + private BigDecimal lthtdz; + + /** + * 最低高潮位注解码 + */ + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + private String lthtdzotm; + + /** + * 平均低潮潮位 + */ + private BigDecimal avltdz; + + /** + * 平均低潮潮位注解码 + */ + private String avltdzrcd; + + /** + * 最高低潮位 + */ + private BigDecimal htltdz; + + /** + * 最高低潮位注解码 + */ + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + private String htltdzotm; + + /** + * 最低低潮位 + */ + private BigDecimal ltltdz; + + /** + * 最低低潮潮位注解码 + */ + private String ltltdzrcd; + + /** + * 最低地潮位出现时间 + */ + private String ltltdzotm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal avftdr; + + /** + * 平均涨潮潮差注解码 + */ + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + private BigDecimal mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + private String mxfltdrhtm; + + /** + * 最小潮潮差 + */ + private BigDecimal mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + private String mnfltdrhtm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal aver; + + /** + * 平均涨潮潮差注解码 + */ + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + private BigDecimal mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + private String mxebtdrht; + + /** + * 最小落潮潮差 + */ + private BigDecimal mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + private String mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + private Integer avftd; + + /** + * 平均涨潮历时注解码 + */ + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + private Integer mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + private String mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + private Integer mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + private String mnfltddrhtm; + + /** + * 平均落潮历时 + */ + private Integer avebbdr; + + /** + * 平均落潮历时注解码 + */ + private String avedrc; + + /** + * 最长落潮历时 + */ + private Integer mxebtddr; + + /** + * 最长落潮历时注解码 + */ + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + private String mxebtddrhtm; + + /** + * 最短落潮历时 + */ + private Integer mnbtddr; + + /** + * 最短落潮历时注解码 + */ + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + private String mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + private BigDecimal hravtdz; + + /** + * 逐时平均潮位注解码 + */ + private String hravtdzrcd; + + /** + * 平均潮差 + */ + private BigDecimal avtdr; + + /** + * 平均潮差注解码 + */ + private String avtdrrcd; + + /** + * 平均历时 + */ + private Integer avtddr; + + /** + * 平均历时注解码 + */ + private String avtddrrcd; + + /** + * 站名 + */ + private String stnm; + + + private String addvcd; + + + /** + * 流域水系码 + */ + private String lysxm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtweEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtweEVo.java new file mode 100644 index 0000000..bae855b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtweEVo.java @@ -0,0 +1,81 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import lombok.Data; + +import java.math.BigDecimal; + +/** + * 月水面蒸发量表 + * @Author al + * @Date 2024/8/27 10:13 + * @Description: TODO + * @Version + */ + +@Data +public class HyMtweEVo { + + /** + * 站码 + */ + private String stcd; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 水面蒸发量(毫米) + */ + private BigDecimal wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + private BigDecimal mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + private String mxdyercd; + + /** + * 最小日水面蒸发量 + */ + private BigDecimal mndye; + + /** + * 最小日水面蒸发量注解码 + */ + private String mndyercd; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtwtEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtwtEVo.java new file mode 100644 index 0000000..4f90214 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtwtEVo.java @@ -0,0 +1,89 @@ +package com.ruoyi.swlscx.month.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 月水温表字段定义 + * @TableName hy_mtwt_e + */ +@Data +public class HyMtwtEVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均水温(摄氏度) + */ + private BigDecimal avwtmp; + + /** + * 平均水温注解码 + */ + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + private BigDecimal mxwtmp; + + /** + * 最高水温注解码 + */ + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + private String mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + private BigDecimal mnwtmp; + + /** + * 最低水温注解码 + */ + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + private String mnwtmpdt; + + + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtzEVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtzEVo.java new file mode 100644 index 0000000..15b1677 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/domain/vo/HyMtzEVo.java @@ -0,0 +1,87 @@ +package com.ruoyi.swlscx.month.domain.vo; + + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName 月水位表 + */ +@Data +public class HyMtzEVo { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 月 + */ + private Integer mth; + + /** + * 平均水位 + */ + private BigDecimal avz; + + /** + * 平均水位注解码 + */ + private String avzrcd; + + /** + * 最高水位 + */ + private BigDecimal htz; + + /** + * 最高水位注解码 + */ + private String htzrcd; + + /** + * 最高水位日期 + */ + private String htzdt; + + /** + * 最低水位 + */ + private BigDecimal mnz; + + /** + * 最低水位注解码 + */ + private String mnzrcd; + + /** + * 最低水位日期 + */ + private String mnzdt; + + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtchpdEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtchpdEMapper.java new file mode 100644 index 0000000..de0ee30 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtchpdEMapper.java @@ -0,0 +1,20 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.month.domain.po.HyMtchpdE; +import org.apache.ibatis.annotations.Mapper; + +/** +* @author 12974 +* @description 针对表【hy_mtchpd_e(月泥沙特征粒径表字段定义)】的数据库操作Mapper +* @createDate 2024-08-06 13:42:04 +* @Entity com.ruoyi.swlscx.domain.po.HyMtchpdE +*/ +@Mapper +public interface HyMtchpdEMapper extends BaseMapper { + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtcsEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtcsEMapper.java new file mode 100644 index 0000000..510b67e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtcsEMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtcsE; +import com.ruoyi.swlscx.month.domain.vo.HyMtcsEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtcs_e(月含沙量表)】的数据库操作Mapper +* @createDate 2024-08-05 14:26:28 +* @Entity com.ruoyi.swlscx.domain.po.HyMtcsE +*/ +@Mapper +public interface HyMtcsEMapper extends BaseMapper { + + IPage selectHyMtcsEDataByPageAndInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtiqEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtiqEMapper.java new file mode 100644 index 0000000..f04defc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtiqEMapper.java @@ -0,0 +1,20 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.month.domain.po.HyMtiqE; +import org.apache.ibatis.annotations.Mapper; + +/** +* @author 12974 +* @description 针对表【hy_mtiq_e(月冰流量表)】的数据库操作Mapper +* @createDate 2024-08-06 13:42:04 +* @Entity com.ruoyi.swlscx.domain.po.HyMtiqE +*/ +@Mapper +public interface HyMtiqEMapper extends BaseMapper { + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpEMapper.java new file mode 100644 index 0000000..d2c6ac1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpEMapper.java @@ -0,0 +1,39 @@ +package com.ruoyi.swlscx.month.mapper; + + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtpE; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtp_e(月降水表)】的数据库操作Mapper +* @createDate 2024-07-23 14:15:25 +* @Entity com/ruoyi/swlscx/domain/po.po.HyMtpE +*/ +@Mapper +public interface HyMtpEMapper extends BaseMapper { + + /**查询月降水表数据**/ + IPage selectHyMtpDataByPageAndInfo(Page page,@Param("map") Map map); + + /**查询月降水表数据**/ + List selectHyMtpData(@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpddbEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpddbEMapper.java new file mode 100644 index 0000000..31a51d9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtpddbEMapper.java @@ -0,0 +1,20 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.month.domain.po.HyMtpddbE; +import org.apache.ibatis.annotations.Mapper; + +/** +* @author 12974 +* @description 针对表【hy_mtpddb_e(月平均泥沙颗粒级配表)】的数据库操作Mapper +* @createDate 2024-08-06 13:42:04 +* @Entity com.ruoyi.swlscx.domain.po.HyMtpddbE +*/ +@Mapper +public interface HyMtpddbEMapper extends BaseMapper { + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqEMapper.java new file mode 100644 index 0000000..8e7e0fe --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqEMapper.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtqE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtq_e(月流量表)】的数据库操作Mapper +* @createDate 2024-08-04 12:45:25 +* @Entity com.ruoyi.swlscx.domain.po.HyMtqE +*/ +@Mapper +public interface HyMtqEMapper extends BaseMapper { + + + IPage selectHyMtqEPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqsEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqsEMapper.java new file mode 100644 index 0000000..bbdad08 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtqsEMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtqsE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqsEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtqs_e(月输沙率表)】的数据库操作Mapper +* @createDate 2024-08-05 16:03:07 +* @Entity com.ruoyi.swlscx.domain.po.HyMtqsE +*/ +@Mapper +public interface HyMtqsEMapper extends BaseMapper { + + IPage selectHyMtqsEByPageAndInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMttdzEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMttdzEMapper.java new file mode 100644 index 0000000..00537d4 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMttdzEMapper.java @@ -0,0 +1,27 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMttdzE; +import com.ruoyi.swlscx.month.domain.vo.HyMttdzEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mttdz_e】的数据库操作Mapper +* @createDate 2024-08-04 12:45:25 +* @Entity com.ruoyi.swlscx.domain.po.HyMttdzE +*/ +@Mapper +public interface HyMttdzEMapper extends BaseMapper { + + IPage selectHyMttdzListPageByInfo(Page page, @Param("map") Map map); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtweEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtweEMapper.java new file mode 100644 index 0000000..4dc9618 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtweEMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtwe_e(月水面蒸发量表)】的数据库操作Mapper +* @createDate 2024-08-01 09:38:45 +* @Entity com.ruoyi.swlscx.domain.po.HyMtweE +*/ +@Mapper +public interface HyMtweEMapper extends BaseMapper { + + + /**查询月水面蒸发量表数据**/ + IPage selectHyMtweEListPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtwtEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtwtEMapper.java new file mode 100644 index 0000000..5b47910 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtwtEMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.month.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtwtE; +import com.ruoyi.swlscx.month.domain.vo.HyMtwtEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtwt_e(月水温表字段定义)】的数据库操作Mapper +* @createDate 2024-08-06 13:42:04 +* @Entity com.ruoyi.swlscx.domain.po.HyMtwtE +*/ +@Mapper +public interface HyMtwtEMapper extends BaseMapper { + + /** + * 查询月水温表数据 + */ + IPage selectHyMtzEListPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtzEMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtzEMapper.java new file mode 100644 index 0000000..f10473f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/mapper/HyMtzEMapper.java @@ -0,0 +1,40 @@ +package com.ruoyi.swlscx.month.mapper; + + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.month.domain.po.HyMtzE; +import com.ruoyi.swlscx.month.domain.vo.HyMtzEVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mtz_e】的数据库操作Mapper +* @createDate 2024-07-23 14:15:25 +* @Entity com/ruoyi/swlscx/domain/po.po.HyMtzE +*/ +@Mapper +public interface HyMtzEMapper extends BaseMapper { + + void batchInsert(@Param("list") List batchList); + + /** + * 查询月水位表 + */ + IPage selectHyMtzEPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtchpdEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtchpdEService.java new file mode 100644 index 0000000..95d03c7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtchpdEService.java @@ -0,0 +1,13 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.month.domain.po.HyMtchpdE; + +/** +* @author 12974 +* @description 针对表【hy_mtchpd_e(月泥沙特征粒径表字段定义)】的数据库操作Service +* @createDate 2024-08-06 13:42:04 +*/ +public interface HyMtchpdEService extends IService { + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtcsEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtcsEService.java new file mode 100644 index 0000000..7b0f17f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtcsEService.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtcsE; +import com.ruoyi.swlscx.month.domain.vo.HyMtcsEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtcs_e(月含沙量表)】的数据库操作Service +* @createDate 2024-08-05 14:26:28 +*/ +public interface HyMtcsEService extends IService { + + /** + * 导入月含沙量表数据 + */ + void importHyMtcsEData(MultipartFile file); + + /** + * 查询月含沙量表数据 + */ + IPage selectHyMtcsEByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出月含沙量表 + */ + R exportMonthSedimentContentList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtiqEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtiqEService.java new file mode 100644 index 0000000..18393fe --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtiqEService.java @@ -0,0 +1,13 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.month.domain.po.HyMtiqE; + +/** +* @author 12974 +* @description 针对表【hy_mtiq_e(月冰流量表)】的数据库操作Service +* @createDate 2024-08-06 13:42:04 +*/ +public interface HyMtiqEService extends IService { + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpEService.java new file mode 100644 index 0000000..8ecb70a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpEService.java @@ -0,0 +1,38 @@ +package com.ruoyi.swlscx.month.service; + + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtpE; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtp_e】的数据库操作Service +* @createDate 2024-07-23 14:15:25 +*/ +public interface HyMtpEService extends IService { + + /** + * 导入月降水表数据 + */ + void importHyMtpEData(MultipartFile file); + + + /** + * 查询月降水表数据 + */ + IPage selectHyMtpDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出月降水表数据 + */ + R exportHyMtpData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpddbEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpddbEService.java new file mode 100644 index 0000000..5dbd438 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtpddbEService.java @@ -0,0 +1,19 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.month.domain.po.HyMtpddbE; +import org.springframework.web.multipart.MultipartFile; + +/** +* @author 12974 +* @description 针对表【hy_mtpddb_e(月平均泥沙颗粒级配表)】的数据库操作Service +* @createDate 2024-08-06 13:42:04 +*/ +public interface HyMtpddbEService extends IService { + + /** + * 导入月平均泥沙颗粒级配表数据 + */ + void importHyMtpddbEData(MultipartFile file); + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqEService.java new file mode 100644 index 0000000..ee3d155 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqEService.java @@ -0,0 +1,31 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtqE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtq_e(月流量表)】的数据库操作Service +* @createDate 2024-08-04 12:45:25 +*/ +public interface HyMtqEService extends IService { + + + /** 导入月流量表数据*/ + void importHyMtqEData(MultipartFile file); + + /**查询月流量表数据*/ + IPage selectHyMtqEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm); + + /**导出月流量表数据*/ + R exportMonthFlowList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqsEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqsEService.java new file mode 100644 index 0000000..1081dd3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtqsEService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtqsE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqsEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtqs_e(月输沙率表)】的数据库操作Service +* @createDate 2024-08-05 16:03:07 +*/ +public interface HyMtqsEService extends IService { + + + /** + * 导入月输沙率表数据 + */ + void importHyMtqsEData(MultipartFile file); + + /** + * 查询月输沙率表数据 + */ + IPage selectHyMtqsEDataByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出月输沙率表 + */ + R exportMonthSedimentDischargeList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMttdzEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMttdzEService.java new file mode 100644 index 0000000..a549802 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMttdzEService.java @@ -0,0 +1,32 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMttdzE; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mttdz_e】的数据库操作Service +* @createDate 2024-08-04 12:45:25 +*/ +public interface HyMttdzEService extends IService { + + + + /** + * 导入月潮位数据 + */ + void importHyMttdzEData(MultipartFile file); + + /**查询月潮位表数据*/ + R selectHyMttdzEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stcd,String stnm); + + /** + * 导出月潮位表数据 + */ + R exportMonthTideLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtweEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtweEService.java new file mode 100644 index 0000000..edb1d37 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtweEService.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtwe_e(月水面蒸发量表)】的数据库操作Service +* @createDate 2024-08-01 09:38:45 +*/ +public interface HyMtweEService extends IService { + + + /**导入月水面蒸发量表数据*/ + void importHyMtweEDate(MultipartFile file); + + + + /**查询月水面蒸发量表数据*/ + IPage selectHyMtweEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm); + + + /** + * 导出月水面蒸发量表 + */ + R exportMonthEvaporationWaterList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtwtEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtwtEService.java new file mode 100644 index 0000000..33dae20 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtwtEService.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.month.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtwtE; +import com.ruoyi.swlscx.month.domain.vo.HyMtwtEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** + * @author 12974 + * @description 针对表【hy_mtwt_e(月水温表字段定义)】的数据库操作Service + * @createDate 2024-08-06 13:42:04 + */ +public interface HyMtwtEService extends IService { + + + /** + * 查询月水温表数据 + */ + IPage selectHyMtweEDataByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm); + + /** + * 导入月水温表数据 + */ + void importHyMtwtEData(MultipartFile file); + + /** + * 导出月水温表 + */ + R exportMonthWaterTemperatureList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtzEService.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtzEService.java new file mode 100644 index 0000000..2c51ded --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/HyMtzEService.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.month.service; + + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.po.HyMtzE; +import com.ruoyi.swlscx.month.domain.vo.HyMtzEVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mtz_e】的数据库操作Service +* @createDate 2024-07-23 14:15:25 +*/ +public interface HyMtzEService extends IService { + + + /**导入月水位表**/ + void importHyMtzEData(MultipartFile file); + + /**查询月水位表**/ + IPage selectHyMtzEPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出月水位表 + */ + R exportMonthWaterLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtchpdEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtchpdEServiceImpl.java new file mode 100644 index 0000000..f484230 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtchpdEServiceImpl.java @@ -0,0 +1,22 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.swlscx.month.domain.po.HyMtchpdE; +import com.ruoyi.swlscx.month.mapper.HyMtchpdEMapper; +import com.ruoyi.swlscx.month.service.HyMtchpdEService; +import org.springframework.stereotype.Service; + +/** +* @author 12974 +* @description 针对表【hy_mtchpd_e(月泥沙特征粒径表字段定义)】的数据库操作Service实现 +* @createDate 2024-08-06 13:42:04 +*/ +@Service +public class HyMtchpdEServiceImpl extends ServiceImpl + implements HyMtchpdEService{ + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtcsEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtcsEServiceImpl.java new file mode 100644 index 0000000..22475ca --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtcsEServiceImpl.java @@ -0,0 +1,363 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtcsEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtcsE; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtcsEVo; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.month.mapper.HyMtcsEMapper; +import com.ruoyi.swlscx.month.service.HyMtcsEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtcs_e(月含沙量表)】的数据库操作Service实现 +* @createDate 2024-08-05 14:26:28 +*/ +@Service +public class HyMtcsEServiceImpl extends ServiceImpl + implements HyMtcsEService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + + + /** + * 导入月含沙量表数据 + */ + @Override + public void importHyMtcsEData(MultipartFile file) { + try { + List hyMtcsEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtcsEBo.class, 2); + // 获取数据 + List dataList = hyMtcsEBos.parallelStream().map(i -> { + HyMtcsE hyMtcsE = new HyMtcsE(); + BeanUtils.copyProperties(i, hyMtcsE); + hyMtcsE.setAvcs(Convert.toBigDecimal(i.getAvcs())); + hyMtcsE.setMth(Convert.toInt(i.getMth())); + hyMtcsE.setMns(Convert.toBigDecimal(i.getMns())); + hyMtcsE.setMnsdt(DateUtils.parseDate(i.getMnsdt())); + hyMtcsE.setYr(Convert.toInt(i.getYr())); + hyMtcsE.setMxs(Convert.toBigDecimal(i.getMxs())); + hyMtcsE.setMxsdt(DateUtils.parseDate(i.getMxsdt())); + return hyMtcsE; + }).collect(Collectors.toList()); + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + @Override + public IPage selectHyMtcsEByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtcsEDataByPageAndInfo(new Query().getPage(map), map); + + } + + /** + * 导出月含沙量表 + */ + @Override + public R exportMonthSedimentContentList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月含沙量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtweEVoList) { + List records = hyMtweEVoList.stream().map(i->{ + HyMtcsE hyMtcsE = new HyMtcsE(); + BeanUtils.copyProperties(i, hyMtcsE); + hyMtcsE.setAvcs(Convert.toBigDecimal(i.getAvcs())); + hyMtcsE.setMth(Convert.toInt(i.getMth())); + hyMtcsE.setMns(Convert.toBigDecimal(i.getMns())); + hyMtcsE.setMnsdt(DateUtils.parseDate(i.getMnsdt())); + hyMtcsE.setYr(Convert.toInt(i.getYr())); + hyMtcsE.setMxs(Convert.toBigDecimal(i.getMxs())); + hyMtcsE.setMxsdt(DateUtils.parseDate(i.getMxsdt())); + return hyMtcsE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月含沙量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月含沙量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtcsEByPageAndInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtcsEServiceImpl hyMtcsEService = context.getBean(HyMtcsEServiceImpl.class); + hyMtcsEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthSedimentConcentration.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMptEVoIPage = this.baseMapper.selectHyMtcsEDataByPageAndInfo(new Query().getPage(map), map); + List records = hyMptEVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtcsEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月含沙量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvcsrcd() != null ? vo.getAvcsrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxs() != null ? CommonUtils.formatNum(vo.getMxs()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxsrcd() != null ? vo.getMxsrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxsdt() != null ? vo.getMxsdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMns() != null ? CommonUtils.formatNum(vo.getMns()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnsrcd() != null ? vo.getMnsrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnsdt() != null ? vo.getMnsdt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + + // 批量查询数据 + IPage hyMptEVoIPage = this.baseMapper.selectHyMtcsEDataByPageAndInfo(new Query().getPage(map), map); + List records = hyMptEVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtcsEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvcsrcd() != null ? vo.getAvcsrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxs() != null ? CommonUtils.formatNum(vo.getMxs()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxsrcd() != null ? vo.getMxsrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxsdt() != null ? vo.getMxsdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMns() != null ? CommonUtils.formatNum(vo.getMns()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnsrcd() != null ? vo.getMnsrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnsdt() != null ? vo.getMnsdt() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtiqEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtiqEServiceImpl.java new file mode 100644 index 0000000..f8fb1de --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtiqEServiceImpl.java @@ -0,0 +1,22 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.swlscx.month.domain.po.HyMtiqE; +import com.ruoyi.swlscx.month.mapper.HyMtiqEMapper; +import com.ruoyi.swlscx.month.service.HyMtiqEService; +import org.springframework.stereotype.Service; + +/** +* @author 12974 +* @description 针对表【hy_mtiq_e(月冰流量表)】的数据库操作Service实现 +* @createDate 2024-08-06 13:42:04 +*/ +@Service +public class HyMtiqEServiceImpl extends ServiceImpl + implements HyMtiqEService{ + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpEServiceImpl.java new file mode 100644 index 0000000..d3acae8 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpEServiceImpl.java @@ -0,0 +1,359 @@ +package com.ruoyi.swlscx.month.service.impl; + + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtpEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtpE; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import com.ruoyi.swlscx.month.domain.vo.HyMtpEVo; +import com.ruoyi.swlscx.month.mapper.HyMtpEMapper; +import com.ruoyi.swlscx.month.service.HyMtpEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.*; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_mtp_e】的数据库操作Service实现 + * @createDate 2024-07-23 14:15:25 + */ +@Service +public class HyMtpEServiceImpl extends ServiceImpl + implements HyMtpEService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入月降水表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtpEData(MultipartFile file) { + try { + List hyMtpEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtpEBo.class, 3); + List hyMtpEList = hyMtpEBos.stream().map(i -> { + HyMtpE hyMtpE = new HyMtpE(); + BeanUtils.copyProperties(i, hyMtpE); + hyMtpE.setMth(Convert.toInt(i.getMth())); + hyMtpE.setMxdypodt(DateUtils.parseDate(i.getMxdypodt())); + hyMtpE.setMxdyp(Convert.toBigDecimal(i.getMxdyp())); + hyMtpE.setMxdyp(Convert.toBigDecimal(i.getMxdyp())); + hyMtpE.setYr(Convert.toInt(i.getYr())); + hyMtpE.setP(Convert.toBigDecimal(i.getP())); + return hyMtpE; + }).collect(Collectors.toList()); + this.saveBatch(hyMtpEList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询月降水表数据 + */ + @Override + public IPage selectHyMtpDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtpDataByPageAndInfo(new Query().getPage(map), map); + } + + + + @Override + public R exportHyMtpData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月降水表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); +// +// Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L,100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response,"monthRain.xls","月降水表",(Sheet sheet, CellStyle style) ->{ +// rainModel(rainMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtpEVoList) { + List records = hyMtpEVoList.stream().map(i->{ + HyMtpE hyMtpE = new HyMtpE(); + BeanUtils.copyProperties(i, hyMtpE); + hyMtpE.setMth(Convert.toInt(i.getMth())); + hyMtpE.setMxdypodt(DateUtils.parseDate(i.getMxdypodt())); + hyMtpE.setMxdyp(Convert.toBigDecimal(i.getMxdyp())); + hyMtpE.setMxdyp(Convert.toBigDecimal(i.getMxdyp())); + hyMtpE.setYr(Convert.toInt(i.getYr())); + hyMtpE.setP(Convert.toBigDecimal(i.getP())); + return hyMtpE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月降水表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月降水表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtpDataByPageAndInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtpEServiceImpl hyMtpEService = context.getBean(HyMtpEServiceImpl.class); + hyMtpEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + List hyMptEVos = this.baseMapper.selectHyMtpData(map); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < hyMptEVos.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, hyMptEVos.size()); + List batchList = hyMptEVos.subList(i, toIndex); + + // 批量写入数据 + for (HyMptEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月降水表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, vo.getYr() != null ? vo.getYr() : "", style); + ExcelUtils.formatCell(row, 5, vo.getMth() != null ? vo.getMth() : "", style); + ExcelUtils.formatCell(row, 6, vo.getP() != null ? vo.getP() : "", style); + ExcelUtils.formatCell(row, 7, vo.getPrcd() != null ? vo.getPrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getPdynum() != null ? vo.getPdynum() : "", style); + ExcelUtils.formatCell(row, 9, vo.getPdynumrcd() != null ? vo.getPdynumrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyp() != null ? vo.getMxdyp() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyprcd() != null ? vo.getMxdyprcd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMxdypodt() != null ? vo.getMxdypodt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void rainModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + List hyMptEVoIPage = this.baseMapper.selectHyMtpData(map); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < hyMptEVoIPage.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, hyMptEVoIPage.size()); + List batchList = hyMptEVoIPage.subList(i, toIndex); + // 批量写入数据 + for (HyMptEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月降水表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, vo.getYr() != null ? vo.getYr() : "", style); + ExcelUtils.formatCell(row, 5, vo.getMth() != null ? vo.getMth() : "", style); + ExcelUtils.formatCell(row, 6, vo.getP() != null ? vo.getP() : "", style); + ExcelUtils.formatCell(row, 7, vo.getPrcd() != null ? vo.getPrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getPdynum() != null ? vo.getPdynum() : "", style); + ExcelUtils.formatCell(row, 9, vo.getPdynumrcd() != null ? vo.getPdynumrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyp() != null ? vo.getMxdyp() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyprcd() != null ? vo.getMxdyprcd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMxdypodt() != null ? vo.getMxdypodt() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpddbEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpddbEServiceImpl.java new file mode 100644 index 0000000..4df7fd2 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtpddbEServiceImpl.java @@ -0,0 +1,54 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.month.domain.bo.HyMtpddbEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtpddbE; +import com.ruoyi.swlscx.month.mapper.HyMtpddbEMapper; +import com.ruoyi.swlscx.month.service.HyMtpddbEService; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import java.util.List; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtpddb_e(月平均泥沙颗粒级配表)】的数据库操作Service实现 +* @createDate 2024-08-06 13:42:04 +*/ +@Service +public class HyMtpddbEServiceImpl extends ServiceImpl + implements HyMtpddbEService{ + + /** + * 导入月平均泥沙颗粒级配表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtpddbEData(MultipartFile file) { + try { + List hyMtpddbEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtpddbEBo.class, 2); + // 获取数据 + List dataList = hyMtpddbEBos.parallelStream().map(i -> { + HyMtpddbE hyMtpddbE = new HyMtpddbE(); + BeanUtils.copyProperties(i, hyMtpddbE); + hyMtpddbE.setAvswpct(Convert.toBigDecimal(i.getAvswpct())); + hyMtpddbE.setYr(Convert.toInt(i.getYr())); + hyMtpddbE.setMth(Convert.toInt(i.getMth())); + hyMtpddbE.setLtpd(Convert.toBigDecimal(i.getLtpd())); + return hyMtpddbE; + }).collect(Collectors.toList()); + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqEServiceImpl.java new file mode 100644 index 0000000..ddf3ec0 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqEServiceImpl.java @@ -0,0 +1,368 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtqEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtqE; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqEVo; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.month.mapper.HyMtqEMapper; +import com.ruoyi.swlscx.month.service.HyMtqEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtq_e(月流量表)】的数据库操作Service实现 +* @createDate 2024-08-04 12:45:25 +*/ +@Service +public class HyMtqEServiceImpl extends ServiceImpl + implements HyMtqEService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入月流量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtqEData(MultipartFile file) { + try { + List hyMtqEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtqEBo.class, 2); + // 获取数据 + List dataList = hyMtqEBos.parallelStream().map(i -> { + HyMtqE hyMtqE = new HyMtqE(); + BeanUtils.copyProperties(i, hyMtqE); + hyMtqE.setMth(Convert.toInt(i.getMth())); + hyMtqE.setYr(Convert.toInt(i.getYr())); + hyMtqE.setAvq(Convert.toBigDecimal(i.getAvq())); + hyMtqE.setMnq(Convert.toBigDecimal(i.getMnq())); + hyMtqE.setMxq(Convert.toBigDecimal(i.getMxq())); + hyMtqE.setMnqdt(DateUtils.parseDate(i.getMnqdt())); + hyMtqE.setMxqdt(DateUtils.parseDate(i.getMxqdt())); + return hyMtqE; + }).collect(Collectors.toList()); + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询月流量表数据 + */ + @Override + public IPage selectHyMtqEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + + } + + /** + * 导出月流量表数据 + */ + @Override + public R exportMonthFlowList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + + Long userId = SecurityUtils.getUserId(); + String tableName = "月流量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map map = CommonUtils.getMonthDataMap(new PageParams(0L,100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response,"monthFlow.xls","月流量表",(Sheet sheet, CellStyle style) ->{ +// dataModel(map, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtqEVoList) { + List records = hyMtqEVoList.stream().map(i->{ + HyMtqE hyMtqE = new HyMtqE(); + BeanUtils.copyProperties(i, hyMtqE); + hyMtqE.setMth(Convert.toInt(i.getMth())); + hyMtqE.setYr(Convert.toInt(i.getYr())); + hyMtqE.setAvq(Convert.toBigDecimal(i.getAvq())); + hyMtqE.setMnq(Convert.toBigDecimal(i.getMnq())); + hyMtqE.setMxq(Convert.toBigDecimal(i.getMxq())); + hyMtqE.setMnqdt(DateUtils.parseDate(i.getMnqdt())); + hyMtqE.setMxqdt(DateUtils.parseDate(i.getMxqdt())); + return hyMtqE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月流量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月流量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtqEListPageByInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtqEServiceImpl hyMtqEService = context.getBean(HyMtqEServiceImpl.class); + hyMtqEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthFlow.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtqEPageByInfo = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyMtqEPageByInfo.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtqEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月流量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvq() != null ? vo.getAvq() : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxq() != null ? vo.getMxq() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxqrcd() != null ? vo.getMxqrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxqdt() != null ? vo.getMxqdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnq() != null ? vo.getMnq() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnqrcd() != null ? vo.getMnqrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnqdt() != null ? vo.getMnqdt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + + // 批量查询数据 + IPage hyMtqEPageByInfo = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyMtqEPageByInfo.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtqEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月流量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvq() != null ? vo.getAvq() : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxq() != null ? vo.getMxq() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxqrcd() != null ? vo.getMxqrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxqdt() != null ? vo.getMxqdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnq() != null ? vo.getMnq() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnqrcd() != null ? vo.getMnqrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnqdt() != null ? vo.getMnqdt() : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqsEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqsEServiceImpl.java new file mode 100644 index 0000000..770b9cc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtqsEServiceImpl.java @@ -0,0 +1,354 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtqsEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtqsE; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtqsEVo; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.month.mapper.HyMtqsEMapper; +import com.ruoyi.swlscx.month.service.HyMtqsEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtqs_e(月输沙率表)】的数据库操作Service实现 +* @createDate 2024-08-05 16:03:07 +*/ +@Service +public class HyMtqsEServiceImpl extends ServiceImpl + implements HyMtqsEService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtqsEData(MultipartFile file) { + try { + List hyMtqsEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtqsEBo.class, 2); + // 获取数据 + List dataList = hyMtqsEBos.parallelStream().map(i -> { + HyMtqsE hyMtqE = new HyMtqsE(); + BeanUtils.copyProperties(i, hyMtqE); + hyMtqE.setYr(Convert.toInt(i.getYr())); + hyMtqE.setMth(Convert.toInt(i.getMth())); + hyMtqE.setAvqs(Convert.toBigDecimal(i.getAvqs())); + hyMtqE.setMxdyqs(Convert.toBigDecimal(i.getMxdyqs())); + hyMtqE.setMxdyqsodt(DateUtils.parseDate(i.getMxdyqsodt())); + return hyMtqE; + }).collect(Collectors.toList()); + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询月输沙率表数据 + */ + @Override + public IPage selectHyMtqsEDataByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtqsEByPageAndInfo(new Query().getPage(map), map); + } + + /** + * 导出月输沙率表 + */ + @Override + public R exportMonthSedimentDischargeList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月输沙率表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map map = CommonUtils.getMonthDataMap(new PageParams(0L,100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response,"momthSedimentTransportRate.xls","月输沙率表",(Sheet sheet, CellStyle style) ->{ +// dataModel(map, sheet, style); +// }); + } + + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtqsEVoList) { + List records = hyMtqsEVoList.stream().map(i->{ + HyMtqsE hyMtqE = new HyMtqsE(); + BeanUtils.copyProperties(i, hyMtqE); + hyMtqE.setYr(Convert.toInt(i.getYr())); + hyMtqE.setMth(Convert.toInt(i.getMth())); + hyMtqE.setAvqs(Convert.toBigDecimal(i.getAvqs())); + hyMtqE.setMxdyqs(Convert.toBigDecimal(i.getMxdyqs())); + hyMtqE.setMxdyqsodt(DateUtils.parseDate(i.getMxdyqsodt())); + return hyMtqE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月输沙率表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月输沙率表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtqsEDataByPageAndInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtqsEServiceImpl hyMtqsEService = context.getBean(HyMtqsEServiceImpl.class); + hyMtqsEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("momthSedimentTransportRate.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtqsEVoIPage = this.baseMapper.selectHyMtqsEByPageAndInfo(new Query().getPage(map), map); + List records = hyMtqsEVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtqsEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getSdtp() != null ? vo.getSdtp() : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqs() != null ? CommonUtils.formatNum(vo.getAvqs()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getAvqsrcd() != null ? vo.getAvqsrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyqs() != null ? CommonUtils.formatNum(vo.getMxdyqs()) : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyqsrcd() != null ? vo.getMxdyqsrcd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMxdyqsodt() != null ? vo.getMxdyqsodt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtqsEVoIPage = this.baseMapper.selectHyMtqsEByPageAndInfo(new Query().getPage(map), map); + List records = hyMtqsEVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtqsEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getSdtp() != null ? vo.getSdtp() : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqs() != null ? CommonUtils.formatNum(vo.getAvqs()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getAvqsrcd() != null ? vo.getAvqsrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyqs() != null ? CommonUtils.formatNum(vo.getMxdyqs()) : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyqsrcd() != null ? vo.getMxdyqsrcd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMxdyqsodt() != null ? vo.getMxdyqsodt() : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMttdzEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMttdzEServiceImpl.java new file mode 100644 index 0000000..3d2d2c9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMttdzEServiceImpl.java @@ -0,0 +1,402 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMttdzEBo; +import com.ruoyi.swlscx.month.domain.po.HyMttdzE; +import com.ruoyi.swlscx.month.domain.vo.HyMttdzEVo; +import com.ruoyi.swlscx.month.mapper.HyMttdzEMapper; +import com.ruoyi.swlscx.month.service.HyMttdzEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.stereotype.Service; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_mttdz_e】的数据库操作Service实现 + * @createDate 2024-08-04 12:45:25 + */ +@Service +public class HyMttdzEServiceImpl extends ServiceImpl + implements HyMttdzEService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + /** + * 导入月潮位数据 + */ + @Override + public void importHyMttdzEData(MultipartFile file) { + try { + List hyMttdzEBos = ExcelUtils.readExcel(file.getInputStream(), HyMttdzEBo.class, 2); + // 获取数据 + List dataList = hyMttdzEBos.parallelStream().map(i -> { + HyMttdzE hyMttdzE = new HyMttdzE(); + BeanUtils.copyProperties(i, hyMttdzE); + hyMttdzE.setYr(Convert.toInt(i.getYr())); + hyMttdzE.setMth(Convert.toInt(i.getMth())); + hyMttdzE.setAvhtdz(Convert.toBigDecimal(i.getAvhtdz())); + hyMttdzE.setHthtdz(Convert.toBigDecimal(i.getHthtdz())); + hyMttdzE.setHthtdzotm(DateUtils.parseDate(i.getHthtdzotm())); + hyMttdzE.setLthtdz(Convert.toBigDecimal(i.getLthtdz())); + hyMttdzE.setLthtdzotm(DateUtils.parseDate(i.getLthtdzotm())); + hyMttdzE.setAvltdz(Convert.toBigDecimal(i.getAvltdz())); + hyMttdzE.setHtltdz(Convert.toBigDecimal(i.getHtltdz())); + hyMttdzE.setHtltdzotm(DateUtils.parseDate(i.getHtltdzotm())); + hyMttdzE.setLtltdz(Convert.toBigDecimal(i.getLtltdz())); + hyMttdzE.setLtltdzotm(DateUtils.parseDate(i.getLtltdzotm())); + hyMttdzE.setAvftdr(Convert.toBigDecimal(i.getAvftdr())); + hyMttdzE.setMxfltdr(Convert.toBigDecimal(i.getMxfltdr())); + hyMttdzE.setMxfltdrhtm(DateUtils.parseDate(i.getMxfltdrhtm())); + hyMttdzE.setMnfltdr(Convert.toBigDecimal(i.getMnfltdr())); + hyMttdzE.setMnfltdrhtm(DateUtils.parseDate(i.getMnfltdrhtm())); + hyMttdzE.setAver(Convert.toBigDecimal(i.getAver())); + hyMttdzE.setMxebtdr(Convert.toBigDecimal(i.getMxebtdr())); + hyMttdzE.setMxebtdrht(DateUtils.parseDate(i.getMxebtdrht())); + hyMttdzE.setMnebtdr(Convert.toBigDecimal(i.getMnebtdr())); + hyMttdzE.setMnebtdrhtm(DateUtils.parseDate(i.getMnebtdrhtm())); + hyMttdzE.setAvftd(Convert.toInt(i.getAvftd())); + hyMttdzE.setMxfltddr(Convert.toInt(i.getMxfltddr())); + hyMttdzE.setMxfltddrhtm(DateUtils.parseDate(i.getMxfltddrhtm())); + hyMttdzE.setMnfltddr(Convert.toInt(i.getMnfltddr())); + hyMttdzE.setMnfltddrhtm(DateUtils.parseDate(i.getMnfltddrhtm())); + hyMttdzE.setAvebbdr(Convert.toInt(i.getAvebbdr())); + hyMttdzE.setMxebtddr(Convert.toInt(i.getMxebtddr())); + hyMttdzE.setMxebtddrhtm(DateUtils.parseDate(i.getMxebtddrhtm())); + hyMttdzE.setMnbtddr(Convert.toInt(i.getMnbtddr())); + hyMttdzE.setMnebtddrhtm(DateUtils.parseDate(i.getMnebtddrhtm())); + hyMttdzE.setHravtdz(Convert.toBigDecimal(i.getHravtdz())); + hyMttdzE.setAvtdr(Convert.toBigDecimal(i.getAvtdr())); + hyMttdzE.setAvtddr(Convert.toInt(i.getAvtddr())); + return hyMttdzE; + }).collect(Collectors.toList()); + this.saveBatch(dataList, 10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + @Override + public R selectHyMttdzEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + IPage hyMtweEListPageByInfo = this.baseMapper.selectHyMttdzListPageByInfo(new Query().getPage(map), map); + return R.ok().put("data", hyMtweEListPageByInfo.getRecords()).put("count", hyMtweEListPageByInfo.getTotal()); + } + + /** + * 导出月潮位表数据 + */ + @Override + public R exportMonthTideLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "月潮位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + + +// Map map = CommonUtils.getMonthDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "monthTideLever.xls", "月潮位表", (Sheet sheet, CellStyle style) -> { +// dataModel(map, sheet, style); +// }); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthTideLever.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtweEListPageByInfo = this.baseMapper.selectHyMttdzListPageByInfo(new Query().getPage(map), map); + List records = hyMtweEListPageByInfo.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMttdzEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月潮位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvhtdz() != null ? CommonUtils.formatNum(vo.getAvhtdz()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvhtdzrcd() != null ? vo.getAvhtdzrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getHthtdz() != null ? CommonUtils.formatNum(vo.getHthtdz()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getHthtdzrcd() != null ? vo.getHthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getHthtdzotm() != null ? vo.getHthtdzotm() : "", style); + ExcelUtils.formatCell(row, 12, vo.getLthtdz() != null ? CommonUtils.formatNum(vo.getLthtdz()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getLthtdzrcd() != null ? vo.getLthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getLthtdzotm() != null ? vo.getLthtdzotm() : "", style); + ExcelUtils.formatCell(row, 15, vo.getAvltdz() != null ? CommonUtils.formatNum(vo.getAvltdz()) : "", style); + ExcelUtils.formatCell(row, 16, vo.getAvltdzrcd() != null ? vo.getAvltdzrcd() : "", style); + ExcelUtils.formatCell(row, 17, vo.getHtltdz() != null ? CommonUtils.formatNum(vo.getHtltdz()) : "", style); + ExcelUtils.formatCell(row, 18, vo.getHtltdzrcd() != null ? vo.getHtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 19, vo.getHtltdzotm() != null ? vo.getHtltdzotm() : "", style); + ExcelUtils.formatCell(row, 20, vo.getLtltdz() != null ? CommonUtils.formatNum(vo.getLtltdz()) : "", style); + ExcelUtils.formatCell(row, 21, vo.getLtltdzrcd() != null ? vo.getLtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 22, vo.getLtltdzotm() != null ? vo.getLtltdzotm() : "", style); + ExcelUtils.formatCell(row, 23, vo.getAvftdr() != null ? CommonUtils.formatNum(vo.getAvftdr()) : "", style); + ExcelUtils.formatCell(row, 24, vo.getAvftdrrcd() != null ? vo.getAvftdrrcd() : "", style); + ExcelUtils.formatCell(row, 25, vo.getMxfltdr() != null ? CommonUtils.formatNum(vo.getMxfltdr()) : "", style); + ExcelUtils.formatCell(row, 26, vo.getMxfltdrrcd() != null ? vo.getMxfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 27, vo.getMxfltdrhtm() != null ? vo.getMxfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 28, vo.getMnfltdr() != null ? CommonUtils.formatNum(vo.getMnfltdr()) : "", style); + ExcelUtils.formatCell(row, 29, vo.getMnfltdrrcd() != null ? vo.getMnfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 30, vo.getMnfltdrhtm() != null ? vo.getMnfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 31, vo.getAver() != null ? CommonUtils.formatNum(vo.getAver()) : "", style); + ExcelUtils.formatCell(row, 32, vo.getAverbbrrcd() != null ? vo.getAverbbrrcd() : "", style); + ExcelUtils.formatCell(row, 33, vo.getMxebtdr() != null ? CommonUtils.formatNum(vo.getMxebtdr()) : "", style); + ExcelUtils.formatCell(row, 34, vo.getMxebtdrrcd() != null ? vo.getMxebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 35, vo.getMxebtdrht() != null ? vo.getMxebtdrht() : "", style); + ExcelUtils.formatCell(row, 36, vo.getMnebtdr() != null ? CommonUtils.formatNum(vo.getMnebtdr()) : "", style); + ExcelUtils.formatCell(row, 37, vo.getMnebtdrrcd() != null ? vo.getMnebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 38, vo.getMnebtdrhtm() != null ? vo.getMnebtdrhtm() : "", style); + ExcelUtils.formatCell(row, 39, vo.getAvftd() != null ? vo.getAvftd().toString() : "", style); + ExcelUtils.formatCell(row, 40, vo.getAvftdrcd() != null ? vo.getAvftdrcd() : "", style); + ExcelUtils.formatCell(row, 41, vo.getMxfltddr() != null ? vo.getMxfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 42, vo.getMxfltddrrcd() != null ? vo.getMxfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 43, vo.getMxfltddrhtm() != null ? vo.getMxfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 44, vo.getMnfltddr() != null ? vo.getMnfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 45, vo.getMnfltddrrcd() != null ? vo.getMnfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 46, vo.getMnfltddrhtm() != null ? vo.getMnfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 47, vo.getAvebbdr() != null ? vo.getAvebbdr().toString() : "", style); + ExcelUtils.formatCell(row, 48, vo.getAvedrc() != null ? vo.getAvedrc() : "", style); + ExcelUtils.formatCell(row, 49, vo.getMxebtddr() != null ? vo.getMxebtddr().toString() : "", style); + ExcelUtils.formatCell(row, 50, vo.getMxebtddrrcd() != null ? vo.getMxebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 51, vo.getMxebtddrhtm() != null ? vo.getMxebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 52, vo.getMnbtddr() != null ? vo.getMnbtddr().toString() : "", style); + ExcelUtils.formatCell(row, 53, vo.getMnebtddrrcd() != null ? vo.getMnebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 54, vo.getMnebtddrhtm() != null ? vo.getMnebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 55, vo.getHravtdz() != null ? CommonUtils.formatNum(vo.getHravtdz()) : "", style); + ExcelUtils.formatCell(row, 56, vo.getHravtdzrcd() != null ? vo.getHravtdzrcd() : "", style); + ExcelUtils.formatCell(row, 57, vo.getAvtdr() != null ? CommonUtils.formatNum(vo.getAvtdr()) : "", style); + ExcelUtils.formatCell(row, 58, vo.getAvtdrrcd() != null ? vo.getAvtdrrcd() : "", style); + ExcelUtils.formatCell(row, 59, vo.getAvtddr() != null ? vo.getAvtddr().toString() : "", style); + ExcelUtils.formatCell(row, 60, vo.getAvtddrrcd() != null ? vo.getAvtddrrcd() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtweEListPageByInfo = this.baseMapper.selectHyMttdzListPageByInfo(new Query().getPage(map), map); + List records = hyMtweEListPageByInfo.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMttdzEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月潮位表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvhtdz() != null ? CommonUtils.formatNum(vo.getAvhtdz()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvhtdzrcd() != null ? vo.getAvhtdzrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getHthtdz() != null ? CommonUtils.formatNum(vo.getHthtdz()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getHthtdzrcd() != null ? vo.getHthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getHthtdzotm() != null ? vo.getHthtdzotm() : "", style); + ExcelUtils.formatCell(row, 12, vo.getLthtdz() != null ? CommonUtils.formatNum(vo.getLthtdz()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getLthtdzrcd() != null ? vo.getLthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getLthtdzotm() != null ? vo.getLthtdzotm() : "", style); + ExcelUtils.formatCell(row, 15, vo.getAvltdz() != null ? CommonUtils.formatNum(vo.getAvltdz()) : "", style); + ExcelUtils.formatCell(row, 16, vo.getAvltdzrcd() != null ? vo.getAvltdzrcd() : "", style); + ExcelUtils.formatCell(row, 17, vo.getHtltdz() != null ? CommonUtils.formatNum(vo.getHtltdz()) : "", style); + ExcelUtils.formatCell(row, 18, vo.getHtltdzrcd() != null ? vo.getHtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 19, vo.getHtltdzotm() != null ? vo.getHtltdzotm() : "", style); + ExcelUtils.formatCell(row, 20, vo.getLtltdz() != null ? CommonUtils.formatNum(vo.getLtltdz()) : "", style); + ExcelUtils.formatCell(row, 21, vo.getLtltdzrcd() != null ? vo.getLtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 22, vo.getLtltdzotm() != null ? vo.getLtltdzotm() : "", style); + ExcelUtils.formatCell(row, 23, vo.getAvftdr() != null ? CommonUtils.formatNum(vo.getAvftdr()) : "", style); + ExcelUtils.formatCell(row, 24, vo.getAvftdrrcd() != null ? vo.getAvftdrrcd() : "", style); + ExcelUtils.formatCell(row, 25, vo.getMxfltdr() != null ? CommonUtils.formatNum(vo.getMxfltdr()) : "", style); + ExcelUtils.formatCell(row, 26, vo.getMxfltdrrcd() != null ? vo.getMxfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 27, vo.getMxfltdrhtm() != null ? vo.getMxfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 28, vo.getMnfltdr() != null ? CommonUtils.formatNum(vo.getMnfltdr()) : "", style); + ExcelUtils.formatCell(row, 29, vo.getMnfltdrrcd() != null ? vo.getMnfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 30, vo.getMnfltdrhtm() != null ? vo.getMnfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 31, vo.getAver() != null ? CommonUtils.formatNum(vo.getAver()) : "", style); + ExcelUtils.formatCell(row, 32, vo.getAverbbrrcd() != null ? vo.getAverbbrrcd() : "", style); + ExcelUtils.formatCell(row, 33, vo.getMxebtdr() != null ? CommonUtils.formatNum(vo.getMxebtdr()) : "", style); + ExcelUtils.formatCell(row, 34, vo.getMxebtdrrcd() != null ? vo.getMxebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 35, vo.getMxebtdrht() != null ? vo.getMxebtdrht() : "", style); + ExcelUtils.formatCell(row, 36, vo.getMnebtdr() != null ? CommonUtils.formatNum(vo.getMnebtdr()) : "", style); + ExcelUtils.formatCell(row, 37, vo.getMnebtdrrcd() != null ? vo.getMnebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 38, vo.getMnebtdrhtm() != null ? vo.getMnebtdrhtm() : "", style); + ExcelUtils.formatCell(row, 39, vo.getAvftd() != null ? vo.getAvftd().toString() : "", style); + ExcelUtils.formatCell(row, 40, vo.getAvftdrcd() != null ? vo.getAvftdrcd() : "", style); + ExcelUtils.formatCell(row, 41, vo.getMxfltddr() != null ? vo.getMxfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 42, vo.getMxfltddrrcd() != null ? vo.getMxfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 43, vo.getMxfltddrhtm() != null ? vo.getMxfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 44, vo.getMnfltddr() != null ? vo.getMnfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 45, vo.getMnfltddrrcd() != null ? vo.getMnfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 46, vo.getMnfltddrhtm() != null ? vo.getMnfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 47, vo.getAvebbdr() != null ? vo.getAvebbdr().toString() : "", style); + ExcelUtils.formatCell(row, 48, vo.getAvedrc() != null ? vo.getAvedrc() : "", style); + ExcelUtils.formatCell(row, 49, vo.getMxebtddr() != null ? vo.getMxebtddr().toString() : "", style); + ExcelUtils.formatCell(row, 50, vo.getMxebtddrrcd() != null ? vo.getMxebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 51, vo.getMxebtddrhtm() != null ? vo.getMxebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 52, vo.getMnbtddr() != null ? vo.getMnbtddr().toString() : "", style); + ExcelUtils.formatCell(row, 53, vo.getMnebtddrrcd() != null ? vo.getMnebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 54, vo.getMnebtddrhtm() != null ? vo.getMnebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 55, vo.getHravtdz() != null ? CommonUtils.formatNum(vo.getHravtdz()) : "", style); + ExcelUtils.formatCell(row, 56, vo.getHravtdzrcd() != null ? vo.getHravtdzrcd() : "", style); + ExcelUtils.formatCell(row, 57, vo.getAvtdr() != null ? CommonUtils.formatNum(vo.getAvtdr()) : "", style); + ExcelUtils.formatCell(row, 58, vo.getAvtdrrcd() != null ? vo.getAvtdrrcd() : "", style); + ExcelUtils.formatCell(row, 59, vo.getAvtddr() != null ? vo.getAvtddr().toString() : "", style); + ExcelUtils.formatCell(row, 60, vo.getAvtddrrcd() != null ? vo.getAvtddrrcd() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtweEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtweEServiceImpl.java new file mode 100644 index 0000000..f5fa951 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtweEServiceImpl.java @@ -0,0 +1,355 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtweEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtweE; +import com.ruoyi.swlscx.month.domain.vo.HyMtweEVo; +import com.ruoyi.swlscx.month.mapper.HyMtweEMapper; +import com.ruoyi.swlscx.month.service.HyMtweEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.io.IOException; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtwe_e(月水面蒸发量表)】的数据库操作Service实现 +* @createDate 2024-08-01 09:38:45 +*/ +@Service +public class HyMtweEServiceImpl extends ServiceImpl + implements HyMtweEService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /**导入月水面蒸发量表数据*/ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtweEDate(MultipartFile file) { + try { + List hyMtweEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtweEBo.class, 3); + List hyMtweEList = hyMtweEBos.stream().map(i -> { + HyMtweE hyMtweE = new HyMtweE(); + BeanUtils.copyProperties(i, hyMtweE); + hyMtweE.setYr(Integer.valueOf(i.getYr())); + hyMtweE.setMth(Integer.valueOf(i.getMth())); + hyMtweE.setWsfe(Convert.toBigDecimal(i.getWsfe())); + hyMtweE.setMxdye(Convert.toBigDecimal(i.getMxdye())); + hyMtweE.setMndye(Convert.toBigDecimal(i.getMndye())); + return hyMtweE; + }).collect(Collectors.toList()); + this.saveBatch(hyMtweEList); + } catch (IOException e) { + throw new RuntimeException(e); + } + + } + + /**查询月水面蒸发量表数据*/ + @Override + public IPage selectHyMtweEListPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stcd,String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtweEListPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出月水面蒸发量表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public R exportMonthEvaporationWaterList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月水面蒸发量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); +// +// Map map = CommonUtils.getMonthDataMap(new PageParams(0L,100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response,"monthEvaporationWater.xls","月水面蒸发量表",(Sheet sheet, CellStyle style) ->{ +// dataModel(map, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtweEVoList) { + List records = hyMtweEVoList.stream().map(i->{ + HyMtweE hyMtweE = new HyMtweE(); + BeanUtils.copyProperties(i, hyMtweE); + hyMtweE.setYr(i.getYr()); + hyMtweE.setMth(i.getMth()); + hyMtweE.setWsfe(Convert.toBigDecimal(i.getWsfe())); + hyMtweE.setMxdye(Convert.toBigDecimal(i.getMxdye())); + hyMtweE.setMndye(Convert.toBigDecimal(i.getMndye())); + return hyMtweE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月水面蒸发量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月水面蒸发量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtweEListPageByInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtweEServiceImpl hyMtweEService = context.getBean(HyMtweEServiceImpl.class); + hyMtweEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthEvaporationWater.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMptEVoIPage = this.baseMapper.selectHyMtweEListPageByInfo(new Query().getPage(map), map); + List records = hyMptEVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtweEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 6, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row,8,vo.getWsfe() != null ? CommonUtils.formatNum(vo.getWsfe()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdye() != null ? CommonUtils.formatNum(vo.getMxdye()) : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyercd() != null ? vo.getMxdyercd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMndye() != null ? CommonUtils.formatNum(vo.getMndye()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMndyercd() != null ? vo.getMndyercd() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + + // 批量查询数据 + IPage hyMptEVoIPage = this.baseMapper.selectHyMtweEListPageByInfo(new Query().getPage(map), map); + List records = hyMptEVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtweEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 6, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row,8,vo.getWsfe() != null ? CommonUtils.formatNum(vo.getWsfe()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdye() != null ? CommonUtils.formatNum(vo.getMxdye()) : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyercd() != null ? vo.getMxdyercd() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMndye() != null ? CommonUtils.formatNum(vo.getMndye()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMndyercd() != null ? vo.getMndyercd() : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtwtEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtwtEServiceImpl.java new file mode 100644 index 0000000..f1536cc --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtwtEServiceImpl.java @@ -0,0 +1,360 @@ +package com.ruoyi.swlscx.month.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtwtEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtwtE; +import com.ruoyi.swlscx.month.domain.vo.HyMtwtEVo; +import com.ruoyi.swlscx.month.mapper.HyMtwtEMapper; +import com.ruoyi.swlscx.month.service.HyMtwtEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_mtwt_e(月水温表字段定义)】的数据库操作Service实现 + * @createDate 2024-08-06 13:42:04 + */ +@Service +public class HyMtwtEServiceImpl extends ServiceImpl + implements HyMtwtEService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 查询月水温表数据 + */ + @Override + public IPage selectHyMtweEDataByPageAndInfo(PageParams pageParams, String stcdIds, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtzEListPageByInfo(new Query().getPage(map), map); + } + + /** + * 导入月水温表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtwtEData(MultipartFile file) { + try { + List hyMtpddbEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtwtEBo.class, 2); + // 获取数据 + List dataList = hyMtpddbEBos.parallelStream().map(i -> { + HyMtwtE hyMtwtE = new HyMtwtE(); + BeanUtils.copyProperties(i, hyMtwtE); + hyMtwtE.setAvwtmp(Convert.toBigDecimal(i.getAvwtmp())); + hyMtwtE.setMth(Convert.toInt(i.getMth())); + hyMtwtE.setMnwtmp(Convert.toBigDecimal(i.getMnwtmp())); + hyMtwtE.setYr(Convert.toInt(i.getYr())); + hyMtwtE.setMnwtmpdt(DateUtils.parseDate(i.getMnwtmpdt())); + hyMtwtE.setMxwtmpdt(DateUtils.parseDate(i.getMxwtmpdt())); + hyMtwtE.setMxwtmp(Convert.toBigDecimal(i.getMxwtmp())); + + return hyMtwtE; + }).collect(Collectors.toList()); + this.saveBatch(dataList, 10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + @Override + public R exportMonthWaterTemperatureList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月水温表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtwtEVoList) { + List records = hyMtwtEVoList.stream().map(i->{ + HyMtwtE hyMtwtE = new HyMtwtE(); + BeanUtils.copyProperties(i, hyMtwtE); + hyMtwtE.setAvwtmp(Convert.toBigDecimal(i.getAvwtmp())); + hyMtwtE.setMth(Convert.toInt(i.getMth())); + hyMtwtE.setMnwtmp(Convert.toBigDecimal(i.getMnwtmp())); + hyMtwtE.setYr(Convert.toInt(i.getYr())); + hyMtwtE.setMnwtmpdt(DateUtils.parseDate(i.getMnwtmpdt())); + hyMtwtE.setMxwtmpdt(DateUtils.parseDate(i.getMxwtmpdt())); + hyMtwtE.setMxwtmp(Convert.toBigDecimal(i.getMxwtmp())); + + return hyMtwtE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月水温表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月水温表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtweEDataByPageAndInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtwtEServiceImpl hyMtwtEService = context.getBean(HyMtwtEServiceImpl.class); + hyMtwtEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,6); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,6); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthWaterTemperature.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtzEVoIPage = this.baseMapper.selectHyMtzEListPageByInfo(new Query().getPage(map), map); + List records = hyMtzEVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtwtEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水温表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvwtmp() != null ? CommonUtils.formatNum(vo.getAvwtmp()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvwtmprcd() != null ? vo.getAvwtmprcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwtmp() != null ? CommonUtils.formatNum(vo.getMxwtmp()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxwtmprcd() != null ? vo.getMxwtmprcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxwtmpdt() != null ? vo.getMxwtmpdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnwtmp() != null ? CommonUtils.formatNum(vo.getMnwtmp()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnwtmprcd() != null ? vo.getMnwtmprcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnwtmpdt() != null ? vo.getMnwtmpdt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + + // 批量查询数据 + IPage hyMtzEVoIPage = this.baseMapper.selectHyMtzEListPageByInfo(new Query().getPage(map), map); + List records = hyMtzEVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtwtEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水温表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvwtmp() != null ? CommonUtils.formatNum(vo.getAvwtmp()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvwtmprcd() != null ? vo.getAvwtmprcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwtmp() != null ? CommonUtils.formatNum(vo.getMxwtmp()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxwtmprcd() != null ? vo.getMxwtmprcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxwtmpdt() != null ? vo.getMxwtmpdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnwtmp() != null ? CommonUtils.formatNum(vo.getMnwtmp()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnwtmprcd() != null ? vo.getMnwtmprcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnwtmpdt() != null ? vo.getMnwtmpdt() : "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtzEServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtzEServiceImpl.java new file mode 100644 index 0000000..d52edb1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/month/service/impl/HyMtzEServiceImpl.java @@ -0,0 +1,360 @@ +package com.ruoyi.swlscx.month.service.impl; + + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.month.domain.bo.HyMtzEBo; +import com.ruoyi.swlscx.month.domain.po.HyMtzE; +import com.ruoyi.swlscx.month.domain.vo.HyMtzEVo; +import com.ruoyi.swlscx.month.mapper.HyMtzEMapper; +import com.ruoyi.swlscx.month.service.HyMtzEService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.io.IOException; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mtz_e】的数据库操作Service实现 +* @createDate 2024-07-23 14:15:25 +*/ +@Service +public class HyMtzEServiceImpl extends ServiceImpl + implements HyMtzEService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + /** + * 导入月水位表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMtzEData(MultipartFile file) { + try { + List HyMtzEBos = ExcelUtils.readExcel(file.getInputStream(), HyMtzEBo.class, 2); + List HyMtzEList = HyMtzEBos.stream().map(i -> { + HyMtzE hyMtweE = new HyMtzE(); + BeanUtils.copyProperties(i, hyMtweE); + hyMtweE.setYr(Integer.valueOf(i.getYr())); + hyMtweE.setMth(Integer.valueOf(i.getMth())); + hyMtweE.setAvz(Convert.toBigDecimal(i.getAvz())); + hyMtweE.setHtz(Convert.toBigDecimal(i.getHtz())); + hyMtweE.setHtzdt(DateUtils.parseDate(i.getHtzdt())); + hyMtweE.setMnz(Convert.toBigDecimal(i.getMnz())); + hyMtweE.setMnzdt(DateUtils.parseDate(i.getMnzdt())); + return hyMtweE; + }).collect(Collectors.toList()); + for (int i = 0; i < HyMtzEList.size(); i += 1000) { + int end = Math.min(i + 1000, HyMtzEList.size()); + List batchList = HyMtzEList.subList(i, end); + baseMapper.batchInsert(batchList); + } + } catch (IOException e) { + throw new RuntimeException(e); + } + + } + + /**查询月水位表*/ + @Override + public IPage selectHyMtzEPageByInfo(PageParams pageParams, String stcdIds, String startTime, String endTime,String stnm,String stcd) { + Map map = CommonUtils.getMonthDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtzEPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出月水位表 + */ + @Override + public R exportMonthWaterLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "月水位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map map = CommonUtils.getMonthDataMap(new PageParams(0L,100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response,"monthWaterLever.xls","月水位表",(Sheet sheet, CellStyle style) ->{ +// dataModel(map, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMtzEVoList) { + List records = hyMtzEVoList.stream().map(i->{ + HyMtzE hyMtweE = new HyMtzE(); + BeanUtils.copyProperties(i, hyMtweE); + hyMtweE.setYr(i.getYr()); + hyMtweE.setMth(i.getMth()); + hyMtweE.setAvz(Convert.toBigDecimal(i.getAvz())); + hyMtweE.setHtz(Convert.toBigDecimal(i.getHtz())); + hyMtweE.setHtzdt(DateUtils.parseDate(i.getHtzdt())); + hyMtweE.setMnz(Convert.toBigDecimal(i.getMnz())); + hyMtweE.setMnzdt(DateUtils.parseDate(i.getMnzdt())); + return hyMtweE; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "月水位表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "月水位表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyMtzEPageByInfo(new PageParams(0L, 1000000000L), null, startTime, endTime, null,null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyMtzEServiceImpl hyMtzEService = context.getBean(HyMtzEServiceImpl.class); + hyMtzEService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String filename = tableName + startTime + "-" + endTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getMonthDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("monthWaterLever.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage HyMtzEVoIPage = this.baseMapper.selectHyMtzEPageByInfo(new Query().getPage(map), map); + List records = HyMtzEVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMtzEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvz() != null ? CommonUtils.formatNum(vo.getAvz()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvzrcd() != null ? vo.getAvzrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getHtz() != null ? CommonUtils.formatNum(vo.getHtz()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getHtzrcd() != null ? vo.getHtzrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getHtzdt() != null ? vo.getHtzdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnz() != null ? CommonUtils.formatNum(vo.getMnz()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnzrcd() != null ? vo.getMnzrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnzdt() != null ? vo.getMnzdt() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage HyMtzEVoIPage = this.baseMapper.selectHyMtzEPageByInfo(new Query().getPage(map), map); + List records = HyMtzEVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMtzEVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row,4,"",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getMth() != null ? vo.getMth().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvz() != null ? CommonUtils.formatNum(vo.getAvz()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvzrcd() != null ? vo.getAvzrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getHtz() != null ? CommonUtils.formatNum(vo.getHtz()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getHtzrcd() != null ? vo.getHtzrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getHtzdt() != null ? vo.getHtzdt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnz() != null ? CommonUtils.formatNum(vo.getMnz()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnzrcd() != null ? vo.getMnzrcd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMnzdt() != null ? vo.getMnzdt() : "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearExportController.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearExportController.java new file mode 100644 index 0000000..9c2ce76 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearExportController.java @@ -0,0 +1,201 @@ +package com.ruoyi.swlscx.year.controller; + +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.thread.ExportThread; +import com.ruoyi.swlscx.year.service.*; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletResponse; + +/** + * @Author al + * @Date 2024/8/7 14:10 + * @Description: TODO + * @Version + */ + +@RequestMapping("/report") +@RequiredArgsConstructor +@RestController +public class YearExportController { + + private final HyYrpFService hyYrpFService; + + private final HyHmxpFService hyHmxpFService; + + private final HyDmxpFService hyDmxpFService; + + private final HyMmxpFService hyMmxpFService; + + private final HyYrzFService hyYrzFService; + + private final HyYrweFService hyYrweFService; + + private final HyYrqFService hyYrqFService; + + private final HyYrcsFService hyYrcsFService; + + private final HyImxfwFService hyImxfwFService; + + private final HyYrqsFService hyYrqsFService; + + private final HyYrwtFService hyYrwtFService; + + private final HyYrtdzFService hyYrtdzFService; + + + + + /** + * 导出年降水表数据 + */ + @RequestMapping("/yearRain") + public R exportYearRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrpFService.exportHyYrpFData(response,startTime,endTime,stcd,stnm); + } + + + + + + /** + * 导出小时时段最大降水量表 + */ + @RequestMapping("/hourMaxRain") + public R exportHourMaxRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyHmxpFService.exportHyHmxpFData(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出日时时段最大降水量表 + */ + @RequestMapping("/dayMaxRain") + public R exportDayMaxRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyDmxpFService.exportDayDmxpFData(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出分钟时段最大降水量表 + */ + @RequestMapping("/minMaxRain") + public R exportMinMaxRainList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyMmxpFService.exportMinMaxRainList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年水位表 + */ + @RequestMapping("/yearWaterLever") + public R exportYearWaterLeverList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrzFService.exportYearWaterLeverList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年水面蒸发量表 + */ + @RequestMapping("/yearEvaporationWater") + public R exportYearEvaporationWaterList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrweFService.exportYearEvaporationWaterList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年流量表 + */ + @RequestMapping("/yearFlow") + public R exportYearFlowList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrqFService.exportYearFlowList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出时段最大洪量表 + */ + @RequestMapping("/maxFloodScale") + public R exportMaxFloodScaleList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyImxfwFService.exportMaxFloodScaleList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年含沙量表 + */ + @RequestMapping("/sedimentConcentration") + public R exportSedimentConcentrationList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrcsFService.exportSedimentConcentrationList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年输沙率表 + */ + @RequestMapping("/sedimentTransportRate") + public R exportSedimentTransportRateList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrqsFService.exportSedimentTransportRateList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年水温表 + */ + @RequestMapping("/yearWaterTemperature") + public R exportYearWaterTemperatureList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrwtFService.exportYearWaterTemperatureList(response,startTime,endTime,stcd,stnm); + } + + /** + * 导出年潮位表 + */ + @RequestMapping("/yearTideLever") + public R exportYearTideLeverList(HttpServletResponse response, + @RequestParam String startTime, + @RequestParam String endTime, + @RequestParam String stcd, + @RequestParam String stnm) { + return hyYrtdzFService.exportYearTideLeverList(response,startTime,endTime,stcd,stnm); + } + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearImportToSoftController.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearImportToSoftController.java new file mode 100644 index 0000000..fda4fbf --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearImportToSoftController.java @@ -0,0 +1,143 @@ +package com.ruoyi.swlscx.year.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.day.service.*; +import com.ruoyi.swlscx.year.domain.vo.*; +import com.ruoyi.swlscx.year.service.*; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.*; + +/** + * @Author al + * @Date 2025/1/5 16:11 + * @Description: TODO + * @Version + */ +@RestController +@RequestMapping("/year/import") +@RequiredArgsConstructor +@Slf4j +public class YearImportToSoftController { + + private final HyHmxpFService hyHmxpFService; + + private final HyYrpFService hyYrpFService; + + private final HyYrweFService hyYrweFService; + + private final HyYrzFService hyYrzFService; + + private final HyYrqFService hyYrqFService; + + private final HyMmxpFService hyMmxpFService; + + private final HyDmxpFService hyDmxpFService; + + private final HyImxfwFService hyImxfwFService; + + private final HyYrcsFService hyYrcsFService; + + private final HyYrqsFService hyYrqsFService; + + private final HyYrwtFService hyYrwtFService; + + private final HyYrtdzFService hyYrtdzFService; + + + /** + * 导入年降水表数据 + */ + @PostMapping("/importYearRainDateToSoft") + public R getMonthRainList(String startTime, String endTime) { + return hyYrpFService.importDateTosoft(startTime,endTime); + } + + /** + * 导入分钟时段最大降水量表数据 + */ + @PostMapping("/importMaxMRainDateToSoft") + public R importMaxMRainDateToSoft(String startTime, String endTime) { + return hyMmxpFService.importDateTosoft(startTime, endTime); + } + + /** + * 导入小时时段最大降水量表数据 + */ + @RequestMapping("/importMaxHRainDateToSoft") + public R importMaxHRainDateToSoft(String startTime, String endTime) { + return hyHmxpFService.importDateTosoft(startTime, endTime); + } + + /** + * 导入日时时段最大降水量表数据 + */ + @RequestMapping("/importMaxDRainDateToSoft") + public R importMaxDRainDateToSoft(String startTime, String endTime) { + return hyDmxpFService.importDateTosoft(startTime,endTime); + } + + /**导入年水面蒸发表数据**/ + @RequestMapping("/importEvaporationWaterDateToSoft") + public R importEvaporationWaterDateToSoft(String startTime, String endTime) { + return hyYrweFService.importDateTosoft(startTime, endTime); + } + + /**导入年水位表数据**/ + @RequestMapping("/importYearWaterLeverDateToSoft") + public R importYearWaterLeverDateToSoft(String startTime, String endTime) { + return hyYrzFService.importDateTosoft(startTime, endTime); + } + + /**导入年流量表数据**/ + @RequestMapping("/importFlowDateToSoft") + public R importFlowDateToSoft(String startTime, String endTime) { + return hyYrqFService.importDateTosoft(startTime, endTime); + } + + + /** + * 导入年水温表数据 + */ + @RequestMapping("/importYearWaterTemperatureDateToSoft") + public R importYearWaterTemperatureDateToSoft(String startTime, String endTime) { + return hyYrwtFService.importDateTosoft(startTime, endTime); + } + + /** + * 导入年输沙率表数据 + */ + @RequestMapping("/importYearsedimentTransportRateDateToSoft") + public R getYearsedimentTransportRateList(String startTime, String endTime) { + return hyYrqsFService.importDateTosoft(startTime, endTime); + } + + + /** + * 导入年含沙量表数据 + */ + @RequestMapping("/importYearSedimentConcentrationDateToSoft") + public R importYearSedimentConcentrationDateToSoft(String startTime, String endTime) { + return hyYrcsFService.importDateTosoft(startTime, endTime); + } + + + /** + * 导入时段最大洪量表数据 + */ + @RequestMapping("/importMaxFloodScaleDateToSoft") + public R importMaxFloodScaleDateToSoft(String startTime, String endTime) { + return hyImxfwFService.importDateTosoft(startTime, endTime); + } + + /** + * 导入年潮位表数据 + */ + @RequestMapping("/importYearTideLeverDateToSoft") + public R importYearTideLeverDateToSoft(String startTime, String endTime) { + return hyYrtdzFService.importDateTosoft(startTime, endTime); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearTableController.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearTableController.java new file mode 100644 index 0000000..36c6aa2 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/controller/YearTableController.java @@ -0,0 +1,184 @@ +package com.ruoyi.swlscx.year.controller; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.vo.*; +import com.ruoyi.swlscx.year.service.*; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; + +/** + * @Author al + * @Date 2024/7/23 14:04 + * @Description: TODO + * @Version + */ + +@RestController +@RequestMapping("/year") +@RequiredArgsConstructor +@Slf4j +public class YearTableController { + + private final HyHmxpFService hyHmxpFService; + + private final HyYrpFService hyYrpFService; + + private final HyYrweFService hyYrweFService; + + private final HyYrzFService hyYrzFService; + + private final HyYrqFService hyYrqFService; + + private final HyMmxpFService hyMmxpFService; + + private final HyDmxpFService hyDmxpFService; + + private final HyImxfwFService hyImxfwFService; + + private final HyYrcsFService hyYrcsFService; + + private final HyYrqsFService hyYrqsFService; + + private final HyYrwtFService hyYrwtFService; + + private final HyYrtdzFService hyYrtdzFService; + + /** + * 导入时段最大降雨量数据 + */ + @PostMapping("/importHourMaxRainData") + public R importHourMaxRainData(MultipartFile file) { + log.info("file:{}", file); + hyHmxpFService.importHourMaxRainData(file); + return R.ok(); + } + + + /** + * 查询年降水表数据 + */ + @GetMapping("/getYearRainList") + public R getMonthRainList(Integer pageNum, Integer pageSize, String startTime, String endTime, String stcd, String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyYrpFVoIPage = hyYrpFService.selectHyYrpFDataByPageAndInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyYrpFVoIPage.getRecords()).put("count", hyYrpFVoIPage.getTotal()); + } + + /**查询年水面蒸发表数据**/ + @RequestMapping("/getEvaporationWaterList") + public R getEvaporationWaterList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyYrweFVoIPage = hyYrweFService.selectHyYrweFListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyYrweFVoIPage.getRecords()).put("count", hyYrweFVoIPage.getTotal()); + } + + /**查询年水位表数据**/ + @GetMapping("/getYearWaterLeverList") + public R getWaterLeverList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stnm,String stcd) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyYrzFVoIPage = hyYrzFService.selectHyYrzFPageByInfo(pageParams, startTime, endTime, stnm, stcd); + return R.ok().put("data", hyYrzFVoIPage.getRecords()).put("count", hyYrzFVoIPage.getTotal()); + } + + + /**查询年流量表数据**/ + @GetMapping("/getYearFlowList") + public R getFlowList(Integer pageNum, Integer pageSize, String startTime, String endTime, String stcd, String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyYrqFVoIPage = hyYrqFService.selecHyYrqFListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyYrqFVoIPage.getRecords()).put("count", hyYrqFVoIPage.getTotal()); + } + + + + /** + * 查询分钟时段最大降水量表数据 + */ + @RequestMapping("/getMaxMRainList") + public R getMaxMRainList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyMmxpFVoIPage = hyMmxpFService.selectMaxMRainListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyMmxpFVoIPage.getRecords()).put("count", hyMmxpFVoIPage.getTotal()); + } + + /** + * 查询小时时段最大降水量表数据 + */ + @RequestMapping("/getMaxHRainList") + public R getMaxHRainList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyHmxpFVoIPage = hyHmxpFService.selectMaxHRainListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyHmxpFVoIPage.getRecords()).put("count", hyHmxpFVoIPage.getTotal()); + } + + /** + * 查询日时时段最大降水量表 + */ + @RequestMapping("/getMaxDRainList") + public R getMaxDRainList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyDmxpFVoIPage = hyDmxpFService.selectMaxDRainListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyDmxpFVoIPage.getRecords()).put("count", hyDmxpFVoIPage.getTotal()); + } + + + /** + * 查询时段最大洪量表数据 + */ + @RequestMapping("/getMaxFloodScaleList") + public R getMaxFloodScaleList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyImxfwFVoIPage = hyImxfwFService.selectMaxFloodScaleListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyImxfwFVoIPage.getRecords()).put("count", hyImxfwFVoIPage.getTotal()); + } + + + /** + * 查询年含沙量表数据 + */ + @RequestMapping("/getYearSedimentConcentrationList") + public R getYearSedimentConcentrationList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage yearSedimentConcentrationList = hyYrcsFService.getYearSedimentConcentrationList(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", yearSedimentConcentrationList.getRecords()).put("count", yearSedimentConcentrationList.getTotal()); + } + + /** + * 查询年输沙率表数据 + */ + @RequestMapping("/getYearsedimentTransportRateList") + public R getYearsedimentTransportRateList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage yearsedimentTransportRateList = hyYrqsFService.getYearsedimentTransportRateList(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", yearsedimentTransportRateList.getRecords()).put("count", yearsedimentTransportRateList.getTotal()); + } + + + /** + * 查询年水温表数据 + */ + @RequestMapping("/getYearWaterTemperatureList") + public R getYearWaterTemperatureList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage yearWaterTemperatureList = hyYrwtFService.getYearWaterTemperatureList(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", yearWaterTemperatureList.getRecords()).put("count", yearWaterTemperatureList.getTotal()); + } + + /** + * 查询年潮位表数据 + */ + @RequestMapping("/getYearTideLeverList") + public R getYearTideLeverList(Integer pageNum,Integer pageSize, String startTime, String endTime,String stcd,String stnm) { + PageParams pageParams = new PageParams(pageNum.longValue(), pageSize.longValue()); + IPage hyYrtdzFVoIPage = hyYrtdzFService.selectHyYrtdzFListPageByInfo(pageParams, startTime, endTime, stcd, stnm); + return R.ok().put("data", hyYrtdzFVoIPage.getRecords()).put("count", hyYrtdzFVoIPage.getTotal()); + } + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyDmxpFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyDmxpFBo.java new file mode 100644 index 0000000..ae75b3c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyDmxpFBo.java @@ -0,0 +1,54 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日时段最大降水量 + * + * @TableName hy_dmxp_f + */ +@Data +public class HyDmxpFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 最大降水量时段长 + */ + @ExcelProperty(value = "最大降水量时段长", index = 7) + private String mxpdr; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 起始日期 + */ + @ExcelProperty(value = "起始日期", index = 6) + private String bgdt; + + /** + * 最大降水量 + */ + @ExcelProperty(value = "最大降水量", index = 8) + private String mxp; + + /** + * 最大降水量注解码 + */ + @ExcelProperty(value = "最大降水量注解码", index = 9) + private String mxprc; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyHmxpFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyHmxpFBo.java new file mode 100644 index 0000000..bebfa8b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyHmxpFBo.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +/** + * @Author al + * @Date 2024/7/30 14:42 + * @Description: TODO + * @Version + */ + + +@Data +public class HyHmxpFBo { + + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + + @ExcelProperty(value = "年", index = 5) + private String yr; + @ExcelProperty(value = "起时间", index = 6) + private String bgtm; + @ExcelProperty(value = "最大降水量时段长", index = 7) + private String mxpdr; + @ExcelProperty(value = "最大降水量", index = 8) + private String mxp; + @ExcelProperty(value = "最大降水量注解码", index = 9) + private String mxprc; + + + + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyImxfwFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyImxfwFBo.java new file mode 100644 index 0000000..e8635f7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyImxfwFBo.java @@ -0,0 +1,54 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 时段最大洪量表 + * + * @TableName hy_imxfw_f + */ +@Data +public class HyImxfwFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 最大洪量时段长 + */ + @ExcelProperty(value = "最大洪量时段长", index = 7) + private String mxwdr; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 起始日期 + */ + @ExcelProperty(value = "起始日期", index = 6) + private String bgdt; + + /** + * 最大洪量 + */ + @ExcelProperty(value = "最大洪量", index = 8) + private String mxw; + + /** + * 最大洪量注解码 + */ + @ExcelProperty(value = "最大洪量注解码", index = 9) + private String mxwrc; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyMmxpFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyMmxpFBo.java new file mode 100644 index 0000000..45b4f97 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyMmxpFBo.java @@ -0,0 +1,54 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 分钟时段最大降水量表 + * + * @TableName hy_mmxp_f + */ +@Data +public class HyMmxpFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 最大降水量时段长 + */ + @ExcelProperty(value = "最大降水量时段长", index = 7) + private String mxpdr; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 起时间 + */ + @ExcelProperty(value = "起时间", index = 6) + private String bgdt; + + /** + * 最大降水量 + */ + @ExcelProperty(value = "最大降水量", index = 8) + private String mxp; + + /** + * 最大降水量注解码 + */ + @ExcelProperty(value = "最大降水量注解码", index = 9) + private String mxprc; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrcsFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrcsFBo.java new file mode 100644 index 0000000..ef05d4d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrcsFBo.java @@ -0,0 +1,78 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 年含沙量表 + * + * @TableName hy_yrcs_f + */ +@Data +public class HyYrcsFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均含沙量(千克每立方米) + */ + @ExcelProperty(value = "平均含沙量", index = 6) + private String avcs; + + /** + * 平均含沙量注解码 + */ + @ExcelProperty(value = "平均含沙量注解码", index = 7) + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + @ExcelProperty(value = "最大含沙量", index = 8) + private String mxs; + + /** + * 最大含沙量注解码 + */ + @ExcelProperty(value = "最大含沙量注解码", index = 9) + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + @ExcelProperty(value = "最大含沙量日期", index = 10) + private String mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + @ExcelProperty(value = "最小含沙量", index = 11) + private String mns; + + /** + * 最小含沙量注解码 + */ + @ExcelProperty(value = "最小含沙量注解码", index = 12) + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + @ExcelProperty(value = "最小含沙量日期", index = 13) + private String mnsdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrpFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrpFBo.java new file mode 100644 index 0000000..e19d373 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrpFBo.java @@ -0,0 +1,75 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 年降水量表 + * @TableName hy_yrp_f + */ +@Data +public class HyYrpFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码",index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年",index = 5) + private String yr; + + /** + * 降水量(毫米) + */ + @ExcelProperty(value = "降水量",index = 6) + private String p; + + /** + * 降水量注解码 + */ + @ExcelProperty(value = "降水量注解码",index = 7) + private String prcd; + + /** + * 降水日数 + */ + @ExcelProperty(value = "降水日数",index = 8) + private String pdynum; + + /** + * 降水日数注解码 + */ + @ExcelProperty(value = "降水日数注解码",index = 9) + private String pdynumrcd; + + /** + * 终霜日期 + */ + @ExcelProperty(value = "终霜日期",index = 10) + private String frdsdt; + + /** + * 初霜日期 + */ + @ExcelProperty(value = "初霜日期",index = 11) + private String frapdt; + + /** + * 终雪日期 + */ + @ExcelProperty(value = "终雪日期",index = 12) + private String sndsdt; + + /** + * 初雪日期 + */ + @ExcelProperty(value = "初雪日期",index = 13) + private String snapdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqFBo.java new file mode 100644 index 0000000..20a6ee4 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqFBo.java @@ -0,0 +1,98 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +/** + * 年流量表 + * + * @TableName hy_yrq_f + */ +@Data +public class HyYrqFBo { + + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均流量 + */ + @ExcelProperty(value = "平均流量", index = 6) + private String avq; + + /** + * 平均流量注解码 + */ + @ExcelProperty(value = "平均流量注解码", index = 7) + private String avqrcd; + + /** + * 最高流量 + */ + @ExcelProperty(value = "最高流量", index = 8) + private String mxq; + + /** + * 最高流量注解码 + */ + @ExcelProperty(value = "最高流量注解码", index = 9) + private String mxqrcd; + + /** + * 最高流量日期 + */ + @ExcelProperty(value = "最高流量日期", index = 10) + private String mxqdt; + + /** + * 最小流量 + */ + @ExcelProperty(value = "最小流量", index = 11) + private String mnq; + + /** + * 最小流量注解码 + */ + @ExcelProperty(value = "最小流量注解码", index = 12) + private String mnqrcd; + + /** + * 最小流量日期 + */ + @ExcelProperty(value = "最小流量日期", index = 13) + private String mnqdt; + + /** + * 径流量(万立方米) + */ + @ExcelProperty(value = "径流量", index = 14) + private String rw; + + /** + * 径流量注解码 + */ + @ExcelProperty(value = "径流量注解码", index = 15) + private String rwrcd; + + /** + * 径流模数(立方分米每秒平方千米) + */ + @ExcelProperty(value = "径流模数", index = 16) + private String rm; + + /** + * 径流深(毫米) + */ + @ExcelProperty(value = "径流深", index = 17) + private String rd; + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqsFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqsFBo.java new file mode 100644 index 0000000..4a7435b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrqsFBo.java @@ -0,0 +1,97 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +/** + * 年输沙率表 + * + * @TableName hy_yrqs_f + */ +@Data +public class HyYrqsFBo { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 泥沙类型 + */ + @ExcelProperty(value = "泥沙类型", index = 6) + private String sdtp; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均输沙率(千克每秒) + */ + @ExcelProperty(value = "平均输沙率", index = 7) + private String avqs; + + /** + * 平均输沙率注解码 + */ + @ExcelProperty(value = "平均输沙率注解码", index = 8) + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + @ExcelProperty(value = "最大日平均沙率", index = 9) + private String mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + @ExcelProperty(value = "最大日平均沙率注解码", index = 10) + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + @ExcelProperty(value = "最大日平均沙率出现日期", index = 11) + private String mxdyqsodt; + + /** + * 输沙量(万吨) + */ + @ExcelProperty(value = "输沙量", index = 12) + private String sw; + + /** + * 输沙量注解码 + */ + @ExcelProperty(value = "输沙量注解码", index = 13) + private String swpc; + + /** + * 输沙数模 + */ + @ExcelProperty(value = "输沙数模", index = 14) + private String sm; + + /** + * 采样仪器型号 + */ + @ExcelProperty(value = "采样仪器型号", index = 15) + private String simn; + + /** + * 采样效率系数 + */ + @ExcelProperty(value = "采样效率系数", index = 16) + private String smec; + + /** + * 备注 + */ + @ExcelProperty(value = "备注", index = 17) + private String nt; + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrtdzFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrtdzFBo.java new file mode 100644 index 0000000..da55e41 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrtdzFBo.java @@ -0,0 +1,353 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 年潮位表 + * + * @TableName hy_yrtdz_f + */ +@Data +public class HyYrtdzFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均高潮潮位 + */ + @ExcelProperty(value = "平均高潮潮位",index = 6) + private String avhtdz; + + /** + * 平均高潮潮位注解码 + */ + @ExcelProperty(value = "平均高潮潮位注解码", index = 7) + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + @ExcelProperty(value = "最高高潮位", index = 8) + private String hthtdz; + + /** + * 最高高潮位注解码 + */ + @ExcelProperty(value = "最高高潮位注解码", index = 9) + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + @ExcelProperty(value = "最高高潮位出现时间", index = 10) + private String hthtdzotm; + + /** + * 最低高潮位 + */ + @ExcelProperty(value = "最低高潮位", index = 11) + private String lthtdz; + + /** + * 最低高潮位注解码 + */ + @ExcelProperty(value = "最低高潮位注解码", index = 12) + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + @ExcelProperty(value = "最低高潮位出现时间", index = 13) + private String lthtdzotm; + + /** + * 平均低潮潮位 + */ + @ExcelProperty(value = "平均低潮潮位", index = 14) + private String avltdz; + + /** + * 平均低潮潮位注解码 + */ + @ExcelProperty(value = "平均低潮潮位注解码", index = 15) + private String avltdzrcd; + + /** + * 最高低潮位 + */ + @ExcelProperty(value = "最高低潮位", index = 16) + private String htltdz; + + /** + * 最高低潮位注解码 + */ + @ExcelProperty(value = "最高低潮位注解码", index = 17) + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + @ExcelProperty(value = "最高低潮潮位出现时间", index = 18) + private String htltdzotm; + + /** + * 最低低潮位 + */ + @ExcelProperty(value = "最低低潮位", index = 19) + private String ltltdz; + + /** + * 最低低潮潮位注解码 + */ + @ExcelProperty(value = "最低低潮潮位注解码", index = 20) + private String ltltdzrcd; + + /** + * 最低低潮位出现时间 + */ + @ExcelProperty(value = "最低低潮位出现时间", index = 21) + private String ltltdzotm; + + /** + * 平均涨潮潮差 + */ + @ExcelProperty(value = "平均涨潮潮差", index = 22) + private String avftdr; + + /** + * 平均涨潮潮差注解码 + */ + @ExcelProperty(value = "平均涨潮潮差注解码", index = 23) + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + @ExcelProperty(value = "最大涨潮潮差", index = 24) + private String mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + @ExcelProperty(value = "最大涨潮潮差注解码", index = 25) + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最大涨潮潮差(高潮)时间", index = 26) + private String mxfltdrhtm; + + /** + * 最小潮潮差 + */ + @ExcelProperty(value = "最小潮潮差", index = 27) + private String mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + @ExcelProperty(value = "最小涨潮潮差注解码", index = 28) + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最小涨潮潮差(高潮)时间", index = 29) + private String mnfltdrhtm; + + + /** + * 平均涨潮潮差 + */ + @ExcelProperty(value = "平均涨潮潮差", index = 30) + private String aver; + + /** + * 平均涨潮潮差注解码 + */ + @ExcelProperty(value = "平均涨潮潮差注解码", index = 31) + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + @ExcelProperty(value = "最大落潮潮差", index = 32) + private String mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + @ExcelProperty(value = "最大落潮潮差注解码", index = 33) + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最大落潮潮差(高潮)时间", index = 34) + private String mxebtdrht; + + /** + * 最小落潮潮差 + */ + @ExcelProperty(value = "最小落潮潮差", index = 35) + private String mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + @ExcelProperty(value = "最小落潮潮差注解码", index = 36) + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + @ExcelProperty(value = "最小落潮潮差(高潮)时间", index = 37) + private String mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + @ExcelProperty(value = "平均涨潮历时", index = 38) + private String avftd; + + /** + * 平均涨潮历时注解码 + */ + @ExcelProperty(value = "平均涨潮历时注解码", index = 39) + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + @ExcelProperty(value = "最长涨潮历时", index = 40) + private String mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + @ExcelProperty(value = "最长涨潮历时注解码", index = 41) + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + @ExcelProperty(value = "最长涨潮历时(高潮)时间", index = 42) + private String mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + @ExcelProperty(value = "最短涨潮历时", index = 43) + private String mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + @ExcelProperty(value = "最短涨潮历时注解码", index = 44) + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + @ExcelProperty(value = "最短涨潮历时(高潮)时间", index = 45) + private String mnfltddrhtm; + + /** + * 平均落潮历时 + */ + @ExcelProperty(value = "平均落潮历时", index = 46) + private String avebbdr; + + /** + * 平均落潮历时注解码 + */ + @ExcelProperty(value = "平均落潮历时注解码", index = 47) + private String avedrc; + + /** + * 最长落潮历时 + */ + @ExcelProperty(value = "最长落潮历时", index = 48) + private String mxebtddr; + + /** + * 最长落潮历时注解码 + */ + @ExcelProperty(value = "最长落潮历时注解码", index = 49) + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + @ExcelProperty(value = "最长落潮历时(高潮)时间", index = 50) + private String mxebtddrhtm; + + /** + * 最短落潮历时 + */ + @ExcelProperty(value = "最短落潮历时", index = 51) + private String mnbtddr; + + /** + * 最短落潮历时注解码 + */ + @ExcelProperty(value = "最短落潮历时注解码", index = 52) + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + @ExcelProperty(value = "最短落潮历时(高潮)时间", index = 53) + private String mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + @ExcelProperty(value = "逐时平均潮位", index = 54) + private String hravtdz; + + /** + * 逐时平均潮位注解码 + */ + @ExcelProperty(value = "逐时平均潮位注解码", index = 55) + private String hravtdzrcd; + + /** + * 平均潮差 + */ + @ExcelProperty(value = "平均潮差", index = 56) + private String avtdr; + + /** + * 平均潮差注解码 + */ + @ExcelProperty(value = "平均潮差注解码", index = 57) + private String avtdrrcd; + + /** + * 平均历时 + */ + @ExcelProperty(value = "平均历时", index = 58) + private String avtddr; + + /** + * 平均历时注解码 + */ + @ExcelProperty(value = "平均历时注解码", index = 59) + private String avtddrrcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrweFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrweFBo.java new file mode 100644 index 0000000..4bb1d82 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrweFBo.java @@ -0,0 +1,106 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 年水面蒸发量表 + * + * @TableName hy_yrwe_f + */ +@Data +public class HyYrweFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 蒸发器型式 + */ + @ExcelProperty(value = "蒸发器型式", index = 5) + private String eetp; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 6) + private String yr; + + /** + * 水面蒸发量(毫米) + */ + @ExcelProperty(value = "水面蒸发量", index = 7) + private String wsfe; + + /** + * 水面蒸发量注解码 + */ + @ExcelProperty(value = "水面蒸发量注解码", index = 8) + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + @ExcelProperty(value = "最大日水面蒸发量", index = 9) + private String mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + @ExcelProperty(value = "最大日水面蒸发量注解码", index = 10) + private String mxdyercd; + + /** + * 最大日水面蒸发量出现日期 + */ + @ExcelProperty(value = "最大日水面蒸发量出现日期", index = 11) + private String mxdyeodt; + + /** + * 最小日水面蒸发量 + */ + @ExcelProperty(value = "最小日水面蒸发量", index = 12) + private String mndye; + + /** + * 最小日水面蒸发量注解码 + */ + @ExcelProperty(value = "最小日水面蒸发量注解码", index = 13) + private String mndyercd; + + /** + * 最小日水面蒸发量出现日期 + */ + @ExcelProperty(value = "最小日水面蒸发量出现日期", index = 14) + private String mndyeodt; + + /** + * 终冰日期 + */ + @ExcelProperty(value = "终冰日期", index = 15) + private String idsdt; + + /** + * 初冰日期 + */ + @ExcelProperty(value = "初冰日期", index = 16) + private String icapd; + + /** + * 蒸发场位置特征 + */ + @ExcelProperty(value = "蒸发场位置特征", index = 17) + private String eslcch; + + /** + * 备注 + */ + @ExcelProperty(value = "备注", index = 18) + private String nt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrwtFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrwtFBo.java new file mode 100644 index 0000000..53979bf --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrwtFBo.java @@ -0,0 +1,75 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 年水温表 + * + * @TableName hy_yrwt_f + */ +@Data +public class HyYrwtFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均水温(摄氏度) + */ + @ExcelProperty(value = "平均水温", index = 6) + private String avwtmp; + + /** + * 平均水温注解码 + */ + @ExcelProperty(value = "平均水温注解码", index = 7) + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + @ExcelProperty(value = "最高水温", index = 8) + private String mxwtmp; + + /** + * 最高水温注解码 + */ + @ExcelProperty(value = "最高水温注解码", index = 9) + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + private String mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + @ExcelProperty(value = "最低水温", index = 10) + private String mnwtmp; + + /** + * 最低水温注解码 + */ + @ExcelProperty(value = "最低水温注解码", index = 11) + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + @ExcelProperty(value = "最低水温日期", index = 12) + private String mnwtmpdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrzFBo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrzFBo.java new file mode 100644 index 0000000..4ffc612 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/bo/HyYrzFBo.java @@ -0,0 +1,76 @@ +package com.ruoyi.swlscx.year.domain.bo; + +import com.alibaba.excel.annotation.ExcelProperty; +import lombok.Data; + +import java.io.Serializable; + +/** + * 年水位表 + * + * @TableName hy_yrz_f + */ +@Data +public class HyYrzFBo implements Serializable { + /** + * 站码 + */ + @ExcelProperty(value = "站码", index = 1) + private String stcd; + + /** + * 年 + */ + @ExcelProperty(value = "年", index = 5) + private String yr; + + /** + * 平均水位 + */ + @ExcelProperty(value = "平均水位", index = 6) + private String avz; + + /** + * 平均水位注解码 + */ + @ExcelProperty(value = "平均水位注解码", index = 7) + private String avzrcd; + + /** + * 最高水位 + */ + @ExcelProperty(value = "最高水位", index = 8) + private String htz; + + /** + * 最高水位注解码 + */ + @ExcelProperty(value = "最高水位注解码", index = 9) + private String htzrcd; + + /** + * 最高水位日期 + */ + @ExcelProperty(value = "最高水位日期", index = 10) + private String htzdt; + + /** + * 最低水位 + */ + @ExcelProperty(value = "最低水位", index = 11) + private String mnz; + + /** + * 最低水位注解码 + */ + @ExcelProperty(value = "最低水位注解码", index = 12) + private String mnzrcd; + + /** + * 最低水位日期 + */ + @ExcelProperty(value = "最低水位日期", index = 13) + private String mnzdt; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyDmxpF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyDmxpF.java new file mode 100644 index 0000000..87603a9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyDmxpF.java @@ -0,0 +1,51 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 日时段最大降水量 + * @TableName hy_dmxp_f + */ +@TableName(value ="hy_dmxp_f") +@Data +public class HyDmxpF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长 + */ + private Integer mxpdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起始日期 + */ + private Date bgdt; + + /** + * 最大降水量 + */ + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + private String mxprc; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyHmxpF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyHmxpF.java new file mode 100644 index 0000000..1d7ab4c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyHmxpF.java @@ -0,0 +1,63 @@ +package com.ruoyi.swlscx.year.domain.po; + + +import com.alibaba.excel.annotation.ExcelProperty; +import com.alibaba.excel.annotation.format.DateTimeFormat; +import com.baomidou.mybatisplus.annotation.FieldStrategy; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @author 12974 + * @TableName hy_hmxp_f + */ +@TableName(value ="hy_hmxp_f") +@Data +public class HyHmxpF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长(小时) + */ + @TableField(updateStrategy = FieldStrategy.IGNORED,insertStrategy = FieldStrategy.IGNORED) + private Integer mxpdr; + + /** + * 年 + */ + @TableField(updateStrategy = FieldStrategy.IGNORED,insertStrategy = FieldStrategy.IGNORED) + private Integer yr; + + /** + * 起时间 + */ + @TableField(updateStrategy = FieldStrategy.IGNORED,insertStrategy = FieldStrategy.IGNORED) + @DateTimeFormat("yyyy-MM-dd HH:mm:ss") + private Date bgtm; + + /** + * 最大降水量(毫米) + */ + @TableField(updateStrategy = FieldStrategy.IGNORED,insertStrategy = FieldStrategy.IGNORED) + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + @TableField(updateStrategy = FieldStrategy.IGNORED,insertStrategy = FieldStrategy.IGNORED) + private String mxprc; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; + + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyImxfwF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyImxfwF.java new file mode 100644 index 0000000..4350eb8 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyImxfwF.java @@ -0,0 +1,51 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 时段最大洪量表 + * @TableName hy_imxfw_f + */ +@TableName(value ="hy_imxfw_f") +@Data +public class HyImxfwF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大洪量时段长 + */ + private String mxwdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起始日期 + */ + private Date bgdt; + + /** + * 最大洪量 + */ + private BigDecimal mxw; + + /** + * 最大洪量注解码 + */ + private String mxwrc; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyMmxpF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyMmxpF.java new file mode 100644 index 0000000..bc12db7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyMmxpF.java @@ -0,0 +1,51 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 分钟时段最大降水量表 + * @TableName hy_mmxp_f + */ +@TableName(value ="hy_mmxp_f") +@Data +public class HyMmxpF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长 + */ + private Integer mxpdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起时间 + */ + private Date bgdt; + + /** + * 最大降水量 + */ + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + private String mxprc; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrcsF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrcsF.java new file mode 100644 index 0000000..c9f8571 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrcsF.java @@ -0,0 +1,71 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年含沙量表 + * @TableName hy_yrcs_f + */ +@TableName(value ="hy_yrcs_f") +@Data +public class HyYrcsF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + private BigDecimal mxs; + + /** + * 最大含沙量注解码 + */ + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + private Date mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + private BigDecimal mns; + + /** + * 最小含沙量注解码 + */ + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + private Date mnsdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrpF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrpF.java new file mode 100644 index 0000000..bc9029c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrpF.java @@ -0,0 +1,71 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年降水量表 + * @TableName hy_yrp_f + */ +@TableName(value ="hy_yrp_f") +@Data +public class HyYrpF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + /** + * 降水日数 + */ + private String pdynum; + + /** + * 降水日数注解码 + */ + private String pdynumrcd; + + /** + * 终霜日期 + */ + private Date frdsdt; + + /** + * 初霜日期 + */ + private Date frapdt; + + /** + * 终雪日期 + */ + private Date sndsdt; + + /** + * 初雪日期 + */ + private Date snapdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqF.java new file mode 100644 index 0000000..1268786 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqF.java @@ -0,0 +1,91 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年流量表 + * @TableName hy_yrq_f + */ +@TableName(value ="hy_yrq_f") +@Data +public class HyYrqF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均流量 + */ + private BigDecimal avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + /** + * 最高流量 + */ + private BigDecimal mxq; + + /** + * 最高流量注解码 + */ + private String mxqrcd; + + /** + * 最高流量日期 + */ + private Date mxqdt; + + /** + * 最小流量 + */ + private BigDecimal mnq; + + /** + * 最小流量注解码 + */ + private String mnqrcd; + + /** + * 最小流量日期 + */ + private Date mnqdt; + + /** + * 径流量(万立方米) + */ + private BigDecimal rw; + + /** + * 径流量注解码 + */ + private String rwrcd; + + /** + * 径流模数(立方分米每秒平方千米) + */ + private BigDecimal rm; + + /** + * 径流深(毫米) + */ + private BigDecimal rd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqsF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqsF.java new file mode 100644 index 0000000..e9e5d02 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrqsF.java @@ -0,0 +1,91 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年输沙率表 + * @TableName hy_yrqs_f + */ +@TableName(value ="hy_yrqs_f") +@Data +public class HyYrqsF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均输沙率(千克每秒) + */ + private BigDecimal avqs; + + /** + * 平均输沙率注解码 + */ + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + private BigDecimal mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + private Date mxdyqsodt; + + /** + * 输沙量(万吨) + */ + private BigDecimal sw; + + /** + * 输沙量注解码 + */ + private String swpc; + + /** + * 输沙数模 + */ + private BigDecimal sm; + + /** + * 采样仪器型号 + */ + private String simn; + + /** + * 采样效率系数 + */ + private BigDecimal smec; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrtdzF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrtdzF.java new file mode 100644 index 0000000..7bbf414 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrtdzF.java @@ -0,0 +1,301 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年潮位表 + * @TableName hy_yrtdz_f + */ +@TableName(value ="hy_yrtdz_f") +@Data +public class HyYrtdzF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均高潮潮位 + */ + private BigDecimal avhtdz; + + /** + * 平均高潮潮位注解码 + */ + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + private BigDecimal hthtdz; + + /** + * 最高高潮位注解码 + */ + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + private Date hthtdzotm; + + /** + * 最低高潮位 + */ + private BigDecimal lthtdz; + + /** + * 最低高潮位注解码 + */ + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + private Date lthtdzotm; + + /** + * 平均低潮潮位 + */ + private BigDecimal avltdz; + + /** + * 平均低潮潮位注解码 + */ + private String avltdzrcd; + + /** + * 最高低潮位 + */ + private BigDecimal htltdz; + + /** + * 最高低潮位注解码 + */ + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + private Date htltdzotm; + + /** + * 最低低潮位 + */ + private BigDecimal ltltdz; + + /** + * 最低低潮潮位注解码 + */ + private String ltltdzrcd; + + /** + * 最低地潮位出现时间 + */ + private Date ltltdzotm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal avftdr; + + /** + * 平均涨潮潮差注解码 + */ + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + private BigDecimal mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + private Date mxfltdrhtm; + + /** + * 最小潮潮差 + */ + private BigDecimal mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + private Date mnfltdrhtm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal aver; + + /** + * 平均涨潮潮差注解码 + */ + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + private BigDecimal mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + private Date mxebtdrht; + + /** + * 最小落潮潮差 + */ + private BigDecimal mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + private Date mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + private Integer avftd; + + /** + * 平均涨潮历时注解码 + */ + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + private Integer mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + private Date mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + private Integer mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + private Date mnfltddrhtm; + + /** + * 平均落潮历时 + */ + private Integer avebbdr; + + /** + * 平均落潮历时注解码 + */ + private String avedrc; + + /** + * 最长落潮历时 + */ + private Integer mxebtddr; + + /** + * 最长落潮历时注解码 + */ + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + private Date mxebtddrhtm; + + /** + * 最短落潮历时 + */ + private Integer mnbtddr; + + /** + * 最短落潮历时注解码 + */ + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + private Date mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + private BigDecimal hravtdz; + + /** + * 逐时平均潮位注解码 + */ + private String hravtdzrcd; + + /** + * 平均潮差 + */ + private BigDecimal avtdr; + + /** + * 平均潮差注解码 + */ + private String avtdrrcd; + + /** + * 平均历时 + */ + private Integer avtddr; + + /** + * 平均历时注解码 + */ + private String avtddrrcd; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrweF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrweF.java new file mode 100644 index 0000000..b2a0f15 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrweF.java @@ -0,0 +1,97 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年水面蒸发量表 + * @TableName hy_yrwe_f + */ +@TableName(value ="hy_yrwe_f") +@Data +public class HyYrweF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 年 + */ + @TableId + private Integer yr; + + /** + * 水面蒸发量(毫米) + */ + private BigDecimal wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + private BigDecimal mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + private String mxdyercd; + + /** + * 最大日水面蒸发量出现日期 + */ + private Date mxdyeodt; + + /** + * 最小日水面蒸发量 + */ + private BigDecimal mndye; + + /** + * 最小日水面蒸发量注解码 + */ + private String mndyercd; + + /** + * 最小日水面蒸发量出现日期 + */ + private Date mndyeodt; + + /** + * 终冰日期 + */ + private Date idsdt; + + /** + * 初冰日期 + */ + private Date icapd; + + /** + * 蒸发场位置特征 + */ + private String eslcch; + + /** + * 备注 + */ + private String nt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrwtF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrwtF.java new file mode 100644 index 0000000..bbc163a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrwtF.java @@ -0,0 +1,71 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年水温表 + * @TableName hy_yrwt_f + */ +@TableName(value ="hy_yrwt_f") +@Data +public class HyYrwtF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均水温(摄氏度) + */ + private BigDecimal avwtmp; + + /** + * 平均水温注解码 + */ + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + private BigDecimal mxwtmp; + + /** + * 最高水温注解码 + */ + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + private Date mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + private BigDecimal mnwtmp; + + /** + * 最低水温注解码 + */ + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + private Date mnwtmpdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrzF.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrzF.java new file mode 100644 index 0000000..a6c0631 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/po/HyYrzF.java @@ -0,0 +1,71 @@ +package com.ruoyi.swlscx.year.domain.po; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; +import lombok.Data; + +/** + * 年水位表 + * @TableName hy_yrz_f + */ +@TableName(value ="hy_yrz_f") +@Data +public class HyYrzF implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均水位 + */ + private BigDecimal avz; + + /** + * 平均水位注解码 + */ + private String avzrcd; + + /** + * 最高水位 + */ + private BigDecimal htz; + + /** + * 最高水位注解码 + */ + private String htzrcd; + + /** + * 最高水位日期 + */ + private Date htzdt; + + /** + * 最低水位 + */ + private BigDecimal mnz; + + /** + * 最低水位注解码 + */ + private String mnzrcd; + + /** + * 最低水位日期 + */ + private Date mnzdt; + + @TableField(exist = false) + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyDmxpFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyDmxpFVo.java new file mode 100644 index 0000000..377fa8f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyDmxpFVo.java @@ -0,0 +1,62 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 日时段最大降水量 + * @TableName hy_dmxp_f + */ +@Data +public class HyDmxpFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长 + */ + private Integer mxpdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起始日期 + */ + private String bgdt; + + /** + * 最大降水量 + */ + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + private String mxprc; + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyHmxpFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyHmxpFVo.java new file mode 100644 index 0000000..c6fe6ce --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyHmxpFVo.java @@ -0,0 +1,70 @@ +package com.ruoyi.swlscx.year.domain.vo; + + +import com.alibaba.excel.annotation.format.DateTimeFormat; +import com.baomidou.mybatisplus.annotation.FieldStrategy; +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import com.fasterxml.jackson.annotation.JsonFormat; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @author 12974 + * @TableName hy_hmxp_f + */ +@Data +public class HyHmxpFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长(小时) + */ + private Integer mxpdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起时间 + */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8") + private String bgtm; + + /** + * 最大降水量(毫米) + */ + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + private String mxprc; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + private static final long serialVersionUID = 1L; + + +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyImxfwFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyImxfwFVo.java new file mode 100644 index 0000000..e2209d3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyImxfwFVo.java @@ -0,0 +1,63 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 时段最大洪量表 + * @TableName hy_imxfw_f + */ +@Data +public class HyImxfwFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大洪量时段长 + */ + private String mxwdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起始日期 + */ + private String bgdt; + + /** + * 最大洪量 + */ + private BigDecimal mxw; + + /** + * 最大洪量注解码 + */ + private String mxwrc; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyMmxpFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyMmxpFVo.java new file mode 100644 index 0000000..bf2cd7d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyMmxpFVo.java @@ -0,0 +1,63 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 分钟时段最大降水量表 + * @TableName hy_mmxp_f + */ +@Data +public class HyMmxpFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 最大降水量时段长 + */ + private Integer mxpdr; + + /** + * 年 + */ + private Integer yr; + + /** + * 起时间 + */ + private String bgdt; + + /** + * 最大降水量 + */ + private BigDecimal mxp; + + /** + * 最大降水量注解码 + */ + private String mxprc; + + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrcsFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrcsFVo.java new file mode 100644 index 0000000..ea662f7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrcsFVo.java @@ -0,0 +1,82 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 年含沙量表 + * @TableName hy_yrcs_f + */ +@Data +public class HyYrcsFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均含沙量(千克每立方米) + */ + private BigDecimal avcs; + + /** + * 平均含沙量注解码 + */ + private String avcsrcd; + + /** + * 最大含沙量(千克每立方米) + */ + private BigDecimal mxs; + + /** + * 最大含沙量注解码 + */ + private String mxsrcd; + + /** + * 最大含沙量日期 + */ + private String mxsdt; + + /** + * 最小含沙量(千克每立方米) + */ + private BigDecimal mns; + + /** + * 最小含沙量注解码 + */ + private String mnsrcd; + + /** + * 最小含沙量日期 + */ + private String mnsdt; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrpFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrpFVo.java new file mode 100644 index 0000000..4f41580 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrpFVo.java @@ -0,0 +1,76 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 年降水量表 + * @TableName hy_yrp_f + */ +@Data +public class HyYrpFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 降水量(毫米) + */ + private BigDecimal p; + + /** + * 降水量注解码 + */ + private String prcd; + + /** + * 降水日数 + */ + private String pdynum; + + /** + * 降水日数注解码 + */ + private String pdynumrcd; + + /** + * 终霜日期 + */ + private Date frdsdt; + + /** + * 初霜日期 + */ + private Date frapdt; + + /** + * 终雪日期 + */ + private Date sndsdt; + + /** + * 初雪日期 + */ + private Date snapdt; + + /**站名**/ + private String stnm; + + /** + * 行政区划码 + */ + private String addvcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqFVo.java new file mode 100644 index 0000000..9080f09 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqFVo.java @@ -0,0 +1,101 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import lombok.Data; + +import java.math.BigDecimal; +import java.util.Date; + +/** + * @Author al + * @Date 2024/9/9 17:02 + * @Description: TODO + * @Version 年流量表 + */ +@Data +public class HyYrqFVo { + + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均流量 + */ + private BigDecimal avq; + + /** + * 平均流量注解码 + */ + private String avqrcd; + + /** + * 最高流量 + */ + private BigDecimal mxq; + + /** + * 最高流量注解码 + */ + private String mxqrcd; + + /** + * 最高流量日期 + */ + private String mxqdt; + + /** + * 最小流量 + */ + private BigDecimal mnq; + + /** + * 最小流量注解码 + */ + private String mnqrcd; + + /** + * 最小流量日期 + */ + private String mnqdt; + + /** + * 径流量(万立方米) + */ + private BigDecimal rw; + + /** + * 径流量注解码 + */ + private String rwrcd; + + /** + * 径流模数(立方分米每秒平方千米) + */ + private BigDecimal rm; + + /** + * 径流深(毫米) + */ + private BigDecimal rd; + + /** + * 站名 + */ + private String stnm; + + + private String addvcd; + + + /** + * 流域水系码 + */ + private String lysxm; + +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqsFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqsFVo.java new file mode 100644 index 0000000..4d7f3ea --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrqsFVo.java @@ -0,0 +1,102 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 年输沙率表 + * @TableName hy_yrqs_f + */ +@Data +public class HyYrqsFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 泥沙类型 + */ + private String sdtp; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均输沙率(千克每秒) + */ + private BigDecimal avqs; + + /** + * 平均输沙率注解码 + */ + private String avqsrcd; + + /** + * 最大日平均沙率(千克每秒) + */ + private BigDecimal mxdyqs; + + /** + * 最大日平均沙率注解码 + */ + private String mxdyqsrcd; + + /** + * 最大日平均沙率出现日期 + */ + private String mxdyqsodt; + + /** + * 输沙量(万吨) + */ + private BigDecimal sw; + + /** + * 输沙量注解码 + */ + private String swpc; + + /** + * 输沙数模 + */ + private BigDecimal sm; + + /** + * 采样仪器型号 + */ + private String simn; + + /** + * 采样效率系数 + */ + private BigDecimal smec; + + /** + * 备注 + */ + private String nt; + + /** + * 站名 + */ + private String stnm; + + + private String addvcd; + + + /** + * 流域水系码 + */ + private String lysxm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrtdzFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrtdzFVo.java new file mode 100644 index 0000000..7da1a8a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrtdzFVo.java @@ -0,0 +1,310 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * + * @TableName hy_yrz_f + */ +@Data +public class HyYrtdzFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均高潮潮位 + */ + private BigDecimal avhtdz; + + /** + * 平均高潮潮位注解码 + */ + private String avhtdzrcd; + + /** + * 最高高潮位 + */ + private BigDecimal hthtdz; + + /** + * 最高高潮位注解码 + */ + private String hthtdzrcd; + + /** + * 最高高潮位出现时间 + */ + private String hthtdzotm; + + /** + * 最低高潮位 + */ + private BigDecimal lthtdz; + + /** + * 最低高潮位注解码 + */ + private String lthtdzrcd; + + /** + * 最低高潮位出现时间 + */ + private String lthtdzotm; + + /** + * 平均低潮潮位 + */ + private BigDecimal avltdz; + + /** + * 平均低潮潮位注解码 + */ + private String avltdzrcd; + + /** + * 最高低潮位 + */ + private BigDecimal htltdz; + + /** + * 最高低潮位注解码 + */ + private String htltdzrcd; + + /** + * 最高低潮潮位出现时间 + */ + private String htltdzotm; + + /** + * 最低低潮位 + */ + private BigDecimal ltltdz; + + /** + * 最低低潮潮位注解码 + */ + private String ltltdzrcd; + + /** + * 最低地潮位出现时间 + */ + private String ltltdzotm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal avftdr; + + /** + * 平均涨潮潮差注解码 + */ + private String avftdrrcd; + + /** + * 最大涨潮潮差 + */ + private BigDecimal mxfltdr; + + /** + * 最大涨潮潮差注解码 + */ + private String mxfltdrrcd; + + /** + * 最大涨潮潮差(高潮)时间 + */ + private String mxfltdrhtm; + + /** + * 最小潮潮差 + */ + private BigDecimal mnfltdr; + + /** + * 最小涨潮潮差注解码 + */ + private String mnfltdrrcd; + + /** + * 最小涨潮潮差(高潮)时间 + */ + private String mnfltdrhtm; + + /** + * 平均涨潮潮差 + */ + private BigDecimal aver; + + /** + * 平均涨潮潮差注解码 + */ + private String averbbrrcd; + + /** + * 最大落潮潮差 + */ + private BigDecimal mxebtdr; + + /** + * 最大落潮潮差注解码 + */ + private String mxebtdrrcd; + + /** + * 最大落潮潮差(高潮)时间 + */ + private String mxebtdrht; + + /** + * 最小落潮潮差 + */ + private BigDecimal mnebtdr; + + /** + * 最小落潮潮差注解码 + */ + private String mnebtdrrcd; + + /** + * 最小落潮潮差(高潮)时间 + */ + private String mnebtdrhtm; + + /** + * 平均涨潮历时 + */ + private Integer avftd; + + /** + * 平均涨潮历时注解码 + */ + private String avftdrcd; + + /** + * 最长涨潮历时 + */ + private Integer mxfltddr; + + /** + * 最长涨潮历时注解码 + */ + private String mxfltddrrcd; + + /** + * 最长涨潮历时(高潮)时间 + */ + private String mxfltddrhtm; + + /** + * 最短涨潮历时 + */ + private Integer mnfltddr; + + /** + * 最短涨潮历时注解码 + */ + private String mnfltddrrcd; + + /** + * 最短涨潮历时(高潮)时间 + */ + private String mnfltddrhtm; + + /** + * 平均落潮历时 + */ + private Integer avebbdr; + + /** + * 平均落潮历时注解码 + */ + private String avedrc; + + /** + * 最长落潮历时 + */ + private Integer mxebtddr; + + /** + * 最长落潮历时注解码 + */ + private String mxebtddrrcd; + + /** + * 最长落潮历时(高潮)时间 + */ + private String mxebtddrhtm; + + /** + * 最短落潮历时 + */ + private Integer mnbtddr; + + /** + * 最短落潮历时注解码 + */ + private String mnebtddrrcd; + + /** + * 最短落潮历时(高潮)时间 + */ + private String mnebtddrhtm; + + /** + * 逐时平均潮位 + */ + private BigDecimal hravtdz; + + /** + * 逐时平均潮位注解码 + */ + private String hravtdzrcd; + + /** + * 平均潮差 + */ + private BigDecimal avtdr; + + /** + * 平均潮差注解码 + */ + private String avtdrrcd; + + /** + * 平均历时 + */ + private Integer avtddr; + + /** + * 平均历时注解码 + */ + private String avtddrrcd; + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrweFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrweFVo.java new file mode 100644 index 0000000..77a8892 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrweFVo.java @@ -0,0 +1,98 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import lombok.Data; + +import java.math.BigDecimal; +import java.util.Date; + +/** + * @Author al + * @Date 2024/8/1 10:24 + * @Description: TODO + * @Version + */ + +@Data +public class HyYrweFVo { + + /** + * 站码 + */ + private String stcd; + + /** + * 蒸发器型式 + */ + private String eetp; + + /** + * 年 + */ + private Integer yr; + + /** + * 水面蒸发量(毫米) + */ + private Double wsfe; + + /** + * 水面蒸发量注解码 + */ + private String wsfercd; + + /** + * 最大日水面蒸发量 + */ + private Double mxdye; + + /** + * 最大日水面蒸发量注解码 + */ + private String mxdyercd; + + /** + * 最大日水面蒸发量出现日期 + */ + private String mxdyeodt; + + /** + * 最小日水面蒸发量 + */ + private Double mndye; + + /** + * 最小日水面蒸发量注解码 + */ + private String mndyercd; + + /** + * 最小日水面蒸发量出现日期 + */ + private String mndyeodt; + + /** + * 终冰日期 + */ + private String idsdt; + + /** + * 初冰日期 + */ + private String icapd; + + /** + * 蒸发场位置特征 + */ + private String eslcch; + + /** + * 备注 + */ + private String nt; + + private String stnm; + + private String lysxm; + + private String addvcd; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrwtFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrwtFVo.java new file mode 100644 index 0000000..2b71d5b --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrwtFVo.java @@ -0,0 +1,74 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import com.baomidou.mybatisplus.annotation.TableField; +import com.baomidou.mybatisplus.annotation.TableName; +import lombok.Data; + +import java.io.Serializable; +import java.math.BigDecimal; +import java.util.Date; + +/** + * 年水温表 + * @TableName hy_yrwt_f + */ +@Data +public class HyYrwtFVo implements Serializable { + /** + * 站码 + */ + private String stcd; + + /** + * 年 + */ + private Integer yr; + + /** + * 平均水温(摄氏度) + */ + private BigDecimal avwtmp; + + /** + * 平均水温注解码 + */ + private String avwtmprcd; + + /** + * 最高水温(摄氏度) + */ + private BigDecimal mxwtmp; + + /** + * 最高水温注解码 + */ + private String mxwtmprcd; + + /** + * 最高水温日期 + */ + private String mxwtmpdt; + + /** + * 最低水温(摄氏度) + */ + private BigDecimal mnwtmp; + + /** + * 最低水温注解码 + */ + private String mnwtmprcd; + + /** + * 最低水温日期 + */ + private Date mnwtmpdt; + + private String stnm; + + private String lysxm; + + private String addvcd; + + private static final long serialVersionUID = 1L; +} \ No newline at end of file diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrzFVo.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrzFVo.java new file mode 100644 index 0000000..0ec2ff4 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/domain/vo/HyYrzFVo.java @@ -0,0 +1,80 @@ +package com.ruoyi.swlscx.year.domain.vo; + +import lombok.Data; + +import java.math.BigDecimal; +import java.util.Date; + +/** + * @Author al + * @Date 2024/9/9 16:21 + * @Description: TODO + * @Version + */ +@Data +public class HyYrzFVo { + /** + * 站点编码 + */ + private String stcd; + + /** + * 年份 + */ + private Integer yr; + + /** + * 平均值 + */ + private BigDecimal avz; + + /** + * 平均值记录 + */ + private String avzrcd; + + /** + * 最大值 + */ + private BigDecimal htz; + + /** + * 最大值记录 + */ + private String htzrcd; + + /** + * 最大值日期时间 + */ + private String htzdt; + + /** + * 最小值 + */ + private BigDecimal mnz; + + /** + * 最小值记录 + */ + private String mnzrcd; + + /** + * 最小值日期时间 + */ + private String mnzdt; + + + /** + * 行政区划码 + */ + private String addvcd; + + /** + * 流域水系码 + */ + private String lysxm; + + + /**站名**/ + private String stnm; +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyDmxpFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyDmxpFMapper.java new file mode 100644 index 0000000..e35bd65 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyDmxpFMapper.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyDmxpF; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.year.domain.vo.HyDmxpFVo; +import org.apache.ibatis.annotations.Param; +import org.springframework.security.core.parameters.P; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_dmxp_f(日时段最大降水量)】的数据库操作Mapper +* @createDate 2024-09-11 10:09:13 +* @Entity com.ruoyi.swlscx.domain.year.po.HyDmxpF +*/ +public interface HyDmxpFMapper extends BaseMapper { + + IPage selecthyDmxpFPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyHmxpFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyHmxpFMapper.java new file mode 100644 index 0000000..33f9504 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyHmxpFMapper.java @@ -0,0 +1,42 @@ +package com.ruoyi.swlscx.year.mapper; + + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_hmxp_f】的数据库操作Mapper +* @createDate 2024-07-23 14:15:25 +* @Entity com/ruoyi/swlscx/domain/po.po.HyHmxpF +*/ +@Mapper +public interface HyHmxpFMapper extends BaseMapper { + + + void batchInsert(@Param("list") List list); + + + /** + * 查询时段最大降雨量表数据 + */ + IPage selecthyMmxpFPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyImxfwFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyImxfwFMapper.java new file mode 100644 index 0000000..e776134 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyImxfwFMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyImxfwF; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.year.domain.vo.HyImxfwFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_imxfw_f(时段最大洪量表)】的数据库操作Mapper +* @createDate 2024-09-09 17:32:19 +* @Entity com.ruoyi.swlscx.domain.year.po.HyImxfwF +*/ +@Mapper +public interface HyImxfwFMapper extends BaseMapper { + + /** + * 查询时段最大洪量表数据 + */ + IPage selectHyImxfwFPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyMmxpFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyMmxpFMapper.java new file mode 100644 index 0000000..5f92e4d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyMmxpFMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyMmxpF; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.ruoyi.swlscx.year.domain.vo.HyMmxpFVo; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_mmxp_f(分钟时段最大降水量表)】的数据库操作Mapper +* @createDate 2024-09-11 10:09:13 +* @Entity com.ruoyi.swlscx.domain.year.po.HyMmxpF +*/ +public interface HyMmxpFMapper extends BaseMapper { + + void batchInsert(@Param("list") List batchList); + + /** + * 查询分钟时段最大降水量表数据 + */ + IPage selectHyMmxpFPageByInfo(Page page,@Param("map") Map map); + + void insertTempData(@Param("list")List batch); + + void deleteTempData(); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrcsFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrcsFMapper.java new file mode 100644 index 0000000..25e6495 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrcsFMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrcsF; +import com.ruoyi.swlscx.year.domain.vo.HyYrcsFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrcs_f(年含沙量表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrcsF +*/ +@Mapper +public interface HyYrcsFMapper extends BaseMapper { + + /** + * 查询年含沙量表数据 + */ + IPage selecthyYrcsFPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrpFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrpFMapper.java new file mode 100644 index 0000000..ec51fd3 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrpFMapper.java @@ -0,0 +1,51 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.basic.domain.po.HyStscA; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import com.ruoyi.swlscx.year.domain.po.HyYrpF; +import com.ruoyi.swlscx.year.domain.vo.HyYrpFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrp_f(年降水量表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrpF +*/ +@Mapper +public interface HyYrpFMapper extends BaseMapper { + + /** + * 批量添加到临时表中 + */ + void batchInsert(@Param("list") List batchList); + + /** + * 刷新数据 + */ + void batchInsertOrUpdate(); + + /** + * 先清空临时表的数据 + */ + void deleteAll(); + + IPage selectHyMtpDataByPageAndInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + + void mergeToData(); + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqFMapper.java new file mode 100644 index 0000000..3ee3ca6 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqFMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrqF; +import com.ruoyi.swlscx.year.domain.vo.HyYrqFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrq_f(年流量表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrqF +*/ +@Mapper +public interface HyYrqFMapper extends BaseMapper { + + IPage selectHyMtqEPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqsFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqsFMapper.java new file mode 100644 index 0000000..308271c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrqsFMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrqsF; +import com.ruoyi.swlscx.year.domain.vo.HyYrqsFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrqs_f(年输沙率表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrqsF +*/ +@Mapper +public interface HyYrqsFMapper extends BaseMapper { + + /** + * 查询年输沙率表数据 + */ + IPage selectHyYrqFPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrtdzFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrtdzFMapper.java new file mode 100644 index 0000000..996fd92 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrtdzFMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrtdzF; +import com.ruoyi.swlscx.year.domain.vo.HyYrtdzFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrtdz_f(年潮位表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrtdzF +*/ +@Mapper +public interface HyYrtdzFMapper extends BaseMapper { + + IPage selectHyYrtdzFListPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrweFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrweFMapper.java new file mode 100644 index 0000000..4c7779a --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrweFMapper.java @@ -0,0 +1,36 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrweF; +import com.ruoyi.swlscx.year.domain.vo.HyYrweFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrwe_f(年水面蒸发量表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrweF +*/ +@Mapper +public interface HyYrweFMapper extends BaseMapper { + + + + IPage selechyYrweFListPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrwtFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrwtFMapper.java new file mode 100644 index 0000000..df4acdb --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrwtFMapper.java @@ -0,0 +1,37 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrwtF; +import com.ruoyi.swlscx.year.domain.vo.HyYrwtFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrwt_f(年水温表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrwtF +*/ +@Mapper +public interface HyYrwtFMapper extends BaseMapper { + + /** + * 查询年水温表数据 + */ + IPage selectHyYrwtFPageByInfo(Page page,@Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrzFMapper.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrzFMapper.java new file mode 100644 index 0000000..27bb5a7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/mapper/HyYrzFMapper.java @@ -0,0 +1,34 @@ +package com.ruoyi.swlscx.year.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.ruoyi.swlscx.year.domain.po.HyYrzF; +import com.ruoyi.swlscx.year.domain.vo.HyYrzFVo; +import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; + +import java.util.List; +import java.util.Map; + +/** +* @author 12974 +* @description 针对表【hy_yrz_f(年水位表)】的数据库操作Mapper +* @createDate 2024-09-05 16:53:36 +* @Entity com.ruoyi.swlscx.domain.po.HyYrzF +*/ +@Mapper +public interface HyYrzFMapper extends BaseMapper { + + IPage selectHyYrzFPageByInfo(Page page, @Param("map") Map map); + + void deleteTempData(); + + void insertTempData(@Param("list") List batch); + + void mergeToData(); +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyDmxpFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyDmxpFService.java new file mode 100644 index 0000000..082937d --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyDmxpFService.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyDmxpF; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.year.domain.vo.HyDmxpFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_dmxp_f(日时段最大降水量)】的数据库操作Service +* @createDate 2024-09-11 10:09:13 +*/ +public interface HyDmxpFService extends IService { + /** 导入日时段最大降水量数据*/ + void importHyDmxpFData(MultipartFile file); + + /** + * 查询小时时段最大降水量表数据 + */ + IPage selectMaxDRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出小时时段最大降水量表数据 + */ + R exportDayDmxpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyHmxpFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyHmxpFService.java new file mode 100644 index 0000000..8f4d0c1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyHmxpFService.java @@ -0,0 +1,42 @@ +package com.ruoyi.swlscx.year.service; + + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_hmxp_f】的数据库操作Service +* @createDate 2024-07-23 14:15:25 +*/ +public interface HyHmxpFService extends IService { + + /** + * 导入时段最大降雨量数据 + */ + void importHourMaxRainData(MultipartFile file); + + + /** + * 查询小时时段最大降水量表数据 + */ + IPage selectMaxHRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + + /** + * 导出小时时段最大降水量表数据 + */ + R exportHyHmxpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + /** + * 导入小时时段最大降水量表数据到sqlserver + */ + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyImxfwFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyImxfwFService.java new file mode 100644 index 0000000..2bee488 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyImxfwFService.java @@ -0,0 +1,38 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyImxfwF; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.year.domain.vo.HyImxfwFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_imxfw_f(时段最大洪量表)】的数据库操作Service +* @createDate 2024-09-09 17:32:19 +*/ +public interface HyImxfwFService extends IService { + + + /** + * 导入时段最大洪量表 + */ + void importHyImxfwFData(MultipartFile file); + + + /** + * 查询时段最大洪量表数据 + */ + IPage selectMaxFloodScaleListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出时段最大洪量表 + */ + R exportMaxFloodScaleList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyMmxpFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyMmxpFService.java new file mode 100644 index 0000000..d9d369e --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyMmxpFService.java @@ -0,0 +1,39 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyMmxpF; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.swlscx.year.domain.vo.HyMmxpFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_mmxp_f(分钟时段最大降水量表)】的数据库操作Service +* @createDate 2024-09-11 10:09:13 +*/ +public interface HyMmxpFService extends IService { + + /** + * 导入分钟时段最大降水量表 + */ + void importHyMmxpFData(MultipartFile file); + + /** + * 查询分钟时段最大降水量表数据 + */ + IPage selectMaxMRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出分钟时段最大降水量表 + */ + R exportMinMaxRainList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + /** + * 导入数据到南方片 + */ + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrcsFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrcsFService.java new file mode 100644 index 0000000..52bf5f2 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrcsFService.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrcsF; +import com.ruoyi.swlscx.year.domain.vo.HyYrcsFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrcs_f(年含沙量表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrcsFService extends IService { + /** + * 导入年含沙量表 + */ + void importHyYrcsFData(MultipartFile file); + + /** + * 查询年含沙量表数据 + */ + IPage getYearSedimentConcentrationList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年含沙量表 + */ + R exportSedimentConcentrationList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrpFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrpFService.java new file mode 100644 index 0000000..2ca7eea --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrpFService.java @@ -0,0 +1,32 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrpF; +import com.ruoyi.swlscx.year.domain.vo.HyYrpFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrp_f(年降水量表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrpFService extends IService { + /** 导入年降水表数据*/ + void importHyYrpFData(MultipartFile file); + /** + * 查询年降水表数据 + */ + IPage selectHyYrpFDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年降水表数据 + */ + R exportHyYrpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqFService.java new file mode 100644 index 0000000..98e12d0 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqFService.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrqF; +import com.ruoyi.swlscx.year.domain.vo.HyYrqFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrq_f(年流量表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrqFService extends IService { + + /** 导入年流量数据*/ + void importHyYrpFData(MultipartFile file); + + /**查询年流量表数据*/ + IPage selecHyYrqFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + + /** + * 导出年流量表 + */ + R exportYearFlowList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqsFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqsFService.java new file mode 100644 index 0000000..993ae0c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrqsFService.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrqsF; +import com.ruoyi.swlscx.year.domain.vo.HyYrqsFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrqs_f(年输沙率表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrqsFService extends IService { + /** 导入年输沙率表据*/ + void importHyYrqsFData(MultipartFile file); + + /** + * 查询年输沙率表数据 + */ + IPage getYearsedimentTransportRateList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年输沙率表 + */ + R exportSedimentTransportRateList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrtdzFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrtdzFService.java new file mode 100644 index 0000000..230f4a8 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrtdzFService.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrtdzF; +import com.ruoyi.swlscx.year.domain.vo.HyYrtdzFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrtdz_f(年潮位表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrtdzFService extends IService { + /** + * 导入年潮位表表 + */ + void importHyMmxpFData(MultipartFile file); + + /**查询年潮位表数据*/ + IPage selectHyYrtdzFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年潮位表 + */ + R exportYearTideLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrweFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrweFService.java new file mode 100644 index 0000000..d22efc9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrweFService.java @@ -0,0 +1,32 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrweF; +import com.ruoyi.swlscx.year.domain.vo.HyYrweFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrwe_f(年水面蒸发量表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrweFService extends IService { + /** 导入年水面蒸发量表数据*/ + void importHyYrweFData(MultipartFile file); + + + /**查询年水面蒸发量表数据*/ + IPage selectHyYrweFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年水面蒸发量表 + */ + R exportYearEvaporationWaterList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrwtFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrwtFService.java new file mode 100644 index 0000000..fafc666 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrwtFService.java @@ -0,0 +1,33 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrwtF; +import com.ruoyi.swlscx.year.domain.vo.HyYrwtFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** +* @author 12974 +* @description 针对表【hy_yrwt_f(年水温表)】的数据库操作Service +* @createDate 2024-09-05 16:53:36 +*/ +public interface HyYrwtFService extends IService { + /** 导入年水温表数据*/ + void importHyYrpFData(MultipartFile file); + + /** + * 查询年水温表数据 + */ + IPage getYearWaterTemperatureList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm); + + /** + * 导出年水温表 + */ + R exportYearWaterTemperatureList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrzFService.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrzFService.java new file mode 100644 index 0000000..691d72f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/HyYrzFService.java @@ -0,0 +1,35 @@ +package com.ruoyi.swlscx.year.service; + +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.IService; +import com.ruoyi.common.core.page.R; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.po.HyYrzF; +import com.ruoyi.swlscx.year.domain.vo.HyYrzFVo; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletResponse; + +/** + * @author 12974 + * @description 针对表【hy_yrz_f(年水位表)】的数据库操作Service + * @createDate 2024-09-05 16:53:36 + */ +public interface HyYrzFService extends IService { + /** + * 导入年水位表数据 + */ + void importHyYrpFData(MultipartFile file); + + /** + * 查询年水位表 + **/ + IPage selectHyYrzFPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd); + + /** + * 导出年水位表数据 + **/ + R exportYearWaterLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm); + + R importDateTosoft(String startTime, String endTime); +} diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyDmxpFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyDmxpFServiceImpl.java new file mode 100644 index 0000000..8337267 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyDmxpFServiceImpl.java @@ -0,0 +1,344 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyDmxpFBo; +import com.ruoyi.swlscx.year.domain.po.HyDmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyDmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.mapper.HyDmxpFMapper; +import com.ruoyi.swlscx.year.service.HyDmxpFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_dmxp_f(日时段最大降水量)】的数据库操作Service实现 + * @createDate 2024-09-11 10:09:13 + */ +@Service +public class HyDmxpFServiceImpl extends ServiceImpl + implements HyDmxpFService { + + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入日时段最大降水量数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyDmxpFData(MultipartFile file) { + try { + List hyDmxpFBos = ExcelUtils.readExcel(file.getInputStream(), HyDmxpFBo.class, 2); + List hyDmxpFList = hyDmxpFBos.stream().map(i -> { + HyDmxpF hyDmxpF = new HyDmxpF(); + BeanUtils.copyProperties(i, hyDmxpF); + hyDmxpF.setYr(Convert.toInt(i.getYr())); + hyDmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyDmxpF.setBgdt(DateUtils.parseDate(i.getBgdt())); + hyDmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyDmxpF; + }).collect(Collectors.toList()); + + this.saveBatch(hyDmxpFList, 5000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + @Override + public IPage selectMaxDRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selecthyDmxpFPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportDayDmxpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "日时时段最大降水量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyDmxpFVoList) { + List records = hyDmxpFVoList.stream().map(i->{ + HyDmxpF hyDmxpF = new HyDmxpF(); + BeanUtils.copyProperties(i, hyDmxpF); + hyDmxpF.setYr(Convert.toInt(i.getYr())); + hyDmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyDmxpF.setBgdt(DateUtils.parseDate(i.getBgdt())); + hyDmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyDmxpF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "日时时段最大降水量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "日时时段最大降水量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectMaxDRainListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + + HyDmxpFServiceImpl hyDmxpFService = context.getBean(HyDmxpFServiceImpl.class); + hyDmxpFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("dayMaxRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDmxpFVoIPage = this.baseMapper.selecthyDmxpFPageByInfo(new Query().getPage(map), map); + List records = hyDmxpFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyDmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? CommonUtils.formatNum(vo.getMxp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyDmxpFVoIPage = this.baseMapper.selecthyDmxpFPageByInfo(new Query().getPage(map), map); + List records = hyDmxpFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyDmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("日时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? CommonUtils.formatNum(vo.getMxp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyHmxpFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyHmxpFServiceImpl.java new file mode 100644 index 0000000..7b5e4a1 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyHmxpFServiceImpl.java @@ -0,0 +1,356 @@ +package com.ruoyi.swlscx.year.service.impl; + + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.basic.service.impl.HyStscAServiceImpl; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.year.domain.bo.HyHmxpFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.mapper.HyHmxpFMapper; +import com.ruoyi.swlscx.year.service.HyHmxpFService; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_hmxp_f】的数据库操作Service实现 + * @createDate 2024-07-23 14:15:25 + */ +@Service +@Slf4j +@RequiredArgsConstructor +public class HyHmxpFServiceImpl extends ServiceImpl + implements HyHmxpFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + @Override + @Transactional(rollbackFor = Exception.class) + public void importHourMaxRainData(MultipartFile file) { + try { + List hyHmxpFExcelBos = ExcelUtils.readExcel(file.getInputStream(), HyHmxpFBo.class, 2); + // 获取数据 + List dataList = hyHmxpFExcelBos.parallelStream().map(i -> { + HyHmxpF hyHmxpF = new HyHmxpF(); + BeanUtils.copyProperties(i, hyHmxpF); + hyHmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyHmxpF.setYr(Convert.toInt(i.getYr())); + hyHmxpF.setBgtm(DateUtils.parseDate(i.getBgtm())); + hyHmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyHmxpF; + }).collect(Collectors.toList()); + for (int i = 0; i < dataList.size(); i += 1000) { + int end = Math.min(i + 1000, dataList.size()); + List batchList = dataList.subList(i, end); + baseMapper.batchInsert(batchList); + } + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + @Override + public IPage selectMaxHRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selecthyMmxpFPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportHyHmxpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "小时时段最大降水量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyHmxpFVoList) { + List records = hyHmxpFVoList.stream().map(i->{ + HyHmxpF hyHmxpF = new HyHmxpF(); + BeanUtils.copyProperties(i, hyHmxpF); + hyHmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyHmxpF.setYr(Convert.toInt(i.getYr())); + hyHmxpF.setBgtm(DateUtils.parseDate(i.getBgtm())); + hyHmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyHmxpF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "小时时段最大降水量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "小时时段最大降水量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyHmxpFServiceImpl hyHmxpFService = context.getBean(HyHmxpFServiceImpl.class); + hyHmxpFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("hourMaxRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + private void exportData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelExport(response, "hourMaxRain.xls", "小时时段最大降水量表", (Sheet sheet, CellStyle style) -> { + dataModel(rainMap, sheet, style); + }); + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyHmxpFVoIPage = this.baseMapper.selecthyMmxpFPageByInfo(new Query().getPage(map), map); + List records = hyHmxpFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyHmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgtm() != null ? vo.getBgtm() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? vo.getMxp().toString() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + rowIndex++; + } + } + } + + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyHmxpFVoIPage = this.baseMapper.selecthyMmxpFPageByInfo( + new Query().getPage(map), map); + List records = hyHmxpFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyHmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgtm() != null ? vo.getBgtm() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? vo.getMxp().toString() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + +} + + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyImxfwFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyImxfwFServiceImpl.java new file mode 100644 index 0000000..b8a4580 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyImxfwFServiceImpl.java @@ -0,0 +1,348 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyImxfwFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.po.HyImxfwF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyImxfwFVo; +import com.ruoyi.swlscx.year.service.HyImxfwFService; +import com.ruoyi.swlscx.year.mapper.HyImxfwFMapper; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_imxfw_f(时段最大洪量表)】的数据库操作Service实现 +* @createDate 2024-09-09 17:32:19 +*/ +@Service +public class HyImxfwFServiceImpl extends ServiceImpl + implements HyImxfwFService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入月平均泥沙颗粒级配表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyImxfwFData(MultipartFile file) { + try { + List hyImxfwFBos = ExcelUtils.readExcel(file.getInputStream(), HyImxfwFBo.class, 2); + // 获取数据 + List dataList = hyImxfwFBos.parallelStream().map(i -> { + HyImxfwF hyImxfwF = new HyImxfwF(); + BeanUtils.copyProperties(i, hyImxfwF); + hyImxfwF.setYr(Convert.toInt(i.getYr())); + hyImxfwF.setBgdt(DateUtils.parseDate(i.getBgdt().split("\\.")[0])); + hyImxfwF.setMxw(Convert.toBigDecimal(i.getMxw())); + return hyImxfwF; + }).collect(Collectors.toList()); + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询时段最大洪量表数据 + */ + @Override + public IPage selectMaxFloodScaleListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyImxfwFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出时段最大洪量表 + */ + @Override + public R exportMaxFloodScaleList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "时段最大洪量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyImxfwFVoList) { + List records = hyImxfwFVoList.stream().map(i->{ + HyImxfwF hyImxfwF = new HyImxfwF(); + BeanUtils.copyProperties(i, hyImxfwF); + hyImxfwF.setYr(Convert.toInt(i.getYr())); + hyImxfwF.setBgdt(DateUtils.parseDate(i.getBgdt().split("\\.")[0])); + hyImxfwF.setMxw(Convert.toBigDecimal(i.getMxw())); + return hyImxfwF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "时段最大洪量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "时段最大洪量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectMaxFloodScaleListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyImxfwFServiceImpl hyImxfwFService = context.getBean(HyImxfwFServiceImpl.class); + hyImxfwFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("maxFloodScale.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyImxfwFVoIPage = this.baseMapper.selectHyImxfwFPageByInfo(new Query().getPage(map), map); + List records = hyImxfwFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyImxfwFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("时段最大洪量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxwdr() != null ? vo.getMxwdr() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxw() != null ? CommonUtils.formatNum(vo.getMxw()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwrc() != null ? vo.getMxwrc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyImxfwFVoIPage = this.baseMapper.selectHyImxfwFPageByInfo(new Query().getPage(map), map); + List records = hyImxfwFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyImxfwFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxwdr() != null ? vo.getMxwdr() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxw() != null ? CommonUtils.formatNum(vo.getMxw()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwrc() != null ? vo.getMxwrc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyMmxpFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyMmxpFServiceImpl.java new file mode 100644 index 0000000..b6dff3f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyMmxpFServiceImpl.java @@ -0,0 +1,349 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyMmxpFBo; +import com.ruoyi.swlscx.year.domain.po.HyMmxpF; +import com.ruoyi.swlscx.year.domain.vo.HyMmxpFVo; +import com.ruoyi.swlscx.year.service.HyMmxpFService; +import com.ruoyi.swlscx.year.mapper.HyMmxpFMapper; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_mmxp_f(分钟时段最大降水量表)】的数据库操作Service实现 +* @createDate 2024-09-11 10:09:13 +*/ +@Service +public class HyMmxpFServiceImpl extends ServiceImpl + implements HyMmxpFService{ + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 导入分钟时段最大降水量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMmxpFData(MultipartFile file) { + try { + List hyMmxpFBos = ExcelUtils.readExcel(file.getInputStream(), HyMmxpFBo.class, 2); + // 获取数据 + List dataList = hyMmxpFBos.parallelStream().map(i -> { + HyMmxpF hyMmxpF = new HyMmxpF(); + BeanUtils.copyProperties(i, hyMmxpF); + hyMmxpF.setYr(Convert.toInt(i.getYr())); + hyMmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyMmxpF.setBgdt(DateUtils.parseDate(i.getBgdt())); + hyMmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyMmxpF; + }).collect(Collectors.toList()); + for (int i = 0; i < dataList.size(); i += 2000) { + int end = Math.min(i + 1000, dataList.size()); + List batchList = dataList.subList(i, end); + baseMapper.batchInsert(batchList); + } + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询分钟时段最大降水量表数据 + */ + @Override + public IPage selectMaxMRainListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMmxpFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出分钟时段最大降水量表 + */ + @Override + public R exportMinMaxRainList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "分钟时段最大降水量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyMmxpFVoList) { + List records = hyMmxpFVoList.stream().map(i->{ + HyMmxpF hyMmxpF = new HyMmxpF(); + BeanUtils.copyProperties(i, hyMmxpF); + hyMmxpF.setYr(Convert.toInt(i.getYr())); + hyMmxpF.setMxpdr(Convert.toInt(i.getMxpdr())); + hyMmxpF.setBgdt(DateUtils.parseDate(i.getBgdt())); + hyMmxpF.setMxp(Convert.toBigDecimal(i.getMxp())); + return hyMmxpF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "分钟时段最大降水量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "分钟时段最大降水量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectMaxMRainListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + HyMmxpFServiceImpl hyMmxpFService = context.getBean(HyMmxpFServiceImpl.class); + hyMmxpFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error("异常:"+e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("minMaxRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtqEPageByInfo = this.baseMapper.selectHyMmxpFPageByInfo(new Query().getPage(map), map); + List records = hyMtqEPageByInfo.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyMmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("分钟时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? CommonUtils.formatNum(vo.getMxp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + rowIndex++; + } + } + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyMtqEPageByInfo = this.baseMapper. + selectHyMmxpFPageByInfo(new Query().getPage(map), map); + List records = hyMtqEPageByInfo.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyMmxpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("分钟时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getBgdt() != null ? vo.getBgdt() : "", style); + ExcelUtils.formatCell(row, 7, vo.getMxpdr() != null ? vo.getMxpdr().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxp() != null ? CommonUtils.formatNum(vo.getMxp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxprc() != null ? vo.getMxprc() : "", style); + ExcelUtils.formatCell(row, 10, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrcsFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrcsFServiceImpl.java new file mode 100644 index 0000000..c8fb20f --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrcsFServiceImpl.java @@ -0,0 +1,366 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrcsFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.po.HyYrcsF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrcsFVo; +import com.ruoyi.swlscx.year.mapper.HyYrcsFMapper; +import com.ruoyi.swlscx.year.service.HyYrcsFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_yrcs_f(年含沙量表)】的数据库操作Service实现 + * @createDate 2024-09-05 16:53:36 + */ +@Service +public class HyYrcsFServiceImpl extends ServiceImpl + implements HyYrcsFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入年含沙量表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrcsFData(MultipartFile file) { + try { + List hyYrcsFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrcsFBo.class, 2); + // 获取数据 + List dataList = hyYrcsFBos.parallelStream().map(i -> { + HyYrcsF hyYrcsF = new HyYrcsF(); + BeanUtils.copyProperties(i, hyYrcsF); + hyYrcsF.setYr(Convert.toInt(i.getYr())); + hyYrcsF.setAvcs(Convert.toBigDecimal(i.getAvcs())); + hyYrcsF.setMxs(Convert.toBigDecimal(i.getMxs())); + hyYrcsF.setMxsdt(DateUtils.parseDate(i.getMxsdt())); + hyYrcsF.setMns(Convert.toBigDecimal(i.getMns())); + hyYrcsF.setMnsdt(DateUtils.parseDate(i.getMnsdt())); + return hyYrcsF; + }).collect(Collectors.toList()); + + + this.saveBatch(dataList, 5000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询年含沙量表数据 + */ + @Override + public IPage getYearSedimentConcentrationList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selecthyYrcsFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出年含沙量表 + */ + @Override + public R exportSedimentConcentrationList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年含沙量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "sedimentConcentration.xls", "年含沙量表", (Sheet sheet, CellStyle style) -> { +// dataModel(rainMap, sheet, style); +// }); + } + + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrcsFVoList) { + List records = hyYrcsFVoList.stream().map(i->{ + HyYrcsF hyYrcsF = new HyYrcsF(); + BeanUtils.copyProperties(i, hyYrcsF); + hyYrcsF.setYr(Convert.toInt(i.getYr())); + hyYrcsF.setAvcs(Convert.toBigDecimal(i.getAvcs())); + hyYrcsF.setMxs(Convert.toBigDecimal(i.getMxs())); + hyYrcsF.setMxsdt(DateUtils.parseDate(i.getMxsdt())); + hyYrcsF.setMns(Convert.toBigDecimal(i.getMns())); + hyYrcsF.setMnsdt(DateUtils.parseDate(i.getMnsdt())); + return hyYrcsF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年含沙量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年含沙量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.getYearSedimentConcentrationList(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyYrcsFServiceImpl hyYrcsFService = context.getBean(HyYrcsFServiceImpl.class); + hyYrcsFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("sedimentConcentration.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrcsFVoIPage = this.baseMapper.selecthyYrcsFPageByInfo(new Query().getPage(map), map); + List records = hyYrcsFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrcsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年含沙量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvcsrcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxs() != null ? CommonUtils.formatNum(vo.getMxs()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxsrcd() != null ? vo.getMxsrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxsdt() != null ? vo.getMxsdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMns() != null ? CommonUtils.formatNum(vo.getMns()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnsrcd() != null ? vo.getMnsrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnsdt() != null ? vo.getMnsdt() : "", style); + ExcelUtils.formatCell(row, 14, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrcsFVoIPage = this.baseMapper.selecthyYrcsFPageByInfo(new Query().getPage(map), map); + List records = hyYrcsFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrcsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年含沙量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvcs() != null ? CommonUtils.formatNum(vo.getAvcs()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvcsrcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxs() != null ? CommonUtils.formatNum(vo.getMxs()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxsrcd() != null ? vo.getMxsrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxsdt() != null ? vo.getMxsdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMns() != null ? CommonUtils.formatNum(vo.getMns()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnsrcd() != null ? vo.getMnsrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnsdt() != null ? vo.getMnsdt() : "", style); + ExcelUtils.formatCell(row, 14, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrpFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrpFServiceImpl.java new file mode 100644 index 0000000..b705d9c --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrpFServiceImpl.java @@ -0,0 +1,364 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import com.ruoyi.swlscx.month.domain.vo.HyMptEVo; +import com.ruoyi.swlscx.year.domain.bo.HyYrpFBo; +import com.ruoyi.swlscx.year.domain.po.HyYrpF; +import com.ruoyi.swlscx.year.domain.vo.HyYrpFVo; +import com.ruoyi.swlscx.year.mapper.HyYrpFMapper; +import com.ruoyi.swlscx.year.service.HyYrpFService; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_yrp_f(年降水量表)】的数据库操作Service实现 + * @createDate 2024-09-05 16:53:36 + */ +@Service +public class HyYrpFServiceImpl extends ServiceImpl + implements HyYrpFService { + + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 导入年降水表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrpFData(MultipartFile file) { + try { + List hyYrpFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrpFBo.class, 2); + List hyYrpFList = hyYrpFBos.stream().map(i -> { + HyYrpF hyYrpF = new HyYrpF(); + BeanUtils.copyProperties(i, hyYrpF); + hyYrpF.setYr(Convert.toInt(i.getYr())); + hyYrpF.setP(Convert.toBigDecimal(i.getP())); + hyYrpF.setFrdsdt(DateUtils.parseDate(i.getFrdsdt())); + hyYrpF.setFrapdt(DateUtils.parseDate(i.getFrapdt())); + hyYrpF.setSndsdt(DateUtils.parseDate(i.getSndsdt())); + hyYrpF.setSnapdt(DateUtils.parseDate(i.getSnapdt())); + return hyYrpF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteAll(); + //先插入到临时表中 从临时表中更新过去 + for (int i = 0; i < hyYrpFList.size(); i += 1000) { + int end = Math.min(i + 1000, hyYrpFList.size()); + List batchList = hyYrpFList.subList(i, end); + this.baseMapper.batchInsert(batchList); + } + //重临时表中把数据刷过去 + this.baseMapper.batchInsertOrUpdate(); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询年降水表数据 + */ + @Override + public IPage selectHyYrpFDataByPageAndInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtpDataByPageAndInfo(new Query().getPage(map), map); + } + + @Override + public R exportHyYrpFData(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + Long userId = SecurityUtils.getUserId(); + String tableName = "年降水表"; + return taskAndExport(tableName, startTime, endTime, stcd, stnm, userId); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年降水表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年降水表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage hyYrpFVoIPage = this.selectHyYrpFDataByPageAndInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + HyYrpFServiceImpl hyYrpFService = context.getBean(HyYrpFServiceImpl.class); + hyYrpFService.importMysqlToSqlserver(hyYrpFVoIPage.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrpFVoList) { + List records = hyYrpFVoList.stream().map(i->{ + HyYrpF hyYrpF = new HyYrpF(); + BeanUtils.copyProperties(i, hyYrpF); + hyYrpF.setYr(Convert.toInt(i.getYr())); + hyYrpF.setP(Convert.toBigDecimal(i.getP())); + hyYrpF.setFrdsdt(DateUtils.parseDate(i.getFrdsdt())); + hyYrpF.setFrapdt(DateUtils.parseDate(i.getFrapdt())); + hyYrpF.setSndsdt(DateUtils.parseDate(i.getSndsdt())); + hyYrpF.setSnapdt(DateUtils.parseDate(i.getSnapdt())); + return hyYrpF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.batchInsert(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + private R taskAndExport(String tableName, String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime = ""; + String endNewTime = ""; + if (ObjectUtils.isNotEmpty(startTime)) { + startNewTime = startTime.substring(0, 4); + } + if (ObjectUtils.isNotEmpty(endTime)) { + endNewTime = endTime.substring(0, 4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator + + DateUtils.parseDateToStr("yyyy-MM-dd", DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm", DateUtils.getNowDate()) + File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearRain.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + + public void rainModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrpFVoIPage = this.baseMapper.selectHyMtpDataByPageAndInfo(new Query().getPage(map), map); + List records = hyYrpFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("月降水表_" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getP() != null ? vo.getP().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getPrcd() != null ? vo.getPrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getPdynum() != null ? vo.getPdynum() : "", style); + ExcelUtils.formatCell(row, 9, vo.getPdynumrcd() != null ? vo.getPdynumrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getFrdsdt() != null ? vo.getFrdsdt().toString() : "", style); + ExcelUtils.formatCell(row, 11, vo.getFrapdt() != null ? vo.getFrapdt().toString() : "", style); + ExcelUtils.formatCell(row, 12, vo.getSndsdt() != null ? vo.getSndsdt().toString() : "", style); + ExcelUtils.formatCell(row, 13, vo.getSnapdt() != null ? vo.getSnapdt().toString() : "", style); + rowIndex++; + } + } + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrpFVoIPage = this.baseMapper.selectHyMtpDataByPageAndInfo(new Query().getPage(map), map); + List records = hyYrpFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrpFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getP() != null ? vo.getP().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getPrcd() != null ? vo.getPrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getPdynum() != null ? vo.getPdynum() : "", style); + ExcelUtils.formatCell(row, 9, vo.getPdynumrcd() != null ? vo.getPdynumrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getFrdsdt() != null ? vo.getFrdsdt().toString() : "", style); + ExcelUtils.formatCell(row, 11, vo.getFrapdt() != null ? vo.getFrapdt().toString() : "", style); + ExcelUtils.formatCell(row, 12, vo.getSndsdt() != null ? vo.getSndsdt().toString() : "", style); + ExcelUtils.formatCell(row, 13, vo.getSnapdt() != null ? vo.getSnapdt().toString() : "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqFServiceImpl.java new file mode 100644 index 0000000..1f17039 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqFServiceImpl.java @@ -0,0 +1,366 @@ +package com.ruoyi.swlscx.year.service.impl; + + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrqFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.po.HyYrqF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrqFVo; +import com.ruoyi.swlscx.year.mapper.HyYrqFMapper; +import com.ruoyi.swlscx.year.service.HyYrqFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_yrq_f(年流量表)】的数据库操作Service实现 +* @createDate 2024-09-05 16:53:36 +*/ +@Service +public class HyYrqFServiceImpl extends ServiceImpl + implements HyYrqFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 年流量表 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrpFData(MultipartFile file) { + try { + List hyYrqFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrqFBo.class, 2); + List hyYrpFList = hyYrqFBos.stream().map(i -> { + HyYrqF hyYrqF = new HyYrqF(); + BeanUtils.copyProperties(i, hyYrqF); + hyYrqF.setYr(Convert.toInt(i.getYr())); + hyYrqF.setAvq(Convert.toBigDecimal(i.getAvq())); + hyYrqF.setMxq(Convert.toBigDecimal(i.getMxq())); + hyYrqF.setMxqdt(DateUtils.parseDate(i.getMxqdt())); + hyYrqF.setMnq(Convert.toBigDecimal(i.getMnq())); + hyYrqF.setMnqdt(DateUtils.parseDate(i.getMnqdt())); + hyYrqF.setRw(Convert.toBigDecimal(i.getRw())); + hyYrqF.setRm(Convert.toBigDecimal(i.getRm())); + hyYrqF.setRd(Convert.toBigDecimal(i.getRd())); + return hyYrqF; + }).collect(Collectors.toList()); + this.saveBatch(hyYrpFList,10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + @Override + public IPage selecHyYrqFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportYearFlowList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年流量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrqFVoList) { + List records = hyYrqFVoList.stream().map(i->{ + HyYrqF hyYrqF = new HyYrqF(); + BeanUtils.copyProperties(i, hyYrqF); + hyYrqF.setYr(Convert.toInt(i.getYr())); + hyYrqF.setAvq(Convert.toBigDecimal(i.getAvq())); + hyYrqF.setMxq(Convert.toBigDecimal(i.getMxq())); + hyYrqF.setMxqdt(DateUtils.parseDate(i.getMxqdt())); + hyYrqF.setMnq(Convert.toBigDecimal(i.getMnq())); + hyYrqF.setMnqdt(DateUtils.parseDate(i.getMnqdt())); + hyYrqF.setRw(Convert.toBigDecimal(i.getRw())); + hyYrqF.setRm(Convert.toBigDecimal(i.getRm())); + hyYrqF.setRd(Convert.toBigDecimal(i.getRd())); + return hyYrqF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年水位表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年水位表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selecHyYrqFListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyYrqFServiceImpl hyYrqFService = context.getBean(HyYrqFServiceImpl.class); + hyYrqFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearFlow.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrqFVoIPage = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyYrqFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrqFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年流量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvq() != null ? CommonUtils.formatNum(vo.getAvq()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxq() != null ? CommonUtils.formatNum(vo.getMxq()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxqrcd() != null ? vo.getMxqrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxqdt() != null ? vo.getMxqdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnq() != null ? CommonUtils.formatNum(vo.getMnq()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnqrcd() != null ? vo.getMnqrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnqdt() != null ? vo.getMnqdt() : "", style); + ExcelUtils.formatCell(row, 14, vo.getRw() != null ? CommonUtils.formatNum(vo.getRw()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getRwrcd() != null ? vo.getRwrcd() : "", style); + ExcelUtils.formatCell(row, 16, vo.getRm() != null ? CommonUtils.formatNum(vo.getRm()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getRd() != null ? CommonUtils.formatNum(vo.getRd()) : "", style); + ExcelUtils.formatCell(row, 18, "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrqFVoIPage = this.baseMapper.selectHyMtqEPageByInfo(new Query().getPage(map), map); + List records = hyYrqFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrqFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvq() != null ? CommonUtils.formatNum(vo.getAvq()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvqrcd() != null ? vo.getAvqrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxq() != null ? CommonUtils.formatNum(vo.getMxq()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxqrcd() != null ? vo.getMxqrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxqdt() != null ? vo.getMxqdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnq() != null ? CommonUtils.formatNum(vo.getMnq()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnqrcd() != null ? vo.getMnqrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnqdt() != null ? vo.getMnqdt() : "", style); + ExcelUtils.formatCell(row, 14, vo.getRw() != null ? CommonUtils.formatNum(vo.getRw()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getRwrcd() != null ? vo.getRwrcd() : "", style); + ExcelUtils.formatCell(row, 16, vo.getRm() != null ? CommonUtils.formatNum(vo.getRm()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getRd() != null ? CommonUtils.formatNum(vo.getRd()) : "", style); + ExcelUtils.formatCell(row, 18, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqsFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqsFServiceImpl.java new file mode 100644 index 0000000..f8cbde7 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrqsFServiceImpl.java @@ -0,0 +1,371 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrqsFBo; +import com.ruoyi.swlscx.year.domain.po.HyYrqsF; +import com.ruoyi.swlscx.year.domain.vo.HyYrqsFVo; +import com.ruoyi.swlscx.year.mapper.HyYrqsFMapper; +import com.ruoyi.swlscx.year.service.HyYrqsFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_yrqs_f(年输沙率表)】的数据库操作Service实现 +* @createDate 2024-09-05 16:53:36 +*/ +@Service +public class HyYrqsFServiceImpl extends ServiceImpl + implements HyYrqsFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrqsFData(MultipartFile file) { + try { + List hyYrqsFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrqsFBo.class, 2); + List hyYrpFList = hyYrqsFBos.stream().map(i -> { + HyYrqsF hyYrqsF = new HyYrqsF(); + BeanUtils.copyProperties(i, hyYrqsF); + hyYrqsF.setYr(Convert.toInt(i.getYr())); + hyYrqsF.setAvqs(Convert.toBigDecimal(i.getAvqs())); + hyYrqsF.setMxdyqs(Convert.toBigDecimal(i.getMxdyqs())); + hyYrqsF.setMxdyqsodt(DateUtils.parseDate(i.getMxdyqsodt())); + hyYrqsF.setSw(Convert.toBigDecimal(i.getSw())); + hyYrqsF.setSm(Convert.toBigDecimal(i.getSm())); + hyYrqsF.setSmec(Convert.toBigDecimal(i.getSmec())); + return hyYrqsF; + }).collect(Collectors.toList()); + this.saveBatch(hyYrpFList, 10000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询年输沙率表数据 + */ + @Override + public IPage getYearsedimentTransportRateList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyYrqFPageByInfo(new Query().getPage(map), map); + + } + + /** + * 导出年输沙率表 + */ + @Override + public R exportSedimentTransportRateList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年输沙率表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + +// +// Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "sedimentTransportRate.xls", "年输沙率表", (Sheet sheet, CellStyle style) -> { +// dataModel(rainMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrqsFVoList) { + List records = hyYrqsFVoList.stream().map(i->{ + HyYrqsF hyYrqsF = new HyYrqsF(); + BeanUtils.copyProperties(i, hyYrqsF); + hyYrqsF.setYr(Convert.toInt(i.getYr())); + hyYrqsF.setAvqs(Convert.toBigDecimal(i.getAvqs())); + hyYrqsF.setMxdyqs(Convert.toBigDecimal(i.getMxdyqs())); + hyYrqsF.setMxdyqsodt(DateUtils.parseDate(i.getMxdyqsodt())); + hyYrqsF.setSw(Convert.toBigDecimal(i.getSw())); + hyYrqsF.setSm(Convert.toBigDecimal(i.getSm())); + hyYrqsF.setSmec(Convert.toBigDecimal(i.getSmec())); + return hyYrqsF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年输沙率表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年输沙率表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.getYearsedimentTransportRateList(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000L), null, null, startTime, endTime); + HyYrqsFServiceImpl hyYrqsFService = context.getBean(HyYrqsFServiceImpl.class); + hyYrqsFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("sedimentTransportRate.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrqsFVoIPage = this.baseMapper.selectHyYrqFPageByInfo(new Query().getPage(map), map); + List records = hyYrqsFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrqsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年输沙率表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getSdtp() != null ? vo.getSdtp() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvqs() != null ? CommonUtils.formatNum(vo.getAvqs()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqsrcd() != null ? vo.getAvqsrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxdyqs() != null ? CommonUtils.formatNum(vo.getMxdyqs()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyqsrcd() != null ? vo.getMxdyqsrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyqsodt() != null ? vo.getMxdyqsodt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getSw() != null ? CommonUtils.formatNum(vo.getSw()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getSwpc() != null ? vo.getSwpc() : "", style); + ExcelUtils.formatCell(row, 14, vo.getSm() != null ? CommonUtils.formatNum(vo.getSm()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getSimn() != null ? vo.getSimn() : "", style); + ExcelUtils.formatCell(row, 16, vo.getSmec() != null ? CommonUtils.formatNum(vo.getSmec()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getNt() != null ? vo.getNt() : "", style); + ExcelUtils.formatCell(row, 18, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrqsFVoIPage = this.baseMapper.selectHyYrqFPageByInfo(new Query().getPage(map), map); + List records = hyYrqsFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrqsFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("小时时段最大降水量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getSdtp() != null ? vo.getSdtp() : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvqs() != null ? CommonUtils.formatNum(vo.getAvqs()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getAvqsrcd() != null ? vo.getAvqsrcd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxdyqs() != null ? CommonUtils.formatNum(vo.getMxdyqs()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyqsrcd() != null ? vo.getMxdyqsrcd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyqsodt() != null ? vo.getMxdyqsodt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getSw() != null ? CommonUtils.formatNum(vo.getSw()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getSwpc() != null ? vo.getSwpc() : "", style); + ExcelUtils.formatCell(row, 14, vo.getSm() != null ? CommonUtils.formatNum(vo.getSm()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getSimn() != null ? vo.getSimn() : "", style); + ExcelUtils.formatCell(row, 16, vo.getSmec() != null ? CommonUtils.formatNum(vo.getSmec()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getNt() != null ? vo.getNt() : "", style); + ExcelUtils.formatCell(row, 18, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrtdzFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrtdzFServiceImpl.java new file mode 100644 index 0000000..80ed2f0 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrtdzFServiceImpl.java @@ -0,0 +1,481 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.bean.BeanUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrtdzFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.po.HyYrtdzF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrtdzFVo; +import com.ruoyi.swlscx.year.mapper.HyYrtdzFMapper; +import com.ruoyi.swlscx.year.service.HyYrtdzFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_yrtdz_f(年潮位表)】的数据库操作Service实现 +* @createDate 2024-09-05 16:53:36 +*/ +@Service +public class HyYrtdzFServiceImpl extends ServiceImpl + implements HyYrtdzFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + + + /** + * 导入年潮位表表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyMmxpFData(MultipartFile file) { + try { + List hyYrtdzFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrtdzFBo.class, 2); + // 获取数据 + List dataList = hyYrtdzFBos.parallelStream().map(i -> { + HyYrtdzF hyYrtdzF = new HyYrtdzF(); + BeanUtils.copyProperties(i, hyYrtdzF); + hyYrtdzF.setYr(Convert.toInt(i.getYr())); + hyYrtdzF.setAvhtdz(Convert.toBigDecimal(i.getAvhtdz())); + hyYrtdzF.setHthtdz(Convert.toBigDecimal(i.getHthtdz())); + hyYrtdzF.setHthtdzotm(DateUtils.parseDate(i.getHthtdzotm().split("\\.")[0])); + hyYrtdzF.setLthtdz(Convert.toBigDecimal(i.getLthtdz())); + hyYrtdzF.setLthtdzotm(DateUtils.parseDate(i.getLthtdzotm().split("\\.")[0])); + hyYrtdzF.setAvltdz(Convert.toBigDecimal(i.getAvltdz())); + hyYrtdzF.setHtltdz(Convert.toBigDecimal(i.getHtltdz())); + hyYrtdzF.setHtltdzotm(DateUtils.parseDate(i.getHtltdzotm().split("\\.")[0])); + hyYrtdzF.setLtltdz(Convert.toBigDecimal(i.getLtltdz())); + hyYrtdzF.setLtltdzotm(DateUtils.parseDate(i.getLtltdzotm().split("\\.")[0])); + hyYrtdzF.setAvftdr(Convert.toBigDecimal(i.getAvftdr())); + hyYrtdzF.setMxfltdr(Convert.toBigDecimal(i.getMxfltdr())); + hyYrtdzF.setMxfltdrhtm(DateUtils.parseDate(i.getMxfltdrhtm().split("\\.")[0])); + hyYrtdzF.setMnfltdr(Convert.toBigDecimal(i.getMnfltdr())); + hyYrtdzF.setMnfltdrhtm(DateUtils.parseDate(i.getMnfltdrhtm().split("\\.")[0])); + hyYrtdzF.setAver(Convert.toBigDecimal(i.getAver())); + hyYrtdzF.setMxebtdr(Convert.toBigDecimal(i.getMxebtdr())); + hyYrtdzF.setMxebtdrht(DateUtils.parseDate(i.getMxebtdrht().split("\\.")[0])); + hyYrtdzF.setMnebtdr(Convert.toBigDecimal(i.getMnebtdr())); + hyYrtdzF.setMnebtdrhtm(DateUtils.parseDate(i.getMnebtdrhtm().split("\\.")[0])); + hyYrtdzF.setAvftd(Convert.toInt(i.getAvftd())); + hyYrtdzF.setMxfltddr(Convert.toInt(i.getMxfltddr())); + hyYrtdzF.setMxfltddrhtm(DateUtils.parseDate(i.getMxfltddrhtm().split("\\.")[0])); + hyYrtdzF.setMnfltddr(Convert.toInt(i.getMnfltddr())); + hyYrtdzF.setMnfltddrhtm(DateUtils.parseDate(i.getMnfltddrhtm().split("\\.")[0])); + hyYrtdzF.setAvebbdr(Convert.toInt(i.getAvebbdr())); + hyYrtdzF.setMxebtddr(Convert.toInt(i.getMxebtddr())); + hyYrtdzF.setMxebtddrhtm(DateUtils.parseDate(i.getMxebtddrhtm().split("\\.")[0])); + hyYrtdzF.setMnbtddr(Convert.toInt(i.getMnbtddr())); + hyYrtdzF.setMnebtddrhtm(DateUtils.parseDate(i.getMnebtddrhtm().split("\\.")[0])); + hyYrtdzF.setHravtdz(Convert.toBigDecimal(i.getHravtdz())); + hyYrtdzF.setAvtdr(Convert.toBigDecimal(i.getAvtdr())); + hyYrtdzF.setAvtddr(Convert.toInt(i.getAvtddr())); + return hyYrtdzF; + }).collect(Collectors.toList()); + + this.saveBatch(dataList,10000); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + /** + * 查询年潮位表数据 + */ + @Override + public IPage selectHyYrtdzFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyYrtdzFListPageByInfo(new Query().getPage(map), map); + } + + @Override + public R exportYearTideLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年潮位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); +// +// Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "yearTideLever.xls", "年潮位表", (Sheet sheet, CellStyle style) -> { +// dataModel(rainMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrtdzFVoList) { + List records = hyYrtdzFVoList.stream().map(i->{ + HyYrtdzF hyYrtdzF = new HyYrtdzF(); + BeanUtils.copyProperties(i, hyYrtdzF); + hyYrtdzF.setYr(Convert.toInt(i.getYr())); + hyYrtdzF.setAvhtdz(Convert.toBigDecimal(i.getAvhtdz())); + hyYrtdzF.setHthtdz(Convert.toBigDecimal(i.getHthtdz())); + hyYrtdzF.setHthtdzotm(DateUtils.parseDate(i.getHthtdzotm().split("\\.")[0])); + hyYrtdzF.setLthtdz(Convert.toBigDecimal(i.getLthtdz())); + hyYrtdzF.setLthtdzotm(DateUtils.parseDate(i.getLthtdzotm().split("\\.")[0])); + hyYrtdzF.setAvltdz(Convert.toBigDecimal(i.getAvltdz())); + hyYrtdzF.setHtltdz(Convert.toBigDecimal(i.getHtltdz())); + hyYrtdzF.setHtltdzotm(DateUtils.parseDate(i.getHtltdzotm().split("\\.")[0])); + hyYrtdzF.setLtltdz(Convert.toBigDecimal(i.getLtltdz())); + hyYrtdzF.setLtltdzotm(DateUtils.parseDate(i.getLtltdzotm().split("\\.")[0])); + hyYrtdzF.setAvftdr(Convert.toBigDecimal(i.getAvftdr())); + hyYrtdzF.setMxfltdr(Convert.toBigDecimal(i.getMxfltdr())); + hyYrtdzF.setMxfltdrhtm(DateUtils.parseDate(i.getMxfltdrhtm().split("\\.")[0])); + hyYrtdzF.setMnfltdr(Convert.toBigDecimal(i.getMnfltdr())); + hyYrtdzF.setMnfltdrhtm(DateUtils.parseDate(i.getMnfltdrhtm().split("\\.")[0])); + hyYrtdzF.setAver(Convert.toBigDecimal(i.getAver())); + hyYrtdzF.setMxebtdr(Convert.toBigDecimal(i.getMxebtdr())); + hyYrtdzF.setMxebtdrht(DateUtils.parseDate(i.getMxebtdrht().split("\\.")[0])); + hyYrtdzF.setMnebtdr(Convert.toBigDecimal(i.getMnebtdr())); + hyYrtdzF.setMnebtdrhtm(DateUtils.parseDate(i.getMnebtdrhtm().split("\\.")[0])); + hyYrtdzF.setAvftd(Convert.toInt(i.getAvftd())); + hyYrtdzF.setMxfltddr(Convert.toInt(i.getMxfltddr())); + hyYrtdzF.setMxfltddrhtm(DateUtils.parseDate(i.getMxfltddrhtm().split("\\.")[0])); + hyYrtdzF.setMnfltddr(Convert.toInt(i.getMnfltddr())); + hyYrtdzF.setMnfltddrhtm(DateUtils.parseDate(i.getMnfltddrhtm().split("\\.")[0])); + hyYrtdzF.setAvebbdr(Convert.toInt(i.getAvebbdr())); + hyYrtdzF.setMxebtddr(Convert.toInt(i.getMxebtddr())); + hyYrtdzF.setMxebtddrhtm(DateUtils.parseDate(i.getMxebtddrhtm().split("\\.")[0])); + hyYrtdzF.setMnbtddr(Convert.toInt(i.getMnbtddr())); + hyYrtdzF.setMnebtddrhtm(DateUtils.parseDate(i.getMnebtddrhtm().split("\\.")[0])); + hyYrtdzF.setHravtdz(Convert.toBigDecimal(i.getHravtdz())); + hyYrtdzF.setAvtdr(Convert.toBigDecimal(i.getAvtdr())); + hyYrtdzF.setAvtddr(Convert.toInt(i.getAvtddr())); + return hyYrtdzF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + @Override + public R importDateTosoft(String startTime, String endTime) { + return null; + } + + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearTideLever.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage HyYrtdzFListPageByInfo = this.baseMapper.selectHyYrtdzFListPageByInfo(new Query().getPage(map), map); + List records = HyYrtdzFListPageByInfo.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrtdzFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年潮位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvhtdz() != null ? CommonUtils.formatNum(vo.getAvhtdz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvhtdzrcd() != null ? vo.getAvhtdzrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getHthtdz() != null ? CommonUtils.formatNum(vo.getHthtdz()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getHthtdzrcd() != null ? vo.getHthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getHthtdzotm() != null ? vo.getHthtdzotm() : "", style); + ExcelUtils.formatCell(row, 11, vo.getLthtdz() != null ? CommonUtils.formatNum(vo.getLthtdz()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getLthtdzrcd() != null ? vo.getLthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getLthtdzotm() != null ? vo.getLthtdzotm() : "", style); + ExcelUtils.formatCell(row, 14, vo.getAvltdz() != null ? CommonUtils.formatNum(vo.getAvltdz()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getAvltdzrcd() != null ? vo.getAvltdzrcd() : "", style); + ExcelUtils.formatCell(row, 16, vo.getHtltdz() != null ? CommonUtils.formatNum(vo.getHtltdz()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getHtltdzrcd() != null ? vo.getHtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 18, vo.getHtltdzotm() != null ? vo.getHtltdzotm() : "", style); + ExcelUtils.formatCell(row, 19, vo.getLtltdz() != null ? CommonUtils.formatNum(vo.getLtltdz()) : "", style); + ExcelUtils.formatCell(row, 20, vo.getLtltdzrcd() != null ? vo.getLtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 21, vo.getLtltdzotm() != null ? vo.getLtltdzotm() : "", style); + ExcelUtils.formatCell(row, 22, vo.getAvftdr() != null ? CommonUtils.formatNum(vo.getAvftdr()) : "", style); + ExcelUtils.formatCell(row, 23, vo.getAvftdrrcd() != null ? vo.getAvftdrrcd() : "", style); + ExcelUtils.formatCell(row, 24, vo.getMxfltdr() != null ? CommonUtils.formatNum(vo.getMxfltdr()) : "", style); + ExcelUtils.formatCell(row, 25, vo.getMxfltdrrcd() != null ? vo.getMxfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 26, vo.getMxfltdrhtm() != null ? vo.getMxfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 27, vo.getMnfltdr() != null ? CommonUtils.formatNum(vo.getMnfltdr()) : "", style); + ExcelUtils.formatCell(row, 28, vo.getMnfltdrrcd() != null ? vo.getMnfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 29, vo.getMnfltdrhtm() != null ? vo.getMnfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 30, vo.getAver() != null ? CommonUtils.formatNum(vo.getAver()) : "", style); + ExcelUtils.formatCell(row, 31, vo.getAverbbrrcd() != null ? vo.getAverbbrrcd() : "", style); + ExcelUtils.formatCell(row, 32, vo.getMxebtdr() != null ? CommonUtils.formatNum(vo.getMxebtdr()) : "", style); + ExcelUtils.formatCell(row, 33, vo.getMxebtdrrcd() != null ? vo.getMxebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 34, vo.getMxebtdrht() != null ? vo.getMxebtdrht() : "", style); + ExcelUtils.formatCell(row, 35, vo.getMnebtdr() != null ? CommonUtils.formatNum(vo.getMnebtdr()) : "", style); + ExcelUtils.formatCell(row, 36, vo.getMnebtdrrcd() != null ? vo.getMnebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 37, vo.getMnebtdrhtm() != null ? vo.getMnebtdrhtm() : "", style); + ExcelUtils.formatCell(row, 38, vo.getAvftd() != null ? vo.getAvftd().toString() : "", style); + ExcelUtils.formatCell(row, 39, vo.getAvftdrcd() != null ? vo.getAvftdrcd() : "", style); + ExcelUtils.formatCell(row, 40, vo.getMxfltddr() != null ? vo.getMxfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 41, vo.getMxfltddrrcd() != null ? vo.getMxfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 42, vo.getMxfltddrhtm() != null ? vo.getMxfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 43, vo.getMnfltddr() != null ? vo.getMnfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 44, vo.getMnfltddrrcd() != null ? vo.getMnfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 45, vo.getMnfltddrhtm() != null ? vo.getMnfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 46, vo.getAvebbdr() != null ? vo.getAvebbdr().toString() : "", style); + ExcelUtils.formatCell(row, 47, vo.getAvedrc() != null ? vo.getAvedrc() : "", style); + ExcelUtils.formatCell(row, 48, vo.getMxebtddr() != null ? vo.getMxebtddr().toString() : "", style); + ExcelUtils.formatCell(row, 49, vo.getMxebtddrrcd() != null ? vo.getMxebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 50, vo.getMxebtddrhtm() != null ? vo.getMxebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 51, vo.getMnbtddr() != null ? vo.getMnbtddr().toString() : "", style); + ExcelUtils.formatCell(row, 52, vo.getMnebtddrrcd() != null ? vo.getMnebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 53, vo.getMnebtddrhtm() != null ? vo.getMnebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 54, vo.getHravtdz() != null ? CommonUtils.formatNum(vo.getHravtdz()) : "", style); + ExcelUtils.formatCell(row, 55, vo.getHravtdzrcd() != null ? vo.getHravtdzrcd() : "", style); + ExcelUtils.formatCell(row, 56, vo.getAvtdr() != null ? CommonUtils.formatNum(vo.getAvtdr()) : "", style); + ExcelUtils.formatCell(row, 57, vo.getAvtdrrcd() != null ? vo.getAvtdrrcd() : "", style); + ExcelUtils.formatCell(row, 58, vo.getAvtddr() != null ? vo.getAvtddr().toString() : "", style); + ExcelUtils.formatCell(row, 59, vo.getAvtddrrcd() != null ? vo.getAvtddrrcd() : "", style); + ExcelUtils.formatCell(row, 60, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage HyYrtdzFListPageByInfo = this.baseMapper.selectHyYrtdzFListPageByInfo(new Query().getPage(map), map); + List records = HyYrtdzFListPageByInfo.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrtdzFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年潮位表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvhtdz() != null ? CommonUtils.formatNum(vo.getAvhtdz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvhtdzrcd() != null ? vo.getAvhtdzrcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getHthtdz() != null ? CommonUtils.formatNum(vo.getHthtdz()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getHthtdzrcd() != null ? vo.getHthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getHthtdzotm() != null ? vo.getHthtdzotm() : "", style); + ExcelUtils.formatCell(row, 11, vo.getLthtdz() != null ? CommonUtils.formatNum(vo.getLthtdz()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getLthtdzrcd() != null ? vo.getLthtdzrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getLthtdzotm() != null ? vo.getLthtdzotm() : "", style); + ExcelUtils.formatCell(row, 14, vo.getAvltdz() != null ? CommonUtils.formatNum(vo.getAvltdz()) : "", style); + ExcelUtils.formatCell(row, 15, vo.getAvltdzrcd() != null ? vo.getAvltdzrcd() : "", style); + ExcelUtils.formatCell(row, 16, vo.getHtltdz() != null ? CommonUtils.formatNum(vo.getHtltdz()) : "", style); + ExcelUtils.formatCell(row, 17, vo.getHtltdzrcd() != null ? vo.getHtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 18, vo.getHtltdzotm() != null ? vo.getHtltdzotm() : "", style); + ExcelUtils.formatCell(row, 19, vo.getLtltdz() != null ? CommonUtils.formatNum(vo.getLtltdz()) : "", style); + ExcelUtils.formatCell(row, 20, vo.getLtltdzrcd() != null ? vo.getLtltdzrcd() : "", style); + ExcelUtils.formatCell(row, 21, vo.getLtltdzotm() != null ? vo.getLtltdzotm() : "", style); + ExcelUtils.formatCell(row, 22, vo.getAvftdr() != null ? CommonUtils.formatNum(vo.getAvftdr()) : "", style); + ExcelUtils.formatCell(row, 23, vo.getAvftdrrcd() != null ? vo.getAvftdrrcd() : "", style); + ExcelUtils.formatCell(row, 24, vo.getMxfltdr() != null ? CommonUtils.formatNum(vo.getMxfltdr()) : "", style); + ExcelUtils.formatCell(row, 25, vo.getMxfltdrrcd() != null ? vo.getMxfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 26, vo.getMxfltdrhtm() != null ? vo.getMxfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 27, vo.getMnfltdr() != null ? CommonUtils.formatNum(vo.getMnfltdr()) : "", style); + ExcelUtils.formatCell(row, 28, vo.getMnfltdrrcd() != null ? vo.getMnfltdrrcd() : "", style); + ExcelUtils.formatCell(row, 29, vo.getMnfltdrhtm() != null ? vo.getMnfltdrhtm() : "", style); + ExcelUtils.formatCell(row, 30, vo.getAver() != null ? CommonUtils.formatNum(vo.getAver()) : "", style); + ExcelUtils.formatCell(row, 31, vo.getAverbbrrcd() != null ? vo.getAverbbrrcd() : "", style); + ExcelUtils.formatCell(row, 32, vo.getMxebtdr() != null ? CommonUtils.formatNum(vo.getMxebtdr()) : "", style); + ExcelUtils.formatCell(row, 33, vo.getMxebtdrrcd() != null ? vo.getMxebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 34, vo.getMxebtdrht() != null ? vo.getMxebtdrht() : "", style); + ExcelUtils.formatCell(row, 35, vo.getMnebtdr() != null ? CommonUtils.formatNum(vo.getMnebtdr()) : "", style); + ExcelUtils.formatCell(row, 36, vo.getMnebtdrrcd() != null ? vo.getMnebtdrrcd() : "", style); + ExcelUtils.formatCell(row, 37, vo.getMnebtdrhtm() != null ? vo.getMnebtdrhtm() : "", style); + ExcelUtils.formatCell(row, 38, vo.getAvftd() != null ? vo.getAvftd().toString() : "", style); + ExcelUtils.formatCell(row, 39, vo.getAvftdrcd() != null ? vo.getAvftdrcd() : "", style); + ExcelUtils.formatCell(row, 40, vo.getMxfltddr() != null ? vo.getMxfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 41, vo.getMxfltddrrcd() != null ? vo.getMxfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 42, vo.getMxfltddrhtm() != null ? vo.getMxfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 43, vo.getMnfltddr() != null ? vo.getMnfltddr().toString() : "", style); + ExcelUtils.formatCell(row, 44, vo.getMnfltddrrcd() != null ? vo.getMnfltddrrcd() : "", style); + ExcelUtils.formatCell(row, 45, vo.getMnfltddrhtm() != null ? vo.getMnfltddrhtm() : "", style); + ExcelUtils.formatCell(row, 46, vo.getAvebbdr() != null ? vo.getAvebbdr().toString() : "", style); + ExcelUtils.formatCell(row, 47, vo.getAvedrc() != null ? vo.getAvedrc() : "", style); + ExcelUtils.formatCell(row, 48, vo.getMxebtddr() != null ? vo.getMxebtddr().toString() : "", style); + ExcelUtils.formatCell(row, 49, vo.getMxebtddrrcd() != null ? vo.getMxebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 50, vo.getMxebtddrhtm() != null ? vo.getMxebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 51, vo.getMnbtddr() != null ? vo.getMnbtddr().toString() : "", style); + ExcelUtils.formatCell(row, 52, vo.getMnebtddrrcd() != null ? vo.getMnebtddrrcd() : "", style); + ExcelUtils.formatCell(row, 53, vo.getMnebtddrhtm() != null ? vo.getMnebtddrhtm() : "", style); + ExcelUtils.formatCell(row, 54, vo.getHravtdz() != null ? CommonUtils.formatNum(vo.getHravtdz()) : "", style); + ExcelUtils.formatCell(row, 55, vo.getHravtdzrcd() != null ? vo.getHravtdzrcd() : "", style); + ExcelUtils.formatCell(row, 56, vo.getAvtdr() != null ? CommonUtils.formatNum(vo.getAvtdr()) : "", style); + ExcelUtils.formatCell(row, 57, vo.getAvtdrrcd() != null ? vo.getAvtdrrcd() : "", style); + ExcelUtils.formatCell(row, 58, vo.getAvtddr() != null ? vo.getAvtddr().toString() : "", style); + ExcelUtils.formatCell(row, 59, vo.getAvtddrrcd() != null ? vo.getAvtddrrcd() : "", style); + ExcelUtils.formatCell(row, 60, "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrweFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrweFServiceImpl.java new file mode 100644 index 0000000..24e28c9 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrweFServiceImpl.java @@ -0,0 +1,375 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrweFBo; +import com.ruoyi.swlscx.year.domain.po.HyYrweF; +import com.ruoyi.swlscx.year.domain.vo.HyYrweFVo; +import com.ruoyi.swlscx.year.mapper.HyYrweFMapper; +import com.ruoyi.swlscx.year.service.HyYrweFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_yrwe_f(年水面蒸发量表)】的数据库操作Service实现 + * @createDate 2024-09-05 16:53:36 + */ +@Service +public class HyYrweFServiceImpl extends ServiceImpl + implements HyYrweFService { + + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入年水面蒸发量表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrweFData(MultipartFile file) { + try { + List hyYrweFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrweFBo.class, 2); + List hyYrweFList = hyYrweFBos.stream().map(i -> { + HyYrweF hyYrweF = new HyYrweF(); + BeanUtils.copyProperties(i, hyYrweF); + hyYrweF.setYr(Convert.toInt(i.getYr())); + hyYrweF.setWsfe(Convert.toBigDecimal(i.getWsfe())); + hyYrweF.setMxdye(Convert.toBigDecimal(i.getMxdye())); + hyYrweF.setMxdyeodt(DateUtils.parseDate(i.getMxdyeodt())); + hyYrweF.setMndye(Convert.toBigDecimal(i.getMndye())); + hyYrweF.setMndyeodt(DateUtils.parseDate(i.getMndyeodt())); + hyYrweF.setIdsdt(DateUtils.parseDate(i.getIdsdt())); + hyYrweF.setIcapd(DateUtils.parseDate(i.getIcapd())); + + return hyYrweF; + }).collect(Collectors.toList()); + this.saveBatch(hyYrweFList, 3000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询年水面蒸发量表数据 + */ + @Override + public IPage selectHyYrweFListPageByInfo(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selechyYrweFListPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出年水面蒸发量表 + */ + @Override + public R exportYearEvaporationWaterList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年水面蒸发量表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); +// +// Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 100000L), startTime, endTime, stcd, stnm); +// ExcelUtils.excelExport(response, "yearEvaporationWater.xls", "年水面蒸发量表", (Sheet sheet, CellStyle style) -> { +// dataModel(rainMap, sheet, style); +// }); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrweFVoList) { + List records = hyYrweFVoList.stream().map(i->{ + HyYrweF hyYrweF = new HyYrweF(); + BeanUtils.copyProperties(i, hyYrweF); + hyYrweF.setYr(Convert.toInt(i.getYr())); + hyYrweF.setWsfe(Convert.toBigDecimal(i.getWsfe())); + hyYrweF.setMxdye(Convert.toBigDecimal(i.getMxdye())); + hyYrweF.setMxdyeodt(DateUtils.parseDate(i.getMxdyeodt())); + hyYrweF.setMndye(Convert.toBigDecimal(i.getMndye())); + hyYrweF.setMndyeodt(DateUtils.parseDate(i.getMndyeodt())); + hyYrweF.setIdsdt(DateUtils.parseDate(i.getIdsdt())); + hyYrweF.setIcapd(DateUtils.parseDate(i.getIcapd())); + return hyYrweF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年水面蒸发量表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年水面蒸发量表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyYrweFListPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); + HyYrweFServiceImpl hyYrweFService = context.getBean(HyYrweFServiceImpl.class); + hyYrweFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearEvaporationWater.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrweFVoIPage = this.baseMapper.selechyYrweFListPageByInfo(new Query().getPage(map), map); + List records = hyYrweFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrweFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 6, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getWsfe() != null ? CommonUtils.formatNum(vo.getWsfe()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxdye() != null ? CommonUtils.formatNum(vo.getMxdye()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyercd() != null ? vo.getMxdyercd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyeodt() != null ? vo.getMxdyeodt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMndye() != null ? CommonUtils.formatNum(vo.getMndye()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMndyercd() != null ? vo.getMndyercd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMndyeodt() != null ? vo.getMndyeodt() : "", style); + ExcelUtils.formatCell(row, 15, vo.getIdsdt() != null ? vo.getIdsdt() : "", style); + ExcelUtils.formatCell(row, 16, vo.getIcapd() != null ? vo.getIcapd() : "", style); + ExcelUtils.formatCell(row, 17, vo.getEslcch() != null ? vo.getEslcch() : "", style); + ExcelUtils.formatCell(row, 18, vo.getNt() != null ? vo.getNt() : "", style); + ExcelUtils.formatCell(row, 19, "", style); + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrweFVoIPage = this.baseMapper.selechyYrweFListPageByInfo(new Query().getPage(map), map); + List records = hyYrweFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrweFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水面蒸发量表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getEetp() != null ? vo.getEetp() : "", style); + ExcelUtils.formatCell(row, 6, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 7, vo.getWsfe() != null ? CommonUtils.formatNum(vo.getWsfe()) : "", style); + ExcelUtils.formatCell(row, 8, vo.getWsfercd() != null ? vo.getWsfercd() : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxdye() != null ? CommonUtils.formatNum(vo.getMxdye()) : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxdyercd() != null ? vo.getMxdyercd() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMxdyeodt() != null ? vo.getMxdyeodt() : "", style); + ExcelUtils.formatCell(row, 12, vo.getMndye() != null ? CommonUtils.formatNum(vo.getMndye()) : "", style); + ExcelUtils.formatCell(row, 13, vo.getMndyercd() != null ? vo.getMndyercd() : "", style); + ExcelUtils.formatCell(row, 14, vo.getMndyeodt() != null ? vo.getMndyeodt() : "", style); + ExcelUtils.formatCell(row, 15, vo.getIdsdt() != null ? vo.getIdsdt() : "", style); + ExcelUtils.formatCell(row, 16, vo.getIcapd() != null ? vo.getIcapd() : "", style); + ExcelUtils.formatCell(row, 17, vo.getEslcch() != null ? vo.getEslcch() : "", style); + ExcelUtils.formatCell(row, 18, vo.getNt() != null ? vo.getNt() : "", style); + ExcelUtils.formatCell(row, 19, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrwtFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrwtFServiceImpl.java new file mode 100644 index 0000000..15dd1ac --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrwtFServiceImpl.java @@ -0,0 +1,356 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrwtFBo; +import com.ruoyi.swlscx.year.domain.po.HyHmxpF; +import com.ruoyi.swlscx.year.domain.po.HyYrwtF; +import com.ruoyi.swlscx.year.domain.vo.HyHmxpFVo; +import com.ruoyi.swlscx.year.domain.vo.HyYrwtFVo; +import com.ruoyi.swlscx.year.mapper.HyYrwtFMapper; +import com.ruoyi.swlscx.year.service.HyYrwtFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** +* @author 12974 +* @description 针对表【hy_yrwt_f(年水温表)】的数据库操作Service实现 +* @createDate 2024-09-05 16:53:36 +*/ +@Service +public class HyYrwtFServiceImpl extends ServiceImpl + implements HyYrwtFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入年水温表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrpFData(MultipartFile file) { + try { + List hyYrwtFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrwtFBo.class, 2); + List hyYrpFList = hyYrwtFBos.stream().map(i -> { + HyYrwtF hyYrwtF = new HyYrwtF(); + BeanUtils.copyProperties(i, hyYrwtF); + hyYrwtF.setYr(Convert.toInt(i.getYr())); + hyYrwtF.setAvwtmp(Convert.toBigDecimal(i.getAvwtmp())); + hyYrwtF.setMxwtmp(Convert.toBigDecimal(i.getMxwtmp())); + hyYrwtF.setMnwtmp(Convert.toBigDecimal(i.getMnwtmp())); + hyYrwtF.setMnwtmpdt(DateUtils.parseDate(i.getMnwtmpdt().split("\\.")[0])); + return hyYrwtF; + }).collect(Collectors.toList()); + this.saveBatch(hyYrpFList,3000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询年水温表数据 + */ + @Override + public IPage getYearWaterTemperatureList(PageParams pageParams, String startTime, String endTime, String stcd, String stnm) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyYrwtFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出年水温表 + */ + @Override + public R exportYearWaterTemperatureList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年水温表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrwtFVoList) { + List records = hyYrwtFVoList.stream().map(i->{ + HyYrwtF hyYrwtF = new HyYrwtF(); + BeanUtils.copyProperties(i, hyYrwtF); + hyYrwtF.setYr(Convert.toInt(i.getYr())); + hyYrwtF.setAvwtmp(Convert.toBigDecimal(i.getAvwtmp())); + hyYrwtF.setMxwtmp(Convert.toBigDecimal(i.getMxwtmp())); + hyYrwtF.setMnwtmp(Convert.toBigDecimal(i.getMnwtmp())); + hyYrwtF.setMnwtmpdt(DateUtils.parseDate(i.getMnwtmpdt())); + return hyYrwtF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年水温表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年水温表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.getYearWaterTemperatureList(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyYrwtFServiceImpl hyYrwtFService = context.getBean(HyYrwtFServiceImpl.class); + hyYrwtFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearWaterTemperature.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrwtFVoIPage = this.baseMapper.selectHyYrwtFPageByInfo(new Query().getPage(map), map); + List records = hyYrwtFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrwtFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水温表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvwtmp() != null ? CommonUtils.formatNum(vo.getAvwtmp()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvwtmprcd() != null ? vo.getAvwtmprcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxwtmp() != null ? CommonUtils.formatNum(vo.getMxwtmp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwtmprcd() != null ? vo.getMxwtmprcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxwtmpdt() != null ? vo.getMxwtmpdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnwtmp() != null ? CommonUtils.formatNum(vo.getMnwtmp()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnwtmprcd() != null ? vo.getMnwtmprcd() : "", style); + ExcelUtils.formatCell(row, 13, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrwtFVoIPage = this.baseMapper.selectHyYrwtFPageByInfo(new Query().getPage(map), map); + List records = hyYrwtFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrwtFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水温表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "",style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvwtmp() != null ? CommonUtils.formatNum(vo.getAvwtmp()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvwtmprcd() != null ? vo.getAvwtmprcd() : "", style); + ExcelUtils.formatCell(row, 8, vo.getMxwtmp() != null ? CommonUtils.formatNum(vo.getMxwtmp()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getMxwtmprcd() != null ? vo.getMxwtmprcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getMxwtmpdt() != null ? vo.getMxwtmpdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnwtmp() != null ? CommonUtils.formatNum(vo.getMnwtmp()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnwtmprcd() != null ? vo.getMnwtmprcd() : "", style); + ExcelUtils.formatCell(row, 13, "", style); + rowIndex++; + } + } + } + +} + + + + diff --git a/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrzFServiceImpl.java b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrzFServiceImpl.java new file mode 100644 index 0000000..ffdad89 --- /dev/null +++ b/swlscx/src/main/java/com/ruoyi/swlscx/year/service/impl/HyYrzFServiceImpl.java @@ -0,0 +1,357 @@ +package com.ruoyi.swlscx.year.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.annotation.DataSource; +import com.ruoyi.common.config.RuoYiConfig; +import com.ruoyi.common.core.page.R; +import com.ruoyi.common.core.text.Convert; +import com.ruoyi.common.enums.DataSourceType; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.swlscx.basic.domain.po.YcExportTask; +import com.ruoyi.swlscx.basic.service.YcExportTaskService; +import com.ruoyi.swlscx.common.PageParams; +import com.ruoyi.swlscx.year.domain.bo.HyYrzFBo; +import com.ruoyi.swlscx.year.domain.po.HyYrzF; +import com.ruoyi.swlscx.year.domain.vo.HyYrzFVo; +import com.ruoyi.swlscx.year.mapper.HyYrzFMapper; +import com.ruoyi.swlscx.year.service.HyYrzFService; +import com.ruoyi.swlscx.common.utils.CommonUtils; +import com.ruoyi.swlscx.common.utils.ExcelUtils; +import com.ruoyi.swlscx.common.utils.Query; +import org.apache.commons.lang3.ObjectUtils; +import org.apache.poi.ss.usermodel.CellStyle; +import org.apache.poi.ss.usermodel.Row; +import org.apache.poi.ss.usermodel.Sheet; +import org.springframework.beans.BeanUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.Date; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ThreadPoolExecutor; +import java.util.stream.Collectors; + +/** + * @author 12974 + * @description 针对表【hy_yrz_f(年水位表)】的数据库操作Service实现 + * @createDate 2024-09-05 16:53:36 + */ +@Service +public class HyYrzFServiceImpl extends ServiceImpl + implements HyYrzFService { + + @Resource + private ThreadPoolExecutor labelThreadPool; + + @Resource + private YcExportTaskService ycExportTaskService; + + /** + * 导入年水位表数据 + */ + @Override + @Transactional(rollbackFor = Exception.class) + public void importHyYrpFData(MultipartFile file) { + try { + List hyYrzFBos = ExcelUtils.readExcel(file.getInputStream(), HyYrzFBo.class, 2); + List hyYrpFList = hyYrzFBos.stream().map(i -> { + HyYrzF hyYrzF = new HyYrzF(); + BeanUtils.copyProperties(i, hyYrzF); + hyYrzF.setYr(Convert.toInt(i.getYr())); + hyYrzF.setAvz(Convert.toBigDecimal(i.getAvz())); + hyYrzF.setHtz(Convert.toBigDecimal(i.getHtz())); + hyYrzF.setHtzdt(DateUtils.parseDate(i.getHtzdt())); + hyYrzF.setMnz(Convert.toBigDecimal(i.getMnz())); + hyYrzF.setMnzdt(DateUtils.parseDate(i.getMnzdt())); + return hyYrzF; + }).collect(Collectors.toList()); + + this.saveBatch(hyYrpFList, 2000); + } catch (Exception e) { + e.printStackTrace(); + } + } + + /** + * 查询年水位表数据 + */ + @Override + public IPage selectHyYrzFPageByInfo(PageParams pageParams, String startTime, String endTime, String stnm, String stcd) { + Map map = CommonUtils.getYearAndDayDataMap(pageParams, startTime, endTime, stcd, stnm); + return this.baseMapper.selectHyYrzFPageByInfo(new Query().getPage(map), map); + } + + /** + * 导出年水位表数据 + */ + @Override + public R exportYearWaterLeverList(HttpServletResponse response, String startTime, String endTime, String stcd, String stnm) { + + Long userId = SecurityUtils.getUserId(); + String tableName = "年水位表"; + return taskAndExport(tableName,startTime, endTime, stcd, stnm, userId); + } + + @Resource + private ThreadPoolExecutor importThreadPool; + + @Autowired + private ApplicationContext context; + + + /** + * 从mysql导入到sqlserver中 + */ + @Transactional(rollbackFor = Exception.class) + @DataSource(DataSourceType.SHARDING) + public void importMysqlToSqlserver(List hyYrzFVoList) { + List records = hyYrzFVoList.stream().map(i->{ + HyYrzF hyYrzF = new HyYrzF(); + BeanUtils.copyProperties(i, hyYrzF); + hyYrzF.setYr(Convert.toInt(i.getYr())); + hyYrzF.setAvz(Convert.toBigDecimal(i.getAvz())); + hyYrzF.setHtz(Convert.toBigDecimal(i.getHtz())); + hyYrzF.setHtzdt(DateUtils.parseDate(i.getHtzdt())); + hyYrzF.setMnz(Convert.toBigDecimal(i.getMnz())); + hyYrzF.setMnzdt(DateUtils.parseDate(i.getMnzdt())); + return hyYrzF; + }).collect(Collectors.toList()); + //先清空临时表的数据 + this.baseMapper.deleteTempData(); + //插入临时表数据 + int batchSize = 50; + for (int i = 0; i < records.size(); i += batchSize) { + int end = Math.min(i + batchSize, records.size()); + List batch = records.subList(i, end); + // 执行批量插入 + this.baseMapper.insertTempData(batch); + } + //从临时表增或修改主表 + this.baseMapper.mergeToData(); + } + + @Override + public R importDateTosoft(String startTime, String endTime) { + Long userId = SecurityUtils.getUserId(); + String filename = "年水位表" + startTime + "-" + endTime+"导入到南方片"; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("status", 0) // 状态为0表示任务未完成 + .like("filename", "年水位表")); + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + YcExportTask ycExportTask = new YcExportTask(); + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + IPage pageByInfo = this.selectHyYrzFPageByInfo(new PageParams(0L, 1000000000L), startTime, endTime, null, null); +// IPage pageByInfo = this.selectMaxHRainListPageByInfo(new PageParams(0L, 1000000000L), null, null, startTime, endTime); + HyYrzFServiceImpl yrzFService = context.getBean(HyYrzFServiceImpl.class); + yrzFService.importMysqlToSqlserver(pageByInfo.getRecords()); + ycExportTask.setStatus(2); + } catch (Exception e) { + ycExportTask.setStatus(4); + log.error(e.getMessage()); + R.error("任务下载失败"); + }finally { + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } + }, importThreadPool); + return R.ok("任务已提交"); + } + + private R taskAndExport(String tableName,String startTime, String endTime, String stcd, String stnm, Long userId) { + String startNewTime=""; + String endNewTime=""; + if (ObjectUtils.isNotEmpty(startTime) ){ + startNewTime=startTime.substring(0,4); + } + if (ObjectUtils.isNotEmpty(endTime)){ + endNewTime=endTime.substring(0,4); + } + String filename = tableName + startNewTime + "-" + endNewTime; + // 查询是否已存在相同条件且未完成的任务 + List ongoingTasks = ycExportTaskService.list( + new QueryWrapper() + .eq("user_id", userId) + .eq("filename", filename) + .eq("status", 0) // 状态为0表示任务未完成 + ); + + if (!ongoingTasks.isEmpty()) { + return R.error("当前任务正在下载中,请稍后重试"); + } + + YcExportTask ycExportTask = new YcExportTask(); + String pathPeffix = RuoYiConfig.getExportPath() + + File.separator + userId + File.separator+ + DateUtils.parseDateToStr("yyyy-MM-dd",DateUtils.getNowDate()) + File.separator + + DateUtils.parseDateToStr("HH_mm",DateUtils.getNowDate())+File.separator; + ycExportTask.setFilename(filename); + ycExportTask.setUserId(userId); + ycExportTask.setStartTime(new Date()); + ycExportTask.setStatus(0); + ycExportTask.setFilepath(pathPeffix); + ycExportTaskService.save(ycExportTask); + CompletableFuture.runAsync(() -> { + try { + File dest = new File(pathPeffix); + if (!dest.exists()) { + dest.mkdirs(); + } + String fullPath = pathPeffix + filename; + exportBlackData(fullPath, startTime, endTime, stcd, stnm); + + ycExportTask.setStatus(1); + ycExportTask.setEndTime(new Date()); + ycExportTaskService.updateById(ycExportTask); + } catch (Exception e) { + log.error(e.getMessage(), e); + R.error("任务下载失败"); + } + }, labelThreadPool); + return R.ok("任务已提交,请稍后下载"); + } + + private void exportBlackData(String tableName, String startTime, String endTime, String stcd, String stnm) { + Map rainMap = CommonUtils.getYearAndDayDataMap(new PageParams(0L, 1000000000L), startTime, endTime, stcd, stnm); + ExcelUtils.excelBackExport2("yearWaterLever.xls", tableName, (Sheet sheet, CellStyle style) -> + dataModel1(rainMap, sheet, style)); + } + + public int dataModel1(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrzFVoIPage = this.baseMapper.selectHyYrzFPageByInfo(new Query().getPage(map), map); + List records = hyYrzFVoIPage.getRecords(); + + // 定义当前sheet的行号 + int rowIndex = 2; // 从第二行开始写入数据 + // 定义全局序号 + int globalIndex = 1; // 从1开始 + + // 定义当前sheet的编号 + int sheetIndex = 1; + + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + + // 复制表头到新sheet的逻辑 + ExcelUtils.copyHeader(sheet, style); + + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + + // 批量写入数据 + for (HyYrzFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水位表" + sheetIndex); + rowIndex = 2; // 重置行号 + // 复制表头到新sheet + ExcelUtils.copyHeader(sheet, style); + } + + // 创建新行并写入数据 + Row row = ExcelUtils.createRow(sheet, rowIndex); + + // 使用 globalIndex 作为序号 + ExcelUtils.formatCell(row, 0, String.valueOf(globalIndex), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvz() != null ? CommonUtils.formatNum(vo.getAvz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvzrcd() != null ? vo.getAvzrcd().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getHtz() != null ? CommonUtils.formatNum(vo.getHtz()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getHtzrcd() != null ? vo.getHtzrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getHtzdt() != null ? vo.getHtzdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnz() != null ? CommonUtils.formatNum(vo.getMnz()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnzrcd() != null ? vo.getMnzrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnzdt() != null ? vo.getMnzdt() : "", style); + ExcelUtils.formatCell(row, 14, "", style); + + // 增加行号和全局序号 + rowIndex++; + globalIndex++; + } + } + return 0; + } + + public void dataModel(Map map, Sheet sheet, CellStyle style) { + // 批量查询数据 + IPage hyYrzFVoIPage = this.baseMapper.selectHyYrzFPageByInfo(new Query().getPage(map), map); + List records = hyYrzFVoIPage.getRecords(); + // 定义当前sheet的行号 + int rowIndex = 2; + // 定义当前sheet的编号 + int sheetIndex = 1; + Row row; + // 最大行数限制 + int maxRowsPerSheet = 65536; // 对于 .xls 文件,最大行数为 65536 + int batchSize = 1000; // 每次批量处理1000行,减少逐行写入的开销 + // 批量处理数据 + for (int i = 0; i < records.size(); i += batchSize) { + // 处理一批数据 + int toIndex = Math.min(i + batchSize, records.size()); + List batchList = records.subList(i, toIndex); + // 批量写入数据 + for (HyYrzFVo vo : batchList) { + // 如果当前行数达到最大行数,创建新sheet + if (rowIndex >= maxRowsPerSheet) { + sheetIndex++; + sheet = sheet.getWorkbook().createSheet("年水位表" + sheetIndex); + rowIndex = 2; // 重置行号 + } + // 创建新行并写入数据 + row = ExcelUtils.createRow(sheet, rowIndex); + //序号 + ExcelUtils.formatCell(row, 0, String.valueOf(rowIndex - 1), style); + ExcelUtils.formatCell(row, 1, vo.getStcd() != null ? vo.getStcd() : "", style); + ExcelUtils.formatCell(row, 2, vo.getStnm() != null ? vo.getStnm() : "", style); + ExcelUtils.formatCell(row, 3, vo.getAddvcd() != null ? vo.getAddvcd() : "", style); + ExcelUtils.formatCell(row, 4, "", style); + ExcelUtils.formatCell(row, 5, vo.getYr() != null ? vo.getYr().toString() : "", style); + ExcelUtils.formatCell(row, 6, vo.getAvz() != null ? CommonUtils.formatNum(vo.getAvz()) : "", style); + ExcelUtils.formatCell(row, 7, vo.getAvzrcd() != null ? vo.getAvzrcd().toString() : "", style); + ExcelUtils.formatCell(row, 8, vo.getHtz() != null ? CommonUtils.formatNum(vo.getHtz()) : "", style); + ExcelUtils.formatCell(row, 9, vo.getHtzrcd() != null ? vo.getHtzrcd() : "", style); + ExcelUtils.formatCell(row, 10, vo.getHtzdt() != null ? vo.getHtzdt() : "", style); + ExcelUtils.formatCell(row, 11, vo.getMnz() != null ? CommonUtils.formatNum(vo.getMnz()) : "", style); + ExcelUtils.formatCell(row, 12, vo.getMnzrcd() != null ? vo.getMnzrcd() : "", style); + ExcelUtils.formatCell(row, 13, vo.getMnzdt() != null ? vo.getMnzdt() : "", style); + ExcelUtils.formatCell(row, 14, "", style); + rowIndex++; + } + } + } +} + + + + diff --git a/swlscx/src/main/resources/mapper/basic/HyStscAMapper.xml b/swlscx/src/main/resources/mapper/basic/HyStscAMapper.xml new file mode 100644 index 0000000..4acde65 --- /dev/null +++ b/swlscx/src/main/resources/mapper/basic/HyStscAMapper.xml @@ -0,0 +1,112 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + stcd,stnm,stct, + bshncd,hnnm,rvnm, + obitmcd,addvcd,wrrgcd, + esstyr,esstmth,wdstyr, + wdstmth,drar,flto, + dstrvm,fdtmnm,admag, + admnst,stlc,lgtd, + lttd,stgrd,frgrd, + nt + + + + + + TRUNCATE TABLE hy_stsc_a_temp; + + + + + + + + MERGE INTO hy_stsc_a AS a + USING hy_stsc_a_temp AS temp + ON (a.stcd = temp.stcd) + WHEN MATCHED THEN + UPDATE SET + a.stnm = temp.stnm, + a.stct = temp.stct, + a.bshncd = temp.bshncd, + a.hnnm = temp.hnnm, + a.rvnm = temp.rvnm, + a.obitmcd = temp.obitmcd, + a.addvcd = temp.addvcd, + a.wrrgcd = temp.wrrgcd, + a.esstyr = temp.esstyr, + a.esstmth = temp.esstmth, + a.wdstyr = temp.wdstyr, + a.wdstmth = temp.wdstmth, + a.drar = temp.drar, + a.flto = temp.flto, + a.dstrvm = temp.dstrvm, + a.fdtmnm = temp.fdtmnm, + a.admag = temp.admag, + a.admnst = temp.admnst, + a.stlc = temp.stlc, + a.lgtd = temp.lgtd, + a.lttd = temp.lttd, + a.stgrd = temp.stgrd, + a.frgrd = temp.frgrd, + a.nt = temp.nt + WHEN NOT MATCHED THEN + INSERT (stcd, stnm, stct, bshncd, hnnm, rvnm, obitmcd, addvcd, wrrgcd, esstyr, esstmth, wdstyr, wdstmth, drar, flto, dstrvm, fdtmnm, admag, admnst, stlc, lgtd, lttd, stgrd, frgrd, nt) + VALUES + (temp.stcd, temp.stnm, temp.stct, temp.bshncd, temp.hnnm, temp.rvnm, temp.obitmcd, temp.addvcd, temp.wrrgcd, temp.esstyr, temp.esstmth, temp.wdstyr, temp.wdstmth, temp.drar, temp.flto, temp.dstrvm, temp.fdtmnm, temp.admag, temp.admnst, temp.stlc, temp.lgtd, temp.lttd, temp.stgrd, temp.frgrd, temp.nt); + + diff --git a/swlscx/src/main/resources/mapper/basic/YcExportTaskMapper.xml b/swlscx/src/main/resources/mapper/basic/YcExportTaskMapper.xml new file mode 100644 index 0000000..b8073b9 --- /dev/null +++ b/swlscx/src/main/resources/mapper/basic/YcExportTaskMapper.xml @@ -0,0 +1,19 @@ + + + + + + + + diff --git a/swlscx/src/main/resources/mapper/day/HyDcsFMapper.xml b/swlscx/src/main/resources/mapper/day/HyDcsFMapper.xml new file mode 100644 index 0000000..cdd5add --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDcsFMapper.xml @@ -0,0 +1,85 @@ + + + + + + + + + + + + + TRUNCATE TABLE hy_dcs_f_temp; + + + + INSERT INTO hy_dcs_f_temp (stcd, dt, avcs, avcsrcd) + VALUES + + (#{item.stcd}, #{item.dt,jdbcType=TIMESTAMP}, #{item.avcs}, #{item.avcsrcd}) + + + + + insert into hy_dcs_f (stcd, dt, avcs, avcsrcd) + select a.stcd, + a.dt, + a.avcs, + a.avcsrcd + from hy_dcs_f_temp a + on duplicate key + update avcs = a.avcs, + avcsrcd = a.avcsrcd + + + + + + + + TRUNCATE TABLE hy_dcs_c_temp; + + + + INSERT INTO hy_dcs_c_temp (stcd, dt, avcs, avcsrcd) + VALUES + + (#{item.stcd}, #{item.dt,jdbcType=TIMESTAMP}, #{item.avcs}, #{item.avcsrcd}) + + + + + MERGE INTO hy_dcs_c AS a + USING hy_dcs_c_temp AS temp + ON (a.stcd = temp.stcd AND a.dt = temp.dt) + WHEN MATCHED THEN + UPDATE SET a.avcs = temp.avcs, a.avcsrcd = temp.avcsrcd + WHEN NOT MATCHED THEN + INSERT (stcd, dt, avcs, avcsrcd) + VALUES (temp.stcd, temp.dt, temp.avcs, temp.avcsrcd); + + diff --git a/swlscx/src/main/resources/mapper/day/HyDpCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDpCMapper.xml new file mode 100644 index 0000000..917c504 --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDpCMapper.xml @@ -0,0 +1,79 @@ + + + + + + + + + + + + + TRUNCATE TABLE hy_dp_c_temp; + + + + INSERT INTO hy_dp_c_temp (stcd, dt, p, prcd) + VALUES + + (#{item.stcd}, #{item.dt,jdbcType=TIMESTAMP}, #{item.p}, #{item.prcd}) + + + + + + + insert into hy_dp_c (stcd, dt, p, prcd) + select a.stcd, + a.dt, + a.p, + a.prcd + from hy_dp_c_temp a + on duplicate key + update p = a.p, + prcd = a.prcd + + + + + + + MERGE INTO hy_dp_c AS hme + USING hy_dp_c_temp AS temp + ON (hme.stcd = temp.stcd AND hme.dt = temp.dt) + WHEN MATCHED THEN + UPDATE SET hme.p = temp.p, + hme.prcd = temp.prcd + WHEN NOT MATCHED THEN + INSERT (stcd, dt, p, prcd) + VALUES (temp.stcd, temp.dt, temp.p, temp.prcd); + + + diff --git a/swlscx/src/main/resources/mapper/day/HyDqCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDqCMapper.xml new file mode 100644 index 0000000..0ce0e75 --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDqCMapper.xml @@ -0,0 +1,75 @@ + + + + + + + + + + + + + TRUNCATE TABLE hy_dq_c_temp; + + + + INSERT INTO hy_dq_c_temp (stcd, dt, avq, avqrcd) + VALUES + + (#{item.stcd}, #{item.dt,jdbcType=TIMESTAMP}, #{item.avq}, #{item.avqrcd}) + + + + + insert into hy_dq_c (stcd, dt, avq, avqrcd) + select a.stcd, + a.dt, + a.avq, + a.avqrcd + from hy_dq_c_temp a + on duplicate key + update avq = a.avq, + avqrcd = a.avqrcd + + + + + + MERGE INTO hy_dq_c AS a + USING hy_dq_c_temp AS temp + ON (a.stcd = temp.stcd AND a.dt = temp.dt) + WHEN MATCHED THEN + UPDATE SET a.avq = temp.avq, a.avqrcd = temp.avqrcd + WHEN NOT MATCHED THEN + INSERT (stcd, dt, avq, avqrcd) + VALUES + (temp.stcd, temp.dt, temp.avq, temp.avqrcd); + + diff --git a/swlscx/src/main/resources/mapper/day/HyDqsCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDqsCMapper.xml new file mode 100644 index 0000000..8070a0a --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDqsCMapper.xml @@ -0,0 +1,19 @@ + + + + + + + + + + + + + + stcd,sdtp,dt, + avqs,avqsrcd + + diff --git a/swlscx/src/main/resources/mapper/day/HyDweCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDweCMapper.xml new file mode 100644 index 0000000..8116bde --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDweCMapper.xml @@ -0,0 +1,70 @@ + + + + + + + + + + + + + + + TRUNCATE TABLE hy_dwe_c_temp; + + + + INSERT INTO hy_dwe_c_temp (stcd, dt, eetp, wsfe,wsfercd) + VALUES + + (#{item.stcd}, #{item.dt}, #{item.eetp}, #{item.wsfe},#{item.wsfercd}) + + + + + + insert into hy_dwe_c (stcd, dt, eetp, wsfe,wsfercd) + select a.stcd, + a.dt, + a.eetp, + a.wsfe, + a.wsfercd + from hy_dwe_c_temp a + on duplicate key + update wsfe = a.wsfe, + wsfercd = a.wsfercd + + + + + + diff --git a/swlscx/src/main/resources/mapper/day/HyDwtCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDwtCMapper.xml new file mode 100644 index 0000000..5574c3b --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDwtCMapper.xml @@ -0,0 +1,33 @@ + + + + + + + diff --git a/swlscx/src/main/resources/mapper/day/HyDzCMapper.xml b/swlscx/src/main/resources/mapper/day/HyDzCMapper.xml new file mode 100644 index 0000000..8e3e177 --- /dev/null +++ b/swlscx/src/main/resources/mapper/day/HyDzCMapper.xml @@ -0,0 +1,75 @@ + + + + + + + + + + + + + TRUNCATE TABLE hy_dz_c_temp; + + + + INSERT INTO hy_dz_c_temp (stcd, dt, avz, avzrcd) + VALUES + + (#{item.stcd}, #{item.dt,jdbcType=TIMESTAMP}, #{item.avz}, #{item.avzrcd}) + + + + + insert into hy_dz_c (stcd, dt, avz, avzrcd) + select a.stcd, + a.dt, + a.avz, + a.avzrcd + from hy_dz_c_temp a + on duplicate key + update avz = a.avz, + avzrcd = a.avzrcd + + + + + + + MERGE INTO hy_dz_c AS a + USING hy_dz_c_temp AS temp + ON (a.stcd = temp.stcd AND a.dt = temp.dt) + WHEN MATCHED THEN + UPDATE SET a.avz = temp.avz, a.avzrcd = temp.avzrcd + WHEN NOT MATCHED THEN + INSERT (stcd, dt, avz, avzrcd) + VALUES (temp.stcd, temp.dt, temp.avz, temp.avzrcd); + + diff --git a/swlscx/src/main/resources/mapper/excerpt/HyFdheexBMapper.xml b/swlscx/src/main/resources/mapper/excerpt/HyFdheexBMapper.xml new file mode 100644 index 0000000..7aea700 --- /dev/null +++ b/swlscx/src/main/resources/mapper/excerpt/HyFdheexBMapper.xml @@ -0,0 +1,82 @@ + + + + + + + + + + + + + + + + TRUNCATE TABLE hy_fdheex_b_temp; + + + + INSERT INTO hy_fdheex_b_temp (stcd, tm, z, zrcd,q,s) + VALUES + + (#{item.stcd}, #{item.tm}, #{item.z}, #{item.zrcd},#{item.q},#{item.s}) + + + + + + + insert into hy_fdheex_b (stcd, tm, z, zrcd, q, s) + select a.stcd, + a.tm, + a.z, + a.zrcd, + a.q, + a.s + from hy_fdheex_b_temp a + on duplicate key + update z = a.z, + zrcd = a.zrcd, + q = a.q, + s = a.s + + + + + + + + MERGE INTO hy_fdheex_b AS a + USING hy_fdheex_b_temp AS temp + ON (a.stcd = temp.stcd AND a.tm = temp.tm) + WHEN MATCHED THEN + UPDATE SET a.z = temp.z, a.zrcd = temp.zrcd, a.q = temp.q, a.s = temp.s + WHEN NOT MATCHED THEN + INSERT (stcd, tm, z, zrcd, q, s) + VALUES (temp.stcd, temp.tm, temp.z, temp.zrcd, temp.q, temp.s); + + diff --git a/swlscx/src/main/resources/mapper/excerpt/HyHltdzBMapper.xml b/swlscx/src/main/resources/mapper/excerpt/HyHltdzBMapper.xml new file mode 100644 index 0000000..a736c41 --- /dev/null +++ b/swlscx/src/main/resources/mapper/excerpt/HyHltdzBMapper.xml @@ -0,0 +1,95 @@ + + + + + + + + + + + + + + + + + TRUNCATE TABLE hy_hltdz_b_temp; + + + + INSERT INTO hy_hltdz_b_temp (stcd, tm, tdtp, tdz,tdzrcd,tdrng,dr,nt) + VALUES + +(#{item.stcd}, #{item.tm,jdbcType=TIMESTAMP}, #{item.tdtp}, #{item.tdz},#{item.tdzrcd},#{item.tdrng},#{item.dr},#{item.nt}) + + + + + + + insert into hy_hltdz_b (stcd, tm, tdtp, tdz, tdzrcd, tdrng, dr, nt) + select a.stcd, + a.tm, + a.tdtp, + a.tdz, + a.tdzrcd, + a.tdrng, + a.dr, + a.nt + from hy_hltdz_b_temp a + on duplicate key + update tdtp = a.tdtp, + tdz = a.tdz, + tdzrcd = a.tdzrcd, + tdrng = a.tdrng, + dr = a.dr, + nt = a.nt + + + + + + + MERGE INTO hy_hltdz_b AS a + USING hy_hltdz_b_temp AS temp + ON (a.stcd = temp.stcd AND a.tm = temp.tm) + WHEN MATCHED THEN + UPDATE SET a.tdtp = temp.tdtp,a.tdz = temp.tdz, a.tdzrcd = temp.tdzrcd, a.tdrng = temp.tdrng, a.dr = temp.dr, a.nt = temp.nt + WHEN NOT MATCHED THEN + INSERT (stcd, tm, tdtp, tdz, tdzrcd, tdrng, dr, nt) + VALUES + (temp.stcd, temp.tm, temp.tdtp, temp.tdz, temp.tdzrcd, temp.tdrng, temp.dr, temp.nt); + + + diff --git a/swlscx/src/main/resources/mapper/excerpt/HyPrexBMapper.xml b/swlscx/src/main/resources/mapper/excerpt/HyPrexBMapper.xml new file mode 100644 index 0000000..f03997b --- /dev/null +++ b/swlscx/src/main/resources/mapper/excerpt/HyPrexBMapper.xml @@ -0,0 +1,77 @@ + + + + + + + + + + + + + + TRUNCATE TABLE hy_prex_b_temp; + + + + INSERT INTO hy_prex_b_temp (stcd, bgtm, endtm, p,prcd) + VALUES + + (#{item.stcd}, #{item.bgtm,jdbcType=TIMESTAMP}, #{item.endtm,jdbcType=TIMESTAMP}, #{item.p},#{item.prcd}) + + + + + + + insert into hy_prex_b (stcd, bgtm, endtm, p, prcd) + select a.stcd, + a.bgtm, + a.endtm, + a.p, + a.prcd + from hy_prex_b_temp a + on duplicate key + update endtm = a.endtm, + p = a.p, + prcd = a.prcd + + + + + + + MERGE INTO hy_prex_b AS a + USING hy_prex_b_temp AS temp + ON (a.stcd = temp.stcd AND a.bgtm = temp.bgtm) + WHEN MATCHED THEN + UPDATE SET a.endtm = temp.endtm, a.p = temp.p, a.prcd = temp.prcd + WHEN NOT MATCHED THEN + INSERT (stcd, bgtm, endtm, p, prcd) + VALUES (temp.stcd, temp.bgtm, temp.endtm, temp.p, temp.prcd); + + + diff --git a/swlscx/src/main/resources/mapper/excerpt/HyRvfhexBMapper.xml b/swlscx/src/main/resources/mapper/excerpt/HyRvfhexBMapper.xml new file mode 100644 index 0000000..289a68b --- /dev/null +++ b/swlscx/src/main/resources/mapper/excerpt/HyRvfhexBMapper.xml @@ -0,0 +1,88 @@ + + + + + + + + + + + + + + + + + + TRUNCATE TABLE hy_rvfhex_b_temp; + + + + INSERT INTO hy_rvfhex_b_temp (stcd, tm, dambhdz, dambhdzrcd,q,s,w,nt) + VALUES + +(#{item.stcd}, #{item.tm,jdbcType=TIMESTAMP}, #{item.dambhdz}, #{item.dambhdzrcd},#{item.q},#{item.s},#{item.w},#{item.nt}) + + + + + + + insert into hy_rvfhex_b (stcd, tm, dambhdz, dambhdzrcd, q, s, w, nt) + select a.stcd, + a.tm, + a.dambhdz, + a.dambhdzrcd, + a.q, + a.s, + a.w, + a.nt + from hy_rvfhex_b_temp a + on duplicate key + update dambhdz = a.dambhdz, + dambhdzrcd = a.dambhdzrcd, + q = a.q, + s = a.s, + w = a.w, + nt = a.nt + + + + + + + MERGE INTO hy_rvfhex_b AS a + USING hy_rvfhex_b_temp AS temp + ON (a.stcd = temp.stcd AND a.tm = temp.tm) + WHEN MATCHED THEN + UPDATE SET a.dambhdz = temp.dambhdz, a.dambhdzrcd = temp.dambhdzrcd, a.q = temp.q, a.s = temp.s, a.w = temp.w, a.nt = temp.nt + WHEN NOT MATCHED THEN + INSERT (stcd, tm, dambhdz, dambhdzrcd, q, s, w, nt) + VALUES (temp.stcd, temp.tm, temp.dambhdz, temp.dambhdzrcd, temp.q, temp.s, temp.w, temp.nt); + + diff --git a/swlscx/src/main/resources/mapper/excerpt/HyWsfhexBMapper.xml b/swlscx/src/main/resources/mapper/excerpt/HyWsfhexBMapper.xml new file mode 100644 index 0000000..5b9640c --- /dev/null +++ b/swlscx/src/main/resources/mapper/excerpt/HyWsfhexBMapper.xml @@ -0,0 +1,61 @@ + + + + + + + + TRUNCATE TABLE hy_wsfhex_b_temp; + + + + INSERT INTO hy_wsfhex_b_temp(stcd, tm, upz, upzrcd, dwz, dwzrcd, q, s) + VALUES + + (#{item.stcd}, #{item.tm,jdbcType=TIMESTAMP}, #{item.upz}, #{item.upzrcd}, #{item.dwz}, #{item.dwzrcd}, #{item.q}, #{item.s}) + + + + + MERGE INTO hy_wsfhex_b AS a + USING hy_wsfhex_b_temp AS temp + ON (a.stcd = temp.stcd AND a.tm = temp.tm AND a.upz = temp.upz AND a.upzrcd = temp.upzrcd AND a.dwz = temp.dwz AND a.dwzrcd = temp.dwzrcd) + WHEN MATCHED THEN + UPDATE SET a.q = temp.q, a.s = temp.s + WHEN NOT MATCHED THEN + INSERT (stcd, tm, upz, upzrcd, dwz, dwzrcd, q, s) + VALUES (temp.stcd, temp.tm, temp.upz, temp.upzrcd, temp.dwz, temp.dwzrcd, temp.q, temp.s); + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtchpdEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtchpdEMapper.xml new file mode 100644 index 0000000..bd15036 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtchpdEMapper.xml @@ -0,0 +1,23 @@ + + + + + + + + + + + + + + + + + stcd,sdtp,yr, + mth,mdpd,avpd, + mxpd,nt + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtcsEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtcsEMapper.xml new file mode 100644 index 0000000..f3e61d3 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtcsEMapper.xml @@ -0,0 +1,79 @@ + + + + + hsa.stnm,hsa.addvcd, + hme.stcd,hme.yr,hme.mth, + hme.avcs,hme.avcsrcd,hme.mxs, + hme.mxsrcd,hme.mxsdt,hme.mns, + hme.mnsrcd,hme.mnsdt + + + + + + + TRUNCATE TABLE hy_mtcs_e_temp; + + + + insert into hy_mtcs_e_temp (stcd, yr, mth, avcs, avcsrcd, mxs, mxsrcd, mxsdt, mns, mnsrcd, mnsdt) + values + + (#{item.stcd}, #{item.yr}, #{item.mth}, #{item.avcs}, #{item.avcsrcd}, #{item.mxs} + , #{item.mxsrcd}, #{item.mxsdt,jdbcType=TIMESTAMP}, + #{item.mns}, #{item.mnsrcd}, #{item.mnsdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_mtcs_e AS a + USING hy_mtcs_e_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr AND a.mth = temp.mth) + WHEN MATCHED THEN + UPDATE SET a.avcs = temp.avcs, a.avcsrcd = temp.avcsrcd, a.mxs = temp.mxs, + a.mxsrcd = temp.mxsrcd, a.mxsdt = temp.mxsdt, + a.mns = temp.mns, a.mnsrcd=temp.mnsrcd, a.mnsdt = temp.mnsdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, mth, avcs, avcsrcd, mxs, mxsrcd, mxsdt, mns, mnsrcd, mnsdt) + VALUES (temp.stcd, temp.yr, temp.mth, temp.avcs, temp.avcsrcd, temp.mxs, temp.mxsrcd, temp.mxsdt, temp.mns, temp.mnsrcd, temp.mnsdt); + + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtiqEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtiqEMapper.xml new file mode 100644 index 0000000..5b29f54 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtiqEMapper.xml @@ -0,0 +1,23 @@ + + + + + + + + + + + + + + + + + stcd,yr,mth, + tiq,tiqrcd,mxiq, + mxiqrcd,mxiqdt + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtpEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtpEMapper.xml new file mode 100644 index 0000000..c41f2e4 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtpEMapper.xml @@ -0,0 +1,98 @@ + + + + + + + + + + + TRUNCATE TABLE hy_mtp_e_temp; + + + INSERT INTO hy_mtp_e_temp (stcd, yr, mth, p, prcd, pdynum, pdynumrcd, mxdyp, mxdyprcd, mxdypodt) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.mth}, #{item.p}, #{item.prcd}, #{item.pdynum}, #{item.pdynumrcd}, #{item.mxdyp}, #{item.mxdyprcd}, #{item.mxdypodt,jdbcType=TIMESTAMP}) + + + + MERGE INTO hy_mtp_e AS a + USING hy_mtp_e_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr AND a.mth = temp.mth) + WHEN MATCHED THEN + UPDATE SET a.p = temp.p, a.prcd = temp.prcd, a.pdynum = temp.pdynum, a.pdynumrcd = temp.pdynumrcd, a.mxdyp =temp.mxdyp, a.mxdyprcd = temp.mxdyprcd, a.mxdypodt = temp.mxdypodt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, mth, p, prcd, pdynum, pdynumrcd, mxdyp, mxdyprcd, mxdypodt) + VALUES (temp.stcd, temp.yr, temp.mth, temp.p, temp.prcd, temp.pdynum, temp.pdynumrcd, temp.mxdyp, temp.mxdyprcd, temp.mxdypodt); + + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtpddbEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtpddbEMapper.xml new file mode 100644 index 0000000..eb033a9 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtpddbEMapper.xml @@ -0,0 +1,20 @@ + + + + + + + + + + + + + + + stcd,sdtp,yr, + mth,ltpd,avswpct + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtqEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtqEMapper.xml new file mode 100644 index 0000000..dcd3529 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtqEMapper.xml @@ -0,0 +1,81 @@ + + + + + + + + + + + + + + + + + + + + + + TRUNCATE TABLE hy_mtq_e_temp + + + + insert into hy_mtq_e_temp (stcd, yr, mth, avq, avqrcd, mxq, mxqrcd, mxqdt, mnq, mnqrcd, mnqdt) + values + + (#{item.stcd}, #{item.yr}, #{item.mth}, #{item.avq}, #{item.avqrcd}, #{item.mxq}, #{item.mxqrcd}, + #{item.mxqdt,jdbcType=TIMESTAMP}, #{item.mnq},#{item.mnqrcd},#{item.mnqdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_mtq_e t + USING ( + SELECT + stcd, yr, mth, avq, avqrcd, mxq, mxqrcd, mxqdt, mnq, mnqrcd, mnqdt + FROM hy_mtq_e_temp + ) s + ON (t.stcd = s.stcd AND t.yr = s.yr AND t.mth = s.mth) + WHEN MATCHED THEN + UPDATE SET t.avq = s.avq, t.avqrcd=s.avqrcd, t.mxq = s.mxq, t.mxqrcd = s.mxqrcd, t.mxqdt = s.mxqdt, + t.mnq = s.mnq, t.mnqrcd = s.mnqrcd, t.mnqdt = s.mnqdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, mth, avq, avqrcd, mxq, mxqrcd,mxqdt, mnq, mnqrcd, mnqdt) + VALUES (s.stcd, s.yr, s.mth, s.avq, s.avqrcd, s.mxq, s.mxqrcd, s.mxqdt, s.mnq, s.mnqrcd, s.mnqdt); + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtqsEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtqsEMapper.xml new file mode 100644 index 0000000..2b92b9d --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtqsEMapper.xml @@ -0,0 +1,83 @@ + + + + + + + + + + + + + + + + + + hme.stcd,hme.sdtp,hme.yr, + hme.mth,hme.avqs,hme.avqsrcd, + hme.mxdyqs,hme.mxdyqsrcd,hme.mxdyqsodt,hsa.stnm,hsa.addvcd + + + + + + TRUNCATE TABLE hy_mtqs_e_temp; + + + + insert into hy_mtqs_e_temp(stcd, sdtp, yr, mth, avqs, avqsrcd, mxdyqs, mxdyqsrcd, mxdyqsodt) + values + + (#{item.stcd}, #{item.sdtp}, #{item.yr}, #{item.mth}, #{item.avqs}, #{item.avqsrcd}, #{item.mxdyqs}, #{item.mxdyqsrcd}, #{item.mxdyqsodt,jdbcType=TIMESTAMP}) + + + + MERGE INTO hy_mtqs_e AS a + USING hy_mtqs_e_temp AS temp + ON (a.stcd = temp.stcd AND a.sdtp = temp.sdtp AND a.yr = temp.yr AND a.mth = temp.mth) + WHEN MATCHED THEN + UPDATE SET a.avqs = temp.avqs, a.avqsrcd = temp.avqsrcd, a.mxdyqs = temp.mxdyqs, a.mxdyqsrcd = temp.mxdyqsrcd + , a.mxdyqsodt = temp.mxdyqsodt + WHEN NOT MATCHED THEN + INSERT (stcd, sdtp, yr, mth, avqs, avqsrcd, mxdyqs, mxdyqsrcd, mxdyqsodt) + VALUES (temp.stcd, temp.sdtp, temp.yr, temp.mth, temp.avqs, temp.avqsrcd, temp.mxdyqs, temp.mxdyqsrcd, temp.mxdyqsodt); + + diff --git a/swlscx/src/main/resources/mapper/month/HyMttdzEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMttdzEMapper.xml new file mode 100644 index 0000000..94cd8ce --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMttdzEMapper.xml @@ -0,0 +1,123 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + stcd,yr,mth, + avhtdz,avhtdzrcd,hthtdz, + hthtdzrcd,hthtdzotm,lthtdz, + lthtdzrcd,lthtdzotm,avltdz, + avltdzrcd,htltdz,htltdzrcd, + htltdzotm,ltltdz,ltltdzrcd, + ltltdzotm,avftdr,avftdrrcd, + mxfltdr,mxfltdrrcd,mxfltdrhtm, + mnfltdr,mnfltdrrcd,mnfltdrhtm, + aver,averbbrrcd,mxebtdr, + mxebtdrrcd,mxebtdrht,mnebtdr, + mnebtdrrcd,mnebtdrhtm,avftd, + avftdrcd,mxfltddr,mxfltddrrcd, + mxfltddrhtm,mnfltddr,mnfltddrrcd, + mnfltddrhtm,avebbdr,avedrc, + mxebtddr,mxebtddrrcd,mxebtddrhtm, + mnbtddr,mnebtddrrcd,mnebtddrhtm, + hravtdz,hravtdzrcd,avtdr, + avtdrrcd,avtddr,avtddrrcd + + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtweEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtweEMapper.xml new file mode 100644 index 0000000..0deda7a --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtweEMapper.xml @@ -0,0 +1,92 @@ + + + + + + + + + + + + + + + + + + + stcd,eetp,yr, + mth,wsfe,wsfercd, + mxdye,mxdyercd,mndye, + mndyercd + + + + + + + TRUNCATE TABLE hy_mtwe_e_temp; + + + + insert into hy_mtwe_e_temp (stcd,eetp,yr,mth,wsfe,wsfercd,mxdye,mxdyercd,mndye,mndyercd) + values + + (#{item.stcd},#{item.eetp},#{item.yr} + ,#{item.mth},#{item.wsfe},#{item.wsfercd},#{item.mxdye},#{item.mxdyercd} + ,#{item.mndye},#{item.mndyercd}) + + + + + MERGE INTO hy_mtwe_e as hme + USING hy_mtwe_e_temp as temp + ON (hme.stcd = temp.stcd AND hme.eetp = temp.eetp AND hme.yr = temp.yr AND hme.mth = temp.mth) + WHEN MATCHED THEN + UPDATE SET hme.wsfe = temp.wsfe, + hme.wsfercd = temp.wsfercd, + hme.mxdye = temp.mxdye, + hme.mxdyercd = temp.mxdyercd, + hme.mndye = temp.mndye, + hme.mndyercd = temp.mndyercd + WHEN NOT MATCHED THEN + INSERT (stcd,eetp,yr,mth,wsfe,wsfercd,mxdye,mxdyercd,mndye,mndyercd) + VALUES (temp.stcd,temp.eetp,temp.yr,temp.mth,temp.wsfe,temp.wsfercd,temp.mxdye,temp.mxdyercd,temp.mndye,temp.mndyercd); + + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtwtEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtwtEMapper.xml new file mode 100644 index 0000000..a437f32 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtwtEMapper.xml @@ -0,0 +1,80 @@ + + + + + + + + + + + + + + + + + + + + stcd,yr,mth, + avwtmp,avwtmprcd,mxwtmp, + mxwtmprcd,mxwtmpdt,mnwtmp, + mnwtmprcd,mnwtmpdt + + + + + + TRUNCATE TABLE hy_mtwt_e_temp; + + + + insert into hy_mtwt_e_temp (stcd,yr,mth,avwtmp,avwtmprcd,mxwtmp,mxwtmprcd,mxwtmpdt,mnwtmp,mnwtmprcd,mnwtmpdt) + values + + (#{item.stcd},#{item.yr},#{item.mth},#{item.avwtmp} + ,#{item.avwtmprcd},#{item.mxwtmp},#{item.mxwtmprcd},#{item.mxwtmpdt,jdbcType=TIMESTAMP}, + #{item.mnwtmp},#{item.mnwtmprcd},#{item.mnwtmpdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_mtwt_e AS t + USING (SELECT stcd,yr,mth,avwtmp,avwtmprcd,mxwtmp,mxwtmprcd,mxwtmpdt,mnwtmp,mnwtmprcd,mnwtmpdt FROM hy_mtwt_e_temp) AS s + ON (t.stcd = s.stcd AND t.yr = s.yr AND t.mth = s.mth) + WHEN MATCHED THEN + UPDATE SET t.avwtmp = s.avwtmp,t.avwtmprcd =s.avwtmprcd,t.mxwtmp = s.mxwtmp,t.mxwtmprcd = s.mxwtmprcd,t.mxwtmpdt = s.mxwtmpdt,t.mnwtmp = s.mnwtmp,t.mnwtmprcd = s.mnwtmprcd,t.mnwtmpdt = s.mnwtmpdt + WHEN NOT MATCHED THEN + INSERT (stcd,yr,mth,avwtmp,avwtmprcd,mxwtmp,mxwtmprcd,mxwtmpdt,mnwtmp,mnwtmprcd,mnwtmpdt) + VALUES (s.stcd,s.yr,s.mth,s.avwtmp,s.avwtmprcd,s.mxwtmp,s.mxwtmprcd,s.mxwtmpdt,s.mnwtmp,s.mnwtmprcd,s.mnwtmpdt); + + diff --git a/swlscx/src/main/resources/mapper/month/HyMtzEMapper.xml b/swlscx/src/main/resources/mapper/month/HyMtzEMapper.xml new file mode 100644 index 0000000..bcae9d5 --- /dev/null +++ b/swlscx/src/main/resources/mapper/month/HyMtzEMapper.xml @@ -0,0 +1,97 @@ + + + + + + INSERT INTO hy_mtz_e (stcd, yr, mth, avz,avzrcd,htz,htzrcd,htzdt,mnz,mnzrcd,mnzdt) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.mth}, #{item.avz}, #{item.avzrcd}, #{item.htz}, + #{item.htzrcd},#{item.htzdt},#{item.mnz},#{item.mnzrcd},#{item.mnzdt}) + + + + + + + TRUNCATE TABLE hy_mtz_e_temp; + + + INSERT INTO hy_mtz_e (stcd, yr, mth, avz,avzrcd,htz,htzrcd,htzdt,mnz,mnzrcd,mnzdt) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.mth}, #{item.avz}, #{item.avzrcd}, #{item.htz}, + #{item.htzrcd},#{item.htzdt,jdbcType=TIMESTAMP},#{item.mnz},#{item.mnzrcd},#{item.mnzdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_mtz_e t + USING ( + SELECT + stcd, + yr, + mth, + avz, + avzrcd, + htz, + htzrcd, + htzdt, + mnz, + mnzrcd, + mnzdt + FROM hy_mtz_e_temp + ) s + ON (t.stcd = s.stcd AND t.yr = s.yr AND t.mth = s.mth) + WHEN MATCHED THEN + UPDATE SET + t.avz = s.avz, + t.avzrcd = s.avzrcd, + t.htz = s.htz, + t.htzrcd = s.htzrcd, + t.htzdt = s.htzdt, + t.mnz = s.mnz, + t.mnzrcd = s.mnzrcd, + t.mnzdt = s.mnzdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, mth, avz,avzrcd,htz,htzrcd,htzdt,mnz,mnzrcd,mnzdt) + VALUES (s.stcd, s.yr, s.mth, s.avz, s.avzrcd, s.htz, s.htzrcd, s.htzdt, s.mnz, s.mnzrcd, s.mnzdt); + + diff --git a/swlscx/src/main/resources/mapper/year/HyDmxpFMapper.xml b/swlscx/src/main/resources/mapper/year/HyDmxpFMapper.xml new file mode 100644 index 0000000..01bedc9 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyDmxpFMapper.xml @@ -0,0 +1,56 @@ + + + + + + + + TRUNCATE TABLE hy_dmxp_f_temp; + + + + insert into hy_dmxp_f_temp (stcd, mxpdr, yr, bgdt, mxp, mxprc) + values + + (#{item.stcd}, #{item.mxpdr}, #{item.yr}, #{item.bgdt,jdbcType=TIMESTAMP}, #{item.mxp}, #{item.mxprc}) + + + + + MERGE INTO hy_dmxp_f AS a + USING hy_dmxp_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr AND a.mxpdr = temp.mxpdr) + WHEN MATCHED THEN + UPDATE SET + a.bgdt = temp.bgdt, a.mxp = temp.mxp, a.mxprc = temp.mxprc + WHEN NOT MATCHED THEN + INSERT (stcd, mxpdr, yr, bgdt, mxp,mxprc) + VALUES (temp.stcd, temp.mxpdr, temp.yr, temp.bgdt, temp.mxp, temp.mxprc); + + + + diff --git a/swlscx/src/main/resources/mapper/year/HyHmxpFMapper.xml b/swlscx/src/main/resources/mapper/year/HyHmxpFMapper.xml new file mode 100644 index 0000000..6bea113 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyHmxpFMapper.xml @@ -0,0 +1,62 @@ + + + + + + INSERT INTO hy_hmxp_f (stcd, yr, bgtm, mxpdr,mxp,mxprc) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.bgtm}, #{item.mxpdr}, #{item.mxp}, #{item.mxprc}) + + + + + + + + TRUNCATE TABLE hy_hmxp_f_temp; + + + + INSERT INTO hy_hmxp_f_temp(stcd, yr, bgtm, mxpdr, mxp, mxprc) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.bgtm,jdbcType=TIMESTAMP}, #{item.mxpdr}, #{item.mxp}, #{item.mxprc}) + + + + + MERGE INTO hy_hmxp_f AS a + USING hy_hmxp_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr AND a.mxpdr = temp.mxpdr) + WHEN MATCHED THEN + UPDATE SET a.bgtm = temp.bgtm, a.mxp = temp.mxp, a.mxprc = temp.mxprc + WHEN NOT MATCHED THEN + INSERT (stcd, yr, bgtm, mxpdr, mxp, mxprc) + VALUES (temp.stcd, temp.yr, temp.bgtm, temp.mxpdr, temp.mxp, temp.mxprc); + + \ No newline at end of file diff --git a/swlscx/src/main/resources/mapper/year/HyImxfwFMapper.xml b/swlscx/src/main/resources/mapper/year/HyImxfwFMapper.xml new file mode 100644 index 0000000..40a8bfb --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyImxfwFMapper.xml @@ -0,0 +1,52 @@ + + + + + + + + TRUNCATE TABLE hy_imxfw_f_temp; + + + + INSERT INTO hy_imxfw_f_temp (stcd, mxwdr, yr, bgdt, mxw, mxwrc) + VALUES + + (#{item.stcd}, #{item.mxwdr}, #{item.yr}, #{item.bgdt,jdbcType=TIMESTAMP}, #{item.mxw}, #{item.mxwrc}) + + + + MERGE INTO hy_imxfw_f AS hme + USING hy_imxfw_f_temp AS ht + ON (hme.stcd = ht.stcd AND hme.yr = ht.yr AND hme.mxwdr = ht.mxwdr) + WHEN MATCHED THEN + UPDATE SET hme.bgdt = ht.bgdt, hme.mxw = ht.mxw, hme.mxwrc = ht.mxwrc + WHEN NOT MATCHED THEN + INSERT (stcd, yr, bgdt, mxwdr, mxw, mxwrc) + VALUES (ht.stcd, ht.yr, ht.bgdt, ht.mxwdr, ht.mxw, ht.mxwrc); + + diff --git a/swlscx/src/main/resources/mapper/year/HyMmxpFMapper.xml b/swlscx/src/main/resources/mapper/year/HyMmxpFMapper.xml new file mode 100644 index 0000000..32edb86 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyMmxpFMapper.xml @@ -0,0 +1,65 @@ + + + + + INSERT INTO hy_mmxp_f (stcd, yr, bgdt, mxpdr,mxp,mxprc) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.bgdt}, #{item.mxpdr}, #{item.mxp}, #{item.mxprc}) + + + + + + + + INSERT INTO hy_mmxp_f_temp (stcd, yr, bgdt, mxpdr,mxp,mxprc) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.bgdt,jdbcType=TIMESTAMP}, #{item.mxpdr}, #{item.mxp}, #{item.mxprc}) + + + + + TRUNCATE TABLE hy_mmxp_f_temp; + + + + MERGE INTO hy_mmxp_f AS a + USING hy_mmxp_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr AND a.mxpdr = temp.mxpdr) + WHEN MATCHED THEN + UPDATE SET + a.bgdt = temp.bgdt, + a.mxp = temp.mxp, + a.mxprc = temp.mxprc + WHEN NOT MATCHED THEN + INSERT (stcd, yr, bgdt, mxpdr, mxp, mxprc) + VALUES + (temp.stcd, temp.yr, temp.bgdt, temp.mxpdr, temp.mxp, temp.mxprc); + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrcsFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrcsFMapper.xml new file mode 100644 index 0000000..4035e14 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrcsFMapper.xml @@ -0,0 +1,78 @@ + + + + + + + + + + TRUNCATE TABLE hy_yrcs_f_temp; + + + + INSERT INTO hy_yrcs_f_temp (stcd, yr, avcs, avcsrcd, mxs, mxsrcd, mxsdt, mns, mnsrcd, mnsdt) + VALUES + + (#{item.stcd},#{item.yr},#{item.avcs},#{item.avcsrcd},#{item.mxs},#{item.mxsrcd},#{item.mxsdt,jdbcType=TIMESTAMP},#{item.mns} + ,#{item.mnsrcd},#{item.mnsdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_yrcs_f t + USING ( + SELECT + stcd, + yr, + avcs, + avcsrcd, + mxs, + mxsrcd, + mxsdt, + mns, + mnsrcd, + mnsdt + FROM hy_yrcs_f_temp + ) s + ON (t.stcd = s.stcd AND t.yr = s.yr) + WHEN MATCHED THEN + UPDATE SET + t.avcs = s.avcs, + t.avcsrcd = s.avcsrcd, + t.mxs = s.mxs, + t.mxsrcd = s.mxsrcd, + t.mxsdt = s.mxsdt, + t.mns = s.mns, + t.mnsrcd = s.mnsrcd, + t.mnsdt = s.mnsdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, avcs, avcsrcd, mxs, mxsrcd, mxsdt, mns, mnsrcd, mnsdt) + VALUES (s.stcd, s.yr, s.avcs, s.avcsrcd, s.mxs, s.mxsrcd, s.mxsdt, s.mns, s.mnsrcd, s.mnsdt); + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrpFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrpFMapper.xml new file mode 100644 index 0000000..2644456 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrpFMapper.xml @@ -0,0 +1,124 @@ + + + + + + + + + + + + + + + + + + + stcd,yr,p, + prcd,pdynum,pdynumrcd, + frdsdt,frapdt,sndsdt, + snapdt + + + + INSERT INTO hy_yrp_f_temp (stcd, yr, p, prcd,pdynum,pdynumrcd,frdsdt,frapdt,sndsdt,snapdt) + VALUES + + (#{item.stcd}, #{item.yr}, #{item.p}, #{item.prcd}, #{item.pdynum}, #{item.pdynumrcd}, #{item.frdsdt}, #{item.frapdt}, #{item.sndsdt}, #{item.snapdt}) + + + + + + + insert into hy_yrp_f (stcd, yr, p, prcd, pdynum, pdynumrcd, frdsdt, frapdt, sndsdt, snapdt) + select a.stcd, + a.yr, + a.p, + a.prcd, + a.pdynum, + a.pdynumrcd, + a.frdsdt, + a.frapdt, + a.sndsdt, + a.snapdt + from hy_yrp_f_temp a + on duplicate key + update p = a.p, + prcd = a.prcd, + pdynum = a.pdynum, + pdynumrcd = a.pdynumrcd, + frdsdt = a.frdsdt, + frapdt = a.frapdt, + sndsdt = a.sndsdt, + snapdt = a.snapdt; + + + + TRUNCATE TABLE hy_yrp_f_temp; + + + + + + + TRUNCATE TABLE hy_yrp_f_temp; + + + + MERGE INTO hy_yrp_f AS a + USING hy_yrp_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr) + WHEN MATCHED THEN + UPDATE SET + a.p = temp.p, + a.prcd = temp.prcd, + a.pdynum = temp.pdynum, + a.pdynumrcd = temp.pdynumrcd, + a.frdsdt = temp.frdsdt, + a.frapdt = temp.frapdt, + a.sndsdt = temp.sndsdt, + a.snapdt = temp.snapdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, p, prcd, pdynum, pdynumrcd, frdsdt, frapdt, sndsdt, snapdt) + VALUES (temp.stcd, temp.yr, temp.p, temp.prcd, temp.pdynum, temp.pdynumrcd, temp.frdsdt, temp.frapdt, temp.sndsdt, temp.snapdt); + + + + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrqFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrqFMapper.xml new file mode 100644 index 0000000..01386bf --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrqFMapper.xml @@ -0,0 +1,69 @@ + + + + + + + + TRUNCATE TABLE hy_yrq_f_temp; + + + insert into hy_yrq_f_temp (stcd, yr, avq, avqrcd, mxq, mxqrcd, mxqdt, mnq, mnqrcd, mnqdt, + rw, rwrcd, rm, rd) + values + + (#{item.stcd}, #{item.yr}, #{item.avq}, #{item.avqrcd}, #{item.mxq}, #{item.mxqrcd}, #{item.mxqdt,jdbcType=TIMESTAMP}, + #{item.mnq}, #{item.mnqrcd}, #{item.mnqdt,jdbcType=TIMESTAMP}, #{item.rw}, #{item.rwrcd}, #{item.rm}, #{item.rd}) + + + + + MERGE INTO hy_yrq_f AS a + USING hy_yrq_f_temp AS b + ON (a.stcd = b.stcd and a.yr = b.yr) + WHEN MATCHED THEN + UPDATE SET a.avq = b.avq, a.avqrcd = b.avqrcd, a.mxq = b.mxq, a.mxqrcd = b.mxqrcd, a.mxqdt = b.mxqdt, + a.mnq = b.mnq, a.mnqrcd = b.mnqrcd, a.mnqdt = b.mnqdt, a.rw = b.rw, a.rwrcd = b.rwrcd, a.rm = b.rm, a.rd =b.rd + WHEN NOT MATCHED THEN + INSERT (stcd, yr, avq, avqrcd, mxq, mxqrcd, mxqdt, mnq, mnqrcd, mnqdt, rw, rwrcd, rm, rd) + VALUES (b.stcd, b.yr, b.avq, b.avqrcd, b.mxq, b.mxqrcd, b.mxqdt, b.mnq, b.mnqrcd, b.mnqdt, b.rw, b.rwrcd, b.rm , b.rd); + + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrqsFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrqsFMapper.xml new file mode 100644 index 0000000..d5fe4a9 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrqsFMapper.xml @@ -0,0 +1,73 @@ + + + + + + + + + + TRUNCATE TABLE hy_yrqs_f_temp; + + + + insert into hy_yrqs_f_temp(stcd, yr, sdtp, avqs, avqsrcd, mxdyqs, mxdyqsrcd, mxdyqsodt, sw, swpc, sm, simn, smec, nt) + values + + (#{item.stcd}, #{item.yr}, #{item.sdtp}, #{item.avqs},#{item.avqsrcd}, + #{item.mxdyqs}, #{item.mxdyqsrcd}, + #{item.mxdyqsodt,jdbcType=TIMESTAMP}, + #{item.sw}, #{item.swpc}, #{item.sm}, + #{item.simn}, #{item.smec}, #{item.nt}) + + + + + MERGE INTO hy_yrqs_f AS t + USING hy_yrqs_f_temp AS s + ON (t.stcd = s.stcd AND t.yr = s.yr AND t.sdtp = s.sdtp) + WHEN MATCHED THEN + UPDATE SET t.avqs = s.avqs, t.avqsrcd = s.avqsrcd, t.mxdyqs = s.mxdyqs, t.mxdyqsrcd = s.mxdyqsrcd, + t.mxdyqsodt = s.mxdyqsodt, t.sw = s.sw, t.swpc = s.swpc, t.sm = s.sm, t.simn = s.simn, t.smec = s.smec, t.nt = s.nt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, sdtp, avqs, avqsrcd, mxdyqs, mxdyqsrcd, mxdyqsodt, sw, swpc, sm, simn, smec, nt) + VALUES (s.stcd, s.yr, s.sdtp, s.avqs, s.avqsrcd, s.mxdyqs, s.mxdyqsrcd, s.mxdyqsodt, s.sw, s.swpc, s.sm, s.simn) + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrtdzFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrtdzFMapper.xml new file mode 100644 index 0000000..cb84299 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrtdzFMapper.xml @@ -0,0 +1,65 @@ + + + + + + TRUNCATE TABLE hy_yrtdz_f_temp; + + + + INSERT INTO hy_yrtdz_f_temp ( + stcd, yr, avhtdz, avhtdzrcd, hthtdz, hthtdzrcd, hthtdzotm, + lthtdz, lthtdzrcd, lthtdzotm, avltdz, avltdzrcd, htltdz, htltdzrcd, + htltdzotm, ltltdz, ltltdzrcd, ltltdzotm, avftdr, avftdrrcd, mxfltdr, + mxfltdrrcd, mxfltdrhtm, mnfltdr, mnfltdrrcd, mnfltdrhtm, aver, + averbbrrcd, mxebtdr, mxebtdrrcd, mxebtdrht, mnebtdr, mnebtdrrcd, + mnebtdrhtm, avftd, avftdrcd, mxfltddr, mxfltddrrcd, mxfltddrhtm, + mnfltddr, mnfltddrrcd, mnfltddrhtm, avebbdr, avedrc, mxebtddr, + mxebtddrrcd, mxebtddrhtm, mnbtddr, mnebtddrrcd, mnebtddrhtm, + hravtdz, hravtdzrcd, avtdr, avtdrrcd, avtddr, avtddrrcd, addvcd, + lysxm, stnm + )VALUES + + ( + #{item.stcd}, #{item.yr}, #{item.avhtdz}, #{item.avhtdzrcd}, #{item.hthtdz}, #{item.hthtdzrcd}, #{item.hthtdzotm,jdbcType=TIMESTAMP}, + #{item.lthtdz}, #{item.lthtdzrcd}, #{item.lthtdzotm,jdbcType=TIMESTAMP}, #{item.avltdz}, #{item.avltdzrcd}, #{item.htltdz}, #{item.htltdzrcd}, + #{item.htltdzotm,jdbcType=TIMESTAMP}, #{item.ltltdz}, #{item.ltltdzrcd}, #{item.ltltdzotm}, #{item.avftdr}, #{item.avftdrrcd}, #{item.mxfltdr}, + #{item.mxfltdrrcd}, #{item.mxfltdrhtm,jdbcType=TIMESTAMP}, #{item.mnfltdr}, #{item.mnfltdrrcd}, #{item.mnfltdrhtm,jdbcType=TIMESTAMP}, #{item.aver}, + #{item.averbbrrcd}, #{item.mxebtdr}, #{item.mxebtdrrcd}, #{item.mxebtdrht}, #{item.mnebtdr}, #{item.mnebtdrrcd}, + #{item.mnebtdrhtm,jdbcType=TIMESTAMP}, #{item.avftd}, #{item.avftdrcd}, #{item.mxfltddr}, #{item.mxfltddrrcd}, #{item.mxfltddrhtm,jdbcType=TIMESTAMP}, + #{item.mnfltddr}, #{item.mnfltddrrcd}, #{item.mnfltddrhtm,jdbcType=TIMESTAMP}, #{item.avebbdr}, #{item.avedrc}, #{item.mxebtddr}, + #{item.mxebtddrrcd}, #{item.mxebtddrhtm,jdbcType=TIMESTAMP}, #{item.mnbtddr}, #{item.mnebtddrrcd}, #{item.mnebtddrhtm,jdbcType=TIMESTAMP}, + #{item.hravtdz}, #{item.hravtdzrcd}, #{item.avtdr}, #{item.avtdrrcd}, #{item.avtddr}, #{item.avtddrrcd}, #{item.addvcd}, + #{item.lysxm}, #{item.stnm} + ) + + + + MERGE INTO hy_yrtdz_f AS a + USING hy_yrtdz_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr) + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrweFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrweFMapper.xml new file mode 100644 index 0000000..54c44a2 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrweFMapper.xml @@ -0,0 +1,140 @@ + + + + + + + + TRUNCATE TABLE hy_yrwe_f_temp; + + + + insert into hy_yrwe_f_temp + (stcd, + eetp, + yr, + wsfe, + wsfercd, + mxdye, + mxdyercd, + mxdyeodt, + mndye, + mndyercd, + mndyeodt, + idsdt, + icapd, + eslcch, + nt) + values + (#{item.stcd}, + #{item.eetp}, + #{item.yr}, + #{item.wsfe}, + #{item.wsfercd}, + #{item.mxdye}, + #{item.mxdyercd}, + #{item.mxdyeodt,jdbcType=TIMESTAMP}, + #{item.mndye}, + #{item.mndyercd}, + #{item.mndyeodt,jdbcType=TIMESTAMP}, + #{item.idsdt,jdbcType=TIMESTAMP}, + #{item.icapd,jdbcType=TIMESTAMP}, + #{item.eslcch}, + #{item.nt}) + + + + + MERGE INTO hy_yrwe_f AS a + USING hy_yrwe_f_temp AS b + + ON (a.stcd = b.stcd AND a.yr = b.yr AND a.eetp=b.eetp) + WHEN MATCHED THEN + UPDATE SET + a.wsfe = b.wsfe, + a.wsfercd = b.wsfercd, + a.mxdye = b.mxdye, + a.mxdyercd = b.mxdyercd, + a.mxdyeodt = b.mxdyeodt, + a.mndye = b.mndye, + a.mndyercd = b.mndyercd, + a.mndyeodt =b.mndyeodt, + a.idsdt = b.idsdt, + a.icapd = b.icapd, + a.eslcch = b.eslcch, + a.nt = b.nt + WHEN NOT MATCHED THEN + INSERT (stcd, + eetp, + yr, + wsfe, + wsfercd, + mxdye, + mxdyercd, + mxdyeodt, + mndye, + mndyercd, + mndyeodt, + idsdt, + icapd, + eslcch, + nt + ) + VALUES (b.stcd, + b.eetp, + b.yr, + b.wsfe, + b.wsfercd, + b.mxdye, + b.mxdyercd, + b.mxdyeodt, + b.mndye, + b.mndyercd, + b.mndyeodt, + b.idsdt, + b.icapd, + b.eslcch, + b.nt + ); + + + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrwtFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrwtFMapper.xml new file mode 100644 index 0000000..29074e3 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrwtFMapper.xml @@ -0,0 +1,72 @@ + + + + + + + + TRUNCATE TABLE hy_yrwt_f_temp; + + + + + insert into hy_yrwt_f_temp (stcd, yr, avwtmp, avwtmprcd, mxwtmp, mxwtmprcd, mxwtmpdt, mnwtmp, mnwtmprcd, mnwtmpdt) + values + + (#{item.stcd}, #{item.yr}, #{item.avwtmp}, + #{item.avwtmprcd}, #{item.mxwtmp}, #{item.mxwtmprcd},#{item.mxwtmpdt,jdbcType=TIMESTAMP}, #{item.mnwtmp}, + #{item.mnwtmprcd},#{item.mnwtmpdt,jdbcType=TIMESTAMP}) + + + + + MERGE INTO hy_yrwt_f AS a + USING hy_yrwt_f_temp AS temp + ON (a.stcd = temp.stcd AND a.yr = temp.yr) + WHEN MATCHED THEN + UPDATE SET + a.avwtmp = temp.avwtmp, + a.avwtmprcd = temp.avwtmprcd, + a.mxwtmp = temp.mxwtmp, + a.mxwtmprcd = temp.mxwtmprcd, + a.mxwtmpdt = temp.mxwtmpdt, + a.mnwtmp = temp.mnwtmp, a.mnwtmprcd = temp.mnwtmprcd, + a.mnwtmpdt = temp.mnwtmpdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, avwtmp, avwtmprcd, mxwtmp, mxwtmprcd, mxwtmpdt, mnwtmp, mnwtmprcd, mnwtmpdt) + VALUES (temp.stcd, temp.yr, temp.avwtmp, temp.avwtmprcd, temp.mxwtmp, temp.mxwtmprcd, temp.mxwtmpdt, temp.mnwtmp, temp.mnwtmprcd, temp.mnwtmpdt); + + diff --git a/swlscx/src/main/resources/mapper/year/HyYrzFMapper.xml b/swlscx/src/main/resources/mapper/year/HyYrzFMapper.xml new file mode 100644 index 0000000..7d6c658 --- /dev/null +++ b/swlscx/src/main/resources/mapper/year/HyYrzFMapper.xml @@ -0,0 +1,72 @@ + + + + + + + + TRUNCATE TABLE hy_yrz_f_temp; + + + + insert into hy_yrz_f_temp (stcd, yr, avz, avzrcd, htz, htzrcd, htzdt, mnz, mnzrcd, mnzdt) + values + + (#{item.stcd}, #{item.yr}, #{item.avz}, #{item.avzrcd}, #{item.htz}, #{item.htzrcd}, #{item.htzdt,jdbcType=TIMESTAMP}, #{item.mnz}, #{item.mnzrcd}, #{item.mnzdt,jdbcType=TIMESTAMP}) + + + + + + MERGE INTO hy_yrz_f AS a + USING hy_yrz_f_temp As b + ON (a.stcd = b.stcd and a.yr = b.yr) + WHEN MATCHED THEN + UPDATE SET a.avz = b.avz, + a.avzrcd = b.avzrcd, + a.htz = b.htz, + a.htzrcd = b.htzrcd, + a.htzdt = b.htzdt, + a.mnz = b.mnz, + a.mnzrcd = b.mnzrcd, + a.mnzdt = b.mnzdt + WHEN NOT MATCHED THEN + INSERT (stcd, yr, avz, avzrcd, htz, htzrcd, htzdt, mnz, mnzrcd, mnzdt) + VALUES (b.stcd, b.yr, b.avz, b.avzrcd, b.htz, b.htzrcd, b.htzdt, b.mnz, b.mnzrcd, b.mnzdt); + + + +