php实现MySQL数据库备份与还原类实例

<?php   
/**  
 * 创建时间： 2012年5月21日  
 *  
 * 说明：分卷文件是以_v1.sql为结尾(20120522021241_all_v1.sql)  
 * 功能：实现mysql数据库分卷备份,选择表进行备份,实现单个sql文件及分卷sql导入  
 * 使用方法：  
 *  
 * ------1. 数据库备份（导出）------------------------------------------------------------  
//分别是主机，用户名，密码，数据库名，数据库编码  
$db = new DBManage ( 'localhost', 'root', 'root', 'test', 'utf8' );  
// 参数：备份哪个表(可选),备份目录(可选，默认为backup),分卷大小(可选,默认2000，即2M)  
$db->backup ();  
 * ------2. 数据库恢复（导入）------------------------------------------------------------  
//分别是主机，用户名，密码，数据库名，数据库编码  
$db = new DBManage ( 'localhost', 'root', 'root', 'test', 'utf8' );  
//参数：sql文件  
$db->restore ( './backup/20120516211738_all_v1.sql');  
 *----------------------------------------------------------------------  
 */  
class DBManage   
{   
    var $db; // 数据库连接   
    var $database; // 所用数据库   
    var $sqldir; // 数据库备份文件夹   
    var $record;   
    // 换行符   
    private $ds = "n";   
    // 存储SQL的变量   
    public $sqlContent = "";   
    // 每条sql语句的结尾符   
    public $sqlEnd = ";";   
    /**  
     * 初始化  
     *  
     * @param string $host  
     * @param string $username  
     * @param string $password  
     * @param string $thisatabase  
     * @param string $charset  
     */  
    function __construct($host = 'localhost', $username = 'root', $password = '', $thisatabase = 'test', $charset = 'utf8')   
    {   
        $this->host = $host;   
        $this->username = $username;   
        $this->password = $password;   
        $this->database = $thisatabase;   
        $this->charset = $charset;   
        // 连接数据库   
        $this->db = mysql_connect ( $this->host, $this->username, $this->password ) or die ( "数据库连接失败." );   
        // 选择使用哪个数据库   
        mysql_select_db ( $this->database, $this->db ) or die ( "无法打开数据库" );   
        // 数据库编码方式   
        mysql_query ( 'SET NAMES ' . $this->charset, $this->db );   
    }   
    
    /*  
      * ------------------------------------------数据库备份start---------------------------------------------------------- 
      */  
    
