1.与数据库建立连接
在DBUtil中设置数据库相关的参数
Class.forName(driverName);
conn = DriverManager.getConnection(url, user, password);
2.获得数据库中的所有表
Mapmaps = new HashMap();
PreparedStatement pstate = conn.prepareStatement("show table status");
ResultSet results = pstate.executeQuery();
while (results.next()) {
String tableName = results.getString("NAME");
String comment = results.getString("COMMENT");
maps.put(tableName, comment);
}
3.获得每个表中的信息
封装一个实体来存储表中的信息
PreparedStatement pstate = conn.prepareStatement("show full fields from " + tableName);
ResultSet results = pstate.executeQuery();
List lists = new ArrayList();
while (results.next()) {
Column column = new Column();
String field = results.getString("FIELD");
column.setFeildMapper(field);
String fieldName = processField(field);
column.setFeild(fieldName);
String type = processType(results.getString("TYPE"));
String comment = results.getString("COMMENT");
column.setType(type);
column.setComment(comment);
lists.add(column);
}
4.拼串输出
将固定代码中需要变动的字段从实体中取拼串写入文档中
例如给表生成po:
public class WritePo {
public static void createPo(String path,String packageName,String upTable, String lowTable, List columns) throws IOException {
File folder = new File(path + lowTable + "/po");
if (!folder.exists()) {
folder.mkdirs();
}
File beanFile = new File(folder + "/" + upTable + ".java");
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(
new FileOutputStream(beanFile)));
bw.write("package "+packageName+lowTable+".po;");
bw.newLine();
bw.write("import java.io.Serializable;");
bw.newLine();
// bw.write("import lombok.Data;");
// bw.write("import javax.persistence.Entity;");
bw.newLine();
bw.write("@SuppressWarnings(\"serial\")");
bw.newLine();
// bw.write("@Entity");
// bw.write("@Data");
// bw.newLine();
bw.write("public class " + upTable + " implements Serializable {");
bw.newLine();
bw.newLine();
int size = columns.size();
for (Column vo : columns) {
// bw.write("\t/**" + comments.get(i) + "**/");
bw.write("\t/**");
bw.newLine();
bw.write("\t *" + vo.getComment());
bw.newLine();
bw.write("\t **/");
bw.newLine();
bw.write("\tprivate " + vo.getType() + " " + vo.getFeild() + ";");
bw.newLine();
bw.newLine();
}
bw.newLine();
// 生成get 和 set方法
String tempField = null;
String _tempField = null;
String tempType = null;
for (Column vo : columns) {
tempType = vo.getType();
_tempField = vo.getFeild();
tempField = _tempField.substring(0, 1).toUpperCase()
+ _tempField.substring(1);
bw.newLine();
// bw.write("\tpublic void set" + tempField + "(" + tempType + " _" +
// _tempField + "){");
bw.write("\tpublic void set" + tempField + "(" + tempType + " "
+ _tempField + "){");
bw.newLine();
// bw.write("\t\tthis." + _tempField + "=_" + _tempField + ";");
bw.write("\t\tthis." + _tempField + " = " + _tempField + ";");
bw.newLine();
bw.write("\t}");
bw.newLine();
bw.newLine();
bw.write("\tpublic " + tempType + " get" + tempField + "(){");
bw.newLine();
bw.write("\t\treturn this." + _tempField + ";");
bw.newLine();
bw.write("\t}");
bw.newLine();
}
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
}
最终的效果
后台的增删改查功能都可以生成!希望对大家有帮助!
