package com.dy.pipIrrApp.config;
|
|
import com.dy.common.multiDataSource.DataSourceContext;
|
import jakarta.servlet.*;
|
import jakarta.servlet.http.HttpServletRequest;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.io.IOException;
|
|
/**
|
* @author ZhuBaoMin
|
* @date 2024-11-29 14:24
|
* @LastEditTime 2024-11-29 14:24
|
* @Description
|
*/
|
|
@Slf4j
|
public class DataSourceFilter implements Filter {
|
|
@Override
|
public void init(FilterConfig filterConfig) throws ServletException {
|
}
|
|
@Override
|
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws ServletException, IOException {
|
HttpServletRequest httpRequest = (HttpServletRequest) servletRequest;
|
String wxDataSourceName = httpRequest.getHeader("tag");
|
if (wxDataSourceName != null && wxDataSourceName.trim().length() > 0) {
|
log.info("APP开发，设置数据源名称为:" + wxDataSourceName);
|
//把组织单位标签作为数据源名称
|
DataSourceContext.set(wxDataSourceName);
|
} else {
|
log.info("用户未选择数据源");
|
}
|
filterChain.doFilter(servletRequest, servletResponse);
|
}
|
|
@Override
|
public void destroy() {
|
}
|
}
|
