/*
 * Created on 2005-11-6
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package com.mycompany.database;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.PropertyResourceBundle;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;

/**
 * @author Administrator
 * 
 * TODO To change the template for this generated type comment go to Window -
 * Preferences - Java - Code Style - Code Templates
 */
public class Database {
	/**
	 * 數(shù)據(jù)庫(kù)訪問(wèn)URL
	 */
	private static String url;

	/**
	 * 數(shù)據(jù)庫(kù)驅(qū)動(dòng)
	 */
	private static String driver;

	/**
	 * 數(shù)據(jù)庫(kù)訪問(wèn)用戶(hù)名
	 */
	private static String username;

	/**
	 * 數(shù)據(jù)庫(kù)訪問(wèn)口令
	 */
	private static String password;

	/**
	 * 訪問(wèn)類(lèi)型
	 */
	private static String type;

	/**
	 * 數(shù)據(jù)源名稱(chēng)
	 */
	private static String datasource;

	/**
	 * 配置文件名稱(chēng)
	 */
	private final static String fileName = "database";

	private static ThreadLocal connection = new ThreadLocal();

	static {
		config();
	}

	private static void config() {
		//讀取系統(tǒng)配置
		PropertyResourceBundle resourceBundle = (PropertyResourceBundle) PropertyResourceBundle
				.getBundle(fileName);
		//將系統(tǒng)設(shè)置賦值給類(lèi)變量
		Enumeration enu = resourceBundle.getKeys();
		while (enu.hasMoreElements()) {
			String propertyName = enu.nextElement().toString();
			if (propertyName.equals("database.url"))
				url = resourceBundle.getString("database.url");
			if (propertyName.equals("database.driver"))
				driver = resourceBundle.getString("database.driver");
			if (propertyName.equals("database.username"))
				username = resourceBundle.getString("database.username");
			if (propertyName.equals("database.password"))
				password = resourceBundle.getString("database.password");
			if (propertyName.equals("database.type"))
				type = resourceBundle.getString("database.type");
			if (propertyName.equals("database.datasource"))
				datasource = resourceBundle.getString("database.datasource");
		}

	}

	/**
	 * 取得數(shù)據(jù)庫(kù)連接
	 * @return
	 * @throws SQLException
	 */
	public synchronized static java.sql.Connection getConnection()
			throws SQLException {
		Connection con = (Connection) connection.get();
		if (con != null && !con.isClosed()) {
			return con;
		}
		if ("pooled".equalsIgnoreCase(type)) {
			//從JNDI中取得數(shù)據(jù)源
			try {
				//此處對(duì)于不同的應(yīng)用服務(wù)器，對(duì)env傳入不同
				Hashtable env = new Hashtable();
				//此處對(duì)于不同的應(yīng)用服務(wù)器，對(duì)env傳入不同
				Context ctx = new InitialContext(env); // 從命名系統(tǒng)中獲取 DataSource 工廠對(duì)象
				DataSource dataSource = (DataSource) ctx.lookup(datasource);
				con= dataSource.getConnection();
				connection.set(con);
				return con;
			} catch (NamingException e) {
				e.printStackTrace();
			}
		} else {
			//直接使用JDBC驅(qū)動(dòng)連接
			try {
				Class providerClass = Class.forName(driver);
				con = DriverManager.getConnection(url, username,
						password);
				con.setAutoCommit(false);
				connection.set(con);
				return con;

			} catch (ClassNotFoundException e) {
				e.printStackTrace();
			}
		}
		return null;
	}

	public static void commit() {
		Connection con = (Connection) connection.get();
		try {
			con.commit();
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public static void rollback() {
		Connection con = (Connection) connection.get();
		try {
			con.rollback();
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public synchronized static void releaseConnection(Connection connection) {

		try {
			if (connection != null && !connection.isClosed())
				connection.close();
		} catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		connection = null;
	}
public static void main(String[] args) {
	
	try {
		Database.getConnection();
	} catch (SQLException e) {
		// TODO Auto-generated catch block
		e.printStackTrace();
	}
}
}