12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- package org.jeecg.config.mybatis.interceptor;
- import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
- import lombok.extern.slf4j.Slf4j;
- import org.apache.commons.lang3.StringUtils;
- import org.springframework.web.servlet.HandlerInterceptor;
- import org.springframework.web.servlet.ModelAndView;
- import javax.servlet.http.HttpServletRequest;
- import javax.servlet.http.HttpServletResponse;
- /**
- * 动态数据源切换拦截器
- *
- * 测试:拦截参数,自动切换数据源
- * 未来规划:后面通过此机制,实现多租户切换数据源功能
- * @author zyf
- */
- @Slf4j
- public class DynamicDatasourceInterceptor implements HandlerInterceptor {
- /**
- * 在请求处理之前进行调用(Controller方法调用之前)
- */
- @Override
- public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
- String requestURI = request.getRequestURI();
- log.info("经过多数据源Interceptor,当前路径是{}", requestURI);
- //获取动态数据源名称
- String dsName = request.getParameter("dsName");
- String dsKey = "master";
- if (StringUtils.isNotEmpty(dsName)) {
- dsKey = dsName;
- }
- DynamicDataSourceContextHolder.push(dsKey);
- return true;
- }
- /**
- * 请求处理之后进行调用,但是在视图被渲染之前(Controller方法调用之后)
- */
- @Override
- public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) {
- }
- /**
- * 在整个请求结束之后被调用,也就是在DispatcherServlet 渲染了对应的视图之后执行(主要是用于进行资源清理工作)
- */
- @Override
- public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
- DynamicDataSourceContextHolder.clear();
- }
- }
|