Skip to content

GitLab

S
schbrain-parent
Commit 52f6fe8d authored by liaozan's avatar liaozan 🏀
Browse files
Options

Update baseDao methodHandle usage

parent ba8388bb
Hide whitespace changes
Inline Side-by-side
Showing with 111 additions and 70 deletions
support/schbrain-base-dao/src/main/java/com/schbrain/framework/dao/mybatis/BaseMethodInvocationHandler.java → support/schbrain-base-dao/src/main/java/com/schbrain/framework/dao/mybatis/BaseDaoInvocationHandler.java
View file @ 52f6fe8d
package com.schbrain.framework.dao.mybatis;
import com.schbrain.common.exception.BaseException;
import com.schbrain.framework.dao.BaseDao;
import com.schbrain.framework.dao.mybatis.mapper.BaseMapper;
import org.apache.ibatis.reflection.ExceptionUtil;
import org.mybatis.spring.SqlSessionTemplate;
import java.io.Serializable;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodHandles.Lookup;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* description
*
* @author liwu on 2019/7/29
* @author liwu
* @since 2019/7/29
*/
public class BaseMethodInvocationHandler<T> implements InvocationHandler, Serializable {
public class BaseDaoInvocationHandler<T> implements InvocationHandler {
private static final long serialVersionUID = -5077848994288044943L;
private final Map<Method, MethodHandle> methodHandleCache = new ConcurrentHashMap<>();
private final T originMapperProxy;
private final BaseMapper baseMapper;
private final Lookup mapperInterfaceLookup;
public BaseMethodInvocationHandler(T originMapperProxy, SqlSessionTemplate sqlSession, Class<T> mapperInterface) {
public BaseDaoInvocationHandler(T originMapperProxy, SqlSessionTemplate sqlSession, Class<T> mapperInterface) throws IllegalAccessException {
this.originMapperProxy = originMapperProxy;
baseMapper = new BaseMapper(sqlSession, mapperInterface);
this.baseMapper = new BaseMapper(sqlSession, mapperInterface);
this.mapperInterfaceLookup = MethodHandles.privateLookupIn(mapperInterface, MethodHandles.lookup());
}
@Override
......@@ -39,21 +42,27 @@ public class BaseMethodInvocationHandler<T> implements InvocationHandler, Serial
} else if (isBaseMethod(method)) {
return invokeBaseMethod(method, args);
}
} catch (Throwable t) {
throw ExceptionUtil.unwrapThrowable(t);
} catch (Throwable throwable) {
throw ExceptionUtil.unwrapThrowable(throwable);
}
return method.invoke(originMapperProxy, args);
}
private Object invokeDefaultMethod(Object proxy, Method method, Object[] args) throws Throwable {
final Constructor<Lookup> constructor = Lookup.class.getDeclaredConstructor(Class.class, int.class);
if (!constructor.isAccessible()) {
constructor.setAccessible(true);
MethodHandle methodHandle = getMethodHandle(proxy, method);
return methodHandle.invokeWithArguments(args);
}
private MethodHandle getMethodHandle(Object proxy, Method method) {
return methodHandleCache.computeIfAbsent(method, key -> bindHandle(proxy, method));
}
private MethodHandle bindHandle(Object proxy, Method method) {
try {
return mapperInterfaceLookup.unreflectSpecial(method, method.getDeclaringClass()).bindTo(proxy);
} catch (IllegalAccessException e) {
throw new BaseException(e.getMessage(), e);
}
final Class<?> declaringClass = method.getDeclaringClass();
return constructor
.newInstance(declaringClass, Lookup.PRIVATE | Lookup.PROTECTED | Lookup.PACKAGE | Lookup.PUBLIC)
.unreflectSpecial(method, declaringClass).bindTo(proxy).invokeWithArguments(args);
}
private boolean isBaseMethod(Method method) {
......@@ -64,4 +73,4 @@ public class BaseMethodInvocationHandler<T> implements InvocationHandler, Serial
return baseMapper.invokeBaseMethod(method, args);
}
}
\ No newline at end of file
}
support/schbrain-base-dao/src/main/java/com/schbrain/framework/dao/mybatis/CustomizeMapperFactoryBean.java
View file @ 52f6fe8d
package com.schbrain.framework.dao.mybatis;
import cn.hutool.aop.ProxyUtil;
import com.github.pagehelper.PageInterceptor;
import com.schbrain.framework.dao.BaseDao;
import org.apache.ibatis.plugin.Interceptor;
......@@ -7,7 +8,6 @@ import org.apache.ibatis.session.Configuration;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.mapper.MapperFactoryBean;
import java.lang.reflect.Proxy;
import java.util.List;
import java.util.Properties;
......@@ -18,16 +18,11 @@ import java.util.Properties;
*/
public class CustomizeMapperFactoryBean<T> extends MapperFactoryBean<T> {
public CustomizeMapperFactoryBean() {
super();
}
public CustomizeMapperFactoryBean(Class<T> mapperInterface) {
super(mapperInterface);
}
@Override
@SuppressWarnings("unchecked")
public T getObject() throws Exception {
T originMapperProxy = super.getObject();
Class<T> mapperInterface = getMapperInterface();
......@@ -38,15 +33,15 @@ public class CustomizeMapperFactoryBean<T> extends MapperFactoryBean<T> {
SqlSessionTemplate sqlSession = getSqlSessionTemplate();
Configuration configuration = sqlSession.getConfiguration();
List<Interceptor> interceptorList = configuration.getInterceptors();
boolean hasPageInterceptor = interceptorList.stream().anyMatch(e -> PageInterceptor.class.isAssignableFrom(e.getClass()));
boolean hasPageInterceptor = interceptorList.stream().anyMatch(PageInterceptor.class::isInstance);
if (!hasPageInterceptor) {
PageInterceptor pageInterceptor = new PageInterceptor();
pageInterceptor.setProperties(new Properties());
configuration.addInterceptor(pageInterceptor);
}
// 创建代理
BaseMethodInvocationHandler<T> handler = new BaseMethodInvocationHandler<>(originMapperProxy, sqlSession, mapperInterface);
return (T) Proxy.newProxyInstance(getMapperInterface().getClassLoader(), new Class[]{mapperInterface}, handler);
BaseDaoInvocationHandler<T> handler = new BaseDaoInvocationHandler<>(originMapperProxy, sqlSession, mapperInterface);
return ProxyUtil.newProxyInstance(mapperInterface.getClassLoader(), handler, mapperInterface);
}
}
\ No newline at end of file
}
support/schbrain-base-dao/src/main/java/com/schbrain/framework/dao/mybatis/mapper/BaseMapperStatement.java
View file @ 52f6fe8d
package com.schbrain.framework.dao.mybatis.mapper;
import cn.hutool.core.util.ArrayUtil;
import cn.hutool.crypto.SecureUtil;
import com.google.common.base.CaseFormat;
import com.google.common.base.Converter;
......@@ -7,6 +8,7 @@ import com.schbrain.framework.dao.mybatis.mapper.sqlsource.*;
import org.apache.ibatis.builder.StaticSqlSource;
import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator;
import org.apache.ibatis.mapping.*;
import org.apache.ibatis.mapping.MappedStatement.Builder;
import org.apache.ibatis.scripting.LanguageDriver;
import org.apache.ibatis.session.Configuration;
......@@ -21,30 +23,22 @@ import java.util.*;
*/
public class BaseMapperStatement {
private static final Converter<String, String> CONVERTER = CaseFormat.LOWER_CAMEL.converterTo(CaseFormat.LOWER_UNDERSCORE);
private static final String GMT_UPDATE_FIELD = "gmtUpdate";
private static final String GMT_CREATE_FIELD = "gmtCreate";
private final Configuration configuration;
private final Map<String, String> fieldColumnMap = new LinkedHashMap<>();
private final List<ResultMap> objectResultMapList = new ArrayList<>(1);
private final List<ResultMap> intResultMapList = new ArrayList<>(1);
private final Configuration configuration;
private final Class<?> mapperInterface;
private final LanguageDriver languageDriver;
private final Map<String, String> fieldColumnMap = new LinkedHashMap<>();
private final String tableName;
private final Class<?> domainClass;
private final Field[] fields;
private final List<ResultMap> objectResultMapList = new ArrayList<>(1);
private final List<ResultMap> intResultMapList = new ArrayList<>(1);
private String selectClause;
private String insertClause;
public BaseMapperStatement(Configuration configuration, Class<?> mapperInterface, Class<?> domainClass, String tableName, Field[] fields) {
......@@ -74,8 +68,7 @@ public class BaseMapperStatement {
}
public String getAddMSId() {
String methodName = "add";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".add";
}
public String getAddListMSId(String... fields) {
......@@ -88,72 +81,58 @@ public class BaseMapperStatement {
}
public String getGetByIdMSId() {
String methodName = "getById";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".getById";
}
public String getListByIdListMSId() {
String methodName = "getByIdList";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".getByIdList";
}
public String getListByConditionMSId() {
String methodName = "listByCondition";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".listByCondition";
}
public String getListByObjectMSId() {
String methodName = "listByObject";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".listByObject";
}
public String getGetOneByObjectMSId() {
String methodName = "getOneByObject";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".getOneByObject";
}
public String getCountByConditionMSId() {
String methodName = "getCountByCondition";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".getCountByCondition";
}
public String getDeleteByIdMSId() {
String methodName = "deleteById";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".deleteById";
}
public String getDeleteByIdListMSId() {
String methodName = "deleteByIdList";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".deleteByIdList";
}
public String getDeleteByConditionMSId() {
String methodName = "deleteByCondition";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".deleteByCondition";
}
public String getUpdateByIdMSId() {
String methodName = "updateById";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".updateById";
}
public String getUpdateByIdWithNullMSId() {
String methodName = "updateByIdWithNull";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".updateByIdWithNull";
}
public String getUpdateByConditionMSId() {
String methodName = "updateByCondition";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".updateByCondition";
}
public String getUpdateByCompleteSqlMSId() {
String methodName = "updateByCompleteSql";
return mapperInterface.getName() + "." + methodName;
return mapperInterface.getName() + ".updateByCompleteSql";
}
private void setClause() {
Converter<String, String> converter = CaseFormat.LOWER_CAMEL.converterTo(CaseFormat.LOWER_UNDERSCORE);
StringBuilder selectClause = new StringBuilder("select ");
StringBuilder insertClause = new StringBuilder("insert into ");
insertClause.append(tableName).append("(");
......@@ -162,7 +141,7 @@ public class BaseMapperStatement {
continue;
}
String fieldName = field.getName();
String columnName = converter.convert(fieldName);
String columnName = CONVERTER.convert(fieldName);
fieldColumnMap.put(fieldName, columnName);
selectClause.append(" ").append(columnName).append(",");
insertClause.append(" ").append(columnName).append(",");
......@@ -218,7 +197,7 @@ public class BaseMapperStatement {
private String getInsertListScript(String... fields) {
String tmpInsertClause;
Set<String> fieldSet;
if (null == fields || 0 == fields.length) {
if (ArrayUtil.isEmpty(fields)) {
tmpInsertClause = insertClause;
fieldSet = fieldColumnMap.keySet();
} else {
......@@ -277,9 +256,7 @@ public class BaseMapperStatement {
///////////////////listByIdList
private void addListByIdListMS() {
String sql = "<script>" + selectClause
+ " where id in <foreach close=')' collection='list' item='id' open='(' separator=','>"
+ "#{id}</foreach></script>";
String sql = "<script>" + selectClause + " where id in <foreach open='(' close=')' collection='list' item='id' separator=','>#{id}</foreach></script>";
SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, List.class);
addSelectMappedStatement(getListByIdListMSId(), sqlSource);
}
......@@ -330,9 +307,7 @@ public class BaseMapperStatement {
///////////////////deleteByIdList
private void addDeleteByIdListMS() {
String sql = "<script>delete from " + tableName
+ " where id in <foreach close=')' collection='list' item='id' open='(' separator=','>"
+ "#{id}</foreach></script>";
String sql = "<script>delete from " + tableName + " where id in <foreach open='(' close=')' collection='list' item='id' separator=','>#{id}</foreach></script>";
SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, List.class);
addUpdateMappedStatement(getDeleteByIdListMSId(), sqlSource, SqlCommandType.DELETE);
}
......@@ -408,8 +383,7 @@ public class BaseMapperStatement {
if (configuration.hasStatement(msId)) {
return;
}
MappedStatement ms = new MappedStatement.Builder(configuration, msId, sqlSource, SqlCommandType.SELECT)
.resultMaps(objectResultMapList).flushCacheRequired(true).build();
MappedStatement ms = new Builder(configuration, msId, sqlSource, SqlCommandType.SELECT).resultMaps(objectResultMapList).flushCacheRequired(true).build();
configuration.addMappedStatement(ms);
}
......@@ -417,8 +391,7 @@ public class BaseMapperStatement {
if (configuration.hasStatement(msId)) {
return;
}
MappedStatement ms = new MappedStatement.Builder(configuration, msId, sqlSource, SqlCommandType.SELECT)
.resultMaps(intResultMapList).build();
MappedStatement ms = new Builder(configuration, msId, sqlSource, SqlCommandType.SELECT).resultMaps(intResultMapList).build();
configuration.addMappedStatement(ms);
}
......@@ -428,14 +401,11 @@ public class BaseMapperStatement {
}
MappedStatement ms;
if (SqlCommandType.INSERT == sqlCommandType) {
ms = new MappedStatement.Builder(configuration, msId, sqlSource, sqlCommandType)
.resultMaps(intResultMapList).keyGenerator(Jdbc3KeyGenerator.INSTANCE)
.keyColumn("id").keyProperty("id").build();
ms = new Builder(configuration, msId, sqlSource, sqlCommandType).resultMaps(intResultMapList).keyGenerator(Jdbc3KeyGenerator.INSTANCE).keyColumn("id").keyProperty("id").build();
} else {
ms = new MappedStatement.Builder(configuration, msId, sqlSource, sqlCommandType)
.resultMaps(intResultMapList).build();
ms = new Builder(configuration, msId, sqlSource, sqlCommandType).resultMaps(intResultMapList).build();
}
configuration.addMappedStatement(ms);
}
}
\ No newline at end of file
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment