代码语言：javascript复制
package com.kaigejava.cms;
import static com.kaigejava.common.web.Constants.UTF8;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
/**
* 安装类
*
* @author liufang
*
*/
public class Install {
public static void dbXml(String fileName, String dbHost, String dbPort,
String dbName, String dbUser, String dbPassword) throws Exception {
String s = FileUtils.readFileToString(new File(fileName));
s = StringUtils.replace(s, "DB_HOST", dbHost);
s = StringUtils.replace(s, "DB_PORT", dbPort);
s = StringUtils.replace(s, "DB_NAME", dbName);
s = StringUtils.replace(s, "DB_USER", dbUser);
s = StringUtils.replace(s, "DB_PASSWORD", dbPassword);
FileUtils.writeStringToFile(new File(fileName), s);
}
public static Connection getConn(String dbHost, String dbPort,
String dbName, String dbUser, String dbPassword) throws Exception {
Class.forName("com.mysql.jdbc.Driver");
Class.forName("com.mysql.jdbc.Driver").newInstance();
String connStr = "jdbc:mysql://" dbHost ":" dbPort "/" dbName
"?user=" dbUser "&password=" dbPassword
"&characterEncoding=utf8";
Connection conn = DriverManager.getConnection(connStr);
return conn;
}
public static void webXml(String fromFile, String toFile) throws Exception {
FileUtils.copyFile(new File(fromFile), new File(toFile));
}
/**
* 创建数据库
*
* @param dbHost
* @param dbName
* @param dbPort
* @param dbUser
* @param dbPassword
* @throws Exception
*/
public static void createDb(String dbHost, String dbPort, String dbName,
String dbUser, String dbPassword) throws Exception {
Class.forName("com.mysql.jdbc.Driver");
Class.forName("com.mysql.jdbc.Driver").newInstance();
String connStr = "jdbc:mysql://" dbHost ":" dbPort "?user="
dbUser "&password=" dbPassword
"&characterEncoding=UTF8";
Connection conn = DriverManager.getConnection(connStr);
Statement stat = conn.createStatement();
String sql = "drop database if exists " dbName;
stat.execute(sql);
sql = "create database " dbName " CHARACTER SET UTF8";
stat.execute(sql);
stat.close();
conn.close();
}
public static void changeDbCharset(String dbHost, String dbPort,
String dbName, String dbUser, String dbPassword) throws Exception {
Connection conn = getConn(dbHost, dbPort, dbName, dbUser, dbPassword);
Statement stat = conn.createStatement();
String sql = "ALTER DATABASE " dbName " CHARACTER SET UTF8";
stat.execute(sql);
stat.close();
conn.close();
}
/**
* 创建表
*
* @param dbHost
* @param dbName
* @param dbPort
* @param dbUser
* @param dbPassword
* @param sqlList
* @throws Exception
*/
public static void createTable(String dbHost, String dbPort, String dbName,
String dbUser, String dbPassword, List<String> sqlList)
throws Exception {
Connection conn = getConn(dbHost, dbPort, dbName, dbUser, dbPassword);
Statement stat = conn.createStatement();
for (String dllsql : sqlList) {
System.out.println(dllsql);
stat.execute(dllsql);
}
stat.close();
conn.close();
}
/**
* 更新配置
*
* @param dbHost
* @param dbName
* @param dbPort
* @param dbUser
* @param dbPassword
* @param domain
* @param cxtPath
* @param port
* @throws Exception
*/
public static void updateConfig(String dbHost, String dbPort,
String dbName, String dbUser, String dbPassword, String domain,
String cxtPath, String port) throws Exception {
Connection conn = getConn(dbHost, dbPort, dbName, dbUser, dbPassword);
Statement stat = conn.createStatement();
String sql = "update jc_site set domain='" domain "'";
stat.executeUpdate(sql);
sql = "update jc_config set context_path='" cxtPath "',port="
port;
stat.executeUpdate(sql);
stat.close();
conn.close();
}
/**
* 读取sql语句。“/*”开头为注释，“;”为sql结束。
*
* @param fileName
* sql文件地址
* @return list of sql
* @throws Exception
*/
public static List<String> readSql(String fileName) throws Exception {
BufferedReader br = new BufferedReader(new InputStreamReader(
new FileInputStream(fileName), UTF8));
List<String> sqlList = new ArrayList<String>();
StringBuilder sqlSb = new StringBuilder();
String s = null;
while ((s = br.readLine()) != null) {
if (s.startsWith("/*") || s.startsWith("#")
|| StringUtils.isBlank(s)) {
continue;
}
if (s.endsWith(";")) {
sqlSb.append(s);
sqlSb.setLength(sqlSb.length() - 1);
sqlList.add(sqlSb.toString());
sqlSb.setLength(0);
} else {
sqlSb.append(s);
}
}
br.close();
return sqlList;
}
}
