import com.qf.sparkproject.sessionanalyze.conf.ConfigurationManager;
import com.qf.sparkproject.sessionanalyze.constant.Constants;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.LinkedList;
import java.util.List;


/**
 * JDBC輔助組件
 * @author Administrator
 *
 */
public class JDBCHelper {
    
    // 第一步：在靜態(tài)代碼塊中皂林，直接加載數(shù)據(jù)庫的驅(qū)動(dòng)
    // 加載驅(qū)動(dòng)础倍，不是直接簡(jiǎn)單的沟启，使用com.mysql.jdbc.Driver就可以了
    // 之所以說犹菇，不要硬編碼，他的原因就在于這里
    // 
    // com.mysql.jdbc.Driver只代表了MySQL數(shù)據(jù)庫的驅(qū)動(dòng)
    // 那么胳搞，如果有一天肌毅，我們的項(xiàng)目底層的數(shù)據(jù)庫要進(jìn)行遷移姑原，比如遷移到Oracle
    // 或者是DB2、SQLServer
    // 那么笨奠，就必須很費(fèi)勁的在代碼中艰躺，找眨八，找到硬編碼了com.mysql.jdbc.Driver的地方，然后改成
    // 其他數(shù)據(jù)庫的驅(qū)動(dòng)類的類名
    // 所以正規(guī)項(xiàng)目页响，是不允許硬編碼的闰蚕，那樣維護(hù)成本很高
    // 
    // 通常，我們都是用一個(gè)常量接口中的某個(gè)常量涩哟，來代表一個(gè)值
    // 然后在這個(gè)值改變的時(shí)候盼玄，只要改變常量接口中的常量對(duì)應(yīng)的值就可以了
    // 
    // 項(xiàng)目埃儿，要盡量做成可配置的
    // 就是說，我們的這個(gè)數(shù)據(jù)庫驅(qū)動(dòng)精钮，更進(jìn)一步剃斧，也不只是放在常量接口中就可以了
    // 最好的方式幼东，是放在外部的配置文件中，跟代碼徹底分離
    // 常量接口中策橘，只是包含了這個(gè)值對(duì)應(yīng)的key的名字
    static {
        try {
            String driver = ConfigurationManager.getProperty(Constants.JDBC_DRIVER);
            Class.forName(driver);
        } catch (Exception e) {
            e.printStackTrace();  
        }
    }

    
    // 第二步娜亿，實(shí)現(xiàn)JDBCHelper的單例模式
    // 為什么要實(shí)現(xiàn)單例模式呢买决？因?yàn)樗膬?nèi)部要封裝一個(gè)簡(jiǎn)單的內(nèi)部的數(shù)據(jù)庫連接池
    // 為了保證數(shù)據(jù)庫連接池有且僅有一份，所以就通過單例的方式
    // 保證JDBCHelper只有一個(gè)實(shí)例嘁灯，實(shí)例中只有一份數(shù)據(jù)庫連接池
    private static JDBCHelper instance = null;
    
    /**
     * 獲取單例
     * @return 單例
     */
    public static JDBCHelper getInstance() {
        if(instance == null) {
            synchronized(JDBCHelper.class) {
                if(instance == null) {
                    instance = new JDBCHelper();
                }
            }
        }
        return instance;
    }
    
    // 數(shù)據(jù)庫連接池
    private LinkedList<Connection> datasource = new LinkedList<Connection>();
    
    /**
     * 
     * 第三步：實(shí)現(xiàn)單例的過程中丑婿，創(chuàng)建唯一的數(shù)據(jù)庫連接池
     * 
     * 私有化構(gòu)造方法
     * 
     * JDBCHelper在整個(gè)程序運(yùn)行聲明周期中，只會(huì)創(chuàng)建一次實(shí)例
     * 在這一次創(chuàng)建實(shí)例的過程中秒旋，就會(huì)調(diào)用JDBCHelper()構(gòu)造方法
     * 此時(shí)迁筛，就可以在構(gòu)造方法中细卧，去創(chuàng)建自己唯一的一個(gè)數(shù)據(jù)庫連接池
     * 
     */
    private JDBCHelper() {
        // 首先第一步筒占，獲取數(shù)據(jù)庫連接池的大小赋铝，就是說革骨，數(shù)據(jù)庫連接池中要放多少個(gè)數(shù)據(jù)庫連接
        // 這個(gè)析恋，可以通過在配置文件中配置的方式助隧，來靈活的設(shè)定
        int datasourceSize = ConfigurationManager.getInteger(
                Constants.JDBC_DATASOURCE_SIZE);
        
        // 然后創(chuàng)建指定數(shù)量的數(shù)據(jù)庫連接并村，并放入數(shù)據(jù)庫連接池中
        for(int i = 0; i < datasourceSize; i++) {
            boolean local = ConfigurationManager.getBoolean(Constants.SPARK_LOCAL);
            String url = null;
            String user = null;
            String password = null;
            
            if(local) {
                url = ConfigurationManager.getProperty(Constants.JDBC_URL);
                user = ConfigurationManager.getProperty(Constants.JDBC_USER);
                password = ConfigurationManager.getProperty(Constants.JDBC_PASSWORD);
            } else {
                url = ConfigurationManager.getProperty(Constants.JDBC_URL_PROD);
                user = ConfigurationManager.getProperty(Constants.JDBC_USER_PROD);
                password = ConfigurationManager.getProperty(Constants.JDBC_PASSWORD_PROD);
            }
            
            try {
                Connection conn = DriverManager.getConnection(url, user, password);
                datasource.push(conn);  
            } catch (Exception e) {
                e.printStackTrace(); 
            }
        }
    }
    
