import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

/**
 * @author
 */
public final class MyJDBC {

    private static String connect;
    private static String driverClassName;
    private static String URL;
    private static String username;
    private static String password;
    private static boolean autoCommit;

    /** 宣告一個 Connection型別的靜態屬性，用來快取一個已經存在的連線物件 */
    private static Connection conn;

    static {
	config();
    }

    /**
     * 開頭配置自己的資料庫資訊
     */
    private static void config() {
	/*
	 * 獲取驅動
	 */
	driverClassName = "com.mysql.jdbc.Driver";
	/*
	 * 獲取URL
	 */
	URL = "jdbc:mysql://localhost:3306/spring?useUnicode=true&characterEncoding=utf8";
	/*
	 * 獲取使用者名稱
	 */
	username = "root";
	/*
	 * 獲取密碼
	 */
	password = "1234";
	/*
	 * 設定是否自動提交，一般為false不用改
	 */
	autoCommit = false;

    }

    /**
     * 載入資料庫驅動類
     */
    private static boolean load() {
	try {
	    Class.forName(driverClassName);
	    return true;
	} catch (ClassNotFoundException e) {
	    System.out.println("驅動類 " + driverClassName + " 載入失敗");
	}

	return false;
    }

    /**
     * 專門檢查快取的連線是否不可以被使用 ，不可以被使用的話，就返回 true
     */
    private static boolean invalid() {
	if (conn != null) {
	    try {
		if (conn.isClosed() || !conn.isValid(3)) {
		    return true;
		    /*
		     * isValid方法是判斷Connection是否有效,如果連線尚未關閉並且仍然有效，則返回 true
		     */
		}
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	    /*
	     * conn 既不是 null 且也沒有關閉 ，且 isValid 返回 true，說明是可以使用的 ( 返回 false )
	     */
	    return false;
	} else {
	    return true;
	}
    }

    /**
     * 建立資料庫連線
     */
    public static Connection connect() {
	if (invalid()) { /* invalid為true時，說明連線是失敗的 */
	    /* 載入驅動 */
	    load();
	    try {
		/* 建立連線 */
		conn = DriverManager.getConnection(URL, username, password);
	    } catch (SQLException e) {
		System.out.println("建立 " + connect + " 資料庫連線失敗 , " + e.getMessage());
	    }
	}
	return conn;
    }

    /**
     * 設定是否自動提交事務
     **/
    public static void transaction() {

	try {
	    conn.setAutoCommit(autoCommit);
	} catch (SQLException e) {
	    System.out.println("設定事務的提交方式為 : " + (autoCommit ? "自動提交" : "手動提交") + " 時失敗: " + e.getMessage());
	}

    }

    /**
     * 建立 Statement 物件
     */
    public static Statement statement() {
	Statement st = null;
	connect();
	/* 如果連線是無效的就重新連線 */
	transaction();
	/* 設定事務的提交方式 */
	try {
	    st = conn.createStatement();
	} catch (SQLException e) {
	    System.out.println("建立 Statement 物件失敗: " + e.getMessage());
	}

	return st;
    }

    /**
     * 根據給定的帶引數佔位符的SQL語句，建立 PreparedStatement 物件
     * 
     * @param SQL
     *            帶引數佔位符的SQL語句
     * @return 返回相應的 PreparedStatement 物件
     */
    private static PreparedStatement prepare(String SQL, boolean autoGeneratedKeys) {

	PreparedStatement ps = null;
	connect();
	/* 如果連線是無效的就重新連線 */
	transaction();
	/* 設定事務的提交方式 */
	try {
	    if (autoGeneratedKeys) {
		ps = conn.prepareStatement(SQL, Statement.RETURN_GENERATED_KEYS);
	    } else {
		ps = conn.prepareStatement(SQL);
	    }
	} catch (SQLException e) {
	    System.out.println("建立 PreparedStatement 物件失敗: " + e.getMessage());
	}

	return ps;

    }


    public static ResultSet query(String SQL, Object... params) {

	if (SQL == null || SQL.trim().isEmpty() || !SQL.trim().toLowerCase().startsWith("select")) {
	    throw new RuntimeException("你的SQL語句為空或不是查詢語句");
	}
	ResultSet rs = null;
	if (params.length > 0) {
	    /* 說明 有引數 傳入，就需要處理引數 */
	    PreparedStatement ps = prepare(SQL, false);
	    try {
		for (int i = 0; i < params.length; i++) {
		    ps.setObject(i + 1, params[i]);
		}
		rs = ps.executeQuery();
	    } catch (SQLException e) {
		System.out.println("執行SQL失敗: " + e.getMessage());
	    }
	} else {
	    /* 說明沒有傳入任何引數 */
	    Statement st = statement();
	    try {
		rs = st.executeQuery(SQL); // 直接執行不帶引數的 SQL 語句
	    } catch (SQLException e) {
		System.out.println("執行SQL失敗: " + e.getMessage());
	    }
	}

	return rs;

    }


    private static Object typeof(Object o) {
	Object r = o;

	if (o instanceof java.sql.Timestamp) {
	    return r;
	}
	// 將 java.util.Date 轉成 java.sql.Date
	if (o instanceof java.util.Date) {
	    java.util.Date d = (java.util.Date) o;
	    r = new java.sql.Date(d.getTime());
	    return r;
	}
	// 將 Character 或 char 變成 String
	if (o instanceof Character || o.getClass() == char.class) {
	    r = String.valueOf(o);
	    return r;
	}
	return r;
    }


    public static boolean execute(String SQL, Object... params) {
	if (SQL == null || SQL.trim().isEmpty() || SQL.trim().toLowerCase().startsWith("select")) {
	    throw new RuntimeException("你的SQL語句為空或有錯");
	}
	boolean r = false;
	/* 表示 執行 DDL 或 DML 操作是否成功的一個標識變數 */

	/* 獲得 被執行的 SQL 語句的 字首 */
	SQL = SQL.trim();
	SQL = SQL.toLowerCase();
	String prefix = SQL.substring(0, SQL.indexOf(" "));
	String operation = ""; // 用來儲存操作型別的 變數
	// 根據字首 確定操作
	switch (prefix) {
	case "create":
	    operation = "create table";
	    break;
	case "alter":
	    operation = "update table";
	    break;
	case "drop":
	    operation = "drop table";
	    break;
	case "truncate":
	    operation = "truncate table";
	    break;
	case "insert":
	    operation = "insert :";
	    break;
	case "update":
	    operation = "update :";
	    break;
	case "delete":
	    operation = "delete :";
	    break;
	}
	if (params.length > 0) { // 說明有引數
	    PreparedStatement ps = prepare(SQL, false);
	    Connection c = null;
	    try {
		c = ps.getConnection();
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	    try {
		for (int i = 0; i < params.length; i++) {
		    Object p = params[i];
		    p = typeof(p);
		    ps.setObject(i + 1, p);
		}
		ps.executeUpdate();
		commit(c);
		r = true;
	    } catch (SQLException e) {
		System.out.println(operation + " 失敗: " + e.getMessage());
		rollback(c);
	    }

	} else { // 說明沒有引數

	    Statement st = statement();
	    Connection c = null;
	    try {
		c = st.getConnection();
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	    // 執行 DDL 或 DML 語句，並返回執行結果
	    try {
		st.executeUpdate(SQL);
		commit(c); // 提交事務
		r = true;
	    } catch (SQLException e) {
		System.out.println(operation + " 失敗: " + e.getMessage());
		rollback(c); // 回滾事務
	    }
	}
	return r;
    }

    /*
     * 
     * @param SQL
     *            需要執行的 INSERT 語句
     * @param autoGeneratedKeys
     *            指示是否需要返回由資料庫產生的鍵
     * @param params
     *            將要執行的SQL語句中包含的引數佔位符的 引數值
     * @return 如果指定 autoGeneratedKeys 為 true 則返回由資料庫產生的鍵； 如果指定 autoGeneratedKeys
     *         為 false 則返回受當前SQL影響的記錄數目
     */
    public static int insert(String SQL, boolean autoGeneratedKeys, Object... params) {
	int var = -1;
	if (SQL == null || SQL.trim().isEmpty()) {
	    throw new RuntimeException("你沒有指定SQL語句，請檢查是否指定了需要執行的SQL語句");
	}
	// 如果不是 insert 開頭開頭的語句
	if (!SQL.trim().toLowerCase().startsWith("insert")) {
	    System.out.println(SQL.toLowerCase());
	    throw new RuntimeException("你指定的SQL語句不是插入語句，請檢查你的SQL語句");
	}
	// 獲得 被執行的 SQL 語句的 字首 ( 第一個單詞 )
	SQL = SQL.trim();
	SQL = SQL.toLowerCase();
	if (params.length > 0) { // 說明有引數
	    PreparedStatement ps = prepare(SQL, autoGeneratedKeys);
	    Connection c = null;
	    try {
		c = ps.getConnection(); // 從 PreparedStatement 物件中獲得 它對應的連線物件
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	    try {
		for (int i = 0; i < params.length; i++) {
		    Object p = params[i];
		    p = typeof(p);
		    ps.setObject(i + 1, p);
		}
		int count = ps.executeUpdate();
		if (autoGeneratedKeys) { // 如果希望獲得資料庫產生的鍵
		    ResultSet rs = ps.getGeneratedKeys(); // 獲得資料庫產生的鍵集
		    if (rs.next()) { // 因為是儲存的是單條記錄，因此至多返回一個鍵
			var = rs.getInt(1); // 獲得值並賦值給 var 變數
		    }
		} else {
		    var = count; // 如果不需要獲得，則將受SQL影像的記錄數賦值給 var 變數
		}
		commit(c);
	    } catch (SQLException e) {
		System.out.println("資料儲存失敗: " + e.getMessage());
		rollback(c);
	    }
	} else { // 說明沒有引數
	    Statement st = statement();
	    Connection c = null;
	    try {
		c = st.getConnection(); // 從 Statement 物件中獲得 它對應的連線物件
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	    // 執行 DDL 或 DML 語句，並返回執行結果
	    try {
		int count = st.executeUpdate(SQL);
		if (autoGeneratedKeys) { // 如果企望獲得資料庫產生的鍵
		    ResultSet rs = st.getGeneratedKeys(); // 獲得資料庫產生的鍵集
		    if (rs.next()) { // 因為是儲存的是單條記錄，因此至多返回一個鍵
			var = rs.getInt(1); // 獲得值並賦值給 var 變數
		    }
		} else {
		    var = count; // 如果不需要獲得，則將受SQL影像的記錄數賦值給 var 變數
		}
		commit(c); // 提交事務
	    } catch (SQLException e) {
		System.out.println("資料儲存失敗: " + e.getMessage());
		rollback(c); // 回滾事務
	    }
	}
	return var;
    }

    /** 提交事務 */
    private static void commit(Connection c) {
	if (c != null && !autoCommit) {
	    try {
		c.commit();
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	}
    }

    /** 回滾事務 */
    private static void rollback(Connection c) {
	if (c != null && !autoCommit) {
	    try {
		c.rollback();
	    } catch (SQLException e) {
		e.printStackTrace();
	    }
	}
    }


    /**
     *  釋放資源
     *   **/
    public static void release(Object cloaseable) {

	if (cloaseable != null) {

	    if (cloaseable instanceof ResultSet) {
		ResultSet rs = (ResultSet) cloaseable;
		try {
		    rs.close();
		} catch (SQLException e) {
		    e.printStackTrace();
		}
	    }

	    if (cloaseable instanceof Statement) {
		Statement st = (Statement) cloaseable;
		try {
		    st.close();
		} catch (SQLException e) {
		    e.printStackTrace();
		}
	    }

	    if (cloaseable instanceof Connection) {
		Connection c = (Connection) cloaseable;
		try {
		    c.close();
		} catch (SQLException e) {
		    e.printStackTrace();
		}
	    }

	}

    }

}