1.使用的是maven項目，添加依賴
<!-- mybatis-plus begin --> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus</artifactId> <version>2.2.0</version> </dependency>
還有數據庫的連接
<dependency> <groupId>mysql</groupId> <artifactId>mysql-connector-java</artifactId> <scope>compile</scope> </dependency>
最後是源碼
import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.config.DataSourceConfig; import com.baomidou.mybatisplus.generator.config.GlobalConfig; import com.baomidou.mybatisplus.generator.config.PackageConfig; import com.baomidou.mybatisplus.generator.config.StrategyConfig; import com.baomidou.mybatisplus.generator.config.rules.DbType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import org.junit.jupiter.api.Test; /** * mybatis-plus 自動生成代碼* * @author Terry * @version 1.0 * @date 2018-05-16 09:35 */ public class SimpleMp { @Test public void generateCode() { //指定包名String packageName = "com.hciot.hhhh"; //user -> UserService, 設置成true: user -> IUserService boolean serviceNameStartWithI = false; //指定生成的表名String[] tableNames = new String[]{"data_air_sensor_co", "order_product", "relation_device_gateway"}; generateByTables(serviceNameStartWithI, packageName, tableNames); } /** * 根據表自動生成* * @param serviceNameStartWithI 默認為false * @param packageName 包名* @param tableNames 表名* @author Terry */ private void generateByTables(boolean serviceNameStartWithI, String packageName, String... tableNames) { //配置數據源DataSourceConfig dataSourceConfig = getDataSourceConfig(); // 策略配置StrategyConfig strategyConfig = getStrategyConfig(tableNames); //全局變量配置GlobalConfig globalConfig = getGlobalConfig(serviceNameStartWithI); //包名配置PackageConfig packageConfig = getPackageConfig(packageName); //自動生成atuoGenerator(dataSourceConfig, strategyConfig, globalConfig, packageConfig); } /** * 集成* * @param dataSourceConfig 配置數據源* @param strategyConfig 策略配置* @param config 全局變量配置* @param packageConfig 包名配置* @author Terry */ private void atuoGenerator(DataSourceConfig dataSourceConfig, StrategyConfig strategyConfig, GlobalConfig config, PackageConfig packageConfig) { new AutoGenerator() .setGlobalConfig(config) .setDataSource(dataSourceConfig) .setStrategy(strategyConfig) .setPackageInfo(packageConfig) .execute(); } /** * 設置包名* * @param packageName 父路徑包名* @return PackageConfig 包名配置* @author Terry */ private PackageConfig getPackageConfig(String packageName) { return new PackageConfig() .setParent(packageName) .setXml("mapper") .setMapper("dao") .setController("controller") .setEntity("entity"); } /** * 全局配置* * @param serviceNameStartWithI false * @return GlobalConfig * @author Terry */ private GlobalConfig getGlobalConfig(boolean serviceNameStartWithI) { GlobalConfig globalConfig = new GlobalConfig(); globalConfig .setBaseColumnList(true) .setBaseResultMap(true) .setActiveRecord(false) .setAuthor("Terry") //設置輸出路徑.setOutputDir(getOutputDir("mybatis-plus")) .setFileOverride(true); if (!serviceNameStartWithI) { //設置service名globalConfig.setServiceName("%sService"); } return globalConfig; } /** * 返回項目路徑* * @param projectName 項目名* @return 項目路徑* @author Terry */ private String getOutputDir(String projectName) { String path = this.getClass().getClassLoader().getResource("").getPath(); int index = path.indexOf(projectName); return path.substring(1, index) + projectName + "/src/main/java/"; } /** * 策略配置* * @param tableNames 表名* @return StrategyConfig * @author Terry */ private StrategyConfig getStrategyConfig(String... tableNames) { return new StrategyConfig() // 全局大寫命名ORACLE 注意.setCapitalMode(true) .setEntityLombokModel(false) // 表名、字段名、是否使用下劃線命名（默認false） .setDbColumnUnderline(true) //從數據庫表到文件的命名策略.setNaming(NamingStrategy.underline_to_camel) //需要生成的的表名，多個表名傳數組.setInclude(tableNames); } /** * 配置數據源* * @return 數據源配置DataSourceConfig * @author Terry */ private DataSourceConfig getDataSourceConfig() { String dbUrl = "jdbc:mysql://localhost:3306/test"; return new DataSourceConfig().setDbType(DbType.MYSQL) .setUrl(dbUrl) .setUsername("root") .setPassword("root") .setDriverName("com.mysql.jdbc.Driver"); } /** * 根據表自動生成* * @param packageName 包名* @param tableNames 表名* @author Terry */ @SuppressWarnings("unused") private void generateByTables(String packageName, String... tableNames) { generateByTables(true, packageName, tableNames); } }總結
以上所述是小編給大家介紹的基於Mybatis plus 自動代碼生成器的實現代碼，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對武林網網站的支持！