日期:2014-05-16  浏览次数:20450 次

帐套管理,新建数据库,拷贝表结构和相应数据

最近做了一个帐套管理的一个功能,需要进行帐套初始化,在初始化的过程中,需要新建一个数据库,同时把上一个帐套的数据库里面的表的表结构全部拿出,在新数据库中全部新建,同时,有些信息在前一个帐套中是需要保留的,如一些基本信息和账目信息,这个时候就会出现表数据的拷贝和更新操作了。逻辑很简单,就是先获取到这些建数据库和建表的语句,以及更新和插入的语句,然后再按照一定的逻辑有顺序的执行它们。在做了这一系列的操作之后就是更改配置文件了,那是属于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.