Db数据库备份和数据库恢复

• ------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’);
  *----------------------------------------------------------------

<?php
/**
 * 说明：分卷文件是以_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');
?>