不同仓库分配不同数据库：filter拦截所有请求，将warehouseCode值存入serlvetContext中

xumiao
1 parent 87b7c26f
Showing 2 changed files with 46 additions and 1 deletions
src/main/java/com/huaheng/framework/config/DruidConfig.java
src/main/java/com/huaheng/framework/config/TestFilter.java
@@ -78,7 +78,11 @@ public class DruidConfig
      */
     @Bean
     public FilterRegistrationBean druidStatFilter() {
-        FilterRegistrationBean bean = new FilterRegistrationBean(new WebStatFilter());
+       // FilterRegistrationBean bean = new FilterRegistrationBean(new WebStatFilter());
+        FilterRegistrationBean<TestFilter> bean = new FilterRegistrationBean<TestFilter>();
+        bean.setFilter(new TestFilter());//注册自定义过滤器
+        bean.setName("flilter1");//过滤器名称
+        bean.setOrder(1);//优先级，最顶级
         //添加过滤规则.
         bean.addUrlPatterns("/*");
         //添加不需要忽略的格式信息.
+package com.huaheng.framework.config;
+
+import com.huaheng.framework.aspectj.lang.constant.DataSourceName;
+import com.huaheng.framework.datasource.DynamicDataSourceContextHolder;
+
+import javax.servlet.*;
+import javax.servlet.FilterConfig;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpSession;
+import java.io.IOException;
+
+public class TestFilter implements Filter {
+    @Override
+    public void init(FilterConfig filterConfig) throws ServletException {
+
+    }
+
+    @Override
+    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
+        HttpServletRequest req=(HttpServletRequest)request;
+        System.out.println("自定义过滤器filter触发,拦截url:"+req.getRequestURI());
+        try {
+            ServletContext servletContext = req.getServletContext();
+            String warehouseCode = request.getParameter("warehouseCode");
+            if(warehouseCode!=null){
+                servletContext.setAttribute("warehouseCode",warehouseCode);
+            }
+            if(servletContext.getAttribute("warehouseCode")!=null&&servletContext.getAttribute("warehouseCode").equals("CS0001")) {
+                DynamicDataSourceContextHolder.setDB(DataSourceName.SLAVE);
+            }
+            chain.doFilter(request, response);
+        } finally {
+            DynamicDataSourceContextHolder.clearDB();
+        }
+    }
+
+    @Override
+    public void destroy() {
+
+    }
+}