    /**
     * 第四步哩牍，提供獲取數(shù)據(jù)庫連接的方法
     * 有可能，你去獲取的時(shí)候丸边，這個(gè)時(shí)候荚孵，連接都被用光了，你暫時(shí)獲取不到數(shù)據(jù)庫連接
     * 所以我們要自己編碼實(shí)現(xiàn)一個(gè)簡(jiǎn)單的等待機(jī)制骄呼，去等待獲取到數(shù)據(jù)庫連接
     * 
     */
    public synchronized Connection getConnection() {
        while(datasource.size() == 0) {
            try {
                Thread.sleep(10);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }  
        }
        return datasource.poll();
    }
    
    /**
     * 第五步：開發(fā)增刪改查的方法
     * 1谒麦、執(zhí)行增刪改SQL語句的方法
     * 2、執(zhí)行查詢SQL語句的方法
     * 3患膛、批量執(zhí)行SQL語句的方法
     */
    
    /**
     * 執(zhí)行增刪改SQL語句
     * @param sql 
     * @param params
     * @return 影響的行數(shù)
     */
    public int executeUpdate(String sql, Object[] params) {
        int rtn = 0;
        Connection conn = null;
        PreparedStatement pstmt = null;
        
        try {
            conn = getConnection();
            // 取消自動(dòng)提交
            conn.setAutoCommit(false);  
            
            pstmt = conn.prepareStatement(sql);
            
            if(params != null && params.length > 0) {
                for(int i = 0; i < params.length; i++) {
                    pstmt.setObject(i + 1, params[i]);  
                }
            }
            
            rtn = pstmt.executeUpdate();
            
            conn.commit();
        } catch (Exception e) {
            e.printStackTrace();  
        } finally {
            if(conn != null) {
                datasource.push(conn);  
            }
        }
        
        return rtn;
    }
    
    /**
     * 執(zhí)行查詢SQL語句
     * @param sql
     * @param params
     * @param callback
     */
    public void executeQuery(String sql, Object[] params, 
            QueryCallback callback) {
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        
        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            
            if(params != null && params.length > 0) {
                for(int i = 0; i < params.length; i++) {
                    pstmt.setObject(i + 1, params[i]);   
                }
            }
            
            rs = pstmt.executeQuery();
            
            callback.process(rs);  
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if(conn != null) {
                datasource.push(conn);  
            }
        }
    }
    
    /**
     * 批量執(zhí)行SQL語句
     * 
     * 批量執(zhí)行SQL語句，是JDBC中的一個(gè)高級(jí)功能
     * 默認(rèn)情況下跃捣，每次執(zhí)行一條SQL語句疚漆，就會(huì)通過網(wǎng)絡(luò)連接刁赦，向MySQL發(fā)送一次請(qǐng)求
     * 
     * 但是甚脉，如果在短時(shí)間內(nèi)要執(zhí)行多條結(jié)構(gòu)完全一模一樣的SQL，只是參數(shù)不同
     * 雖然使用PreparedStatement這種方式狡耻，可以只編譯一次SQL，提高性能猴凹，但是夷狰，還是對(duì)于每次SQL
     * 都要向MySQL發(fā)送一次網(wǎng)絡(luò)請(qǐng)求
     * 
     * 可以通過批量執(zhí)行SQL語句的功能優(yōu)化這個(gè)性能
     * 一次性通過PreparedStatement發(fā)送多條SQL語句，比如100條郊霎、1000條孵淘，甚至上萬條
     * 執(zhí)行的時(shí)候，也僅僅編譯一次就可以
     * 這種批量執(zhí)行SQL語句的方式歹篓，可以大大提升性能
     * 
     * @param sql
     * @param paramsList
     * @return 每條SQL語句影響的行數(shù)
     */
    public int[] executeBatch(String sql, List<Object[]> paramsList) {
        int[] rtn = null;
        Connection conn = null;
        PreparedStatement pstmt = null;
        
        try {
            conn = getConnection();
            
            // 第一步：使用Connection對(duì)象瘫证，取消自動(dòng)提交
            conn.setAutoCommit(false);  
            
            pstmt = conn.prepareStatement(sql);
            
            // 第二步：使用PreparedStatement.addBatch()方法加入批量的SQL參數(shù)
            if(paramsList != null && paramsList.size() > 0) {
                for(Object[] params : paramsList) {
                    for(int i = 0; i < params.length; i++) {
                        pstmt.setObject(i + 1, params[i]);  
                    }
                    pstmt.addBatch();
                }
            }
            
            // 第三步：使用PreparedStatement.executeBatch()方法，執(zhí)行批量的SQL語句
            rtn = pstmt.executeBatch();
            
            // 最后一步：使用Connection對(duì)象庄撮，提交批量的SQL語句
            conn.commit();
        } catch (Exception e) {
            e.printStackTrace();  
        } finally {
            if(conn != null) {
                datasource.push(conn);  
            }
        }
        
        return rtn;
    }
    
    /**
     * 靜態(tài)內(nèi)部類：查詢回調(diào)接口
     * @author Administrator
     *
     */
    public static interface QueryCallback {
        
        /**
         * 處理查詢結(jié)果
         * @param rs 
         * @throws Exception
         */
        void process(ResultSet rs) throws Exception;
        
    }
    
}