CreateDBOperation.jsp 6.24 KB

Raw Blame History Permalink

<%@ page import="weaver.conn.mybatis.MysqlScriptRunner,weaver.general.GCONST,weaver.general.TimeUtil,weaver.general.Util" %>
<%@ page import="java.io.*" %>
<%@ page import="java.sql.Connection" %>
<%@ page import="java.sql.Driver" %>
<%@ page import="java.sql.DriverManager" %>
<%@ page import="java.sql.Statement" %>
<%@ page import="java.util.Properties" %>
<jsp:useBean id="EditProperties" class="weaver.system.EditProperties" scope="page" />

<%@ page language="java" contentType="text/html; charset=UTF-8" %>

<HTML><HEAD>
<LINK href="/css/Weaver_wev8.css" type=text/css rel=STYLESHEET>
<SCRIPT language="javascript" src="../../js/weaver_wev8.js"></script>
</head>
<BODY>
<%

String sqlFilePath = GCONST.getSqlPath() ;

String dbserver = Util.null2String(request.getParameter("dbserver"));
String dbport = Util.null2String(request.getParameter("dbport"));
String dbname = Util.null2String(request.getParameter("dbname"));
String username = Util.null2String(request.getParameter("username"));
String password = Util.null2String(request.getParameter("password"));
String dbtype = Util.null2String(request.getParameter("dbtype"));
String isexist = Util.null2String(request.getParameter("isexist"));
String code = Util.null2String(request.getParameter("code"));
String characterSet = Util.null2String(request.getParameter("characterSet"));
char[]  c_code=new char[16];
new FileReader(GCONST.getRootPath()+File.separator+"WEB-INF"+File.separator+"code.key").read(c_code);
String realcode=new String(c_code).trim();
if(!realcode.equals(code)){
%>
无效的验证码！
<%return;}
if(dbtype.equals("")) dbtype="1";
String url = "" ;
String forname = "" ;
Connection conn = null ;
String message="" ;
String path = "";
String linepara = "";
String lastsqlname = "" ;
boolean isOracle12c = false;

try {

	if(dbtype.equals("1")){
		url = "jdbc:sqlserver://"+dbserver+":"+dbport+";DatabaseName=master" ;
		forname = "com.microsoft.sqlserver.jdbc.SQLServerDriver" ;
		path = request.getRealPath("/data/SQLServer");
		linepara = "go" ;
	}else if(dbtype.equals("2")){
		url = "jdbc:oracle:thin:@"+dbserver+":"+dbport+":"+dbname ;
		forname = "oracle.jdbc.OracleDriver" ;
		path = request.getRealPath("/data/Oracle");
		linepara = "/" ;
	}else if(dbtype.equals("3")){
		url = "jdbc:db2:"+dbserver+":"+dbname ;
		forname = "COM.ibm.db2.jdbc.net.DB2Driver" ;
		path = request.getRealPath("/data/DB2");
		linepara = ";" ;
	} else if (dbtype.equals("4")) {
		url = "jdbc:mysql://"+dbserver+":"+dbport+"/"+dbname+"?characterEncoding="+characterSet+"&useSSL=false&autoReconnect=true&failOverReadOnly=false";
		forname = "com.mysql.jdbc.Driver";
		path = request.getRealPath("/data/Mysql");
	}
    boolean isMysql = dbtype.equals("4");

	Driver driver = (Driver)Class.forName(forname).newInstance();
	DriverManager.registerDriver(driver);

	Properties props = new Properties();
	props.put("user", username);
	props.put("password", password);
	if (!isMysql) {
		props.put("CHARSET", "ISO");
	}
	try{
			conn = DriverManager.getConnection(url, props);
		}catch(Exception e){
			String urlTem = "";

			if(dbtype.equals("2")){
				try{
					urlTem = "jdbc:oracle:thin:@"+dbserver+":"+dbport+"/"+dbname ;
					conn = DriverManager.getConnection(urlTem, props);
					isOracle12c = true;
				}catch(Exception e2){
					e2.printStackTrace();
					throw e;
				}
			}else{
				throw e;
			}
		}

	Statement st = conn.createStatement() ;

	String sqlstr = "" ;
	String sqlfile = "";
	String readline = "" ;

	if(isexist.equals("1")){
		if(dbtype.equals("1")){
			sqlstr = " use master ";
				st.executeUpdate(sqlstr) ;
			sqlstr = " use "+dbname;
				st.executeUpdate(sqlstr) ;
		}
		message="";
	}else{
		if(dbtype.equals("1")){
			sqlstr = " use master ";
				st.executeUpdate(sqlstr) ;
			//sqlstr = " create database "+dbname;
			//	st.executeUpdate(sqlstr) ;
			sqlstr = " use "+dbname;
				st.executeUpdate(sqlstr) ;
		}
		File dir = new File(path);
		String fileList[] = dir.list();

		for(int i=0;i<fileList.length;i++){
			for(int j=i+1;j<fileList.length;j++){
				if(fileList[i].compareTo(fileList[j])>0){
					String tmpfileList=fileList[i];
					fileList[i]=fileList[j];
					fileList[j]=tmpfileList;
				}
			}
		}

		for(int i=0;i<fileList.length;i++){

			sqlfile = path + "/" + fileList[i] ;

			System.out.println(TimeUtil.getCurrentTimeString()+"--->"+fileList[i] +" is running .... <br>") ;
//			out.flush() ;

			//out.println(fileList[i]+" ");
			readline = "" ;
			sqlstr = "" ;
			File fin = new File(sqlfile);

//            boolean isProc = "5function.sql".equals(fin.getName()) || "6trigger.sql".equals(fin.getName()) || "8procedure.sql".equals(fin.getName()) ;
			BufferedReader is;
            is  = new BufferedReader(new InputStreamReader(new FileInputStream(fin),"GBK"));
            if (isMysql) {
                MysqlScriptRunner runner = new MysqlScriptRunner(conn,false,false);
                runner.runScript(is);
            } else {
                while ((readline = is.readLine()) != null)   {
                    readline = readline.trim() ;
                    if(!readline.equalsIgnoreCase(linepara)) sqlstr += " "+ readline ;
                    else if( !(sqlstr.trim()).equals("") ) {
                        message += sqlstr+"<br>" ;
                        st.executeUpdate(sqlstr) ;
                        sqlstr = "" ;
                        message = "" ;
                    }
                }
            }
			is.close();
			lastsqlname = fileList[i];

			System.out.println(TimeUtil.getCurrentTimeString()+"--->"+lastsqlname+" execute success ! <br>") ;
//			out.flush() ;
		}
		message="";
	}

}catch(Exception e) { message += e.toString(); }
finally {
	try { conn.close() ;} catch(Exception ex) {}
}

if(message.equals("")){
	EditProperties.editProp(dbserver,dbport,dbname,username,password,dbtype, characterSet,isOracle12c);
	EditProperties.editIndex();
	if(!lastsqlname.equals("")) EditProperties.editDblog(dbserver,dbname,dbtype,lastsqlname);
	message=" Success ! " ;
	message+="<br><br>请重启Resin服务，在Resin启动成功后再点击下方的登入系统！";
	message+="<br><br><a href=/index.htm target=_top>登入系统</a>";

	//response.sendRedirect("/index.htm");
}else {
	message=" error ! message is below : <br>" + message;
	message+="<br><br><a href=/system/CreateDB.jsp>返回</a>";
}

%>
<%=message%>
</BODY>
</HTML>