From e959811fa2c9be8ea8604a2b6fafa4bc06f894de Mon Sep 17 00:00:00 2001
From: PIGCLOUD <wangiegie@gmail.com>
Date: Mon, 29 Dec 2025 14:53:41 +0800
Subject: [PATCH] =?UTF-8?q?feat(codegen):=20=E6=94=AF=E6=8C=81=E5=8A=A8?=
 =?UTF-8?q?=E6=80=81=E6=95=B0=E6=8D=AE=E6=BA=90=E5=88=87=E6=8D=A2=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在GenTableController的export方法中添加DynamicDataSourceContextHolder支持
- 实现根据table参数动态切换数据源功能
---
 .../pig4cloud/pigx/codegen/controller/GenTableController.java  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/controller/GenTableController.java b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/controller/GenTableController.java
index 3da6ac9b2..f0cf9f628 100644
--- a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/controller/GenTableController.java
+++ b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/controller/GenTableController.java
@@ -17,6 +17,7 @@
 
 package com.pig4cloud.pigx.codegen.controller;
 
+import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.pig4cloud.pigx.codegen.entity.GenTable;
@@ -179,6 +180,8 @@ public class GenTableController {
     @ResponseExcel
     @GetMapping("/export")
     public List<GenTable> export(GenTable table) {
+        // 切换至对应数据源
+        DynamicDataSourceContextHolder.push(table.getDsName());
         return tableService.list(Wrappers.query(table));
     }