zm
2020-05-18 a18bfacbf56b401f6e0fdae8710fbca4df8cff77
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package com.changhong.epc.config.sqlfilter;
 
import com.changhong.epc.config.tool.MybatisTool;
import com.changhong.epc.constter.base.Context;
import com.iemsoft.framework.cloud.core.tools.StringUtil;
import com.iemsoft.framework.cloud.mybatis.interceptor.MybatisFilter;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
 
import java.lang.reflect.Method;
import java.util.Optional;
 
/**
 * 基础字段sql拦截
 */
@Component
@Slf4j
@Order(Integer.MIN_VALUE)
public class BaseParamSqlFilter implements MybatisFilter, SqlFilter {
 
    @Override
    public void filter(StatementHandler rsh, Object[] args, Method method) {
        try {
            MybatisTool.setSql(rsh, filter(MybatisTool.getSql(rsh)));
        }catch (Exception e){
            log.error("基础数据添加错误:{}", e.getMessage());
            log.error(e.getMessage(), e);
        }
    }
 
    @Override
    public String filter(String sql) {
        return sql.replace("{companyId}", Context.getCompanyIdStr())
                .replace("{tenantId}", Context.getTenantIdStr())
                .replace("{openId}", Optional.ofNullable(Context.getOpenId()).orElse(""))
                .replace("{rule}"
                        , StringUtil.toString(Context.getTenantId(), "") +"_"+ StringUtil.toString(Context.getCompanyId(), ""));
    }
 
    @Override
    public void filter(Executor e, Object[] args, Method method) {
 
    }
}