24282 2 жил өмнө
parent
commit
c52020777d
20 өөрчлөгдсөн 60 нэмэгдсэн , 57 устгасан
  1. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java
  2. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
  3. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/CacheController.java
  4. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/ServerController.java
  5. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysLogininforController.java
  6. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysOperlogController.java
  7. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysUserOnlineController.java
  8. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysConfigController.java
  9. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDeptController.java
  10. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictDataController.java
  11. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictTypeController.java
  12. 3 0
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysIndexController.java
  13. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java
  14. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysMenuController.java
  15. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysNoticeController.java
  16. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysPostController.java
  17. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java
  18. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRegisterController.java
  19. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRoleController.java
  20. 3 3
      ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysUserController.java

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java

@@ -1,13 +1,13 @@
 package com.ruoyi.web.controller.common;
 
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.google.code.kaptcha.Producer;
-import com.ruoyi.common.annotation.DataSource;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.constant.CacheConstants;
 import com.ruoyi.common.constant.Constants;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.common.core.redis.RedisCache;
-import com.ruoyi.common.enums.DataSourceType;
 import com.ruoyi.common.utils.sign.Base64;
 import com.ruoyi.common.utils.uuid.IdUtils;
 import com.ruoyi.system.service.ISysConfigService;
