package com.itheima.utils;

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

import javax.sql.DataSource;

import com.mchange.v2.c3p0.ComboPooledDataSource;

public class DataSourceUtils {
    private static ComboPooledDataSource ds = new
 
 ComboPooledDataSource();
    private static ThreadLocal<Connection> tl = new ThreadLocal<>();

    /**
     * 獲取資料來源
     *
     * @return 連線池
     */
    public static DataSource getDataSource() {
        return ds;
    }

    /**
     * 從當前執行緒上獲取連線
     *
     * @return 連線
     * @throws
 
 SQLException
     */
    public static Connection getConnection() throws SQLException {
        Connection conn = tl.get();
        if (conn == null) {
            //第一次獲取 建立一個連線和當前的執行緒繫結
            conn = ds.getConnection();

            //繫結
            tl.set(conn);
        }
        return conn;
    }


    /**
     * 釋放資源
     *
     * @param
 
 conn 連線
     * @param st   語句執行者
     * @param rs   結果集
     */
    public static void closeResource(Connection conn, Statement st, ResultSet rs) {
        closeResource(st, rs);
        closeConn(conn);
    }

    public static void closeResource(Statement st, ResultSet rs) {
        closeResultSet(rs);
        closeStatement(st);

    }

    /**
     * 釋放連線
     *
     * @param conn 連線
     */
    public static void closeConn(Connection conn) {
        if (conn != null) {
            try {
                conn.close();
                //和當前執行緒解綁
                tl.remove();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            conn = null;
        }

    }

    /**
     * 釋放語句執行者
     *
     * @param st 語句執行者
     */
    private static void closeStatement(Statement st) {
        if (st != null) {
            try {
                st.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            st = null;
        }

    }

    /**
     * 釋放結果集
     *
     * @param rs 結果集
     */
    private static void closeResultSet(ResultSet rs) {
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            rs = null;
        }
    }

    /**
     * 開啟事務
     *
     * @throws SQLException
     */
    public static void startTransaction() throws SQLException {
        //獲取連線//開啟事務
        getConnection().setAutoCommit(false);
    }

    /**
     * 事務提交
     */
    public static void commitAndClose() {
        try {
            //獲取連線
            Connection conn = getConnection();

            //提交事務
            conn.commit();

            //釋放資源
            conn.close();

            //解除繫結
            tl.remove();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    /**
     * 事務回滾
     */
    public static void rollbackAndClose() {
        try {
            //獲取連線
            Connection conn = getConnection();

            //提交回滾
            conn.rollback();

            //釋放資源
            conn.close();

            //解除繫結
            tl.remove();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}