12/29 河门口大屏特殊修改

101captain

2021-12-29 018af6bed3da8be2e7b846aa3e654f16f8f71654

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package com.panzhihua.service_dangjian.config;
 
import com.baomidou.mybatisplus.core.toolkit.PluginUtils;
import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
 
import java.lang.reflect.Field;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
 
public class MybatisHmkInterceptor implements InnerInterceptor {
    @Override
    public void beforeQuery(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
        String sql=boundSql.getSql();
        if(parameter instanceof Long){
            if(sql.contains("community_id = ?")&&(long)parameter==10086){
                sql=sql.replace("community_id = ?","community_id in (?)");
                parameter="30,31";
                boundSql.setAdditionalParameter("communityId",parameter);
                PluginUtils.mpBoundSql(boundSql).sql(sql);
            }
        }
        if(parameter instanceof Map){
            if(sql.contains("community_id = ?")){
                List<ParameterMapping> parameterMappingList=new ArrayList<>();
                if(((Map<?, ?>) parameter).containsValue(10086L)){
                    List<ParameterMapping> parameterMappings=boundSql.getParameterMappings();
                    parameterMappings.forEach(parameterMapping -> {
                        if (parameterMapping.getProperty().equals("communityId")){
                            ParameterMapping parameterMapping1=new ParameterMapping.Builder(ms.getConfiguration(),parameterMapping.getProperty(),Object.class).build();
                            parameterMappingList.add(parameterMapping1);
                        }
                        else {
                            parameterMappingList.add(parameterMapping);
                        }
                    });
                    sql=sql.replace("community_id = ?","community_id in (?)");
                    ((Map<?,?>) parameter).replaceAll((k,v)->{
                        if(v.toString().contains("10086")){
                            boundSql.setAdditionalParameter(k.toString(),"30,31");
                        }
                        return v;
                    });
                    PluginUtils.mpBoundSql(boundSql).parameterMappings(parameterMappingList);
                    PluginUtils.mpBoundSql(boundSql).sql(sql);
                }
                else{
                    if(((Map<?, ?>) parameter).containsKey("arg0")){
                        Map<?, ?> param= (Map<?, ?>) boundSql.getAdditionalParameter("_parameter");
                        String finalSql = sql;
                        param.forEach((k, v)->{
                            try {
                                Field field= v.getClass().getDeclaredField("communityId");
                                if(field!=null){
                                    field.setAccessible(true);
                                    if((long)field.get(v)==10086L){
                                        String sql1= finalSql.replace("community_id = ?","community_id in (30,31)");
                                        boundSql.getParameterMappings().forEach(parameterMapping -> {
                                            if(!parameterMapping.getProperty().contains("communityId")){
                                                parameterMappingList.add(parameterMapping);
                                            }
                                        });
                                        PluginUtils.mpBoundSql(boundSql).parameterMappings(parameterMappingList);
                                        PluginUtils.mpBoundSql(boundSql).sql(sql1);
                                    }
                                }
                            } catch (IllegalAccessException e) {
                                e.printStackTrace();
                            } catch (NoSuchFieldException e) {
 
                            }
                        });
                    }
                }
            }
        }
        InnerInterceptor.super.beforeQuery(executor, ms, parameter, rowBounds, resultHandler, boundSql);
    }
 
}