databasetask.java

sping下勇Xdoclet开发hibernate的程序实例,版本hibernate2.0,开发工具eclipse3.0

package org.ggyy.util;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.io.DataInputStream;
import java.io.FileInputStream;
import java.io.IOException;
/**
 * 改类负责执行xdoclet生成的sql语句
 * @author jiangyubao
 *
 */
public class DataBaseTask {

	String ddlFile = "src\\sql.ddl";

	public static void main(String[] args) {
		DataBaseTask task = new DataBaseTask();
		task.runTask1();
	}

	public void runTask1() {
		createTableFromDllFileTask();
		//this.test();
		//initDataBase();
	}
	/**
	 * 该函数执行nosqldb_schema.ddl的SQL语句
	 * 
	 * @author jiangyb
	 *  
	 */
	public void createTableFromDllFileTask() {
		DataInputStream dis = null;
		FileInputStream fis = null;
		Connection con = null;
		Statement stmt = null;
		try {
			Class.forName("com.mysql.jdbc.Driver").newInstance();
			con = DriverManager.getConnection("jdbc:mysql:///test");
			stmt = con.createStatement();
			fis = new FileInputStream(this.ddlFile);
			String sqlStr = "";
			dis = new DataInputStream(fis);
			String strRead = null;
			List sqlList = new ArrayList();
			String sql = "";
			while ((strRead = dis.readLine()) != null
					) {
				if (strRead.indexOf("drop") != -1) {
					stmt.addBatch(strRead);
					System.out.println(strRead + ";");
				} else if(strRead.indexOf("alter") != -1) {
					stmt.addBatch(strRead);
					System.out.println(strRead + ";");
				}else if (strRead.indexOf("create") != -1) {
					sql = "";
					sql += strRead;
				} else if (strRead.equals(")")) {
					sql += strRead;
					stmt.addBatch(sql);
					System.out.println(sql + ";");
				} else {
					sql += strRead;
				}
			}
			
			stmt.addBatch("ALTER TABLE tbl_cat TYPE = InnoDB");
			stmt.addBatch("ALTER TABLE tbl_owner TYPE = InnoDB");
			
			stmt.executeBatch();
			System.out.println("execute Success");
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} finally {
			try {
				fis.close();
			} catch (IOException e1) {
				// TODO Auto-generated catch block
				e1.printStackTrace();
			}
			try {
				dis.close();
			} catch (IOException e2) {
				// TODO Auto-generated catch block
				e2.printStackTrace();
			}
			try {
				stmt.close();
			} catch (SQLException e3) {
				// TODO Auto-generated catch block
				e3.printStackTrace();
			}
			try {
				con.close();
			} catch (SQLException e4) {
				// TODO Auto-generated catch block
				e4.printStackTrace();
			}

		}

	}

}