com.fly.re.read.FlyReadMySql Maven / Gradle / Ivy

package com.fly.re.read;

import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import com.fly.re.CodeCfg;
import com.fly.re.model.Column;
import com.fly.re.model.MkTable;

public class FlyReadMySql implements FlyRead{

	// 配置信息
	CodeCfg codeCfg;
	public FlyRead setCodeCfg(CodeCfg codeCfg){
		this.codeCfg = codeCfg;
		return this;
	}
		

	// 开始读取 
	@Override
	public void readInfo() {
		Map tcMap = ReadUtil.getTcMap(codeCfg.sqlFly);
		for (String tableName : codeCfg.tableNameList) {
			MkTable table = new MkTable();
			table.name = tableName;		// 表名字 
			table.comment = tcMap.get(tableName);	// 表注释
			getColumnList(table);	// 获取字段信息
			codeCfg.tableList.add(table);	// 添加进集合 
		}
	}
	
	
	
	// 获取指定表的所有列信息 
	public void getColumnList(MkTable table){
		List columns = new ArrayList<>();
		try {
			Map jtMap = ReadUtil.getJtMap(codeCfg.sqlFly, table.name);
			ResultSet rs = codeCfg.sqlFly.getResultSet("show full columns from " + table.name);
			while (rs.next()) {
				Column column = new Column();
				column.name = rs.getString("Field"); 			// 字段名
				column.dbType = rs.getString("Type"); 		// DB类型
				column.javaType = jtMap.get(column.name); 					// java类型
				column.comment = rs.getString("Comment"); 	// 注释
				columns.add(column);
			}
			table.columnList = columns; 
			table.pk = "id";	// 主键列
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	

	

	
	
}