日期:2014-05-16 浏览次数:20477 次
最近做了一个帐套管理的一个功能,需要进行帐套初始化,在初始化的过程中,需要新建一个数据库,同时把上一个帐套的数据库里面的表的表结构全部拿出,在新数据库中全部新建,同时,有些信息在前一个帐套中是需要保留的,如一些基本信息和账目信息,这个时候就会出现表数据的拷贝和更新操作了。逻辑很简单,就是先获取到这些建数据库和建表的语句,以及更新和插入的语句,然后再按照一定的逻辑有顺序的执行它们。在做了这一系列的操作之后就是更改配置文件了,那是属于xml的操作,就不赘述了!
以下代码中用到的db是一个对Spring封装的jdbc的封装
?
主逻辑
List<String> tableNames = this.getTableNames(db); List<String> sqls = this.getCreateTableSql(db, dataBaseName, tableNames); String createDdSql = "create database "+dataBaseName; sqls.add(0, createDdSql); sqls.add(0, "set character_set_results=gbk;"); sqls.add(0, "set character_set_connection=gbk;"); sqls.add(0, "set character_set_client=gbk;"); List<String> insertTableNames = getInsertTableNames(); List<String> insertSqls = this.getInsertSql(dataBaseName, insertTableNames, db); List<String> updateSqls = this.getUpdateSql(dataBaseName, db, cover); sqls.addAll(insertSqls); sqls.addAll(updateSqls); this.executeSql(db, sqls);?
?
/** * 判断数据库是否存在,如果已经存在,就表示该帐套已经初始化了,需要给用户一个提示 * @param db * @param dbName 数据库名 * @return */ public static boolean existDatabase(DB db, String dbName) { boolean flag = false; String sql = "show databases"; SqlRowSet srs = db.executeQuery(sql); while (srs.next()) { String databaseName = srs.getString(1); if (databaseName.equals(dbName)) { flag = true; break; } } return flag; }
?
/** * 获取当前数据库的名字,有了当前数据库的名字可以更精确的从当前数据库中拿数据 * @param tableName * @param db * @return */ public static String getDatabaseName(String tableName, DB db) { String sql = "select * from "+tableName+" limit 0,1"; SqlRowSet srs = db.executeQuery(sql); return srs.getMetaData().getCatalogName(1); }
?
/** * 获得所有的表名 * @param db * @return */ private List<String> getTableNames(DB db) { List<String> tableNames = new ArrayList<String>(); String sql = "show tables"; SqlRowSet srs = db.executeQuery(sql); while (srs.next()) tableNames.add(srs.getString(1)); return tableNames; }
?
?
/** * 获取所有的需要插入数据的表 * * @return */ private List<String> getInsertTableNames() { List<String> tableNames = new ArrayList<String>(); String tables = "t1,t2,t3"; String[] tnames = tables.split(","); for (String tableName:tnames) tableNames.add(tableName); return tableNames; }
?
?
/** * 获取建表语句 * @param dbName新建数据库的名称 * @param tableName 表的名字 * @param db * @return */ public static String createTable(String dbName,String tableName,DB db) { StringBuffer buffer = new StringBuffer("create table ").append(dbName+"."+tableName).append("("); String sql = "select * from " + tableName + " limit 0,1"; SqlRowSet srs = db.executeQuery(sql); SqlRowSetMetaData metaData = srs.getMetaData(); for (int i=0;i<metaData.getColumnCount();i++) { String columnName = metaData.getColumnName((i+1));//字段名 String columnType = metaData.getColumnTypeName((i+1));//字段类型 int displaySize = metaData.getColumnDisplaySize((i+1));//长度 int precision = metaData.getPrecision((i+1));//整数位 int scale = metaData.getScale((i+1));//小数位 if (displaySize > 300) columnType = "text"; if (columnName.equals("s_id")) { buffer.append("s_id int primary key,"); } else if (scale > 0 && "decimal".equalsIgnoreCase(columnType)) { buffer.append(columnName+" "+columnType+"("+precision+","+scale+"),"); } else if ("double".equalsIgnoreCase(columnType) || columnType.toLowerCase().startsWith("date") || columnType.toLowerCase().startsWith("int")) { buffer.append(columnName+" "+columnType+","); } else { buffer.append(columnName+" "+columnType+"("+displaySize+"),"); } // System.