Update to Connector/J 5.1.

[irc/rizon/bncbot.git] / src / net / rizon / DatabaseConnection.java

package net.rizon;\r
\r
import java.sql.Connection;\r
import java.sql.DriverManager;\r
import java.sql.SQLException;\r
import java.util.Properties;\r
\r
public class DatabaseConnection {\r
	private static Connection con = null;\r
	private static Properties props = null;\r
\r
	public static Connection getConnection() {\r
		if (props == null) throw new RuntimeException("DatabaseConnection not initialized");\r
		\r
		try {\r
			if((con != null) && con.isValid(2)) {\r
				return con;\r
			} else { // Attempt reconnect.\r
				if (con != null) con.close();\r
				\r
				DatabaseConnection.connect();\r
				\r
				if(con == null) {\r
					throw new RuntimeException("MySQL server has gone away. Attempted reconnection failed.");\r
				}\r
				\r
				return con;\r
			}\r
		}\r
		catch (SQLException e) {\r
			e.printStackTrace();\r
		}\r
		catch (AbstractMethodError e) {\r
			DatabaseConnection.connect();\r
			\r
			if(con == null) {\r
				throw new RuntimeException("MySQL server has gone away. Attempted reconnection failed.");\r
			}\r
			\r
			return con;\r
		}\r
		return con;\r
	}\r
	\r
	public static boolean isInitialized() {\r
		return props != null;\r
	}\r
	\r
	public static void setProps(Properties aProps) {\r
		props = aProps;\r
	}\r
	\r
	private static void connect() {\r
		String driver = props.getProperty("sqlDriver");\r
		String url = props.getProperty("sqlUrl");\r
		String user = props.getProperty("sqlUser");\r
		String password = props.getProperty("sqlPass");\r
		\r
		try {\r
			Class.forName(driver);\r
		} catch (ClassNotFoundException e) {\r
			e.printStackTrace();\r
		}\r
		try {\r
			con = DriverManager.getConnection(url, user, password);\r
		} catch (SQLException e) {\r
			e.printStackTrace();\r
			throw new RuntimeException("Error connecting to MySQL: " + e.getMessage());\r
		}\r
	}\r
}\r