前言
MyBatis官方提供了MyBatis Generator插件用來生成model、mapper和xml怯屉，但是如果在生成代碼的時(shí)候想加入一些自定義的內(nèi)容蔚舀，也就是plugin，就要把自定義的內(nèi)容打成包锨络，再把配置放到MyBatis Generator插件里才行赌躺，這樣相當(dāng)麻煩。于是采用JAVA執(zhí)行MyBatis Generator的方法就不需要再打成包了羡儿。
開始配置
generatorConfig.xml
JSR303Plugin是自定義的內(nèi)容
<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE generatorConfiguration PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN" "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd" >
<generatorConfiguration>
<!-- 引入配置文件 -->
<properties resource="generatorConfig.properties"/>
<!-- 數(shù)據(jù)庫驅(qū)動(dòng) -->
<!--classPathEntry location="${classPath}"/-->
<!-- 數(shù)據(jù)庫驅(qū)動(dòng) -->
<classPathEntry location="${db.oracle.kf.classPath}"/>
<!-- oralce -->
<context id="oracle" targetRuntime="MyBatis3" defaultModelType="flat">
<property name="javaFileEncoding" value="UTF-8"/>
<!-- This plugin is only valid for MyBatis3 target runtime -->
<plugin type="org.mybatis.generator.plugins.RowBoundsPlugin" />
<plugin type="org.mybatis.generator.plugins.ToStringPlugin"/>
<plugin type="com.asiainfo.aigov.fdapp.common.JSR303Plugin" />
<!-- 數(shù)據(jù)庫連接 -->
<jdbcConnection
driverClass="${db.oracle.kf.driverClass}"
connectionURL="${db.oracle.kf.jdbcUrl}"
userId="${db.oracle.kf.user}"
password="${db.oracle.kf.password}">
<!-- 針對(duì)oracle數(shù)據(jù)庫 -->
<property name="remarksReporting" value="true"></property>
<!-- 針對(duì)mysql數(shù)據(jù)庫 -->
<property name="useInformationSchema" value="true"></property>
</jdbcConnection>
<!-- model -->
<javaModelGenerator
targetPackage="${targetPackage}.model"
targetProject="${targetProject}" />
<!-- mapper.xml -->
<sqlMapGenerator
targetPackage="${targetPackage}.dao.mapper.oracle"
targetProject="${targetProject}" />
<!-- dao或mapper接口 -->
<javaClientGenerator
targetPackage="${targetPackage}.dao"
targetProject="${targetProject}"
type="XMLMAPPER" />
<!-- 表配置 -->
<table tableName="FD_ATTENTION" domainObjectName="" />
</context>
</generatorConfiguration>
generatorConfig.properties
#oralce
db.oracle.kf.driverClass=oracle.jdbc.driver.OracleDriver
db.oracle.kf.jdbcUrl=數(shù)據(jù)庫鏈接
db.oracle.kf.user=用戶名
db.oracle.kf.password=密碼
db.oracle.kf.classPath=/Users/pany/.m2/repository/com/oracle/ojdbc14/10.2.0.4.0/ojdbc14-10.2.0.4.0.jar
#這里要用全路徑
targetProject=/Users/pany/Documents/workspace/aigov/familydoctor-webapp/src/main/java
targetPackage=com.asiainfo.aigov.familydoctor
JSR303Plugin
package com.asiainfo.aigov.fdapp.common;
import java.util.List;
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.Field;
import org.mybatis.generator.api.dom.java.TopLevelClass;
/**
* 功能描述：jsr303 代碼生成插件
*/
public class JSR303Plugin extends PluginAdapter {
@Override
public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass,
IntrospectedTable introspectedTable) {
// 增加類注解
topLevelClass.addImportedType("io.swagger.annotations.ApiModel");
topLevelClass.addAnnotation("@ApiModel");
return super.modelBaseRecordClassGenerated(topLevelClass, introspectedTable);
}
@Override
public boolean modelFieldGenerated(Field field,
TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn,
IntrospectedTable introspectedTable, ModelClassType modelClassType) {
if (false == introspectedColumn.isNullable()) {
if (false == introspectedColumn.isIdentity()) {
boolean isPrimarkKey = false;//是否為主鍵
for(IntrospectedColumn column : introspectedTable.getPrimaryKeyColumns()) {
if(introspectedColumn==column) {
isPrimarkKey = true;
}
}
if(!isPrimarkKey) {
topLevelClass.addImportedType("javax.validation.constraints.NotNull");
field.addAnnotation("@NotNull");
}
}
}
if (introspectedColumn.isStringColumn()) {
topLevelClass.addImportedType("io.swagger.annotations.ApiModelProperty");
field.addAnnotation("@ApiModelProperty(\"" + introspectedColumn.getRemarks() + "\")");
topLevelClass.addImportedType("javax.validation.constraints.Size");
field.addAnnotation("@Size(min = 0, max = " + introspectedColumn.getLength() + " , message = \"長度必須在{min}和{max}之間\")");
} else if ("DECIMAL".equals(introspectedColumn.getJdbcTypeName())) {
topLevelClass.addImportedType("io.swagger.annotations.ApiModelProperty");
field.addAnnotation("@ApiModelProperty(value=\"" + introspectedColumn.getRemarks() + "\", example=\"0\")");
}
return super.modelFieldGenerated(field, topLevelClass, introspectedColumn,
introspectedTable, modelClassType);
}
@Override
public boolean validate(List<String> warnings) {
return true;
}
}
MyBatisGenTest
右鍵執(zhí)行單元測試即可
package com.asiainfo.aigov;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.api.ProgressCallback;
import org.mybatis.generator.api.VerboseProgressCallback;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
@RunWith(SpringRunner.class)
@SpringBootTest
public class MyBatisGenTest {
@Test
public void gen() {
//讀取文件
File configFile = new File(this.getClass().getClassLoader().getResource("generatorConfig.xml").getFile());
List<String> warnings = new ArrayList<String>();
ConfigurationParser cp = new ConfigurationParser(warnings);
//true:覆蓋生成
DefaultShellCallback callback = new DefaultShellCallback(true);
try {
Configuration config = cp.parseConfiguration(configFile);
MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
ProgressCallback progressCallback = new VerboseProgressCallback();
myBatisGenerator.generate(progressCallback);
System.out.println("代碼成功生成!");
} catch (Exception e) {
e.printStackTrace();
}
}
}
結(jié)后語
略
