package com.dataquery.model;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

//import oracle.jdbc.driver.OracleDriver;

public class DBConnection {
	
	private Connection conn;

	private Properties pro = new Properties();

	private static final String DB_IP = "DB.IP";

	private static final String DB_PORT = "DB.PORT";

	private static final String DB_USER = "DB.USER";

	private static final String DB_PASSWORD = "DB.PASSWORD";

	private static final String DB_SID = "DB.SID";

	public DBConnection() {
		// new OracleDriver();
		try {
			// 使用IO流打開屬性文件,并將屬性文件中的值對存儲至pro對象中
			pro.load(DBConnection.class
					.getResourceAsStream("DbConfig.properties"));
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	public Connection getConnection() {
		String ip = pro.getProperty(DB_IP);
		String port = pro.getProperty(DB_PORT);
		String user = pro.getProperty(DB_USER);
		String password = pro.getProperty(DB_PASSWORD);
		String sid = pro.getProperty(DB_SID);
		String url = "jdbc:mysql://" + ip + ":" + port + "/" + sid
				+ "?useUnicode=true&characterEncoding=GB2312";

		// String url = "jdbc:oracle:thin:@"+ip+":"+port+":"+sid;
		try {
			if (conn == null) {
				Class.forName("com.mysql.jdbc.Driver");
				//conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/test", "root", "root");
				conn = DriverManager.getConnection(url, user, password);
			}

		} catch (SQLException e) {
			e.printStackTrace();
		} catch (ClassNotFoundException e) {
			e.printStackTrace();
		}
		return conn;
	}

	public void closeConnection() {
		if (conn != null) {
			try {
				conn.close();
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
	}

}