代码语言:javascript复制
<?php
define('MSB_VERSION', '1.0.0');
define('MSB_NL', "rn");
define('MSB_STRING', 0);
define('MSB_DOWNLOAD', 1);
define('MSB_SAVE', 2);
define('__SEP__', "/*sep*/" );
set_time_limit( 600 );
class MySQL_Backup {
var $server = 'localhost';
var $port = 3306;
var $username = 'root';
var $password = '';
var $database = '';
var $link_id = -1;
var $connected = false;
var $tables = array();
var $drop_tables = true;
var $struct_only = false;
var $comments = true;
var $backup_dir = '';
var $fname_format = 'd_m_y__H_i_s';
var $error = '';
var $complete_inserts = false;
var $inserts_block = 200;
function Execute($task = MSB_STRING, $fname = '', $compress = false) {
if ( !( $sql = $this->_Retrieve() ) ) {
return false;
}
if ( $task == MSB_SAVE ) {
if (empty($fname)) {
$fname = $this->backup_dir;
$fname .= date($this->fname_format);
$fname .= ($compress ? '.sql.gz' : '.sql');
}
return $this->_SaveToFile($fname, $sql, $compress);
} elseif ($task == MSB_DOWNLOAD) {
if ( empty( $fname ) ) {
$fname = date($this->fname_format);
$fname .= ($compress ? '.sql.gz' : '.sql');
}
return $this->_DownloadFile($fname, $sql, $compress);
} else {
return $sql;
}
}
function _Connect() {
$value = false;
if (!$this->connected) {
$host = $this->server . ':' . $this->port;
$this->link_id = mysql_connect($host, $this->username, $this->password);
}
if ($this->link_id) {
if (empty($this->database)) {
$value = true;
} elseif ($this->link_id !== -1) {
$value = mysql_select_db($this->database, $this->link_id);
} else {
$value = mysql_select_db($this->database);
}
}
if (!$value) {
$this->error = mysql_error();
}
return $value;
}
function _Query($sql) {
if ($this->link_id !== -1) {
$result = mysql_query($sql, $this->link_id);
} else {
$result = mysql_query($sql);
}
if (!$result) {
$this->error = mysql_error();
}
return $result;
}
function _GetTables() {
$value = array();
if ( !( $result = $this->_Query('SHOW TABLES') ) ) {
return false;
}
while ( $row = mysql_fetch_row( $result ) ) {
if ( empty( $this->tables) || in_array( $row[0], $this->tables ) ) {
$value[] = $row[0];
}
}
if (!sizeof($value)) {
$this->error = 'No tables found in database.';
return false;
}
return $value;
}
function _DumpTable( $table ) {
$value = '';
$this->_Query( 'LOCK TABLES ' . $table . ' WRITE' );
if ( $this->comments ) {
$value .= '#' . MSB_NL;
$value .= '# Table structure for table `' . $table . '`' . MSB_NL;
$value .= '#' . MSB_NL . MSB_NL;
}
if ( $this->drop_tables ) {
$value .= 'DROP TABLE IF EXISTS `' . $table . '`;' . __SEP__ . MSB_NL;
}
if ( !( $result = $this->_Query('SHOW CREATE TABLE ' . $table) ) ) {
return false;
}
$row = mysql_fetch_assoc($result);
$value .= str_replace("n", MSB_NL, $row['Create Table']) . ';' . __SEP__;
$value .= MSB_NL . MSB_NL;
if (!$this->struct_only) {
if ($this->comments) {
$value .= '#' . MSB_NL;
$value .= '# Dumping data for table `' . $table . '`' . MSB_NL;
$value .= '#' . MSB_NL . MSB_NL;
}
$value .= $this->_GetInserts($table);
}
$value .= MSB_NL . MSB_NL;
$this->_Query('UNLOCK TABLES');
return $value;
}
function _GetInserts($table) {
$value = '';
if (!($result = $this->_Query('SELECT * FROM ' . $table))) {
return false;
}
if ( $this->complete_inserts ) {
while ($row = mysql_fetch_row($result)) {
$values = '';
foreach ($row as $data) {
$values .= ''' . addslashes($data) . '', ';
}
$values = substr($values, 0, -2);
$value .= 'INSERT INTO ' . $table . ' VALUES (' . $values . ');' . __SEP__ . MSB_NL;
}
} else {
$blocks_counter = 0;
$blocks = array();
while ($row = mysql_fetch_row($result)) {
$values = array();
foreach ($row as $data) {
$values[] = ''' . addslashes($data) . ''';
}
$blocks[] = '(' . implode( ',', $values ) . ')';
if ( $blocks_counter < $this->inserts_block ) {
$blocks_counter ;
} else {
$value .= 'INSERT INTO ' . $table . ' VALUES ' . implode( ',', $blocks ) . ";" . __SEP__ . MSB_NL;
$blocks = array();
$blocks_counter = 0;
}
}
if ( count( $blocks ) ) {
$value .= 'INSERT INTO ' . $table . ' VALUES ' . implode( ',', $blocks ) . ";" . __SEP__ . MSB_NL;
}
}
return $value;
}
function _Retrieve() {
$value = '';
if (!$this->_Connect()) {
return false;
}
if ($this->comments) {
$value .= '#' . MSB_NL;
$value .= '# MySQL database dump' . MSB_NL;
$value .= '# Created by MySQL_Backup class, ver. ' . MSB_VERSION . MSB_NL;
$value .= '#' . MSB_NL;
$value .= '# Host: ' . $this->server . MSB_NL;
$value .= '# Generated: ' . date('M j, Y') . ' at ' . date('H:i') . MSB_NL;
$value .= '# MySQL version: ' . mysql_get_server_info() . MSB_NL;
$value .= '# PHP version: ' . phpversion() . MSB_NL;
if (!empty($this->database)) {
$value .= '#' . MSB_NL;
$value .= '# Database: `' . $this->database . '`' . MSB_NL;
}
$value .= '#' . MSB_NL . MSB_NL . MSB_NL;
}
if (!($tables = $this->_GetTables())) {
return false;
}
foreach ($tables as $table) {
if (!($table_dump = $this->_DumpTable($table))) {
$this->error = mysql_error();
return false;
}
$value .= $table_dump;
}
return $value;
}
function _SaveToFile($fname, $sql, $compress) {
if ($compress) {
if (!($zf = gzopen($fname, 'w9'))) {
$this->error = 'Can't create the output file.';
return false;
}
gzwrite($zf, $sql);
gzclose($zf);
} else {
if (!($f = fopen($fname, 'w'))) {
$this->error = 'Can't create the output file.';
return false;
}
fwrite($f, $sql);
fclose($f);
}
return true;
}
function _DownloadFile($fname, $sql, $compress) {
header('Content-disposition: filename=' . $fname);
header('Content-type: application/octetstream');
header('Pragma: no-cache');
header('Expires: 0');
echo ($compress ? gzencode($sql) : $sql);
return true;
}
}
?></pre>