dbconfig配置

在QMaster分类里记录的相关源码是个人认为写得比较好、比较有用（可复用程度比较高）的一部分，并非QMaster的全部源码。欢迎评论。

dbconfig.xml

<?xml version=”1.0″ encoding=”GB2312″?>
<!–数据库配置文件–>
<dbconfig>
<jdbcdriver>com.mysql.jdbc.Driver</jdbcdriver><!–JDBC Driver–>
<jdbcurl>jdbc:mysql://localhost:3306/questionnaire</jdbcurl><!–JDBC URL–>
<dbuser>root</dbuser><!–数据库用户名–>
<dbpwd>toot</dbpwd><!–数据库密码–>
</dbconfig>

DBConfig.java

/**
* 功能：解析dbconfig.xml文件，提供数据库配置。
*/
package db;

import java.io.IOException;
import java.io.InputStream;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

 

public class DBConfig {
//对应dbconfig.xml中的配置
private String jdbcDriver;
private String jdbcUrl;
private String dbUser;
private String dbPwd;

DBConfig() throws ParserConfigurationException, SAXException, IOException {
DocumentBuilderFactory docBuildFac = DocumentBuilderFactory.newInstance();//生成一个DOM解析器工厂
InputStream is = this.getClass().getClassLoader().getResourceAsStream(“dbconfig.xml”);//相对位置src/dbconfig.xml
Document doc = docBuildFac.newDocumentBuilder().parse(is);//生成一个DOM解析器，并解析dbconfig.xml生成Doc。
Element root = doc.getDocumentElement();//根结点
NodeList nodeList = root.getChildNodes();//根结点的子结点链表
for(int i = 0; i < nodeList.getLength(); i++) {
Node node = nodeList.item(i);//子结点
if(node.getNodeType() == Node.ELEMENT_NODE) {
if(node.getNodeName().equals(“jdbcdriver”)) {
jdbcDriver = node.getFirstChild().getNodeValue();//DOM把<node>value</node>当成两层结点
} else if(node.getNodeName().equals(“jdbcurl”)) {
jdbcUrl = node.getFirstChild().getNodeValue();
} else if(node.getNodeName().equals(“dbuser”)) {
dbUser = node.getFirstChild().getNodeValue();
} else if(node.getNodeName().equals(“dbpwd”)) {
dbPwd = node.getFirstChild().getNodeValue();
}
}
}
}

public String getDbPwd() {
return dbPwd;
}

public String getDbUser() {
return dbUser;
}

public String getJdbcDriver() {
return jdbcDriver;
}

public String getJdbcUrl() {
return jdbcUrl;
}

}

DB.java

/**
* 功能：连接、关闭数据库
*/
package db;

import java.io.IOException;
import java.sql.*;
import javax.xml.parsers.ParserConfigurationException;
import org.xml.sax.SAXException;

public class DB {
private DBConfig config = null;
private Connection conn = null;

public DB() throws ClassNotFoundException, ParserConfigurationException, SAXException, IOException {
config = new DBConfig();
Class.forName(config.getJdbcDriver());
}

public Connection getConnection() throws SQLException {
conn = DriverManager.getConnection(config.getJdbcUrl(), config.getDbUser(), config.getDbPwd());
return conn;
}

public void close() throws SQLException {
if(conn != null)
conn.close();
}
}

DBBak.java

/**
* 功能：备份、还原数据库
*/
package db;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Calendar;
import javax.xml.parsers.ParserConfigurationException;
import org.xml.sax.SAXException;

public class DBBak {
private DBConfig config = null;

public DBBak() throws ParserConfigurationException, SAXException, IOException {
config = new DBConfig();
}

public String getFileName() {
Calendar today = Calendar.getInstance();
String year = String.valueOf(today.get(Calendar.YEAR));
String month = String.valueOf(today.get(Calendar.MONTH) + 1);//0~11表示1~12月
String date = String.valueOf(today.get(Calendar.DATE));
String time = String.valueOf(today.getTimeInMillis());
return year + “-” + month + “-” + date + “-” + time + “.sql”;
}

public void backUp(String path) {
try {
String jdbcUrl = config.getJdbcUrl();
int pos = jdbcUrl.lastIndexOf(“/”);
String dbName = jdbcUrl.substring(pos+1);
String cmd = “mysqldump -u” + config.getDbUser() + ” -p” + config.getDbPwd() + ” ” + dbName;
Runtime rt = Runtime.getRuntime();
Process child = rt.exec(cmd);
InputStream in = child.getInputStream();
InputStreamReader xx = new InputStreamReader(in, “ISO-8859-1″);
String inStr;
StringBuffer sb = new StringBuffer(“”);
String outStr;
BufferedReader br = new BufferedReader(xx);
while ((inStr = br.readLine()) != null) {
sb.append(inStr + “/r/n”);
}
outStr = new String(sb.toString().getBytes(“ISO-8859-1″), “UTF-8″);
FileOutputStream fout = new FileOutputStream(path);
OutputStreamWriter writer = new OutputStreamWriter(fout,”UTF-8″);
writer.write(outStr);
writer.flush();
in.close();
xx.close();
br.close();
writer.close();
fout.close();
}
catch (Exception e) {
e.printStackTrace();
}
}

public void load(String path){
try{
File f = new File(path);
FileInputStream fis = new FileInputStream(f);
String jdbcUrl = config.getJdbcUrl();
int pos = jdbcUrl.lastIndexOf(“/”);
String dbName = jdbcUrl.substring(pos+1);
String cmd = “mysql -u” + config.getDbUser() + ” -p” + config.getDbPwd() + ” ” + dbName;
Runtime rt = Runtime.getRuntime();
Process child = rt.exec(cmd);
OutputStream out = child.getOutputStream();
String inStr;
StringBuffer sb = new StringBuffer(“”);
String outStr;
BufferedReader br = new BufferedReader(new InputStreamReader(fis, “UTF-8″));
while ((inStr = br.readLine()) != null) {
sb.append(inStr + “/r/n”);
}
outStr =new String(sb.toString().getBytes(“UTF-8″), “ISO-8859-1″);
OutputStreamWriter writer = new OutputStreamWriter(out, “ISO-8859-1″);
writer.write(outStr);
writer.flush();
out.close();
br.close();
writer.close();
fis.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}

FileList.java

/**
* 功能：列出数据库备份文件
*/
package db;

import java.io.File;
import java.util.ArrayList;

public class FileList {
public ArrayList tree(File f) {//传入目录
ArrayList files = new ArrayList();
File[] list = f.listFiles();
if(list != null){
for(int i = 0; i < list.length; i++) {
if(list[i].isFile()) {
if(list[i].getName().endsWith(“.sql”)) {
files.add(list[i].getName());
}
}
}
}
return files;
}
}