Search in sources :

Example 1 with IdField

use of com.github.mengweijin.generator.entity.IdField in project code-generator-maven-plugin by mengweijin.

the class DefaultAutoGenerator method getIdField.

private static IdField getIdField(TableInfo tableInfo) {
    TableField tableField = tableInfo.getFields().stream().filter(TableField::isKeyFlag).findFirst().orElse(null);
    IdField idField = new IdField();
    if (tableField != null) {
        idField.setColumnName(tableField.getName());
        idField.setPropertyName(tableField.getPropertyName());
        idField.setPropertyType(tableField.getColumnType().getType());
    }
    return idField;
}
Also used : IdField(com.github.mengweijin.generator.entity.IdField) TableField(com.baomidou.mybatisplus.generator.config.po.TableField)

Aggregations

TableField (com.baomidou.mybatisplus.generator.config.po.TableField)1 IdField (com.github.mengweijin.generator.entity.IdField)1