writer.java

使用Java Swing开发的一个自动生成POJO的软件,能够生成Appfuse框架需要的带有XDoclet代码的POJO.

package org.eclipse.ve.model;

import java.io.*;
import java.util.ArrayList;

/**
 * <p>Title: 写Model文件</p>
 * <p>Description: 创建Model文件,并把表的属性和配置写入到文件</p>
 * <p>Copyright: Copyright (c) 2006</p>
 * <p>Company: www.use.com</p>
 * @author jiangyujie
 * @version 1.0
 */

public class Writer {

	//6个参数
	String obj;
	String typ;
	String len;
	String key;
	String nul;
	String uni;

	//定义upperFirstName为首字母大写的name,lowwerFirstName为首字母小写
	String lowerFirstName;

	String upperFirstName;

	//定义firstName为name除首字母之外的其他字符串,firstName为首字母,upperFirst为大写的首字母,lowerFirst为小写的首字母
	String otherName;
	char firstName;
	char upperFirst;
	char lowerFirst;

	void fileCreate(String name, ArrayList al)// code为代码框的输入,flagForeignKey为设置了外键的标志
	{
		char ch = '\"';

		String fileName = getFileName(name) + ".java";

		String tableName = getTableName(name);

		try {
			// 获得路径，创建model文件
			File model = new File(Auto.path, fileName);
			model.createNewFile();

			FileWriter fw = new FileWriter(model);
			BufferedWriter bfw = new BufferedWriter(fw);

			// 开始写文件头
			bfw.write("package com.ufida.appfrm.model;");

			bfw.newLine();
			bfw.newLine();
			bfw.write("import java.io.Serializable;");
			bfw.newLine();
			bfw.newLine();
			bfw.write("/**");
			bfw.newLine();
			bfw.write(" * @author zhengqian");
			bfw.newLine();
			bfw.write(" * @struts.form extends=" + ch + "BaseForm" + ch + "");
			bfw.newLine();
			bfw.write(" * @hibernate.class table=" + ch + "" + tableName + ch);
			bfw.newLine();
			bfw.write(" */");
			bfw.newLine();
			// 根据flagForeignKey确定是否有parent,由此确定继承的类和定义parent
			if (Auto.flagForeignKey == 0) {
				bfw.write("public class " + getFileName(name)
						+ " extends CommonObject implements Serializable {");
				bfw.newLine();
				bfw.newLine();

			} else {
				bfw.write("public class " + getFileName(name)
						+ " extends SubCommonObject implements Serializable {");
				bfw.newLine();
				bfw.newLine();
				bfw.write("    private " + Auto.parentModel + " parent;");
				bfw.newLine();
			}

			// 定义id
			bfw.write("    private Long id;");
			bfw.newLine();

			/*
			 * 根据属性的输入，写属性定义
			 * obj为设置对话框输入的名称,typ表示类型,len为长度,key是外键
			 * nul是非空与否,nui是unique唯一
			 */
			for (int i = 0; i < al.size(); i++) {

				//将al中的ArrayList取出
				obj = ((ArrayList) al.get(i)).get(0).toString();
				typ = ((ArrayList) al.get(i)).get(1).toString();
				len = ((ArrayList) al.get(i)).get(2).toString();
				key = ((ArrayList) al.get(i)).get(3).toString();
				nul = ((ArrayList) al.get(i)).get(4).toString();
				uni = ((ArrayList) al.get(i)).get(5).toString();
				
				if(typ.equals("Timestamp")){
					typ = "java.sql.Timestamp";
				}

				// 定义upperFirstName为首字母大写的name,lowwerFirstName为首字母小写
				firstName = obj.charAt(0);
				if ('0' <= firstName && firstName <= '9') {
					upperFirstName = obj;
					lowerFirstName = obj;
				} else {
					otherName = obj.substring(1);
					upperFirst = Character.toUpperCase(firstName);
					lowerFirst = Character.toLowerCase(firstName);

					upperFirstName = upperFirst + otherName;
					lowerFirstName = firstName + otherName;
				}

				if (key.equals("没有外键") == true) {
					bfw.write("    private" + " " + typ + " " + lowerFirstName
							+ ";");
				} else
					bfw.write("    private" + " " + key + " " + lowerFirstName
							+ ";");
				bfw.newLine();

			}

			bfw.newLine();
			bfw.write("    /**");
			bfw.newLine();
			bfw.write("     * @return Returns the id.");
			bfw.newLine();
			bfw.write("     * @hibernate.id column=" + ch + "id" + ch
					+ " generator-class=" + ch + "increment" + ch);
			bfw.newLine();
			bfw.write("     *               unsaved-value=" + ch + "null" + ch);
			bfw.newLine();
			bfw.write("     * @struts.form-field");
			bfw.newLine();
			bfw.write("     */");
			bfw.newLine();
			bfw.write("    public Long getId() {");
			bfw.newLine();
			bfw.write("        return id;");
			bfw.newLine();
			bfw.write("    }");
			bfw.newLine();
			bfw.newLine();
			bfw.write("    public void setId(Long id) {");
			bfw.newLine();
			bfw.write("        this.id = id;");
			bfw.newLine();
			bfw.write("    }");
			bfw.newLine();
			bfw.newLine();

			/*
			 * 根据属性的输入，写属性定义
			 * obj为设置对话框输入的名称,typ表示类型,len为长度,key是外键
			 * nul是非空与否,nui是unique唯一
			 */
			for (int i = 0; i < al.size(); i++) {
				//将al中的ArrayList取出
				obj = ((ArrayList) al.get(i)).get(0).toString();
				typ = ((ArrayList) al.get(i)).get(1).toString();
				len = ((ArrayList) al.get(i)).get(2).toString();
				key = ((ArrayList) al.get(i)).get(3).toString();
				nul = ((ArrayList) al.get(i)).get(4).toString();
				uni = ((ArrayList) al.get(i)).get(5).toString();
				
				if(typ.equals("Timestamp")){
					typ = "java.sql.Timestamp";
				}

				// 写此参数的set,get方法
				// 定义upperFirstName为首字母大写的name,lowwerFirstName为首字母小写
				firstName = obj.charAt(0);
				if ('0' <= firstName && firstName <= '9') {
					upperFirstName = obj;
					lowerFirstName = obj;
				} else {
					otherName = obj.substring(1);
					upperFirst = Character.toUpperCase(firstName);
					lowerFirst = Character.toLowerCase(firstName);

					upperFirstName = upperFirst + otherName;
					lowerFirstName = firstName + otherName;
				}
				//得到列名
				String columnName = getTableName(obj);

				// 外键为空的情况
				if (key.equals("没有外键") == true) {

					bfw.write("    /**");
					bfw.newLine();
					bfw.write("     * @return Returns the " + lowerFirstName
							+ ".");
					bfw.newLine();
					bfw.write("     * @hibernate.property column=" + ch
							+ columnName + ch + " " + "not-null=" + ch + "true"
							+ ch);

					// 如果设置了唯一
					if (uni.equals("true") == true) {
						bfw.write(" unique=" + ch + "true" + ch);
					}

					// 如果长度不为空
					if (len.equals("") == false) {
						bfw.write(" " + "length=" + ch + len + ch);
					}
					bfw.newLine();

					// 如果不为空
					if (nul.equals("true") == true) {
						bfw.write("     * @struts.validator type=" + ch
								+ "required" + ch);
						bfw.newLine();
					}
					bfw.write("     * @struts.form-field");
					bfw.newLine();
					bfw.write("     */");
					bfw.newLine();

					// get方法
					bfw.write("    public " + typ + " get" + upperFirstName
							+ "() {");
					bfw.newLine();
					bfw.write("        return " + lowerFirstName + ";");
					bfw.newLine();
					bfw.write("    }");
					bfw.newLine();
					bfw.newLine();

					// set方法
					bfw.write("    public void set" + upperFirstName + "("
							+ typ + " " + lowerFirstName + ") {");
					bfw.newLine();
					bfw.write("        this." + lowerFirstName + " = "
							+ lowerFirstName + ";");
					bfw.newLine();
					bfw.write("    }");
					bfw.newLine();
					bfw.newLine();
				}

				// 当设置了外键的时候
				else {
					bfw.write("    /**");
					bfw.newLine();
					bfw.write("     * @hibernate.many-to-one column=" + ch
							+ columnName + "_id" + ch + " class=" + ch
							+ "com.ufida.appfrm.model." + key + ch
							+ " cascade=" + ch + "all" + ch);
					bfw.newLine();
					bfw.write("     * @hibernate.column name=" + ch
							+ columnName + "_id" + ch);
					bfw.newLine();
					bfw.write("     * @struts.form-field");
					bfw.newLine();
					bfw.write("     */");
					bfw.newLine();
					// get方法
					bfw.write("    public " + key + " get" + upperFirstName
							+ "() {");
					bfw.newLine();
					bfw.write("        return " + lowerFirstName + ";");
					bfw.newLine();
					bfw.write("    }");
					bfw.newLine();
					bfw.newLine();

					// set方法
					bfw.write("    public void set" + upperFirstName + "("
							+ key + " " + lowerFirstName + ") {");
					bfw.newLine();
					bfw.write("        this." + lowerFirstName + " = "
							+ lowerFirstName + ";");
					bfw.newLine();
					bfw.write("    }");
					bfw.newLine();
					bfw.newLine();
				}

			}
			// 根据flagForeignKey确定是否有外键,由此写set,get和getParentObject方法
			if (Auto.flagForeignKey != 0) {
				bfw.write("    /**");
				bfw.newLine();
				bfw.write("     * @hibernate.many-to-one cascade=" + ch + "all"
						+ ch);
				bfw.newLine();
				bfw.write("     * @hibernate.column name=" + ch + "parent_id"
						+ ch);
				bfw.newLine();
				bfw.write("     * @struts.form-field");
				bfw.newLine();
				bfw.write("     */");
				bfw.newLine();
				bfw.write("    public " + Auto.parentModel + " getParent() {");
				bfw.newLine();
				bfw.write("        return parent;");
				bfw.newLine();
				bfw.write("    }");
				bfw.newLine();
				bfw.newLine();

				bfw.write("    public void setParent(" + Auto.parentModel
						+ " parent) {");
				bfw.newLine();
				bfw.write("        this.parent = parent;");
				bfw.newLine();
				bfw.write("	    }");
				bfw.newLine();
				bfw.newLine();

				bfw.write("    public CommonObject getParentObject(){");
				bfw.newLine();
				bfw.write("        return parent;");
				bfw.newLine();
				bfw.write("    }");
				bfw.newLine();
				bfw.newLine();
			}
			bfw.write("}");
			bfw.flush();
			bfw.close();
		} catch (Exception ex) {
			System.out.println(ex.getMessage());
		}
	}

