trunk/java/com/l2jserver/L2DatabaseFactory.java

@@ -0,0 +1,274 @@
+/*
+ * Copyright (C) 2004-2014 L2J Server
+ * 
+ * This file is part of L2J Server.
+ * 
+ * L2J Server is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * L2J Server is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package com.l2jserver;
+
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import com.l2jserver.gameserver.ThreadPoolManager;
+import com.mchange.v2.c3p0.ComboPooledDataSource;
+
+/**
+ * This class manages the database connections.
+ */
+public class L2DatabaseFactory
+{
+	private static final Logger _log = Logger.getLogger(L2DatabaseFactory.class.getName());
+	
+	private static L2DatabaseFactory _instance;
+	private static volatile ScheduledExecutorService _executor;
+	private ComboPooledDataSource _source;
+	
+	/**
+	 * Instantiates a new l2 database factory.
+	 * @throws SQLException the SQL exception
+	 */
+	public L2DatabaseFactory() throws SQLException
+	{
+		try
+		{
+			if (Config.DATABASE_MAX_CONNECTIONS < 2)
+			{
+				Config.DATABASE_MAX_CONNECTIONS = 2;
+				_log.warning("A minimum of " + Config.DATABASE_MAX_CONNECTIONS + " db connections are required.");
+			}
+			
+			_source = new ComboPooledDataSource();
+			_source.setAutoCommitOnClose(true);
+			
+			_source.setInitialPoolSize(10);
+			_source.setMinPoolSize(10);
+			_source.setMaxPoolSize(Math.max(10, Config.DATABASE_MAX_CONNECTIONS));
+			
+			_source.setAcquireRetryAttempts(0); // try to obtain connections indefinitely (0 = never quit)
+			_source.setAcquireRetryDelay(500); // 500 milliseconds wait before try to acquire connection again
+			_source.setCheckoutTimeout(0); // 0 = wait indefinitely for new connection
+			// if pool is exhausted
+			_source.setAcquireIncrement(5); // if pool is exhausted, get 5 more connections at a time
+			// cause there is a "long" delay on acquire connection
+			// so taking more than one connection at once will make connection pooling
+			// more effective.
+			
+			// this "connection_test_table" is automatically created if not already there
+			_source.setAutomaticTestTable("connection_test_table");
+			_source.setTestConnectionOnCheckin(false);
+			
+			// testing OnCheckin used with IdleConnectionTestPeriod is faster than testing on checkout
+			
+			_source.setIdleConnectionTestPeriod(3600); // test idle connection every 60 sec
+			_source.setMaxIdleTime(Config.DATABASE_MAX_IDLE_TIME); // 0 = idle connections never expire
+			// *THANKS* to connection testing configured above
+			// but I prefer to disconnect all connections not used
+			// for more than 1 hour
+			
+			// enables statement caching, there is a "semi-bug" in c3p0 0.9.0 but in 0.9.0.2 and later it's fixed
+			_source.setMaxStatementsPerConnection(100);
+			
+			_source.setBreakAfterAcquireFailure(false); // never fail if any way possible
+			// setting this to true will make
+			// c3p0 "crash" and refuse to work
+			// till restart thus making acquire
+			// errors "FATAL" ... we don't want that
+			// it should be possible to recover
+			_source.setDriverClass(Config.DATABASE_DRIVER);
+			_source.setJdbcUrl(Config.DATABASE_URL);
+			_source.setUser(Config.DATABASE_LOGIN);
+			_source.setPassword(Config.DATABASE_PASSWORD);
+			
+			/* Test the connection */
+			_source.getConnection().close();
+			
+			if (Config.DEBUG)
+			{
+				_log.fine("Database Connection Working");
+			}
+		}
+		catch (SQLException x)
+		{
+			if (Config.DEBUG)
+			{
+				_log.fine("Database Connection FAILED");
+			}
+			// re-throw the exception
+			throw x;
+		}
+		catch (Exception e)
+		{
+			if (Config.DEBUG)
+			{
+				_log.fine("Database Connection FAILED");
+			}
+			throw new SQLException("Could not init DB connection:" + e.getMessage());
+		}
+	}
+	
+	/**
+	 * Shutdown.
+	 */
+	public void shutdown()
+	{
+		try
+		{
+			_source.close();
+		}
+		catch (Exception e)
+		{
+			_log.log(Level.INFO, "", e);
+		}
+		try
+		{
+			_source = null;
+		}
+		catch (Exception e)
+		{
+			_log.log(Level.INFO, "", e);
+		}
+	}
+	
+	/**
+	 * Gets the single instance of L2DatabaseFactory.
+	 * @return single instance of L2DatabaseFactory
+	 * @throws SQLException the SQL exception
+	 */
+	public static L2DatabaseFactory getInstance() throws SQLException
+	{
+		synchronized (L2DatabaseFactory.class)
+		{
+			if (_instance == null)
+			{
+				_instance = new L2DatabaseFactory();
+			}
+		}
+		return _instance;
+	}
+	
+	/**
+	 * Gets the connection.
+	 * @return the connection
+	 */
+	public Connection getConnection()
+	{
+		Connection con = null;
+		while (con == null)
+		{
+			try
+			{
+				con = _source.getConnection();
+				if (Server.serverMode == Server.MODE_GAMESERVER)
+				{
+					ThreadPoolManager.getInstance().scheduleGeneral(new ConnectionCloser(con, new RuntimeException()), Config.CONNECTION_CLOSE_TIME);
+				}
+				else
+				{
+					getExecutor().schedule(new ConnectionCloser(con, new RuntimeException()), Config.CONNECTION_CLOSE_TIME, TimeUnit.MILLISECONDS);
+				}
+			}
+			catch (SQLException e)
+			{
+				_log.log(Level.WARNING, "L2DatabaseFactory: getConnection() failed, trying again " + e.getMessage(), e);
+			}
+		}
+		return con;
+	}
+	
+	/**
+	 * The Class ConnectionCloser.
+	 */
+	private static class ConnectionCloser implements Runnable
+	{
+		private static final Logger _log = Logger.getLogger(ConnectionCloser.class.getName());
+		
+		/** The connection. */
+		private final Connection c;
+		
+		/** The exception. */
+		private final RuntimeException exp;
+		
+		/**
+		 * Instantiates a new connection closer.
+		 * @param con the con
+		 * @param e the e
+		 */
+		public ConnectionCloser(Connection con, RuntimeException e)
+		{
+			c = con;
+			exp = e;
+		}
+		
+		@Override
+		public void run()
+		{
+			try
+			{
+				if (!c.isClosed())
+				{
+					_log.log(Level.WARNING, "Unclosed connection! Trace: " + exp.getStackTrace()[1], exp);
+				}
+			}
+			catch (SQLException e)
+			{
+				_log.log(Level.WARNING, "", e);
+			}
+		}
+	}
+	
+	/**
+	 * Gets the executor.
+	 * @return the executor
+	 */
+	private static ScheduledExecutorService getExecutor()
+	{
+		if (_executor == null)
+		{
+			synchronized (L2DatabaseFactory.class)
+			{
+				if (_executor == null)
+				{
+					_executor = Executors.newSingleThreadScheduledExecutor();
+				}
+			}
+		}
+		return _executor;
+	}
+	
+	/**
+	 * Gets the busy connection count.
+	 * @return the busy connection count
+	 * @throws SQLException the SQL exception
+	 */
+	public int getBusyConnectionCount() throws SQLException
+	{
+		return _source.getNumBusyConnectionsDefaultUser();
+	}
+	
+	/**
+	 * Gets the idle connection count.
+	 * @return the idle connection count
+	 * @throws SQLException the SQL exception
+	 */
+	public int getIdleConnectionCount() throws SQLException
+	{
+		return _source.getNumIdleConnectionsDefaultUser();
+	}
+}