Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.jeesuite.mybatis.codegen.CrudSupportPlugin Maven / Gradle / Ivy
/**
*
*/
package com.jeesuite.mybatis.codegen;
import java.util.HashSet;
import java.util.List;
import java.util.Properties;
import java.util.Set;
import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.api.dom.java.Interface;
import org.mybatis.generator.api.dom.java.Method;
import org.mybatis.generator.api.dom.java.TopLevelClass;
import org.mybatis.generator.api.dom.xml.XmlElement;
import org.mybatis.generator.config.CommentGeneratorConfiguration;
import org.mybatis.generator.config.Context;
import org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl;
import org.mybatis.generator.internal.util.StringUtility;
public class CrudSupportPlugin extends PluginAdapter {
private Set mappers = new HashSet();
private boolean caseSensitive = false;
//开始的分隔符，例如mysql为`，sqlserver为[
private String beginningDelimiter = "";
//结束的分隔符，例如mysql为`，sqlserver为]
private String endingDelimiter = "";
//数据库模式
private String schema;
//注释生成器
private CommentGeneratorConfiguration commentCfg;
private JavaTypeResolverDefaultImpl javaTypeResolver = new JavaTypeResolverDefaultImpl();
@Override
public void setContext(Context context) {
super.setContext(context);
//设置默认的注释生成器
commentCfg = new CommentGeneratorConfiguration();
commentCfg.setConfigurationType(MapperCommentGenerator.class.getCanonicalName());
context.setCommentGeneratorConfiguration(commentCfg);
//支持oracle获取注释#114
context.getJdbcConnectionConfiguration().addProperty("remarksReporting", "true");
}
@Override
public void setProperties(Properties properties) {
super.setProperties(properties);
String mappers = this.properties.getProperty("mappers");
if (StringUtility.stringHasValue(mappers)) {
for (String mapper : mappers.split(",")) {
this.mappers.add(mapper);
}
} else {
throw new RuntimeException("Mapper插件缺少必要的mappers属性!");
}
String caseSensitive = this.properties.getProperty("caseSensitive");
if (StringUtility.stringHasValue(caseSensitive)) {
this.caseSensitive = caseSensitive.equalsIgnoreCase("TRUE");
}
String beginningDelimiter = this.properties.getProperty("beginningDelimiter");
if (StringUtility.stringHasValue(beginningDelimiter)) {
this.beginningDelimiter = beginningDelimiter;
}
commentCfg.addProperty("beginningDelimiter", this.beginningDelimiter);
String endingDelimiter = this.properties.getProperty("endingDelimiter");
if (StringUtility.stringHasValue(endingDelimiter)) {
this.endingDelimiter = endingDelimiter;
}
commentCfg.addProperty("endingDelimiter", this.endingDelimiter);
String schema = this.properties.getProperty("schema");
if (StringUtility.stringHasValue(schema)) {
this.schema = schema;
}
}
public String getDelimiterName(String name) {
StringBuilder nameBuilder = new StringBuilder();
if (StringUtility.stringHasValue(schema)) {
nameBuilder.append(schema);
nameBuilder.append(".");
}
nameBuilder.append(beginningDelimiter);
nameBuilder.append(name);
nameBuilder.append(endingDelimiter);
return nameBuilder.toString();
}
@Override
public boolean validate(List warnings) {
return true;
}
/**
* 生成的Mapper接口
*
* @param interfaze
* @param topLevelClass
* @param introspectedTable
* @return
*/
@Override
public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
//获取实体类
FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
//主键
FullyQualifiedJavaType idType = null;
List columns = introspectedTable.getBaseColumns();
for (IntrospectedColumn col : columns) {
if(!col.isIdentity())continue;
idType = javaTypeResolver.calculateJavaType(col);
break;
}
if(idType == null){
for (IntrospectedColumn col : columns) {
if("id".equalsIgnoreCase(col.getActualColumnName())){
idType = javaTypeResolver.calculateJavaType(col);
break;
}
}
}
//import接口
for (String mapper : mappers) {
interfaze.addImportedType(new FullyQualifiedJavaType(mapper));
String idJavaType = idType != null ? idType.getShortName() : "String";
interfaze.addSuperInterface(new FullyQualifiedJavaType(mapper + "<" + entityType.getShortName() + ","+idJavaType+">"));
}
//import实体类
interfaze.addImportedType(entityType);
return true;
}
/**
* 处理实体类的包和@Table注解
*
* @param topLevelClass
* @param introspectedTable
*/
private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
//引入JPA注解
topLevelClass.addImportedType("javax.persistence.*");
String tableName = introspectedTable.getFullyQualifiedTableNameAtRuntime();
//如果包含空格，或者需要分隔符，需要完善
if (StringUtility.stringContainsSpace(tableName)) {
tableName = context.getBeginningDelimiter()
+ tableName
+ context.getEndingDelimiter();
}
//是否忽略大小写，对于区分大小写的数据库，会有用
if (caseSensitive && !topLevelClass.getType().getShortName().equals(tableName)) {
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
} else if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) {
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
} else if (StringUtility.stringHasValue(schema)
|| StringUtility.stringHasValue(beginningDelimiter)
|| StringUtility.stringHasValue(endingDelimiter)) {
topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
}
}
/**
* 生成基础实体类
*
* @param topLevelClass
* @param introspectedTable
* @return
*/
@Override
public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
processEntityClass(topLevelClass, introspectedTable);
return true;
}
/**
* 生成实体类注解KEY对象
*
* @param topLevelClass
* @param introspectedTable
* @return
*/
@Override
public boolean modelPrimaryKeyClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
processEntityClass(topLevelClass, introspectedTable);
return true;
}
/**
* 生成带BLOB字段的对象
*
* @param topLevelClass
* @param introspectedTable
* @return
*/
@Override
public boolean modelRecordWithBLOBsClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
processEntityClass(topLevelClass, introspectedTable);
return false;
}
//下面所有return false的方法都不生成。这些都是基础的CRUD方法，使用通用Mapper实现
@Override
public boolean clientDeleteByPrimaryKeyMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientInsertMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientInsertSelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientSelectByPrimaryKeyMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeySelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeyWithoutBLOBsMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientDeleteByPrimaryKeyMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientInsertMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientInsertSelectiveMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientSelectAllMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientSelectAllMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientSelectByPrimaryKeyMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeySelectiveMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean clientUpdateByPrimaryKeyWithoutBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapDeleteByPrimaryKeyElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapInsertElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapInsertSelectiveElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapSelectAllElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapSelectByPrimaryKeyElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapUpdateByPrimaryKeySelectiveElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapUpdateByPrimaryKeyWithBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean sqlMapUpdateByPrimaryKeyWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean providerGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean providerApplyWhereMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean providerInsertSelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
@Override
public boolean providerUpdateByPrimaryKeySelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
return false;
}
}
