数据库逆向生成实体类和映射文件

最新推荐文章于 2024-07-07 16:15:55 发布

Gemini_Kanon

最新推荐文章于 2024-07-07 16:15:55 发布

阅读量1.2k

点赞数 1

分类专栏： Java

本文链接：https://blog.csdn.net/gemini_kanon/article/details/80324544

版权

Java 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

首先需要几个jar包：freemarker-2.3.23.jar

log4j-1.2.16.jar

mybatis-3.2.3.jar

mybatis-generator-core-1.3.2.jar

mysql-connector-java-5.1.28-bin.jar

ojdbc14.jar

这些jar包网上都有下载的地方，小编是直接拿过来用的，链接就麻烦大家手动下载了~~~

xml配置文件：generatorConfig.xml，代码如下

<?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>
   <context id="testTables" targetRuntime="MyBatis3">
       <commentGenerator>
           
           <property name="suppressAllComments" value="true" />
       </commentGenerator>
       
       <jdbcConnection driverClass="oracle.jdbc.driver.OracleDriver"
           connectionURL="jdbc:oracle:thin:@localhost:ORACLE端口一般为1521:你的数据库名称" userId="自己的ID"
           password="自己的密码">
       </jdbcConnection>
       连接MySQL
       连接Oracle
       
       <javaTypeResolver>
           <property name="forceBigDecimals" value="false" />
       </javaTypeResolver>

       
       <javaModelGenerator targetPackage="XXX.pojo"
           targetProject=".\src">
           
           <property name="enableSubPackages" value="false" />
           
           <property name="trimStrings" value="true" />
       </javaModelGenerator>
        
       <sqlMapGenerator targetPackage="XXX.mapper"
           targetProject=".\src">
           
           <property name="enableSubPackages" value="false" />
       </sqlMapGenerator>
       
       <javaClientGenerator type="XMLMAPPER"
           targetPackage="XXX.mapper"
           targetProject=".\src">
           
           <property name="enableSubPackages" value="false" />
       </javaClientGenerator>
       
       <table schema="" tableName="你的数据库表名"></table>此地方可以写多个表名，一起生成文件

此处更新一段代码：如此生成java代码后会有个问题，数据库里number型字段会被生成bigdecimal型，
其他也有可能会改变，暂时还没遇到，但是可以有规避方法~
此处是配置你数据库想要的对应的表名：
<table schema="" tableName="NDJFMXK">
    <!-- 此处可以配置上你想要生成的对应的java代码类型，以number类型举例，这样生成的java实体类该列的类型就是integer了 -->
    <columnOverride column="列名" javaType="java.lang.Integer"></columnOverride>
</table>

</context>
</generatorConfiguration>

运行Java文件，代码如下

一、FreemarkerUtil.java

public class FreemarkerUtil {

   public static String ControllerPackage = "XXX.controller";

   public static String ServicePackage = "XXX.service";

   public static List<String> TableNames = new ArrayList<String>();

   private static FreemarkerUtil fu = new FreemarkerUtil();
   public static void main(String[] args) {

       //添加表名
       TableNames.add("你自己的表名");//lzcwbh


       fu.CreateController();
       fu.CreateService();
   }

   //生成Controller
   public void CreateController(){

       for (String tablename : TableNames) {
           String[] split = tablename.toLowerCase().split("_");
           tablename = "";
           for (int i = 0; i < split.length; i++) {
               tablename += split[i].substring(0, 1).toUpperCase() + split[i].substring(1);
           }
           //1、创建数据模型
            Map<String,Object> root = new HashMap<String,Object>();
            //2、为数据模型添加值
            root.put("TableName", tablename);
            root.put("Package",ControllerPackage);
            root.put("ServicePackage", ServicePackage);
            //3、将数据模型和模板组合的数据输出到控制台
            //fu.print("controller.ftl", root);
            //生成模板
            fu.fprint("controller.ftl", root, tablename+"Controller.java");
       }
   }

       //生成Controller
       public void CreateService(){

           for (String tablename : TableNames) {
               String[] split = tablename.toLowerCase().split("_");
               tablename = "";
               for (int i = 0; i < split.length; i++) {
                   tablename += split[i].substring(0, 1).toUpperCase() + split[i].substring(1);
               }
               //1、创建数据模型
                Map<String,Object> root = new HashMap<String,Object>();
                //2、为数据模型添加值
                root.put("TableName", tablename);
                root.put("Package",ServicePackage);
                //3、将数据模型和模板组合的数据输出到控制台
                //fu.print("controller.ftl", root);
                //生成模板
                fu.fprint("service.ftl", root, tablename+"Service.java");
                root.put("Package",ServicePackage+".Impl");
                fu.fprint("serviceimpl.ftl", root, tablename+"ServiceImpl.java");
           }
       }

   public Template getTemplate(String name) {
       try {
           // 通过Freemaker的Configuration读取相应的ftl
           Configuration cfg = new Configuration();
           // 设定去哪里读取相应的ftl模板文件
           cfg.setClassForTemplateLoading(this.getClass(), "/ftl");
           // 在模板文件目录中找到名称为name的文件
           Template temp = cfg.getTemplate(name);
           return temp;
       } catch (IOException e) {
           e.printStackTrace();
       }
       return null;
   }

   public void print(String name, Map<String, Object> root) {
       try {
           // 通过Template可以将模板文件输出到相应的流
           Template temp = this.getTemplate(name);
           temp.process(root, new PrintWriter(System.out));
       } catch (TemplateException e) {
           e.printStackTrace();
       } catch (IOException e) {
           e.printStackTrace();
       }
   }

   public void fprint(String name, Map<String, Object> root, String outFile) {
       FileWriter out = null;
       try {
           String replace = root.get("Package").toString().replace(".", "/");
           String property = System.getProperty("user.dir").replace("\\", "/")+"/src/"+replace;
           File file = new File(property);
           if(!file.exists()){
               file.mkdirs();
           }
           // 通过一个文件输出流，就可以写到相应的文件中
           out = new FileWriter(new File(property, outFile));
           Template temp = this.getTemplate(name);
           temp.process(root, out);
       } catch (IOException e) {
           e.printStackTrace();
       } catch (TemplateException e) {
           e.printStackTrace();
       } finally {
           try {
               if (out != null)
                   out.close();
           } catch (IOException e) {
               e.printStackTrace();
           }
       }
   }
}

二、GeneratorSqlmap.java

public class GeneratorSqlmap {

   public void generator() throws Exception{

       List<String> warnings = new ArrayList<String>();
       boolean overwrite = true;
       //指定逆向工程配置文件
       File configFile = new File("generatorConfig.xml");
       ConfigurationParser cp = new ConfigurationParser(warnings);
       Configuration config = cp.parseConfiguration(configFile);
       DefaultShellCallback callback = new DefaultShellCallback(overwrite);
       MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config,
               callback, warnings);
       myBatisGenerator.generate(null);

   }
   public static void main(String[] args) throws Exception {
       try {
           GeneratorSqlmap generatorSqlmap = new GeneratorSqlmap();
           generatorSqlmap.generator();
       } catch (Exception e) {
           e.printStackTrace();
       }
   }
}