MybatisPlusConfig.java
4.47 KB
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
88
89
90
91
92
93
94
95
96
package com.tianting.infoloop.config;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.autoconfigure.MybatisPlusAutoConfiguration;
import com.baomidou.mybatisplus.core.injector.ISqlInjector;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
import com.fasterxml.jackson.databind.JsonNode;
import com.mysql.jdbc.PreparedStatement;
import com.tianting.infoloop.handler.MultiTenantHandler;
import com.tianting.infoloop.inject.MybatisPlusSqlInjector;
import com.tianting.infoloop.utils.MysqlTypeTransferUtil;
import java.util.List;
import java.util.Map;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.TypeHandler;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;
@Configuration
@EnableTransactionManagement
@EnableConfigurationProperties(TenantProperties.class)
@MapperScan(basePackages = "com.tianting.infoloop.mapper")
public class MybatisPlusConfig {
@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor(TenantProperties tenantProperties){
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
if (Boolean.TRUE.equals(tenantProperties.getEnable())) {
interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new MultiTenantHandler(tenantProperties)));
}
// 启用分页插件
PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
paginationInnerInterceptor.setDbType(DbType.MYSQL);
interceptor.addInnerInterceptor(paginationInnerInterceptor);
// 注册自定义类型处理器
TypeHandler<JsonNode> jsonNodeTypeHandler = new MysqlTypeTransferUtil();
// 创建拦截器并添加到MybatisPlusInterceptor
// interceptor.addInnerInterceptor(new MybatisPlusInterceptor() {
// @Override
// public Object intercept(MybatisPlusInterceptor.Chain chain) throws Throwable {
// SqlSessionFactory sqlSessionFactory = chain.getSqlSessionFactory();
// SqlSession sqlSession = sqlSessionFactory.openSession();
//
// try {
// Object[] args = chain.getArgs();
// if (args!= null && args.length > 0) {
// Object arg = args[0];
// if (arg instanceof java.util.Map) {
// java.util.Map<?,?> map = (java.util.Map<?,?>) arg;
// Object mealDetailJsonValue = map.get("mealDetailJson");
// if (mealDetailJsonValue instanceof JsonNode) {
// // 如果是JsonNode类型，使用自定义类型处理器进行处理
// PreparedStatement ps = sqlSession.getConnection().prepareStatement(chain.getSql());
// int paramIndex = 0;
// for (Object key : map.keySet()) {
// if ("mealDetailJson".equals(key) && mealDetailJsonValue instanceof JsonNode) {
// break;
// }
// paramIndex++;
// }
// jsonNodeTypeHandler.setNonNullParameter(ps, paramIndex, (JsonNode) mealDetailJsonValue, JdbcType.JSON);
// }
// }
// }
// // 继续执行后续的数据库操作逻辑，并返回结果
// return chain.proceed();
// } finally {
// sqlSession.close();
// }
// }
// });
return interceptor;
}
@Bean
public ISqlInjector sqlInjector() {
return new MybatisPlusSqlInjector();
}
@Bean
public MysqlTypeTransferUtil mysqlTypeTransferUtil() {
return new MysqlTypeTransferUtil();
}
}