    /**  
     * 数据库备份  
     * 参数：备份哪个表(可选),备份目录(可选，默认为backup),分卷大小(可选,默认2000，即2M)  
     *  
     * @param $string $dir  
     * @param int $size  
     * @param $string $tablename  
     */  
    function backup($tablename = '', $dir = '', $size = 2000)   
    {   
        //$dir = $dir ? $dir : 'backup/';   
        //$size = $size ? $size : 2000;   
        $sql = '';   
        // 只备份某个表   
        if (! emptyempty ( $tablename ))   
        {   
            echo '正在备份表' . $tablename . '<br />';   
            // 插入dump信息   
            $sql = $this->_retrieve();   
            // 插入表结构信息   
            $sql .= $this->_insert_table_structure ( $tablename );   
            // 插入数据   
            $data = mysql_query ( "select * from " . $tablename );   
            // 文件名前面部分   
            $filename = date ( 'YmdHis' ) . "_" . $tablename;   
            // 字段数量   
            $num_fields = mysql_num_fields ( $data );   
            // 第几分卷   
            $p = 1;   
            // 循环每条记录   
            while ( $record = mysql_fetch_array ( $data ) )   
            {   
                // 单条记录   
                $sql .= $this->_insert_record ( $tablename, $num_fields, $record );   
                // 如果大于分卷大小，则写入文件   
                if (strlen ( $sql ) >= $size * 1000)   
                {   
                    $file = $filename . "_v" . $p . ".sql";   
                    if ($this->_write_file ( $sql, $file, $dir ))   
                    {   
                        echo "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename</span><br />";   
                    }   
                    else  
                    {   
                        echo "备份表-" . $tablename . "-失败<br />";   
                    }   
                    // 下一个分卷   
                    $p ++;   
                    // 重置$sql变量为空，重新计算该变量大小   
                    $sql = "";   
                }   
            }   
            // sql大小不够分卷大小   
            if ($sql != "")   
            {   
                $filename .= "_v" . $p . ".sql";   
                if ($this->_write_file ( $sql, $filename, $dir ))   
                {   
                    echo "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename</span><br />";   
                }   
                else  
                {   
                    echo "备份卷-" . $p . "-失败<br />";   
                }   
            }   
        }   
        else  
        { // 备份全部表   
            if ($tables = mysql_query ( "show table status from " . $this->database ))   
            {   
                echo "读取数据库结构成功！<br />";   
            }   
            else  
            {   
                exit ( "读取数据库结构成功！<br />" );   
            }   
            // 插入dump信息   
            $sql .= $this->_retrieve();   
            // 文件名前面部分   
            $filename = date ( 'YmdHis' ) . "_all";   
            // 查出所有表   
            $tables = mysql_query ( 'SHOW TABLES' );   
            // 第几分卷   
            $p = 1;   
            // 循环所有表   
            while ( $table = mysql_fetch_array ( $tables ) )   
            {   
                // 获取表名   
                $tablename = $table [0];   
                // 获取表结构   
                $sql .= $this->_insert_table_structure ( $tablename );   
                $data = mysql_query ( "select * from " . $tablename );   
                $num_fields = mysql_num_fields ( $data );   
    
                // 循环每条记录   
                while ( $record = mysql_fetch_array ( $data ) )   
                {   
                    // 单条记录   
                    $sql .= $this->_insert_record ( $tablename, $num_fields, $record );   
                    // 如果大于分卷大小，则写入文件   
                    if (strlen ( $sql ) >= $size * 1000)   
                    {   
    
                        $file = $filename . "_v" . $p . ".sql";   
                        // 写入文件   
                        if ($this->_write_file ( $sql, $file, $dir ))   
                        {   
                            echo "-卷-" . $p . "-数据备份完成,生成备份文件<span style='color:#f00;'>$dir$file</span><br />";   
                        }   
                        else  
                        {   
                            echo "备份卷-" . $p . "-失败<br />";   
                        }   
                        // 下一个分卷   
                        $p ++;   
                        // 重置$sql变量为空，重新计算该变量大小   
                        $sql = "";   
                    }   
                }   
            }   
            // sql大小不够分卷大小   
            if ($sql != "")   
            {   
                $filename .= "_v" . $p . ".sql";   
                if ($this->_write_file ( $sql, $filename, $dir ))   
                {   
                    echo "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename<br />";   
                }   
                else  
                {   
                    echo "备份卷-" . $p . "-失败<br />";   
                }   
            }   
        }   
    }   
    
    /**  
     * 插入数据库备份基础信息  
     *  
     * @return string  
     */  
    private function _retrieve() {   
        $value = '';   
        $value .= '--' . $this->ds;   
        $value .= '-- MySQL database dump' . $this->ds;   
        $value .= '-- Created by DBManage class, Power By yanue. ' . $this->ds;   
        $value .= '-- http://yanue.net ' . $this->ds;   
        $value .= '--' . $this->ds;   
        $value .= '-- 主机: ' . $this->host . $this->ds;   
        $value .= '-- 生成日期: ' . date ( 'Y' ) . ' 年  ' . date ( 'm' ) . ' 月 ' . date ( 'd' ) . ' 日 ' . date ( 'H:i' ) . $this->ds;   
        $value .= '-- MySQL版本: ' . mysql_get_server_info () . $this->ds;   
        $value .= '-- PHP 版本: ' . phpversion () . $this->ds;   
        $value .= $this->ds;   
        $value .= '--' . $this->ds;   
        $value .= '-- 数据库: `' . $this->database . '`' . $this->ds;   
        $value .= '--' . $this->ds . $this->ds;   
        $value .= '-- -------------------------------------------------------';   
        $value .= $this->ds . $this->ds;   
        return $value;   
    }   
    
