fix: @ColumnMask,typeHandler 一起使用时冲突的问题 close #I8EXFW

This commit is contained in:
开源海哥 2023-11-23 10:40:54 +08:00
parent 41aaeb69cb
commit 9bc74d7e4e
2 changed files with 86 additions and 15 deletions

View File

@ -0,0 +1,55 @@
/*
* Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
* <p>
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.mybatisflex.core.mask;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.TypeHandler;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
public class CompositeMaskTypeHandler implements TypeHandler<Object> {
private final String maskType;
private final TypeHandler<Object> typeHandler;
public CompositeMaskTypeHandler(String maskType, TypeHandler<Object> typeHandler) {
this.maskType = maskType;
this.typeHandler = typeHandler;
}
@Override
public void setParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
typeHandler.setParameter(ps, i, parameter, jdbcType);
}
@Override
public Object getResult(ResultSet rs, String columnName) throws SQLException {
return MaskManager.mask(maskType, typeHandler.getResult(rs, columnName));
}
@Override
public Object getResult(ResultSet rs, int columnIndex) throws SQLException {
return MaskManager.mask(maskType, typeHandler.getResult(rs, columnIndex));
}
@Override
public Object getResult(CallableStatement cs, int columnIndex) throws SQLException {
return MaskManager.mask(maskType, typeHandler.getResult(cs, columnIndex));
}
}

View File

@ -16,6 +16,7 @@
package com.mybatisflex.core.table;
import com.mybatisflex.core.FlexGlobalConfig;
import com.mybatisflex.core.mask.CompositeMaskTypeHandler;
import com.mybatisflex.core.mask.MaskTypeHandler;
import com.mybatisflex.core.util.StringUtil;
import org.apache.ibatis.session.Configuration;
@ -54,6 +55,11 @@ public class ColumnInfo {
*/
protected TypeHandler typeHandler;
/**
* 最终使用和构建出来的 typeHandler
*/
protected TypeHandler buildTypeHandler;
/**
* 数据脱敏类型
*/
@ -107,23 +113,33 @@ public class ColumnInfo {
public TypeHandler buildTypeHandler(Configuration configuration) {
//优先使用自定义的 typeHandler
if (typeHandler != null) {
return typeHandler;
if (buildTypeHandler != null) {
return buildTypeHandler;
}
//脱敏规则配置
else if (StringUtil.isNotBlank(maskType)) {
if (typeHandler != null) {
buildTypeHandler = new CompositeMaskTypeHandler(maskType, typeHandler);
} else {
buildTypeHandler = new MaskTypeHandler(maskType);
}
}
//用户自定义的 typeHandler
else if (typeHandler != null) {
buildTypeHandler = typeHandler;
}
//枚举
else if (propertyType.isEnum()) {
if (configuration == null) {
configuration = FlexGlobalConfig.getDefaultConfig().getConfiguration();
}
this.typeHandler = configuration.getTypeHandlerRegistry().getTypeHandler(propertyType);
}
//若用户未定义 typeHandler而配置了数据脱敏则使用脱敏的 handler 处理
else if (StringUtil.isNotBlank(maskType)) {
typeHandler = new MaskTypeHandler(maskType);
buildTypeHandler = configuration.getTypeHandlerRegistry().getTypeHandler(propertyType);
}
return typeHandler;
return buildTypeHandler;
}
public void setTypeHandler(TypeHandler typeHandler) {