Files
luos/gulliver/system/class.dbMaintenance.php
2011-01-14 23:11:13 +00:00

656 lines
14 KiB
PHP
Executable File

<?php
/**
* pakeGulliver.php
*
* ProcessMaker Open Source Edition
* Copyright (C) 2004 - 2008 Colosa Inc.23
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* For more information, contact Colosa Inc, 2566 Le Jeune Rd.,
* Coral Gables, FL, 33134, USA, or email info@colosa.com.
*
*/
/**
*
* Database Maintenance class
*
* author Erik A. Ortiz <erik@colosa.com, aortiz.erik@gmail.com>
* date May 17th, 2010
*
* @package gulliver.system
*/
class DataBaseMaintenance
{
private $host;
private $user;
private $passwd;
private $link;
private $dbName;
public $result;
protected $tmpDir;
protected $outfile;
protected $infile;
/**
* __construct
*
* @param string $host is null
* @param string $user is null
* @param string $passwd is null
*
* @return none
*/
public function __construct($host = null, $user = null, $passwd = null)
{
$this->tmpDir = './';
$this->link = null;
$this->dbName = null;
if( isset($host) && isset($user) && isset($passwd) ) {
$this->host = $host;
$this->user = $user;
$this->passwd = $passwd;
}
}
/**
* setUser
*
* @param string $user
*
* @return none
*/
public function setUser($user)
{
$this->user = $user;
}
/**
* setPasswd
*
* @param string $passwd
*
* @return none
*/
public function setPasswd($passwd)
{
$this->passwd = $passwd;
}
/**
* setHost
*
* @param string $host
*
* @return none
*/
public function setHost($host)
{
$this->host = $host;
}
/**
* setTempDir
*
* @param string $tmpDir
*
* @return none
*/
public function setTempDir($tmpDir)
{
$this->tmpDir = $tmpDir;
if (!file_exists($tmpDir))
mkdir($this->tmpDir);
}
/**
* getTempDir
*
* @return $this->tmpDir
*/
public function getTempDir()
{
return $this->tmpDir;
}
/**
* status
*
* @return $this->link
*/
public function status()
{
return $$this->link;
}
/**
* connect
*
* @param string $dbname is null
*
* @return none
*/
public function connect($dbname = null)
{
if( $this->link != null ) {
mysql_close($this->link);
$this->link = null;
}
if( isset($dbname) ) {
$this->dbName = $dbname;
}
$this->link = mysql_connect($this->host, $this->user, $this->passwd);
if( ! $this->link ) {
throw new Exception("Couldn't connect to host {$this->host} with user {$this->user}");
}
if( $this->dbName != null ) {
$this->selectDataBase($this->dbName);
}
}
/**
* setDbName
*
* @param string $dbname is null
*
* @return none
*/
public function setDbName($dbname)
{
$this->dbName = $dbname;
}
/**
* selectDataBase
*
* @param string $dbname
*
* @return none
*/
public function selectDataBase($dbname)
{
$this->setDbName($dbname);
If( ! @mysql_select_db($this->dbName, $this->link) ) {
throw new Exception("Couldn't select database $dbname");
}
}
/**
* query
*
* @param string $sql
*
* @return $aRows
*/
public function query($sql)
{
$this->result = @mysql_query($sql);
if($this->result){
$aRows = Array();
while( $aRow = @mysql_fetch_assoc($this->result) ) {
array_push($aRows, $aRow);
}
return $aRows;
} else {
return false;
}
}
/**
* error
*
* @return @mysql_error()
*/
public function error()
{
return @mysql_error($this->link);
}
/**
* getTablesList
*
* @return $aRows
*/
public function getTablesList()
{
$this->result = @mysql_query("SHOW TABLES;");
$aRows = Array ();
while( $aRow = mysql_fetch_row($this->result) ) {
array_push($aRows, $aRow[0]);
}
return $aRows;
}
/**
* dumpData
*
* @param string $table
*
* @return boolean true or false
*/
function dumpData($table)
{
$this->outfile = $this->tmpDir . $table . '.dump';
//if the file exists delete it
if( is_file($this->outfile) ) {
@unlink($this->outfile);
}
$sql = "SELECT * INTO OUTFILE '{$this->outfile}' FIELDS TERMINATED BY '\t|\t' OPTIONALLY ENCLOSED BY '\"' LINES TERMINATED BY '\t\t\r\r\n' FROM $table";
mysql_escape_string("';");
if( ! @mysql_query($sql) ) {
echo mysql_error() . "\n";
return false;
}
return true;
}
/**
* restoreData
*
* @param string $backupFile
*
* @return boolean true or false
*/
function restoreData($backupFile)
{
$tableName = str_replace('.dump', '', basename($backupFile));
$sql = "LOAD DATA INFILE '$backupFile' INTO TABLE $tableName FIELDS TERMINATED BY '\t|\t' OPTIONALLY ENCLOSED BY '\"' LINES TERMINATED BY '\t\t\r\r\n'";
if( ! @mysql_query($sql) ) {
print mysql_error() . "\n";
return false;
}
return true;
}
/**
* backupData
*
* @return boolean true or false
*/
function backupData()
{
$aTables = $this->getTablesList();
foreach( $aTables as $table ) {
if( $this->dumpData($table) !== false ) {
printf("%20s %s %s\n", 'Dump of table:', $table, " in file {$this->outfile}");
} else {
return false;
}
}
return true;
}
/**
* backupSqlData
*
* @return boolean true or false
*/
function backupSqlData()
{
$aTables = $this->getTablesList();
foreach( $aTables as $table ) {
$fsize = $this->dumpSqlInserts($table);
$file = basename($this->outfile);
}
return true;
}
/**
* restoreAllData
*
* @param string $type default value null
*
* @return none
*/
function restoreAllData($type = null)
{
$aTables = $this->getTablesList();
foreach( $aTables as $table ) {
if( isset($type) && $type == 'sql' ) {
$this->infile = $this->tmpDir . $table . ".sql";
if( is_file($this->infile) ) {
$queries = $this->restoreFromSql($this->infile, true);
if (!isset($queries))
$queries = "unknown";
printf("%-59s%20s", "Restored table $table", "$queries queries\n");
}
} else {
$this->infile = $this->tmpDir . $table . ".dump";
if( is_file($this->infile) ) {
$this->restoreData($this->infile);
printf("%20s %s %s\n", 'Restoring data from ', $this->infile, " in table $table");
}
}
}
}
/**
* createDb
*
* @param string $dbname
* @param string $drop default value false
*
* @return none
*/
function createDb($dbname, $drop = false)
{
if( $drop ) {
$sql = "DROP DATABASE IF EXISTS $dbname;";
if( ! @mysql_query($sql) ) {
throw new Exception(mysql_error());
}
}
$sql = "CREATE DATABASE IF NOT EXISTS $dbname DEFAULT CHARACTER SET utf8 COLLATE utf8_general_ci;";
if( ! @mysql_query($sql) ) {
throw new Exception(mysql_error());
}
}
/**
* restoreFromSql2
*
* @param string $sqlfile
*
* @return none
*/
function restoreFromSql2($sqlfile)
{
ini_set('memory_limit', '512M');
if( ! is_file($sqlfile) ) {
throw new Exception("the $sqlfile doesn't exist!");
}
$query = file_get_contents($sqlfile);
$mysqli = new mysqli($this->host, $this->user, $this->passwd, $this->dbName);
/* check connection */
if( mysqli_connect_errno() ) {
printf("Connect failed: %s\n", mysqli_connect_error());
exit();
}
/* execute multi query */
if( $mysqli->multi_query($query) ) {
do {
/* store first result set */
if( $result = $mysqli->store_result() ) {
while( $row = $result->fetch_row() ) {
}
$result->free();
}
} while( $mysqli->next_result() );
}
/* close connection */
$mysqli->close();
}
function lockTables() {
$aTables = $this->getTablesList();
if (empty($aTables))
return false;
printf("%-70s", "LOCK TABLES");
if( @mysql_query("LOCK TABLES ".implode(" READ, ", $aTables)." READ; ") ) {
echo " [OK]\n";
return true;
} else {
echo "[FAILED]\n".mysql_error()."\n";
return false;
}
}
function unlockTables() {
printf("%-70s", "UNLOCK TABLES");
if( @mysql_query("UNLOCK TABLES;") )
echo " [OK]\n";
else
echo "[FAILED]\n".mysql_error()."\n";
}
/**
* dumpSqlInserts
*
* @param string $table
*
* @return integer $bytesSaved;
*/
function dumpSqlInserts($table)
{
$bytesSaved = 0;
$result = @mysql_query("SELECT * FROM `$table`");
$num_rows = mysql_num_rows($result);
$num_fields = mysql_num_fields($result);
$data = "";
for( $i = 0; $i < $num_rows; $i ++ ) {
$row = mysql_fetch_object($result);
$data .= "INSERT INTO `$table` VALUES (";
for( $x = 0; $x < $num_fields; $x ++ ) {
$field_name = mysql_field_name($result, $x);
$data .= ($row->$field_name === null) ? 'NULL' : "'" . mysql_real_escape_string($row->$field_name) . "'";
$data .= ($x < ($num_fields - 1)) ? ", " : false;
}
$data .= ");\n";
}
printf("%-59s%20s", "Dump of table $table", strlen($data) . " Bytes Saved\n");
return $data;
}
/**
* backupDataBaseSchema
*
* @param string $outfile
*
* @return none
*/
function backupDataBase($outfile)
{
$aTables = $this->getTablesList();
$ws = explode('_', $this->dbName);
$ws = isset($ws[1]) ? $ws[1] : $this->dbName;
$schema = "\n";
$schema .= " -- Processmaker Data Base Maintenance Tool\n";
$schema .= " --\n";
$schema .= " -- Workspace: " . $ws . "\n";
$schema .= " -- Data Base: " . $this->dbName . "\n";
$schema .= " -- Tables: " . (count($aTables)) . "\n";
$schema .= " -- Date: " . (date('l jS \of F Y h:i:s A')) . "\n";
$schema .= " --\n\n";
$file = fopen($outfile, "w+");
fwrite($file, $schema);
foreach( $aTables as $table ) {
$tableSchema = "\nDROP TABLE IF EXISTS `$table`;\n\n";
$tableSchema .= $this->getSchemaFromTable($table);
$data = $this->dumpSqlInserts($table);
fwrite($file, $tableSchema);
fwrite($file, $data);
}
fclose($file);
}
/**
* restoreFromSql
*
* @param string $sqlfile
*
* @return boolean false or true
*/
function restoreFromSql($sqlfile, $type='file')
{
ini_set('memory_limit', '64M');
if( $type == 'file' && !is_file($sqlfile) ) {
throw new Exception("the $sqlfile doesn't exist!");
}
$metaFile = str_replace('.sql', '.meta', $sqlfile);
$queries = 0;
if( is_file($metaFile) ) {
echo "Using $metaFile as metadata.\n";
$fp = fopen($sqlfile, 'rb');
$fpmd = fopen($metaFile, 'r');
while( $offset = fgets($fpmd, 1024) ) {
$buffer = intval($offset); //reading the size of $oData
$query = fread($fp, $buffer); //reading string $oData
$queries += 1;
if( ! @mysql_query($query) ) {
echo mysql_error() . "\n";
echo "==>" . $query . "<==\n";
}
}
} else {
$queries = NULL;
try{
$mysqli = new mysqli($this->host, $this->user, $this->passwd, $this->dbName);
/* check connection */
if( mysqli_connect_errno() ) {
printf("Connect failed: %s\n", mysqli_connect_error());
exit();
}
if($type == 'file'){
$query = file_get_contents($sqlfile);
} else if($type == 'string'){
$query = $sqlfile;
} else {
return false;
}
if( trim($query) == "" )
return false;
/* execute multi query */
if( $mysqli->multi_query($query) ) {
do {
/* store first result set */
if( $result = $mysqli->store_result() ) {
while( $row = $result->fetch_row() ) {
//printf("%s\n", $row[0]);
}
$result->free();
}
/* print divider */
if( $mysqli->more_results() ) {
//printf("-----------------\n");
}
} while( $mysqli->next_result() );
} else throw new Exception(mysqli_error($mysqli));
/* close connection */
$mysqli->close();
} catch(Exception $e){
echo $query;
echo $e->getMessage();
}
}
return $queries;
}
/**
* getSchemaFromTable
*
* @param string $tablename
*
* @return string $tableSchema
*/
function getSchemaFromTable($tablename)
{
//$tableSchema = "/* Structure for table `$tablename` */\n";
//$tableSchema .= "DROP TABLE IF EXISTS `$tablename`;\n\n";
$tableSchema = "";
$sql = "show create table `$tablename`; ";
$result = @mysql_query($sql);
if( $result ) {
if( $row = mysql_fetch_assoc($result) ) {
$tableSchema .= $row['Create Table'] . ";\n\n";
}
mysql_free_result($result);
} else
echo mysql_error();
return $tableSchema;
}
/**
* removeCommentsIntoString
*
* @param string $str
*
* @return string $str
*/
function removeCommentsIntoString($str)
{
$str = preg_replace('/\/\*[\w\W]*\*\//', '', $str);
$str = preg_replace("/--[\w\W]*\\n/", '', $str);
$str = preg_replace("/\/\/[\w\W]*\\n/", '', $str);
$str = preg_replace("/\#[\w\W]*\\n/", '', $str);
return $str;
}
}
/*
// Sample to use
$oDbMaintainer = new DataBaseMaintenance('localhost', 'root', 'atopml2005');
$oDbMaintainer->setTempDir('/home/erik/backs/');
$oDbMaintainer->setDbName('rb_os');
$oDbMaintainer->connect();
$oDbMaintainer->backupDataBaseSchema('/home/erik/backs/schema_os.sql');
$oDbMaintainer->backupSqlData();
$oDbMaintainer->createDb('neyek12', true);
$o2 = new DataBaseMaintenance('localhost', 'root', 'atopml2005');
$o2->setTempDir('/home/erik/backs/');
$o2->setDbName('neyek12');
$o2->connect();
$o2->restoreFromSql('/home/erik/backs/schema_os.sql');
$o2->restoreAllData('sql');
*/