This commit is contained in:
2023-06-02 18:12:51 +08:00
parent 603b8c1c9f
commit 951a08a0a1
7 changed files with 44 additions and 6 deletions

View File

@ -1,5 +1,6 @@
package com.qiaoba.common.database.filters;
import com.qiaoba.common.base.constants.BaseConstant;
import com.qiaoba.common.base.context.BaseContext;
import com.qiaoba.common.database.config.DynamicDataSourceContext;
import com.qiaoba.common.database.constants.DynamicDatasourceConstant;
@ -32,7 +33,7 @@ public class DynamicDataSourceFilter extends OncePerRequestFilter {
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
before();
before(request);
filterChain.doFilter(request, response);
after();
}
@ -43,12 +44,13 @@ public class DynamicDataSourceFilter extends OncePerRequestFilter {
return false;
}
private void before() {
private void before(HttpServletRequest request) {
String tenantId = request.getHeader(BaseConstant.HEADER_KEY_TENANT);
// todo
//设置当前租户对应的数据源
BaseContext.setDataSource(DynamicDatasourceConstant.DEFAULT_MASTER_DATASOURCE_KEY);
//设置当前租户对应的租户ID
BaseContext.setTenantId("1");
BaseContext.setTenantId(tenantId);
//设置当前租户对应的数据库类型
BaseContext.setDatabaseType("MySQL");
}