yangys
2024-05-18 040976de6f9934b99f30268a28e2ecf42260e217
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
package com.qianwen.core.tenant.dynamic;
 
import com.baomidou.dynamic.datasource.processor.DsProcessor;
import org.aopalliance.intercept.MethodInvocation;
import com.qianwen.core.secure.utils.AuthUtil;
 
 
public class DsTenantIdProcessor extends DsProcessor {
    public static final String TENANT_ID_KEY = "#token.tenantId";
 
    public boolean matches(String key) {
        return key.equals(TENANT_ID_KEY);
    }
 
    public String doDetermineDatasource(MethodInvocation invocation, String key) {
        return AuthUtil.getTenantId();
    }
}