package com.iplatform.generator;
|
|
import com.iplatform.base.LocalDatabaseMetaEngine;
|
import com.walker.db.TableInfo;
|
import com.walker.db.page.GenericPager;
|
import com.walker.dbmeta.FieldInfo;
|
import com.walker.infrastructure.utils.StringUtils;
|
import org.apache.commons.io.IOUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import java.io.FileOutputStream;
|
import java.util.List;
|
|
//@Component
|
public class JdbcGeneratorEngine {
|
|
// @Autowired
|
private JdbcGeneratorServiceImpl jdbcGeneratorService = null;
|
|
private LocalDatabaseMetaEngine localDatabaseMetaEngine = null;
|
|
@Autowired
|
public JdbcGeneratorEngine(JdbcGeneratorServiceImpl jdbcGeneratorService, LocalDatabaseMetaEngine localDatabaseMetaEngine){
|
this.jdbcGeneratorService = jdbcGeneratorService;
|
this.localDatabaseMetaEngine = localDatabaseMetaEngine;
|
}
|
|
/**
|
* 返回字段信息集合
|
* @param tableName 表名称
|
* @return
|
* @date 2022-11-22
|
*/
|
public List<FieldInfo> getFieldsObject(String tableName){
|
return this.localDatabaseMetaEngine.getFieldsObject(tableName);
|
}
|
|
/**
|
* 返回表名称集合。
|
* @param tableNameLike 模糊查询表前缀
|
* @return
|
* @date 2022-11-22
|
*/
|
public List<String> getTableNamesByLike(String tableNameLike){
|
return this.localDatabaseMetaEngine.getTableNamesByLike(tableNameLike);
|
}
|
|
/**
|
* 分页返回表信息集合。当前页信息通过对象: ListPageContext.getCurrentPageIndex()方法获得。
|
* @param tableNameLike 表名模糊查询，如果为空字符串则查询所有。
|
* @return
|
* @date 2022-11-22
|
*/
|
public GenericPager<TableInfo> queryPageTableNamesByLike(String tableNameLike
|
// , int pageIndex, int pageSize
|
){
|
return this.localDatabaseMetaEngine.queryPageTableNamesByLike(tableNameLike);
|
}
|
|
/**
|
* 返回指定表元数据对象。
|
* @param tableName 表名称
|
* @return
|
* @date 2022-11-26
|
*/
|
public TableInfo queryOneTableInfo(String tableName){
|
return this.localDatabaseMetaEngine.queryOneTableInfo(tableName);
|
}
|
|
/**
|
* 生成相应前缀的表PO对象。
|
* @param tableNamePrefix 前缀名，儒：sys_
|
* @param mbbs 生成标识，默认为：po,mapper
|
* @param outputFile 生成的文件，如：d:/po.zip
|
* @throws Exception 如果失败抛出异常
|
*/
|
public void generatePoFile(String tableNamePrefix, String mbbs, String outputFile) throws Exception{
|
if(StringUtils.isEmpty(mbbs)){
|
// mbbs = "po,mapper";
|
mbbs = TEMPLATE_PO_MBBS;
|
}
|
byte[] fileContent = this.jdbcGeneratorService.generatorPoCode(tableNamePrefix, mbbs);
|
IOUtils.write(fileContent, new FileOutputStream(outputFile));
|
// Files.newOutputStream(Paths.get(URI.create(outputFile)));
|
}
|
|
/**
|
* 获得一个表的PO生成文件，返回一个zip压缩包。<p></p>
|
* 该方法在代码生成功能中，让用户下载一个表数据库PO文件。
|
* @param tableName 表名称
|
* @return
|
* @date 2022-11-28
|
*/
|
public byte[] generateOnePoZipFile(String tableName){
|
return this.jdbcGeneratorService.generatorPoCode(tableName, TEMPLATE_PO_MBBS, true);
|
}
|
|
/**
|
* 生成一个表的po文件。
|
* @param tableName 具体的表名称，不是前缀
|
* @param outputFile
|
* @throws Exception
|
*/
|
public void generateOnePoFile(String tableName, String outputFile) throws Exception{
|
byte[] fileContent = this.jdbcGeneratorService.generatorPoCode(tableName, TEMPLATE_PO_MBBS, true);
|
IOUtils.write(fileContent, new FileOutputStream(outputFile));
|
}
|
|
public static final String TEMPLATE_PO_MBBS = "po,mapper";
|
|
/*@PostConstruct
|
public void init(){
|
System.out.println("~~~~~~~~~~~~~~~ 开始生成数据库代码：");
|
try {
|
this.generatePoFile("auto_", "po,mapper", "f:/项目/springboot/po.zip");
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
}*/
|
}
|