    /**  
     * 插入表结构  
     *  
     * @param unknown_type $table  
     * @return string  
     */  
    private function _insert_table_structure($table) {   
        $sql = '';   
        $sql .= "--" . $this->ds;   
        $sql .= "-- 表的结构" . $table . $this->ds;   
        $sql .= "--" . $this->ds . $this->ds;   
    
        // 如果存在则删除表   
        $sql .= "DROP TABLE IF EXISTS `" . $table . '`' . $this->sqlEnd . $this->ds;   
        // 获取详细表信息   
        $res = mysql_query ( 'SHOW CREATE TABLE `' . $table . '`' );   
        $row = mysql_fetch_array ( $res );   
        $sql .= $row [1];   
        $sql .= $this->sqlEnd . $this->ds;   
        // 加上   
        $sql .= $this->ds;   
        $sql .= "--" . $this->ds;   
        $sql .= "-- 转存表中的数据 " . $table . $this->ds;   
        $sql .= "--" . $this->ds;   
        $sql .= $this->ds;   
        return $sql;   
    }   
    
    /**  
     * 插入单条记录  
     *  
     * @param string $table  
     * @param int $num_fields  
     * @param array $record  
     * @return string  
     */  
    private function _insert_record($table, $num_fields, $record) {   
        // sql字段逗号分割   
        $insert = $comma = "";   
        $insert .= "INSERT INTO `" . $table . "` VALUES(";   
        // 循环每个子段下面的内容   
        for($i = 0; $i < $num_fields; $i ++) {   
            $insert .= ($comma . "'" .mysql_real_escape_string ( $record [$i] ) . "'");   
            $comma = ",";   
        }   
        $insert .= ");" . $this->ds;   
        return $insert;   
    }   
    
    /**  
     * 写入文件  
     *  
     * @param string $sql  
     * @param string $filename  
     * @param string $dir  
     * @return boolean  
     */  
    private function _write_file($sql, $filename, $dir) {   
        $dir = $dir ? $dir : './backup/';   
        // 不存在文件夹则创建   
        if (! file_exists ( $dir )) {   
            mkdir ( $dir );   
        }   
        $re = true;   
        if (! @$fp = fopen ( $dir . $filename, "w+" )) {   
            $re = false;   
            echo "打开文件失败！";   
        }   
        if (! @fwrite ( $fp, $sql )) {   
            $re = false;   
            echo "写入文件失败，请文件是否可写";   
        }   
        if (! @fclose ( $fp )) {   
            $re = false;   
            echo "关闭文件失败！";   
        }   
        return $re;   
    }   
    
    /*  
      *  
      * -------------------------------上：数据库导出-----------分割线----------下：数据库导入--------------------------------  
      */  
    
