package test.jdbc;

import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;

import com.mysql.jdbc.Driver;

/*
 * 這個程式用來測試資料庫和JDBC驅動器是否被正確配置和使用；
 */
public class TestDB {

	public static void main(String[] args) throws IOException {
		try {
			runTest();
		} catch (SQLException e) {
			for (Throwable t : e) {
				t.printStackTrace();
			}
		}
	}

	/*
	 * 執行一個測試程式，用來建立一個表、插入一個值、展示表中的內容、移除表
	 */
	public static void runTest() throws SQLException, IOException {
		try (Connection conn = getConnection()) {
			Statement stat = conn.createStatement();
			stat.executeUpdate("CREATE TABLE Greetings(Message CHAR(20))");
			stat.executeUpdate("INSERT INTO Greetings VALUES('Hello, World!')");
			try (ResultSet rs = stat.executeQuery("SELECT * FROM Greetings")) {
				while (rs.next()) {
					System.out.println(rs.getString(1));
				}
			}
			stat.executeUpdate("DROP TABLE Greetings");
		}
	}

	/*
	 * 從指定的database.properties檔案中獲取資料庫連線
	 * 
	 * @return 對應的資料庫連線
	 */
	public static Connection getConnection() throws IOException, SQLException {

		Properties props = new Properties();
		// 從database.properties檔案中載入資料庫連線資訊
		try (InputStream in = Files.newInputStream(Paths.get("src/database.properties"))) {
			props.load(in);
		}

		String driver = props.getProperty("jdbc.driver");
		String url = props.getProperty("jdbc.url");
		String username = props.getProperty("jdbc.username");
		String password = props.getProperty("jdbc.password");

		// 註冊MySql驅動器程式
		if (driver != null) {
			System.setProperty("jdbc.drivers", driver);
		} else {
			DriverManager.registerDriver(new Driver());
		}

		// 獲取資料庫連線
		return DriverManager.getConnection(url, username, password);
	}
}