文章目录
1.创建通用接口
在通用接口中使用@InsertProvider,@UpdateProvider,@DeleteProvider,@SelectProvider注解配置增删改查的类和方法,并添加通用方法
public interface BaseMapper<T, E> {
@InsertProvider(type = BaseMapperProvider.class, method = "insert")
int insert(T entity);
@DeleteProvider(type = BaseMapperProvider.class, method = "deleteByPrimary")
int deleteByPrimary(E id);
@DeleteProvider(type = BaseMapperProvider.class, method = "delete")
int delete(@Param("wrapper") Wrapper wrapper);
@UpdateProvider(type = BaseMapperProvider.class, method = "updateByPrimary")
int updateByPrimary(T entity);
@DeleteProvider(type = BaseMapperProvider.class, method = "update")
int update(@Param("entity") T entity, @Param("wrapper") Wrapper wrapper);
@SelectProvider(type = BaseMapperProvider.class, method = "selectByPrimary")
T selectByPrimary(@Param("id") E id);
@SelectProvider(type = BaseMapperProvider.class, method = "select")
T selectOne(@Param("wrapper") Wrapper wrapper);
@SelectProvider(type = BaseMapperProvider.class, method = "select")
List<T> select();
@SelectProvider(type = BaseMapperProvider.class, method = "select")
List<T> selectList(@Param("wrapper") Wrapper wrapper);
}
2.自定义注解
通过注解配置表的一些信息
1.@Table 配置表
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface Table {
/**
* 配置表名
*
* @return 表名
*/
String value() default "";
}
2.@Column 配置字段
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Column {
/**
* 配置字段名称
*
* @return 字段名称
*/
String value() default "";
/**
* 配置字段排序
*
* @return 字段排序
*/
boolean order() default false;
/**
* 配置字段排序类型
*
* @return 段排序类型
*/
String orderType() default "desc";
}
3.@JoinColumn 配置关联表
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface JoinColumn {
/**
* 配置要关联的表
*
* @return 表名
*/
String tableName();
/**
* 配置字段名称
*
* @return 字段名称
*/
String column() default "";
/**
* 配置两表之间的关系
*
* @return 两表之间的关系
*/
Association[] relations();
}
4.@Association 配置两表之间的关系
public @interface Association {
/**
* 配置当前表字段名称
*
* @return 当前表字段名称
*/
String target();
/**
* 配置关联表字段名称
*
* @return 关联表字段名称
*/
String association();
}
5.@Primary 配置主键
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Primary {
/**
* 配置主键
*
* @return 是否是主键
*/
boolean value() default true;
}
3.编写BaseMapperProvider通过注解动态拼接SQL
这个类的方法可以加入ProviderContext类型的参数,可以通过 ProviderContext 来获取mapper的信息,通过泛型获取实体,通过注解动态生成SQL
public class BaseMapperProvider {
/**
* 用来缓存已经生成的SQL
*/
private static final Map<String, SQLBuilder> CONTEXT = new ConcurrentHashMap<>();
public String insert(ProviderContext context) {
return getSqlBuilder(context).insert().build();
}
public String deleteByPrimary(Object o, ProviderContext context) {
return getSqlBuilder(context).select().byPrimary().build();
}
public String delete(Wrapper wrapper, ProviderContext context) {
return getSqlBuilder(context).delete().where(wrapper).build();
}
public String updateByPrimary(Object o, ProviderContext context) {
return getSqlBuilder(context).update().byPrimary().build();
}
public String update(Object entity, Wrapper wrapper, ProviderContext context) {
return getSqlBuilder(context).update().where(wrapper).build();
}
public String select(Wrapper wrapper, ProviderContext context) {
return getSqlBuilder(context).select().where(wrapper).build();
}
public String selectByPrimary(Object o, ProviderContext context) {
return getSqlBuilder(context).select().byPrimary().build();
}
private Class getEntityType(ProviderContext context) {
return (Class) ((ParameterizedType) (context.getMapperType().getGenericInterfaces()[0])).getActualTypeArguments()[0];
}
private SQLBuilder getSqlBuilder(ProviderContext context) {
String name = getEntityType(context).getName();
SQLBuilder sqlBuilder = CONTEXT.get(name);
if (sqlBuilder == null) {
sqlBuilder = new SQLBuilder(getEntityType(context));
CONTEXT.put(name, sqlBuilder);
}
return sqlBuilder.reset();
}
}
4.动态SQL类编写
public class SQLBuilder {
private final TableInfo tableInfo;
private final Class<?> type;
private SQL sql;
public SQLBuilder(Class<?> type) {
this.tableInfo = new TableInfo(type);
this.type = type;
}
public SQLBuilder reset() {
sql = new SQL();
return this;
}
public SQLBuilder insert() {
List<ColumnInfo> columns = tableInfo.getColumns();
sql.INSERT_INTO(getTableName());
StringBuilder value = new StringBuilder();
StringBuilder values = new StringBuilder();
for (int i = 0; i < columns.size(); i++) {
ColumnInfo columnInfo = columns.get(i);
String columnName = columnInfo.getColumnName();
String fieldName = columnInfo.getFieldName();
String separator = ",";
if (i + 1 == columns.size()) {
separator = "";
}
value.append(ifScript(fieldName, columnName, separator));
values.append(ifScript(fieldName, sharp(fieldName), separator));
}
sql.VALUES(getPrimaryColumnName(), sharp(getPrimaryFieldName()));
sql.VALUES(value.toString(), values.toString());
return this;
}
public SQLBuilder delete() {
sql.DELETE_FROM(getTableName());
return this;
}
public SQLBuilder update() {
List<ColumnInfo> columns = tableInfo.getColumns();
sql.UPDATE(getTableName());
StringBuilder set = new StringBuilder();
for (int i = 0; i < columns.size(); i++) {
ColumnInfo columnInfo = columns.get(i);
String fieldName = columnInfo.getFieldName();
String columnName = columnInfo.getColumnName();
String separator = ",";
if (i + 1 == columns.size()) {
separator = "";
}
set.append(ifScript(fieldName, set(columnName, fieldName), separator));
}
sql.SET(set.toString());
return this;
}
public SQLBuilder select() {
String tableName = getTableName();
sql.SELECT(select(tableName, getPrimaryColumnName(), getPrimaryFieldName()));
for (ColumnInfo columnInfo : tableInfo.getColumns()) {
sql.SELECT(select(tableName, columnInfo.getColumnName(), columnInfo.getFieldName()));
}
sql.FROM(tableName);
for (JoinColumnInfo joinColumnInfo : tableInfo.getJoinColumns()) {
String joinTableName = joinColumnInfo.getTableName();
sql.FROM(joinTableName);
sql.SELECT(select(joinTableName, joinColumnInfo.getColumnName(), joinColumnInfo.getFieldName()));
List<AssociationInfo> associations = joinColumnInfo.getAssociations();
for (AssociationInfo associationInfo : associations) {
sql.WHERE(condition(tableName, associationInfo.getTarget(), joinTableName, associationInfo.getAssociation()));
}
}
return this;
}
public SQLBuilder byPrimary() {
sql.WHERE(condition(getTableName(), getPrimaryColumnName(), "id"));
return this;
}
public SQLBuilder where(Wrapper wrapper) {
if (wrapper != null) {
List<Condition> conditions = wrapper.getConditions();
for (Condition condition : conditions) {
if (condition.isAnd()) {
sql.AND();
}
if (!condition.isAnd()) {
sql.OR();
}
sql.WHERE(condition.render());
}
}
return this;
}
public TableInfo getTableInfo() {
return tableInfo;
}
public Class<?> getType() {
return type;
}
private String getTableName() {
return tableInfo.getTableName();
}
private PrimaryInfo getPrimaryInfo() {
return tableInfo.getPrimaryInfo();
}
private String getPrimaryColumnName() {
return getPrimaryInfo().getColumnName();
}
private String getPrimaryFieldName() {
return getPrimaryInfo().getFieldName();
}
private String select(String tableName, String columnName, String fieldName) {
return String.format("%s.%s \"%s\"", tableName, columnName, fieldName);
}
private String sharp(String fieldName) {
return String.format("#{%s}", fieldName);
}
private String set(String columnName, String fieldName) {
return String.format("%s = #{%s}", columnName, fieldName);
}
private String ifScript(String fieldName, String content, String separator) {
return String.format("<if test=\"%s != null and %s != ''\">%s%s</if>", fieldName, fieldName, content, separator);
}
private String condition(String tableName, String columnName, String fieldName) {
return String.format("%s.%s = #{%s}", tableName, columnName, fieldName);
}
private String condition(String tableName, String left, String joinTableName, String right) {
return String.format("%s.%s = %s.%s", tableName, left, joinTableName, right);
}
/**
* <script> MyBatis Java动态SQL不能直接用OGNL,需要加入<script>标签
*
* @return SQL
*/
public String build() {
return "<script>" + sql.toString() + "</script>";
}
}
有几个对应注解的实体我就不粘了,这些代码只是给大家提供些思路,这种方式是完全支持xml写法的,具体实现可以看看这个项目
common-mapper MyBatis通用Mapper