webtool自动生成数据库代码

package com.sqlite; import java.io.BufferedOutputStream; import java.io.File; import java.io.FileOutputStream; import java.sql.ResultSet; import java.sql.Statement; import java.util.Iterator; import java.util.List; import java.util.Map.Entry; import com.cjnetwork.webtool.common.Globel; import com.cjnetwork.webtool.util.NamingUtil; import com.cjnetwork.webtool.util.PropertyUtil; public class DaoGenerator implements Generator { StringBuilder implSb; // private @Override public boolean generate() { boolean result = false; try { generateBaseDao(); Iterator<Entry<String, List<Entry<String, String>>>> iter = Globel.tables.entrySet().iterator(); while (iter.hasNext()) { Entry<String, List<Entry<String, String>>> entry = iter.next(); generate(entry.getKey(), entry.getValue()); } result = true; } catch (Exception e) { e.printStackTrace(); } return result; } private void generateBaseDao() { String res = "resource/SqlHelper.java"; String tar = PropertyUtil.getProperty("daoFileFolder") + File.separator + "SqlHelper.java"; Globel.fileCopyPool.put(res, tar); } /** * 生成dao接口 YanHuan 2011-1-2下午05:01:26 */ private void generate(String tableName, List<Entry<String, String>> columns) throws Exception { implSb = new StringBuilder(); implSb.append("package " + PropertyUtil.getProperty("sqlitepackage") + ".basedao;\n"); implSb.append("\n"); // implSb.append("import " + PropertyUtil.getProperty("sqlitepackage")+"." +NamingUtil.getClassName(tableName)+ ";\n"); implSb.append("import java.util.Date;\n"); implSb.append("import android.text.TextUtils;\n"); implSb.append("import android.database.Cursor;\n"); implSb.append("import android.content.ContentValues;\n"); implSb.append("import android.database.sqlite.SQLiteDatabase;\n"); implSb.append("import android.database.sqlite.SQLiteOpenHelper;\n"); implSb.append("import java.text.SimpleDateFormat;\n"); implSb.append("import android.util.SparseArray;\n"); implSb.append("import android.database.sqlite.SQLiteStatement;\n"); implSb.append("import java.text.ParseException;\n"); implSb.append("import android.util.Log;\n"); implSb.append("import java.util.concurrent.locks.ReentrantReadWriteLock;\n"); implSb.append("\n\n"); implSb.append("public class " + NamingUtil.getClassName(tableName) + "Dao{\n"); implSb.append(" ReentrantReadWriteLock rwl = new ReentrantReadWriteLock();\n"); implSb.append(" SimpleDateFormat dfu = new SimpleDateFormat(\"yyyy-MM-dd HH:mm:ss\");\n"); implSb.append(" public static final String TABLENAME =\"" + NamingUtil.getClassName(tableName) + "\";\n"); implSb.append(" COLUMNINDEXS cOLUMNINDEXS=new COLUMNINDEXS();\n"); implSb.append(" COLUMNS cOLUMNS=new COLUMNS();\n"); implSb.append(" private final SQLiteOpenHelper mOpenHelper;\n"); implSb.append(" public " + NamingUtil.getClassName(tableName) + "Dao(SQLiteOpenHelper openHelper){\n"); implSb.append(" mOpenHelper=openHelper;\n "); Statement stat = Launch.con.createStatement(); ResultSet rs = stat.executeQuery("SELECT type,sql FROM sqlite_master where (type='table' or type= 'index') and tbl_name='" + tableName + "'"); while (rs.next()) { String type = rs.getString("type"); if(type.equals("table") ){ String sql = rs.getString("sql"); sql= sql.replace("CREATE TABLE", "CREATE TABLE IF NOT EXISTS"); sql=sql.replaceAll("\r\n", ""); sql=sql.replaceAll("\n", ""); // sql= sql.replaceAll("\r\n" ,"\" + \n \"" ); implSb.append(" mOpenHelper.getWritableDatabase().execSQL(\"" +sql + "\"); \n"); } else if(type.equals("index")){ String sql = rs.getString("sql"); sql= sql.replace(" INDEX ", " INDEX IF NOT EXISTS "); implSb.append(" mOpenHelper.getWritableDatabase().execSQL(\"" +sql + "\"); \n"); } } rs.close(); implSb.append(" }\n"); // 查询 implSb.append(" public Cursor query(String whereClause, String []whereArgs){\n"); implSb.append(" final String sql = \"SELECT *\"\n"); implSb.append(" + \" FROM \" + (TextUtils.isEmpty(whereClause)? TABLENAME : TABLENAME+\" WHERE \"+whereClause);\n"); implSb.append(" return mOpenHelper.getReadableDatabase().rawQuery(sql, whereArgs);\n"); implSb.append(" }\n\n\n"); // + cOLUMNS.username + ","+ cOLUMNS.ID //-------------insertList implSb.append(" public boolean insertList( SparseArray<"+NamingUtil.getClassName(tableName)+"> list) {\n"); implSb.append(" if ( null == list || list.size() <= 0) {\n"); implSb.append(" return false;\n"); implSb.append(" }\n"); implSb.append(" SQLiteDatabase db = mOpenHelper.getWritableDatabase();\n"); implSb.append(" try {\n"); implSb.append(" String sql =\"insert into " +tableName + "(\" \n"); String wenhao=""; for (int i = 0; i < columns.size(); i++) { Entry<String, String> entry = columns.get(i); String columnName = NamingUtil.getInstanceName(entry.getKey()); if(i==columns.size()-1) implSb.append(" + cOLUMNS." + columnName); else implSb.append(" + cOLUMNS." + columnName + "+\",\""); wenhao+=",?"; } implSb.append(" + \") \" + \"values("+wenhao.substring(1)+")\";\n"); implSb.append(" String createdate= dfu.format(new Date());\n"); implSb.append(" SQLiteStatement stat = db.compileStatement(sql);\n"); implSb.append(" db.beginTransaction();\n"); implSb.append(" for (int i=0;i<list.size();i++) {\n"); implSb.append(" "+NamingUtil.getClassName(tableName)+" entity = list.get(list.keyAt(i)); \n"); for (int i = 0; i < columns.size(); i++) { Entry<String, String> entry = columns.get(i); String columnName = NamingUtil.getInstanceName(entry.getKey()); if (entry.getValue().contains("Int")||entry.getValue().startsWith("Long")) { implSb.append(" if(null==entity."+ columnName + ") "); if(columnName.toLowerCase().equals("lifestatus")){ implSb.append(" stat.bindLong("+(i+1)+",1);else\n"); }else if(columnName.toLowerCase().equals("upgradeflag")){ implSb.append(" stat.bindLong("+(i+1)+",1);else\n"); }else{ implSb.append(" stat.bindNull("+(i+1)+"); else\n"); } implSb.append(" stat.bindLong(" +(i+1)+",entity."+ columnName + ");\n"); } else if (entry.getValue().startsWith("String")) { implSb.append(" if(null==entity."+ columnName + "||entity."+ columnName +".length()==0) "); implSb.append(" stat.bindNull("+(i+1)+");else\n"); implSb.append(" stat.bindString(" +(i+1)+",entity."+ columnName + ");\n"); } else if (entry.getValue().toLowerCase().startsWith("float")||entry.getValue().toLowerCase().startsWith("double")) { implSb.append(" stat.bindDouble(" +(i+1)+",entity."+ columnName + ");\n"); } else if (entry.getValue().contains("Date")) { // implSb.append(" if(null==entity."+ columnName + ") "); // implSb.append(" stat.bindNull("+(i+1)+"); else\n"); implSb.append(" stat.bindString(" +(i+1)+",createdate);\n"); } } implSb.append(" long result = stat.executeInsert();\n"); implSb.append(" if (result < 0) {\n"); implSb.append(" return false;\n"); implSb.append(" }\n"); implSb.append(" entity.id=(int)result;\n"); implSb.append(" }\n"); implSb.append(" db.setTransactionSuccessful();\n"); implSb.append(" } catch (Exception e) {\n"); implSb.append(" e.printStackTrace();\n"); implSb.append(" return false;\n"); implSb.append(" } finally {\n"); implSb.append(" try {\n"); implSb.append(" if (null != db) {\n"); implSb.appe