php数据库导入导出类,mysql备份恢复分卷处理

2013年5月12日 分类: Mysql, PHP (182 个脚步)

实现mysql数据库分卷备份,选择表进行备份,实现单个sql文件及分卷sql导入。

分卷导出思路:统计sql语句变量的长度,按1个字符当成1 字节比较,如果大于设定分卷大小,则写入一个sql文件。

分卷导入思路:按行读取sql文件,将每一行当作完整的sql语句存到数组再循环执行插入数据库就可以了,但是在创建表语句分了多行,这个需要单独处理;

导出后的sql文件格式如下:

--
-- MySQL database dump
-- Created by DBManage class, Power By yanue.
-- http://yanue.net
--
-- 主机: localhost
-- 生成日期: 2012 年 10 月 06 日 22:32
-- MySQL版本: 5.1.50-community
-- PHP 版本: 5.3.9-ZS5.6.0

--
-- 数据库: `test`
--

-- -------------------------------------------------------

--
-- 表的结构aa
--

DROP TABLE IF EXISTS `aa`;
CREATE TABLE `aa` (
`id` int(10) unsigned NOT NULL AUTO_INCREMENT,
`content` text NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;

--
-- 转存表中的数据 aa
--

INSERT INTO `aa` VALUES('1','<p id=\"test\"><span class=\'hahh\' style=\"line-height:;\">我是测试数据 呵呵</span></p>');

下面是类代码:

<?php
/**
 * @author yanue
 * @copyright  Copyright (c) 2012 yanue.net
 * @link  http://yanue.net/archives/174.html
 * @version 1.1
 * 创建时间: 2012年5月21日

更新时间: 2012年10月6日
更新说明: 1.去除sql导入的时候排除sql文件里面的注释'-- ' 从而解决sql中单双引号不能导入
		   2.单行读取后的sql直接执行,避免重新将sql语句组合到数组中再从数组中读取导入sql,提高效率

 * 说明:分卷文件是以_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;
	var $msg; // 提示信息
	          // 换行符
	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 );
	}

	/*
	 * 新增查询数据库表
	 */
	function getTables() {
		$res = mysql_query ( "SHOW TABLES" );
		$tables = array ();
		while ( $row = mysql_fetch_array ( $res ) ) {
			$tables [] = $row [0];
		}
		return $tables;
	}

	/*
	 *
	 * ------------------------------------------数据库备份start----------------------------------------------------------
	 */

	/**
	 * 数据库备份
	 * 参数:备份哪个表(可选),备份目录(可选,默认为backup),分卷大小(可选,默认2000,即2M)
	 *
	 * @param $string $dir        	
	 * @param int $size        	
	 * @param $string $tablename        	
	 */
	function backup($tablename = '', $dir, $size) {
		$dir = $dir ? $dir : './backup/';
		// 创建目录
		if (! is_dir ( $dir )) {
			mkdir ( $dir, 0777, true ) or die ( '创建文件夹失败' );
		}
		$size = $size ? $size : 2000;
		$sql = '';
		// 只备份某个表
		if (! empty ( $tablename )) {
			$this->msg .= '正在备份表' . $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 )) {
						$this->msg .= "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename</span><br />";
					} else {
						$this->msg .= "备份表-" . $tablename . "-失败<br />";
						return false;
					}
					// 下一个分卷
					$p ++;
					// 重置$sql变量为空,重新计算该变量大小
					$sql = "";
				}
			}
			// sql大小不够分卷大小
			if ($sql != "") {
				$filename .= "_v" . $p . ".sql";
				if ($this->_write_file ( $sql, $filename, $dir )) {
					$this->msg .= "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename</span><br />";
				} else {
					$this->msg .= "备份卷-" . $p . "-失败<br />";
					return false;
				}
			}
		} else { // 备份全部表
			if ($tables = mysql_query ( "show table status from " . $this->database )) {
				$this->msg .= "读取数据库结构成功!<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 )) {
							$this->msg .= "-卷-" . $p . "-数据备份完成,生成备份文件<span style='color:#f00;'>$dir$file</span><br />";
						} else {
							$this->msg .= "备份卷-" . $p . "-失败<br />";
							return false;
						}
						// 下一个分卷
						$p ++;
						// 重置$sql变量为空,重新计算该变量大小
						$sql = "";
					}
				}
			}
			// sql大小不够分卷大小
			if ($sql != "") {
				$filename .= "_v" . $p . ".sql";
				if ($this->_write_file ( $sql, $filename, $dir )) {
					$this->msg .= "-卷-" . $p . "-数据备份完成,生成备份文件 <span style='color:#f00;'>$dir$filename<br />";
				} else {
					$this->msg .= "备份卷-" . $p . "-失败<br />";
					return false;
				}
			}
		}
		return true;
	}

	/**
	 * 插入数据库备份基础信息
	 *
	 * @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_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 (! is_dir ( $dir )) {
			mkdir ( $dir, 0777, true );
		}
		$re = true;
		if (! @$fp = fopen ( $dir . $filename, "w+" )) {
			$re = false;
			$this->msg .= "打开文件失败!";
		}
		if (! @fwrite ( $fp, $sql )) {
			$re = false;
			$this->msg .= "写入文件失败,请文件是否可写";
		}
		if (! @fclose ( $fp )) {
			$re = false;
			$this->msg .= "关闭文件失败!";
		}
		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];
		$this->msg .= "请勿刷新及关闭浏览器以防止程序被中止,如有不慎!将导致数据库结构受损<br />";
		$this->msg .= "正在导入备份数据,请稍等!<br />";
		if (empty ( $volume [1] )) {
			$this->msg .= "正在导入sql:<span style='color:#f00;'>" . $sqlfile . '</span><br />';
			// 没有分卷
			if ($this->_import ( $sqlfile )) {
				$this->msg .= "数据库导入成功!";
			} else {
				exit ( '数据库导入失败!' );
			}
		} else {
			// 存在分卷,则获取当前是第几分卷,循环执行余下分卷
			$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 )) {
					// 执行导入方法
					$this->msg .= "正在导入分卷 $volume_id :<span style='color:#f00;'>" . $tmpfile . '</span><br />';
					if ($this->_import ( $tmpfile )) {

					} else {
						$volume_id = $volume_id ? $volume_id :1;
						exit ( "导入分卷:<span style='color:#f00;'>" . $tmpfile . '</span>失败!可能是数据库结构已损坏!请尝试从分卷1开始导入' );
					}
				} else {
					$this->msg .= "此分卷备份全部导入成功!<br />";
					return;
				}
				$volume_id ++;
			}
		}
	}

	/**
	 * 将sql导入到数据库(普通导入)
	 *
	 * @param string $sqlfile        	
	 * @return boolean
	 */
	private function _import($sqlfile) {
		// sql文件包含的sql语句数组
		$sqls = array ();
		$f = fopen ( $sqlfile, "rb" );
		// 创建表缓冲变量
		$create_table = '';
		while ( ! feof ( $f ) ) {
			// 读取每一行sql
			$line = fgets ( $f );
			// 这一步为了将创建表合成完整的sql语句
			// 如果结尾没有包含';'(即为一个完整的sql语句,这里是插入语句),并且不包含'ENGINE='(即创建表的最后一句)
			if (! preg_match ( '/;/', $line ) || preg_match ( '/ENGINE=/', $line )) {
				// 将本次sql语句与创建表sql连接存起来
				$create_table .= $line;
				// 如果包含了创建表的最后一句
				if (preg_match ( '/ENGINE=/', $create_table)) {
			        //执行sql语句创建表
					$this->_insert_into($create_table);
					// 清空当前,准备下一个表的创建
					$create_table = '';
				}
				// 跳过本次
				continue;
			}
			//执行sql语句
			$this->_insert_into($line);
		}
		fclose ( $f );
		return true;
	}

	//插入单条sql语句
	private function _insert_into($sql){
		if (! mysql_query ( trim ( $sql ) )) {
			$this->msg .= mysql_error ();
			return false;
		}
	}

	/*
	 * -------------------------------数据库导入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 );
	}
}

【说明】 本文参考自:http://yanue.net/archives/174.html , 由柠之漠然(枫之落叶)整理编辑.
本文链接地址:https://blog.shiniv.com/2013/05/php%e6%95%b0%e6%8d%ae%e5%ba%93%e5%af%bc%e5%85%a5%e5%af%bc%e5%87%ba%e7%b1%bb%ef%bc%8cmysql%e5%a4%87%e4%bb%bd%e6%81%a2%e5%a4%8d%e5%88%86%e5%8d%b7%e5%a4%84%e7%90%86/ , 转载请保留本说明!

标签: , , ,
目前还没有任何评论.
你必须要启用 Javascript