	String getFileName(String fileName) {
		// 保证文件名首字母大写
		String lowwerFile = fileName.toLowerCase();// lowwerFile为全部小写的输入

		char lowwerFirstFile = lowwerFile.charAt(0);// lowwerFirstFile为小写的首字母
		String otherFile = fileName.substring(1);// otherFile为除首字母之外的其他输入
		char upperFirstFile;

		if (lowwerFirstFile >= 97)// 为小写的时候,避开为数字的时候
			upperFirstFile = Character.toUpperCase(lowwerFirstFile);// upperFirstFile为大写的首字母
		else
			upperFirstFile = lowwerFirstFile;

		return upperFirstFile + otherFile;// 返回首字母大写
	}

	String getTableName(String fileName) {
		// 根据代码名大小写转换表名
		StringBuffer codeCache = new StringBuffer(fileName);
		StringBuffer tableNameCache = new StringBuffer();
		int t = 0;

		if (fileName.length() == 1)
			return fileName.toLowerCase();
		else {
			for (int s = 1; s < fileName.length(); s++) {
				char test = fileName.charAt(s);
				if ('A' <= test && test <= 'Z') {
					tableNameCache = codeCache.insert(s + t, "_");// 如果test此字符为大写,则在前面加入_,转换为表名
					t += 1;
				} else
					tableNameCache = codeCache;// 如果为小写,则直接赋值
			}
			return tableNameCache.toString().toLowerCase();// 表名由加入_的stringCache变为小写得到
		}
	}
}