快速生成实体类介绍

快速生成实体类介绍

快速生成实体类有多种方式，这次主要介绍两种：

    1、idea 从数据库快速生成Spring Data JPA实体类。

    2、通过Mybatis generator工具生成实体类。

• idea 从数据库快速生成Spring Data JPA实体类

 方法一  通过groovy模板快速生成

     配置数据库

    view -> Tool Windows -> Database

    + -> Data source -> Oracle(其他数据库同理)

配置数据库信息

建立存放实体的包

导入必要maven依赖

  如果是oracle数据库，添加oracle依赖

<!-- 添加oracle驱动依赖 -->
<dependency>
    <groupId>com.oracle</groupId>
    <artifactId>ojdbc6</artifactId>
    <version>11.2.0.1.0</version>
</dependency>

   如果是mysql，添加mysql依赖

    <!--mysql-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>5.1.21</version>
        </dependency>
       

    其他依赖

<!--jpa-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>
        <!-- lombok -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>

在database视图区域任意地方右键，然后Scripted Extensions -> Go to Scripts Directory

 

若没有Generate POJOs.groovy文件，新增Generate POJOs.groovy文件。 Groovy 模板引擎

文件内容为:

import com.intellij.database.model.DasTable
import com.intellij.database.model.ObjectKind
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil


/*
* Available context bindings:
*   SELECTION   Iterable<DasObject>
*   PROJECT     project
*   FILES       files helper
*/


packageName = "com.sample;"
typeMapping = [
  (~/(?i)int/)                      : "long",
  (~/(?i)float|double|decimal|real/): "double",
  (~/(?i)datetime|timestamp/)       : "java.sql.Timestamp",
  (~/(?i)date/)                     : "java.sql.Date",
  (~/(?i)time/)                     : "java.sql.Time",
  (~/(?i)/)                         : "String"
]


FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
  SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }.each { generate(it, dir) }
}


def generate(table, dir) {
  def className = javaName(table.getName(), true)
  def fields = calcFields(table)
  new File(dir, className + ".java").withPrintWriter { out -> generate(out, className, fields) }
}


def generate(out, className, fields) {
  out.println "package $packageName"
  out.println ""
  out.println ""
  out.println "public class $className {"
  out.println ""
  fields.each() {
    if (it.annos != "") out.println "  ${it.annos}"
    out.println "  private ${it.type} ${it.name};"
  }
  out.println ""
  fields.each() {
    out.println ""
    out.println "  public ${it.type} get${it.name.capitalize()}() {"
    out.println "    return ${it.name};"
    out.println "  }"
    out.println ""
    out.println "  public void set${it.name.capitalize()}(${it.type} ${it.name}) {"
    out.println "    this.${it.name} = ${it.name};"
    out.println "  }"
    out.println ""
  }
  out.println "}"
}


def calcFields(table) {
  DasUtil.getColumns(table).reduce([]) { fields, col ->
    def spec = Case.LOWER.apply(col.getDataType().getSpecification())
    def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
    fields += [[
                 name : javaName(col.getName(), false),
                 type : typeStr,
                 annos: ""]]
  }
}


