atablebeancreator.java

java bean生成器

package turing;

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.util.Vector;

class ATableBeanCreator  extends ARecorderBeanCreator
{
	ATableBeanCreator(String tableName,String packageName)
	{
		super(tableName , packageName );	
	}
	
	Column getTableIdentity(Vector v_col)
	{
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			if(col.isIdentity)
				return col;			
		}		
		return null;
	}
	
	void createATableBean(Vector v_col) throws Exception
	{
        BufferedWriter out = new BufferedWriter(new FileWriter("com/"+tableName+"Biao.java"));
        
 		out.write("package "+packageName+";\n\n");
 		
 		out.write("import java.sql.*;\n");
 		out.write("import java.util.*;\n");
 		out.write("import "+packageName+".Dao;\n\n");	
 		
		out.write("public class "+tableName+"Biao\n{\n");
		
		createColumnDeclare(v_col,out);
		out.write("\t private Dao dao;\n");
	
		out.write("\n\t public "+tableName+"Biao()\n\t {\n");
		createDefaultValue(v_col,out);
		out.write("\t\t dao=new Dao();\n");				
		out.write("\t }\n\n");
		
		
		createQueryFunction( v_col, out);
		out.write("\n");		
		createQueryOneFunction( v_col, out);
		out.write("\n");				
		createAddFunction( v_col, out);	
		out.write("\n");			
		createDeleteFunction( v_col, out);
		out.write("\n");				
		createUpdateFunction( v_col, out);			
		out.write("\n");
				
		createSetGet(v_col,out);
		
		out.write("}\n");
        out.close();
	}
	
	void createQueryFunction(Vector v_col,BufferedWriter out) throws Exception
	{
		Column col_identity=getTableIdentity(v_col);
		
		out.write("\t public  ArrayList query(String tiaoJian)\n");
		out.write("\t {\n");
		out.write("\t\t String sql=\"select * from "+tableName+" where \"+tiaoJian ;\n");


		out.write("\t\t ResultSet rs=dao.query(sql);\n");
	   	out.write("\t\t ArrayList al=new ArrayList();	\n");
	   	out.write("\t\t try\n");
		out.write("\t\t {\n");
		out.write("\t\t\t while(rs.next())\n");
		out.write("\t\t\t {\n");
		out.write("\t\t\t\t "+tableName +" a=new "+tableName +"();\n");
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			out.write("\t\t\t\t a.set"+ abc2Abc(col.name)+"(rs.get"+abc2Abc((String)ht_columnType.get(col.type)) +"(\""+col.name+"\"));\n");
		}		
		out.write("\t\t\t\t al.add(a);\n");
		out.write("\t\t\t }\n");
		out.write("\t\t }\n");
		out.write("\t\t catch(Exception e){}\n");
			
		out.write("\t\t return al;		\n");
		out.write("\t }		\n");
	}
	
	void createQueryOneFunction(Vector v_col,BufferedWriter out) throws Exception
	{
		Column col_identity=getTableIdentity(v_col);
		
		out.write("\t public  "+tableName+" queryOne()\n");
		out.write("\t {\n");
		out.write("\t\t String sql=\"select * from "+tableName+" where "+col_identity.name+"=\"+"+col_identity.name+" ;\n");



		out.write("\t\t ResultSet rs=dao.query(sql);\n");
	   	out.write("\t\t try\n");
		out.write("\t\t {\n");
		out.write("\t\t\t if(rs.next())\n");
		out.write("\t\t\t {\n");
		out.write("\t\t\t\t "+tableName +" a=new "+tableName +"();\n");
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			out.write("\t\t\t\t a.set"+ abc2Abc(col.name)+"(rs.get"+abc2Abc((String)ht_columnType.get(col.type)) +"(\""+col.name+"\"));\n");
		}		
		out.write("\t\t\t\t return a;\n");
		out.write("\t\t\t }\n");
		out.write("\t\t }\n");
		out.write("\t\t catch(Exception e){}\n");
			
		out.write("\t\t return null;		\n");
		out.write("\t }		\n");		
	}
	
	void createAddFunction(Vector v_col,BufferedWriter out) throws Exception
	{
		Column col_identity=getTableIdentity(v_col);
		out.write("\n");		
		out.write("\t public int add()\n");
		out.write("\t {\n");

		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			if(ht_columnType.get(col.type).equals("String")==true)
			{
			    out.write("\t\t if(this."+col.name+"==null) "+col.name+"=\"\";\n");		
			}
		}
		
		out.write("\t\t String sql=\"insert into "+tableName+"(");
		String cols="";
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			if(col.isIdentity==false)
			{
			    cols=cols + col.name+",";		
			}
		}
		out.write(cols.substring(0,cols.length()-1));				
		out.write(") values(");
		
		cols="";
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			if(col.isIdentity==false)
			{
			   if( ht_columnType.get(col.type).equals("String")==true)
			   {
					cols=cols + "'\"+"+col.name+"+\"',";		   	
			   }
			   else
			   {
					cols=cols +"\"+"+ col.name+"+\",";	 	
			   }
			}
		}			
		out.write(cols.substring(0,cols.length()-1)+")\";\n");		
					      
		out.write("\t\t int tiao=dao.update(sql);\n");
		out.write("\t\t return tiao;\n");
		out.write("\t }	\n");			
	}	
	
	void createDeleteFunction(Vector v_col,BufferedWriter out) throws Exception
	{
		Column col_identity=getTableIdentity(v_col);
		out.write("\n");		
		out.write("\t public int delete("+ht_columnType.get(col_identity.type)+" "+col_identity.name+")\n");
		out.write("\t {\n");

		out.write("\t\t String sql=\"delete   from "+tableName+" where "+col_identity.name+"=\"+"+col_identity.name+";\n");

		out.write("\t\t int tiao=dao.update(sql);\n");
		out.write("\t\t return tiao;\n");
		out.write("\t }	\n");
	}
	
	void createUpdateFunction(Vector v_col,BufferedWriter out) throws Exception
	{
		Column col_identity=getTableIdentity(v_col);
		out.write("\n");		
		out.write("\t public int update()\n");
		out.write("\t {\n");


		out.write("\t\t String sql=\"update "+tableName+" set \";\n");
	
		for(int i=0;i<v_col.size();i++)
		{
			Column col=(Column)v_col.get(i);
			if(col.isIdentity==false)
			{			
				if(ht_columnType.get(col.type).equals("String")==true)
				{
				    out.write("\t\t if(this."+col.name+"==null) "+col.name+"=\"\";\n");
				    out.write("\t\t sql=sql+\""+col.name+"='\"+"+col.name+"+\"' ,\";\n");			
				}
				else
				{
				    out.write("\t\t sql=sql+\""+col.name+"=\"+"+col.name+"+\" ,\";\n");						
				}
			}
		}
		
		out.write("\t\t sql=sql.substring(0,sql.length()-1)+\" where "+col_identity.name+"=\"+"+col_identity.name+";\n");
		      
		out.write("\t\t int tiao=dao.update(sql);\n");
		out.write("\t\t return tiao;\n");
		out.write("\t }	\n");	
	}					
}