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;
}
Aggregations