From 3dc2929d22a5f437fbee133df66aaff26a10d243 Mon Sep 17 00:00:00 2001
From: zhubaomin <zhubaomin>
Date: 星期五, 13 九月 2024 09:47:10 +0800
Subject: [PATCH] Merge branch 'master' of http://8.140.179.55:20000/r/pipIrr-SV

---
 pipIrr-platform/pipIrr-web/pipIrr-web-app/src/main/java/com/dy/pipIrrApp/captcha/CaptchaCtrl.java |   12 ++++++++++++
 1 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/pipIrr-platform/pipIrr-web/pipIrr-web-app/src/main/java/com/dy/pipIrrApp/captcha/CaptchaCtrl.java b/pipIrr-platform/pipIrr-web/pipIrr-web-app/src/main/java/com/dy/pipIrrApp/captcha/CaptchaCtrl.java
index 95abb7a..1015dd7 100644
--- a/pipIrr-platform/pipIrr-web/pipIrr-web-app/src/main/java/com/dy/pipIrrApp/captcha/CaptchaCtrl.java
+++ b/pipIrr-platform/pipIrr-web/pipIrr-web-app/src/main/java/com/dy/pipIrrApp/captcha/CaptchaCtrl.java
@@ -1,5 +1,6 @@
 package com.dy.pipIrrApp.captcha;
 
+import com.dy.common.multiDataSource.DataSourceContext;
 import com.dy.pipIrrApp.captcha.dto.CaptchaDomain;
 import com.dy.pipIrrGlobal.pojoBa.BaCaptcha;
 import jakarta.servlet.ServletOutputStream;
@@ -35,6 +36,17 @@
     @GetMapping("/get")
     @ResponseBody
     public CaptchaDomain getCaptcha(HttpServletRequest request, @RequestParam(value = "type", required = false, defaultValue = "char") String type) {
+        // 閰嶇疆鏁版嵁婧�
+        HttpServletRequest httpRequest = (HttpServletRequest) request;
+        String wxDataSourceName = httpRequest.getHeader("tag");
+        if(wxDataSourceName != null && wxDataSourceName.trim().length() > 0){
+            log.info("寰俊寮�鍙戯紝璁剧疆鏁版嵁婧愬悕绉颁负:" + wxDataSourceName);
+            //鎶婄粍缁囧崟浣嶆爣绛句綔涓烘暟鎹簮鍚嶇О
+            DataSourceContext.set(wxDataSourceName);
+        } else {
+            log.info("鐢ㄦ埛鏈�夋嫨鏁版嵁婧�");
+        }
+
         // 鐢熸垚楠岃瘉鐮佸疄浣�
         CaptchaDomain captchaDomain = captchaSV.createGoogleCaptcha(type);
         if (null != captchaDomain) {

--
Gitblit v1.8.0