def javaName(str, capitalize) {
  def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
    .collect { Case.LOWER.apply(it).capitalize() }
    .join("")
    .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
  capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

在database视图区域选择你想要生成的表，然后Scripted Extensions -> Generate POJOs.groovy 可以使用Shift和Ctrl多选

 方法二 通过Persistence快速生成

配置数据库连接同上(方法一中的数据库连接)

调出 Persistence 窗口 

配置数据库连接同上(方法一中的数据库连接)

调出 Persistence 窗口

File—>Project Structure—>model—> + —>JPA

打开 Persistence窗口

配置生成实体类的参数

1.数据源,即链接数据库的信息,按要求填写数据库信息即可

2.生成实体类的位置.

3.实体类名称前缀.

4.实体类名称后缀. 这里我写Entity. 比如数据库表名为 user,那么生成实体类为 UserEntity

5.选择哪些表的哪些字段生成实体类.这里我全选.

6.生成的实体类自动添加 JPA注解.

• 通过Mybatis generator工具生成实体类

 

 

方法一:通过idea自动生成

选择上面配置的数据源，并在中找到需要生成的对应的数据库表。

选择mybatis-generator，并制定对应entity,dao,mapper的生成位置。

 

方法二:

1、IDEA创建maven工程（略）

2、 在maven项目的pom.xml 添加mybatis-generator-maven-plugin 插件和MySQL数据库驱动依赖

 <build>
    <plugins>
      <plugin>
        <groupId>org.mybatis.generator</groupId>
        <artifactId>mybatis-generator-maven-plugin</artifactId>
        <version>1.3.5</version>
        <configuration>
          <verbose>true</verbose>
          <overwrite>true</overwrite>
        </configuration>
      </plugin>
    </plugins>
  </build>

<dependency>
      <groupId>mysql</groupId>
      <artifactId>mysql-connector-java</artifactId>
      <version>5.1.25</version>
</dependency>

3、在maven项目下的src/main/resources 目录下建立名为 generatorConfig.xml的配置文件，作为mybatis-generator-maven-plugin 插件的执行目标，模板如下：

<?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>
    <!-- 配置mysql 驱动jar包路径.用了绝对路径 -->
    <classPathEntry location="C:\Users\zhangbo\.m2\repository\mysql\mysql-connector-java\5.1.25\mysql-connector-java-5.1.25.jar" />
    <context id="emil_mysql_tables">
        <!-- 防止生成的代码中有很多注释，加入下面的配置控制 -->
        <commentGenerator>
            <!-- 删除代码中带有 代码生成器的注释信息  -->
            <property name="suppressAllComments" value="true" />
            <property name="suppressDate" value="true" />
        </commentGenerator>

        <jdbcConnection connectionURL="jdbc:mysql://127.0.0.1:3306/emiltest" driverClass="com.mysql.jdbc.Driver" password="123456" userId="root" />

        <!-- 数据表对应的model层  -->
        <javaModelGenerator targetPackage="com.emil.shiro.entity" targetProject="src/main/java">
            <property name="enableSubPackages" value="true" />
            <property name="trimStrings" value="true" />
        </javaModelGenerator>

        <!-- sql mapper 映射配置文件 -->
        <sqlMapGenerator targetPackage="mapper"  targetProject="src/main/resources">
            <property name="enableSubPackages" value="true" />
        </sqlMapGenerator>

        <!-- mybatis3中的mapper接口 -->
        <javaClientGenerator type="XMLMAPPER" targetPackage="com.emil.shiro.dao"  targetProject="src/main/java">
            <property name="enableSubPackages" value="true" />
        </javaClientGenerator>

<!--
        <table tableName=""></table>
--> <!-- 数据表进行生成操作 schema:相当于库名; tableName:表名; domainObjectName:对应的DO  domainObjectName="SysAccessPermission"-->
        <table tableName="om_menu" schema="emiltest"   enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" />
        <table tableName="om_user" schema="emiltest" enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" ></table>
        <table tableName="om_role" schema="emiltest" enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" ></table>
        <table tableName="om_permis" schema="emiltest" enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" ></table>
        <table tableName="role_permis" schema="emiltest" enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" ></table>
        <table tableName="user_role" schema="emiltest" enableSelectByExample="false" enableDeleteByExample="false" enableCountByExample="false" enableUpdateByExample="fasle" ></table>
    </context>
</generatorConfiguration>

MyBatis Generator 详解。参考：

https://blog.youkuaiyun.com/isea533/article/details/42102297

4、使用maven运行mybatis-generator-maven-plugin插件：工程名->Plugins->mybatis-generator->mybatis-generator:generate->Run Maven Build

5、自动生成的结构如下：

如果你觉得这篇文章对你有帮助或启发，也可以来请我喝咖啡。