    /**  
     * 导入备份数据  
     * 说明：分卷文件格式20120516211738_all_v1.sql  
     * 参数：文件路径(必填)  
     *  
     * @param string $sqlfile  
     */  
    function restore($sqlfile)   
    {   
        // 检测文件是否存在   
        if (! file_exists ( $sqlfile ))   
        {   
            exit ( "文件不存在！请检查" );   
        }   
        $this->lock ( $this->database );   
        // 获取数据库存储位置   
        $sqlpath = pathinfo ( $sqlfile );   
        $this->sqldir = $sqlpath ['dirname'];   
        // 检测是否包含分卷，将类似20120516211738_all_v1.sql从_v分开,有则说明有分卷   
        $volume = explode ( "_v", $sqlfile );   
        $volume_path = $volume [0];   
        echo "请勿刷新及关闭浏览器以防止程序被中止，如有不慎！将导致数据库结构受损<br />";   
        echo "正在导入备份数据，请稍等！<br />";   
        if (emptyempty ( $volume [1] ))   
        {   
            echo "正在导入sql：<span style='color:#f00;'>" . $sqlfile . '</span><br />';   
            // 没有分卷   
            if ($this->_import ( $sqlfile )) {   
                echo "数据库导入成功！";   
            }   
            else  
            {   
                exit ( '数据库导入失败！' );   
            }   
        }   
        else  
        {   
            //$volume_id = array();   
            // 存在分卷，则获取当前是第几分卷，循环执行余下分卷   
            $volume_id = explode ( ".sq", $volume [1] );   
            // 当前分卷为$volume_id   
            $volume_id = intval ( $volume_id [0] );   
            while ( $volume_id )   
            {   
                $tmpfile = $volume_path . "_v" . $volume_id . ".sql";   
                // 存在其他分卷，继续执行   
                if (file_exists ( $tmpfile )) {   
                    // 执行导入方法   
                    echo "正在导入分卷<span style='color:#f00;'>" . $tmpfile . '</span><br />';   
                    if ($this->_import ( $tmpfile ))   
                    {   
    
                    }   
                    else  
                    {   
                        exit ( "导入分卷<span style='color:#f00;'>" . $tmpfile . '</span>失败！可能是数据库结构已损坏！请尝试从分卷1开始导入' );   
                    }   
                }   
                else  
                {   
                    echo "此分卷备份全部导入成功！<br />";   
                    return;   
                }   
                $volume_id++;   
            }   
        }   
    }   
    
    /**  
     * 将sql导入到数据库（普通导入）  
     *  
     * @param string $sqlfile  
     * @return boolean  
     */  
    private function _import($sqlfile) {   
        // sql文件包含的sql语句数组   
        $sqls = array ();   
        $f = fopen ( $sqlfile, "rb" );   
        // 创建表缓冲变量   
        $create = '';   
        while ( ! feof ( $f ) ) {   
            // 读取每一行sql   
            $line = fgets ( $f );   
            // 如果包含'-- '等注释，或为空白行，则跳过   
            if (trim ( $line ) == '' || preg_match ( '/--*?/', $line, $match )) {   
                continue;   
            }   
            // 如果结尾包含';'(即为一个完整的sql语句，这里是插入语句)，并且不包含'ENGINE='(即创建表的最后一句)，   
            if (! preg_match ( '/;/', $line, $match ) || preg_match ( '/ENGINE=/', $line, $match )) {   
                // 将本次sql语句与创建表sql连接存起来   
                $create .= $line;   
                // 如果包含了创建表的最后一句   
                if (preg_match ( '/ENGINE=/', $create, $match )) {   
                    // 则将其合并到sql数组   
                    $sqls [] = $create;   
                    // 清空当前，准备下一个表的创建   
                    $create = '';   
                }   
                // 跳过本次   
                continue;   
            }   
            $sqls [] = $line;   
        }   
        fclose ( $f );   
        // 循环sql语句数组，分别执行   
        foreach ( $sqls as $sql ) {   
            str_replace ( "n", "", $sql );   
            if (! mysql_query ( trim ( $sql ) )) {   
                echo mysql_error ();   
                return false;   
            }   
        }   
        return true;   
    }   
    
    /*  
      * -------------------------------数据库导入end---------------------------------  
      */  
    
    // 关闭数据库连接   
    private function close() {   
        mysql_close ( $this->db );   
    }   
    
    // 锁定数据库，以免备份或导入时出错   
    private function lock($tablename, $op = "WRITE") {   
        if (mysql_query ( "lock tables " . $tablename . " " . $op ))   
            return true;   
        else  
            return false;   
    }   
    
    // 解锁   
    private function unlock() {   
        if (mysql_query ( "unlock tables" ))   
            return true;   
        else  
            return false;   
    }   
    
    // 析构   
    function __destruct() {   
        mysql_query ( "unlock tables", $this->db );   
        mysql_close ( $this->db );   
    }
} 
 
$db = new DBManage ( 'localhost', 'root', '', 'tao', 'gbk' );  
//$db->backup ('tao_admin');   
$db->restore ( './backup/20140228222713_tao_admin_v1.sql'); 
?>