Examples with TableDetails com.hccake.ballcat.codegen.model.bo.TableDetails used on opensource projects

Example 1 with TableDetails

use of com.hccake.ballcat.codegen.model.bo.TableDetails in project ballcat-codegen by ballcat-projects.

the class GeneratorServiceImpl method getStringFileEntryMap.

/**
 * 获得生成后的 代码地址：代码文件 的 map
 * @param generateOptionDTO 生成参数
 * @param templateFiles 模板文件
 * @return Map<String, FileEntry>
 */
private Map<String, FileEntry> getStringFileEntryMap(GeneratorOptionDTO generateOptionDTO, List<TemplateFile> templateFiles) {
    Map<String, FileEntry> map = new HashMap<>(templateFiles.size());
    for (String tableName : generateOptionDTO.getTableNames()) {
        // 查询表详情
        TableDetails tableDetails = tableInfoQuery.queryTableDetails(tableName);
        // 生成代码
        Map<String, FileEntry> fileEntryMap = generatorCode(tableDetails, generateOptionDTO.getTablePrefix(), generateOptionDTO.getGenProperties(), templateFiles);
        map.putAll(fileEntryMap);
    }
    return map;
}

Example 2 with TableDetails

use of com.hccake.ballcat.codegen.model.bo.TableDetails in project ballcat-codegen by ballcat-projects.

the class TableInfoQuery method queryTableDetails.

/**
 * 查询指定表的详细信息
 * @param tableName 表名称
 * @return TableDetails
 */
public TableDetails queryTableDetails(String tableName) {
    DbType dbType = getDbType();
    TableInfoMapper baseMapper = tableInfoMapperMap.get(dbType);
    // 查询表信息
    TableInfo tableInfo = baseMapper.queryTableInfo(tableName);
    // 查询列信息
    List<ColumnInfo> columnInfoList = baseMapper.listColumnInfo(tableName);
    TableDetails tableDetails = new TableDetails();
    tableDetails.setDbType(dbType);
    tableDetails.setTableName(tableInfo.getTableName());
    tableDetails.setEngine(tableInfo.getEngine());
    tableDetails.setTableComment(tableInfo.getTableComment());
    tableDetails.setCreateTime(tableInfo.getCreateTime());
    tableDetails.setColumnInfos(columnInfoList);
    return tableDetails;
}