TrackManiaControl/application/core/Database/Database.php

234 lines
5.8 KiB
PHP
Raw Normal View History

2013-11-09 17:24:03 +01:00
<?php
2014-04-27 21:31:55 +02:00
namespace ManiaControl\Database;
2014-03-12 13:27:14 +01:00
use ManiaControl\Callbacks\TimerListener;
2014-08-05 01:03:48 +02:00
use ManiaControl\Logger;
2014-04-27 21:31:55 +02:00
use ManiaControl\ManiaControl;
2013-11-09 17:24:03 +01:00
/**
* Database Connection Class
2013-11-09 17:24:03 +01:00
*
2014-05-02 17:40:47 +02:00
* @author ManiaControl Team <mail@maniacontrol.com>
* @copyright 2014 ManiaControl Team
* @license http://www.gnu.org/licenses/ GNU General Public License, Version 3
2013-11-09 17:24:03 +01:00
*/
class Database implements TimerListener {
/*
* Public properties
2013-11-09 17:24:03 +01:00
*/
/** @var \mysqli $mysqli */
/** @deprecated see getMysqli() */
2013-11-09 17:24:03 +01:00
public $mysqli = null;
2014-05-02 17:40:47 +02:00
/*
* Private properties
2013-11-09 17:24:03 +01:00
*/
/** @var ManiaControl $maniaControl */
private $maniaControl = null;
2014-06-20 19:06:55 +02:00
/** @var Config $config */
private $config = null;
/** @var MigrationHelper $migrationHelper */
private $migrationHelper = null;
2013-11-09 17:24:03 +01:00
/**
2014-05-09 17:30:31 +02:00
* Construct a new Database Connection
*
* @param ManiaControl $maniaControl
2013-11-09 17:24:03 +01:00
*/
public function __construct(ManiaControl $maniaControl) {
$this->maniaControl = $maniaControl;
2014-05-02 17:40:47 +02:00
// Enable mysqli Reconnect
2014-03-12 13:27:14 +01:00
ini_set('mysqli.reconnect', 'on');
2014-05-02 17:40:47 +02:00
2013-11-09 17:24:03 +01:00
// Open database connection
$this->loadConfig();
$this->mysqli = @new \mysqli($this->config->host, $this->config->user, $this->config->pass, null, $this->config->port);
if ($connectError = $this->getMysqli()->connect_error) {
$message = "Couldn't connect to Database: '{$connectError}'";
2014-05-09 14:08:39 +02:00
$this->maniaControl->quit($message, true);
2013-11-09 17:24:03 +01:00
}
$this->getMysqli()->set_charset("utf8");
2014-05-02 17:40:47 +02:00
2013-11-09 17:24:03 +01:00
$this->initDatabase();
$this->optimizeTables();
2014-05-02 17:40:47 +02:00
// Register Method which checks the Database Connection every 5 seconds
$this->maniaControl->getTimerManager()->registerTimerListening($this, 'checkConnection', 5000);
2014-05-02 17:40:47 +02:00
// Children
2014-04-27 21:31:55 +02:00
$this->migrationHelper = new MigrationHelper($maniaControl);
2014-03-12 13:27:14 +01:00
}
2013-11-09 17:24:03 +01:00
/**
* Load the Database Config
*/
private function loadConfig() {
$databaseElements = $this->maniaControl->getConfig()->xpath('database');
if (!$databaseElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit('No Database configured!', true);
}
$databaseElement = $databaseElements[0];
// Host
$hostElements = $databaseElement->xpath('host');
if (!$hostElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit("Invalid database configuration (Host).", true);
}
$host = (string)$hostElements[0];
// Port
$portElements = $databaseElement->xpath('port');
if (!$portElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit("Invalid database configuration (Port).", true);
}
$port = (string)$portElements[0];
// User
$userElements = $databaseElement->xpath('user');
if (!$userElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit("Invalid database configuration (User).", true);
}
$user = (string)$userElements[0];
// Pass
$passElements = $databaseElement->xpath('pass');
if (!$passElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit("Invalid database configuration (Pass).", true);
}
$pass = (string)$passElements[0];
// Name
$nameElements = $databaseElement->xpath('name');
if (!$nameElements) {
$nameElements = $databaseElement->xpath('db_name');
}
if (!$nameElements) {
2014-08-05 02:08:11 +02:00
$this->maniaControl->quit("Invalid database configuration (Name).", true);
}
$name = (string)$nameElements[0];
// Create config object
2014-06-20 19:06:55 +02:00
$config = new Config($host, $port, $user, $pass, $name);
if (!$config->validate()) {
$this->maniaControl->quit("Your config file doesn't seem to be maintained properly. Please check the database configuration again!", true);
}
$this->config = $config;
}
/**
* Connect to the defined Database
*
* @return bool
2013-11-09 17:24:03 +01:00
*/
private function initDatabase() {
// Try to connect
$result = $this->getMysqli()->select_db($this->config->name);
2014-05-02 17:40:47 +02:00
if ($result) {
return true;
}
Logger::logInfo("Database '{$this->config->name}' doesn't exist! Trying to create it...");
2014-05-02 17:40:47 +02:00
// Create database
$databaseQuery = "CREATE DATABASE " . $this->getMysqli()->escape_string($this->config->name) . ";";
$this->getMysqli()->query($databaseQuery);
if ($this->getMysqli()->error) {
$this->maniaControl->quit($this->getMysqli()->error, true);
return false;
2013-11-09 17:24:03 +01:00
}
2014-05-02 17:40:47 +02:00
// Connect to new database
$this->getMysqli()->select_db($this->config->name);
if ($error = $this->getMysqli()->error) {
$message = "Couldn't select database '{$this->config->name}'. {$error}";
2014-05-09 14:08:39 +02:00
$this->maniaControl->quit($message, true);
return false;
2013-11-09 17:24:03 +01:00
}
return true;
2013-11-09 17:24:03 +01:00
}
/**
* Optimize all existing Tables
2013-11-09 17:24:03 +01:00
*
* @return bool
2013-11-09 17:24:03 +01:00
*/
private function optimizeTables() {
$showQuery = 'SHOW TABLES;';
$result = $this->getMysqli()->query($showQuery);
if ($error = $this->getMysqli()->error) {
Logger::logError($error);
return false;
2013-11-09 17:24:03 +01:00
}
$count = $result->num_rows;
if ($count <= 0) {
$result->free();
return true;
2013-11-09 17:24:03 +01:00
}
$optimizeQuery = 'OPTIMIZE TABLE ';
2014-05-02 17:40:47 +02:00
$index = 0;
while ($row = $result->fetch_row()) {
$tableName = $row[0];
$optimizeQuery .= "`{$tableName}`";
if ($index < $count - 1) {
$optimizeQuery .= ',';
2013-11-09 17:24:03 +01:00
}
$index++;
2013-11-09 17:24:03 +01:00
}
$result->free();
$optimizeQuery .= ';';
$this->getMysqli()->query($optimizeQuery);
if ($error = $this->getMysqli()->error) {
Logger::logError($error);
return false;
2013-11-09 17:24:03 +01:00
}
return true;
2013-11-09 17:24:03 +01:00
}
2014-05-02 17:40:47 +02:00
/**
* Return the mysqli instance
*
* @return \mysqli
*/
public function getMysqli() {
return $this->mysqli;
}
/**
* Return the database config
*
* @return Config
*/
public function getConfig() {
return $this->config;
}
/**
* Return the migration helper
*
* @return MigrationHelper
*/
public function getMigrationHelper() {
return $this->migrationHelper;
}
2014-05-02 17:40:47 +02:00
/**
* Check whether the Database Connection is still open
2014-05-02 17:40:47 +02:00
*/
public function checkConnection() {
if (!$this->getMysqli() || !$this->getMysqli()->ping()) {
$this->maniaControl->quit('The MySQL Server has gone away!', true);
2014-05-02 17:40:47 +02:00
}
}
/**
* Destruct Database Connection
2014-05-02 17:40:47 +02:00
*/
public function __destruct() {
if ($this->getMysqli() && !$this->getMysqli()->connect_error) {
$this->getMysqli()->close();
2014-05-02 17:40:47 +02:00
}
}
2013-11-09 17:24:03 +01:00
}