java文件自动生成模板

package yangpu.jdbc.mysql; import java.io.BufferedReader; import java.io.BufferedWriter; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.IOException; import java.util.ArrayList; import java.util.List; import org.apache.commons.lang3.StringUtils; import yangpu.jdbc.mysql.model.ColumnInfo; import yangpu.util.CharUtil; import yangpu.util.ReaderWriter; /** * <p> * yangpu.jdbc.mysql.ModelCreate.java * </p> * <p> * description： * </p> * * @author YangPu * @createTime 2016年7月21日 下午3:50:33 */ public class MapperCreate { private BufferedWriter writer; private List<ColumnInfo> list; /** * 加载模板 * * @param modelName */ public MapperCreate(String tableName, List<ColumnInfo> list) { try { this.list = list; String modelName = CharUtil.UnderlineToUppercaseTitle(tableName); String fileName = null; if (Constant.isTable) { fileName = Constant.path + modelName + File.separator + modelName + "Mapper.xml"; } else { fileName = Constant.path + Constant.package_mapper + File.separator + modelName + "Mapper.xml"; } File file = new File(fileName); if (!file.getParentFile().exists()) { file.getParentFile().mkdir(); } if (!file.exists()) { file.createNewFile(); } writer = new BufferedWriter(new FileWriter(file)); BufferedReader reader = new BufferedReader(new FileReader(Constant.mapper)); reader = ReaderWriter.readWrite(reader, list, tableName); String line = null; while ((line = reader.readLine()) != null) { if (line.contains("-modelName-")) { line = line.replaceAll("-modelName-", modelName); writer.write(line + "\n"); } else if (line.contains("-columns-")) { createColumns(); } else if (line.contains("-where-")) { createWhere(); } else if (line.contains("-insert-")) { createInsert(); } else if (line.contains("-update-")) { createUpdate(); } else if (line.contains("-deleteById-")) { createDeleteById(); } else if (line.contains("-deleteByEntity-")) { createDeleteByEntity(); } else if (line.contains("-getById-")) { createGetById(); } else if (line.contains("-getList-")) { createGetList(); } else if (line.contains("-getList_count-")) { createGetListCount(); } else { writer.write(line + "\n"); } } writer.close(); reader.close(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void createColumns() { try { int MaxLength = 0; String temp = null; List<String[]> fields = new ArrayList<>(); for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); String[] strings = new String[3]; temp = new String("\t\t" + cloumn); if (temp.length() > MaxLength) { MaxLength = temp.length(); } strings[0] = temp; temp = new String("AS"); strings[1] = temp; if (index == (list.size() - 1)) { temp = new String(field); } else { temp = new String(field + " ,"); } strings[2] = temp; fields.add(strings); } for (String[] strings : fields) { writer.write(strings[0] + CharUtil.getWhitespace(MaxLength - strings[0].length()) + "\t " + strings[1] + "\t " + strings[2] + "\n"); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void createWhere() { String line = new String(); try { for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); line = new String("\t\t\t<if test=\"" + field + "!=null and " + field + "!=''\">"); writer.write(line.toString() + "\n"); line = new String("\t\t\t\tAND " + cloumn + " = #{" + field + "}"); writer.write(line.toString() + "\n"); line = new String("\t\t\t</if>"); writer.write(line.toString() + "\n"); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void createInsert() { String line = null; try { line = new String("\t\tINSERT INTO " + list.get(0).getTableName().toUpperCase() + " ("); writer.write(line + "\n"); for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); if (index == (list.size() - 1)) { line = new String("\t\t\t" + cloumn); } else { line = new String("\t\t\t" + cloumn + " ,"); } writer.write(line + "\n"); } writer.write("\t\t) VALUES (\n"); for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); if (index == (list.size() - 1)) { line = new String("\t\t\t#{" + field + "}"); } else { line = new String("\t\t\t#{" + field + "} ,"); } writer.write(line + "\n"); } writer.write("\t\t)\n"); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void createUpdate() { String line = null; try { line = new String("\t\tUPDATE " + list.get(0).getTableName().toUpperCase() + " SET"); writer.write(line + "\n"); List<ColumnInfo> pris = new ArrayList<>(); int MaxLength = 0; String temp = null; List<String[]> fields = new ArrayList<>(); for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); if (StringUtils.isNotBlank(columninfo.getColumnKey()) && "PRI".equals(columninfo.getColumnKey().toUpperCase())) { pris.add(columninfo); continue; } String[] strings = new String[3]; temp = new String("\t\t\t" + cloumn); if (MaxLength < temp.length()) { MaxLength = temp.length(); } strings[0] = temp; temp = new String("="); strings[1] = temp; if (index == (list.size() - 1)) { temp = new String("#{" + field + "}"); } else { temp = new String("#{" + field + "} ,"); } strings[2] = temp; fields.add(strings); } for (String[] strings : fields) { writer.write(strings[0] + CharUtil.getWhitespace(MaxLength - strings[0].length()) + "\t" + strings[1] + "\t" + strings[2] + "\n"); } writer.write("\t\tWHERE\n"); if (pris.size() != 0) { for (ColumnInfo columninfo : pris) { String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn); writer.write("\t\t\t" + cloumn + " = #{" + field + "}\n"); } } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void createDeleteById() { String line = null; try { line = new String("\t\tDELETE FROM " + list.get(0).getTableName().toUpperCase()); writer.write(line + "\n"); line = new String("\t\t\tWHERE"); writer.write(line + "\n"); for (int index = 0; index < list.size(); index++) { ColumnInfo columninfo = list.get(index); if (StringUtils.isNotBlank(columninfo.getColumnKey()) && "PRI".equals(columninfo.getColumnKey().toUpperCase())) { String cloumn = columninfo.getColumnName(); String field = CharUtil.UnderlineToUppercase(cloumn)