IntelliJ IDEA 10 generate entity (POJO) from DB model

半城伤御伤魂 提交于 2019-11-27 17:14:29

UPDATE:
In IntelliJ 16 this feature in now implemented. The steps to do it are:
1. Database view context menu
2. Scripted Extensions
3. Generate POJOs


You can read more here:
Feature request: allow "generate classes from database schema" for plain-JDBC developers


First you need tell intelliJ that you are using Hibernate (I guess you are if you need the orm pojo of the table)

  1. Go to "Project structure" (alt+ctrl+shift+s)
  2. In "Project settings" select "Modules"
  3. Press + and add the Hibernate facet in your module.

Now you have setup your hibernate configuration facet you can extract your pojos.

  1. At your bottom right horizontal panel you will now see a tab called "Persistence" (ιf you can't find Persistence tab you may show it by choosing View > Tool Windows > Persistence)
  2. There you can right click on the hibernate icon named like your module
  3. Go to "Generate Persistence Mapping"-"by database schema"
  4. Now I guess you can find your way...
  5. In general settings select the datasource that you want to use and now you can see all the tables in your datasource object
  6. Now you can do many things, add relationships with the + sign, change the name and type of the pojo's properties etc. note: if you get an error and the "OK" is disabled its probably because the data type that intelliJ found for your pojo is invalid. Just change it to the one you need and you are ready to go!

The default Scripted Extensions Generate POJOs.groovy is not very good when dealing with tables with underscore(which is very common).

So I make some modifications.

The main code

def calcFields(DasObject table) {
    DasUtil.getColumns(table).reduce([]) { fields, col ->
        def spec = Case.LOWER.apply(col.dataType.specification)
        def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
        fields += [[
                       name : javaName(col.name, false),
                       type : typeStr,
                       annos: """
    /**
     * $col.comment
     */"""]]
    }
}

static String javaName(String str, boolean capitalize) {
    def s = CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.UPPER_CAMEL, str);
    capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

You can find the whole gist here https://gist.github.com/aristotll/ad799a7462e8b705b26103944cca24a6

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!