|
import java.sql.*;
import java.io.*;
/**
* 執行SQL語句的demo.
**/
public class ExecuteSQL {
public static void main(String[] args) {
Connection conn = null; // 定義連接數據庫的JDBC Connection對象
try {
String driver = null, url = null, user = "", password = "";
// 讀取輸入的參數
for(int n = 0; n < args.length; n++) {
if (args[n].equals("-d")) driver = args[++n];
else if (args[n].equals("-u")) user = args[++n];
else if (args[n].equals("-p")) password = args[++n];
else if (url == null) url = args[n];
else throw new IllegalArgumentException("Unknown argument.");
}
// 數據庫的url參數必須輸入.
if (url == null)
throw new IllegalArgumentException("No database specified");
// 如果啟動程序的時候輸入了驅動，那么使用用戶輸入的參數動態注冊數據庫驅動
if (driver != null) Class.forName(driver);
// 建立到數據庫的連接。jdbc會使用它所知道的所有的數據庫驅動來建立與數據庫的連接。
conn = DriverManager.getConnection(url, user, password);
// 創建statement對象來與數據庫進行交互
Statement s = conn.createStatement();
//從命令提示符中讀取sql命令
BufferedReader in =
new BufferedReader(new InputStreamReader(System.in));
//循環執行用戶輸入的sql語句
while(true) {
System.out.print("sql> "); // 提示用戶可以繼續輸入
System.out.flush(); // 顯示提示符.
String sql = in.readLine(); // 讀取用戶輸入
// 輸入“quit”，退出程序.
if ((sql == null) || sql.equals("quit")) break;
// 忽略空行輸入
if (sql.length() == 0) continue;
// 執行輸入的sql命令，并顯示執行結果.
try {
// 使用execute進行判斷用戶輸入的是查詢還是更新，如果是更新則返回”假“
boolean status = s.execute(sql);
// 一些復雜的查詢返回不只一個記錄集，所以循環讀取所有的記錄集
do {
if (status) { //status為”真“，查詢并返回記錄集
ResultSet rs = s.getResultSet(); // 讀取記錄集
printResultsTable(rs, System.out); // 顯示記錄集
}
else {
// status為”假“，進行更新操作，并返回操作所影響的行數
int numUpdates = s.getUpdateCount();
System.out.println("Ok. " + numUpdates +
" rows affected.");
}
// 判斷是否有更多的記錄集
status = s.getMoreResults();
} while(status || s.getUpdateCount() != -1);
}
// 如果sql語句執行出錯，打印出錯信息
catch (SQLException e) {
System.err.println("SQLException: " + e.getMessage()+ ":" +
e.getSQLState());
}
finally { // 打印所有的警告信息
SQLWarning w;
for(w=conn.getWarnings(); w != null; w=w.getNextWarning())
System.err.println("WARNING: " + w.getMessage() +
":" + w.getSQLState());
}
}
}
//打印連接數據庫失敗等錯誤信息，并顯示出來
catch (Exception e) {
System.err.println(e);
if (e instanceof SQLException)
System.err.println("SQL State: " +
((SQLException)e).getSQLState());
System.err.println("Usage: java ExecuteSQL [-d <driver>] " +
"[-u <user>] [-p <password>] <database URL>");
}
// 每次退出程序或程序出現錯誤時關閉數據庫連接。關閉數據庫連接的同時也關閉了所有打開的statements和記錄集
finally {
try { conn.close(); } catch (Exception e) {}
}
}
/**
* printResultsTable方法輸出記錄集的內容，此方法依賴于ResultSetMetaData類
**/
static void printResultsTable(ResultSet rs, OutputStream output)
throws SQLException
{
// 建立輸出流
PrintWriter out = new PrintWriter(output);
// 讀取記錄集
ResultSetMetaData metadata = rs.getMetaData();
int numcols = metadata.getColumnCount(); // 列的數目
String[] labels = new String[numcols]; // 列標題
int[] colwidths = new int[numcols];
int[] colpos = new int[numcols];
int linewidth;
linewidth = 1;
for(int i = 0; i < numcols; i++) { // 保存每一列的位置和標簽
colpos[i] = linewidth;
labels[i] = metadata.getColumnLabel(i+1);
int size = metadata.getColumnDisplaySize(i+1);
if (size == -1) size = 30;
if (size > 500) size = 30;
int labelsize = labels[i].length();
if (labelsize > size) size = labelsize;
colwidths[i] = size + 1;
linewidth += colwidths[i] + 2;
}
StringBuffer divider = new StringBuffer(linewidth);
StringBuffer blankline = new StringBuffer(linewidth);
for(int i = 0; i < linewidth; i++) {
divider.insert(i, '-');
blankline.insert(i, " ");
}
for(int i=0; i<numcols; i++) divider.setCharAt(colpos[i]-1,'+');
divider.setCharAt(linewidth-1, '+');
out.println(divider);
StringBuffer line = new StringBuffer(blankline.toString());
line.setCharAt(0, '|');
for(int i = 0; i < numcols; i++) {
int pos = colpos[i] + 1 + (colwidths[i]-labels[i].length())/2;
overwrite(line, pos, labels[i]);
overwrite(line, colpos[i] + colwidths[i], " |");
}
out.println(line);
out.println(divider);
// 使用next()方法遍歷記錄集，使用getObject方法獲得列的數值并輸出
while(rs.next()) {
line = new StringBuffer(blankline.toString());
line.setCharAt(0, '|');
for(int i = 0; i < numcols; i++) {
Object value = rs.getObject(i+1);
if (value != null)
overwrite(line, colpos[i] + 1, value.toString().trim());
overwrite(line, colpos[i] + colwidths[i], " |");
}
out.println(line);
}
out.println(divider);
out.flush();
}
static void overwrite(StringBuffer b, int pos, String s) {
int slen = s.length();
int blen = b.length();
if (pos+slen > blen) slen = blen-pos;
for(int i = 0; i < slen; i++)
b.setCharAt(pos+i, s.charAt(i));
}
}
執行時如下所示：
javac ExecuteSQL.java
java ExecuteSQL -d com.microsoft.jdbc.sqlserver.SQLServerDrive
r -u username -p password jdbc:microsoft:sqlserver://localhost:1433;databaseName=pubs |