ZhuDeKang 3 месяцев назад
Родитель
Сommit
72939a8047

+ 8 - 0
ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/controller/SysMetaDatasourceController.java

@@ -3,6 +3,7 @@ package com.ruoyi.interfaces.controller;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
+import com.ruoyi.common.utils.SM4Util;
 import com.ruoyi.interfaces.domain.vo.SysMetaQueryVo;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -35,6 +36,9 @@ public class SysMetaDatasourceController extends BaseController {
     @Autowired
     private ISysMetaDatasourceService sysMetaDatasourceService;
 
+
+    @Autowired
+    private SM4Util sm4Util;
     /**
      * 查询数据源管理列表
      */
@@ -70,6 +74,8 @@ public class SysMetaDatasourceController extends BaseController {
     @Log(title = "数据源管理", businessType = BusinessType.INSERT)
     @PostMapping
     public AjaxResult add(@RequestBody SysMetaDatasource sysMetaDatasource) {
+        sysMetaDatasource.setDsUser(sm4Util.encrypt(sysMetaDatasource.getDsUser()));
+        sysMetaDatasource.setDsPass(sm4Util.encrypt(sysMetaDatasource.getDsPass()));
         return toAjax(sysMetaDatasourceService.insertSysMetaDatasource(sysMetaDatasource));
     }
 
@@ -79,6 +85,8 @@ public class SysMetaDatasourceController extends BaseController {
     @Log(title = "数据源管理", businessType = BusinessType.UPDATE)
     @PutMapping
     public AjaxResult edit(@RequestBody SysMetaDatasource sysMetaDatasource) {
+        sysMetaDatasource.setDsUser(sm4Util.encrypt(sysMetaDatasource.getDsUser()));
+        sysMetaDatasource.setDsPass(sm4Util.encrypt(sysMetaDatasource.getDsPass()));
         return toAjax(sysMetaDatasourceService.updateSysMetaDatasource(sysMetaDatasource));
     }
 

+ 16 - 2
ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/service/impl/SysMetaDatasourceServiceImpl.java

@@ -1,5 +1,6 @@
 package com.ruoyi.interfaces.service.impl;
 
+import com.ruoyi.common.utils.SM4Util;
 import org.springframework.jdbc.core.JdbcTemplate;
 
 import java.sql.Connection;
@@ -36,6 +37,8 @@ public class SysMetaDatasourceServiceImpl implements ISysMetaDatasourceService {
     @Autowired
     private DynamicDataSourceManager dataSourceManager;
 
+    @Autowired
+    private SM4Util sm4Util;
 
     /**
      * 查询数据源管理
@@ -45,9 +48,14 @@ public class SysMetaDatasourceServiceImpl implements ISysMetaDatasourceService {
      */
     @Override
     public SysMetaDatasource selectSysMetaDatasourceByDsCode(String dsCode) {
-        return sysMetaDatasourceMapper.selectSysMetaDatasourceByDsCode(dsCode);
+        SysMetaDatasource sysMetaDatasource = sysMetaDatasourceMapper.selectSysMetaDatasourceByDsCode(dsCode);
+        sysMetaDatasource.setDsUser(sm4Util.decrypt(sysMetaDatasource.getDsUser()));
+        sysMetaDatasource.setDsPass(sm4Util.decrypt(sysMetaDatasource.getDsPass()));
+        return sysMetaDatasource;
     }
 
+
+
     /**
      * 查询数据源管理列表
      *
@@ -56,7 +64,13 @@ public class SysMetaDatasourceServiceImpl implements ISysMetaDatasourceService {
      */
     @Override
     public List<SysMetaDatasource> selectSysMetaDatasourceList(SysMetaDatasource sysMetaDatasource) {
-        return sysMetaDatasourceMapper.selectSysMetaDatasourceList(sysMetaDatasource);
+        List<SysMetaDatasource> sysMetaDatasources = sysMetaDatasourceMapper.selectSysMetaDatasourceList(sysMetaDatasource);
+        sysMetaDatasources.forEach(p->{
+            p.setDsUser(sm4Util.decrypt(p.getDsUser()));
+            p.setDsPass(sm4Util.decrypt(p.getDsPass()));
+        });
+
+        return sysMetaDatasources;
     }
 
     /**