@@ -27,8 +27,8 @@ import java.util.concurrent.TimeUnit;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
-@DataSource(DataSourceType.SLAVE)
 public class CaptchaController {
     @Resource(name = "captchaProducer")
     private Producer captchaProducer;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java

@@ -1,10 +1,10 @@
 package com.ruoyi.web.controller.common;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.constant.Constants;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.domain.AjaxResult;
-import com.ruoyi.common.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.file.FileUploadUtils;
 import com.ruoyi.common.utils.file.FileUtils;
@@ -29,9 +29,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/common")
-@DataSource(DataSourceType.SLAVE)
 public class CommonController {
     private static final Logger log = LoggerFactory.getLogger(CommonController.class);
     private static final String FILE_DELIMETER = ",";

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/CacheController.java

@@ -1,9 +1,9 @@
 package com.ruoyi.web.controller.monitor;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.constant.CacheConstants;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.domain.AjaxResult;
-import com.ruoyi.common.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.domain.SysCache;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -19,9 +19,9 @@ import java.util.*;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/monitor/cache")
-@DataSource(DataSourceType.SLAVE)
 public class CacheController {
     private final static List<SysCache> caches = new ArrayList<SysCache>();
     @Autowired

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/ServerController.java

@@ -1,8 +1,8 @@
 package com.ruoyi.web.controller.monitor;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.domain.AjaxResult;
-import com.ruoyi.common.enums.DataSourceType;
 import com.ruoyi.framework.web.domain.Server;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -14,9 +14,9 @@ import org.springframework.web.bind.annotation.RestController;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/monitor/server")
-@DataSource(DataSourceType.SLAVE)
 public class ServerController {
     @PreAuthorize("@ss.hasPermi('monitor:server:list')")
     @GetMapping()

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysLogininforController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.monitor;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.web.service.SysPasswordService;
 import com.ruoyi.system.domain.SysLogininfor;
@@ -23,9 +23,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/monitor/logininfor")
-@DataSource(DataSourceType.SLAVE)
 public class SysLogininforController extends BaseController {
     @Autowired
     private ISysLogininforService logininforService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysOperlogController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.monitor;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.system.domain.SysOperLog;
 import com.ruoyi.system.service.ISysOperLogService;
@@ -22,9 +22,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/monitor/operlog")
-@DataSource(DataSourceType.SLAVE)
 public class SysOperlogController extends BaseController {
     @Autowired
     private ISysOperLogService operLogService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/monitor/SysUserOnlineController.java

@@ -1,15 +1,15 @@
 package com.ruoyi.web.controller.monitor;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.constant.CacheConstants;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.domain.SysUserOnline;
 import com.ruoyi.system.service.ISysUserOnlineService;
@@ -27,9 +27,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/monitor/online")
-@DataSource(DataSourceType.SLAVE)
 public class SysUserOnlineController extends BaseController {
     @Autowired
     private ISysUserOnlineService userOnlineService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysConfigController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.system.domain.SysConfig;
 import com.ruoyi.system.service.ISysConfigService;
@@ -23,9 +23,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/config")
-@DataSource(DataSourceType.SLAVE)
 public class SysConfigController extends BaseController {
     @Autowired
     private ISysConfigService configService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDeptController.java

@@ -1,13 +1,13 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.service.ISysDeptService;
 import org.apache.commons.lang3.ArrayUtils;
@@ -23,9 +23,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/dept")
-@DataSource(DataSourceType.SLAVE)
 public class SysDeptController extends BaseController {
     @Autowired
     private ISysDeptService deptService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictDataController.java

@@ -1,13 +1,13 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.system.service.ISysDictDataService;
@@ -26,9 +26,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/dict/data")
-@DataSource(DataSourceType.SLAVE)
 public class SysDictDataController extends BaseController {
     @Autowired
     private ISysDictDataService dictDataService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysDictTypeController.java

@@ -1,13 +1,13 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.system.service.ISysDictTypeService;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -23,9 +23,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/dict/type")
-@DataSource(DataSourceType.SLAVE)
 public class SysDictTypeController extends BaseController {
     @Autowired
     private ISysDictTypeService dictTypeService;

+ 3 - 0
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysIndexController.java

@@ -1,6 +1,8 @@
 package com.ruoyi.web.controller.system;
 
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.config.RuoYiConfig;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.utils.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -11,6 +13,7 @@ import org.springframework.web.bind.annotation.RestController;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 public class SysIndexController {
     /**

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.constant.Constants;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.framework.web.service.SysLoginService;
 import com.ruoyi.framework.web.service.SysPermissionService;
@@ -26,8 +26,8 @@ import java.util.Set;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
-@DataSource(DataSourceType.SLAVE)
 public class SysLoginController {
     @Autowired
     private SysLoginService loginService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysMenuController.java

@@ -1,13 +1,13 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.service.ISysMenuService;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -22,9 +22,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/menu")
-@DataSource(DataSourceType.SLAVE)
 public class SysMenuController extends BaseController {
     @Autowired
     private ISysMenuService menuService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysNoticeController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.system.domain.SysNotice;
 import com.ruoyi.system.service.ISysNoticeService;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -21,9 +21,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/notice")
-@DataSource(DataSourceType.SLAVE)
 public class SysNoticeController extends BaseController {
     @Autowired
     private ISysNoticeService noticeService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysPostController.java

@@ -1,12 +1,12 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.system.domain.SysPost;
 import com.ruoyi.system.service.ISysPostService;
@@ -23,9 +23,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/post")
-@DataSource(DataSourceType.SLAVE)
 public class SysPostController extends BaseController {
     @Autowired
     private ISysPostService postService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java

@@ -1,14 +1,14 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.config.RuoYiConfig;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.file.FileUploadUtils;
@@ -24,9 +24,9 @@ import org.springframework.web.multipart.MultipartFile;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/user/profile")
-@DataSource(DataSourceType.SLAVE)
 public class SysProfileController extends BaseController {
     @Autowired
     private ISysUserService userService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRegisterController.java

@@ -1,10 +1,10 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
+import com.ruoyi.common.constant.DatasourceConstant;
 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.framework.web.service.SysRegisterService;
 import com.ruoyi.system.service.ISysConfigService;
@@ -18,8 +18,8 @@ import org.springframework.web.bind.annotation.RestController;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
-@DataSource(DataSourceType.SLAVE)
 public class SysRegisterController extends BaseController {
     @Autowired
     private SysRegisterService registerService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysRoleController.java

@@ -1,7 +1,8 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.common.core.domain.entity.SysDept;
@@ -10,7 +11,6 @@ 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.enums.DataSourceType;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.web.service.SysPermissionService;
@@ -32,9 +32,9 @@ import java.util.List;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/role")
-@DataSource(DataSourceType.SLAVE)
 public class SysRoleController extends BaseController {
     @Autowired
     private ISysRoleService roleService;

+ 3 - 3
ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysUserController.java

@@ -1,7 +1,8 @@
 package com.ruoyi.web.controller.system;
 
-import com.ruoyi.common.annotation.DataSource;
+import com.baomidou.dynamic.datasource.annotation.DS;
 import com.ruoyi.common.annotation.Log;
+import com.ruoyi.common.constant.DatasourceConstant;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.common.core.domain.entity.SysDept;
@@ -9,7 +10,6 @@ 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.enums.DataSourceType;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
@@ -33,9 +33,9 @@ import java.util.stream.Collectors;
  *
  * @author ruoyi
  */
+@DS(DatasourceConstant.SLAVE_NAME)
 @RestController
 @RequestMapping("/system/user")
-@DataSource(DataSourceType.SLAVE)
 public class SysUserController extends BaseController {
     @Autowired
     private ISysUserService userService;