From 1c50cb5546715fa21496fbdf2bdacb7ae8836b8a Mon Sep 17 00:00:00 2001
From: hdy <1105738590@qq.com>
Date: 星期二, 11 三月 2025 13:17:53 +0800
Subject: [PATCH] 修改
---
billion-generator/src/main/java/com/billion/generator/controller/GenController.java | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/billion-generator/src/main/java/com/billion/generator/controller/GenController.java b/billion-generator/src/main/java/com/billion/generator/controller/GenController.java
index 164a57b..410d037 100644
--- a/billion-generator/src/main/java/com/billion/generator/controller/GenController.java
+++ b/billion-generator/src/main/java/com/billion/generator/controller/GenController.java
@@ -6,6 +6,9 @@
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;
+
+import com.billion.common.annotation.DataSource;
+import com.billion.common.enums.DataSourceType;
import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
@@ -84,6 +87,7 @@
*/
@PreAuthorize("@ss.hasPermi('tool:gen:list')")
@GetMapping("/db/list")
+ @DataSource(value = DataSourceType.SLAVE)
public TableDataInfo dataList(GenTable genTable)
{
startPage();
@@ -250,7 +254,7 @@
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.setHeader("Content-Disposition", "attachment; filename=\"billion.zip\"");
response.addHeader("Content-Length", "" + data.length);
response.setContentType("application/octet-stream; charset=UTF-8");
IOUtils.write(data, response.getOutputStream());
--
Gitblit v1.9.3