数据库导入导出是一个后台必要拥有的功能,网上一搜,有很多关于数据库导入导出的,但基本上一个大的系统,包含了许多我们并不需要的,而且他们都是自己的后台的形式,我并不喜欢的是拿人家的东西整合到自己的后台,我需要的是自己东西。于是参照了很多,自己写了一个关于分卷导入类。以方便调用。欢迎大家拍砖。

具体完整实现mysql导入导出功能见:http://yanue.net/post-19.html

//分别是主机,用户名,密码,数据库名,数据库编码
$db = new DataManage ( 'localhost', 'root', 'root', 'test', 'utf8' );
//sql文件,是否只导入单个sql(即如果有其他分卷也不导入)
$db->restore ( './backup/20120516211738_all_v1.sql', false );

这里针对分卷文件是以‘_v1.sql’为结尾,实现单个sql文件及分卷sql导入,分卷导入可选择是否当前分卷导入余下分卷,我们只需要直接调用类即可完成

对应如何去列出备份的sql文件或选择sql之类的,自己去实现,那个不在这个范畴了,也很简单的。

实际演示效果:

还有目前只实现了数据库导入,关于数据库导出的,正在编写功能。

下面是完整的类代码:(具体思路及实现代码里面都有说明,这里不在赘述)

<?php
/**
 * @author yanue
 * @copyright	Copyright (c) 2012 www.yanue.net
 * @link  http://www.yanue.net/archives/150.html
 * 说明:分卷文件是以_v1.sql为结尾
 * 功能:实现单个sql文件及分卷sql导入,分卷导入可选择是否当前分卷导入余下分卷
 * 使用方法:
 *
 *
 * ------------------------------------------------------------------
//分别是主机,用户名,密码,数据库名,数据库编码
$db = new DataManage ( 'localhost', 'root', 'root', 'test', 'utf8' );
//sql文件,是否只导入单个sql(即如果有其他分卷也不导入)
$db->restore ( './backup/20120516211738_all_v1.sql', false );
 *----------------------------------------------------------------------
 */
class DataManage {
	var $db; // 数据库连接
	var $database; // 所用数据库
	var $sqldir; // 数据库备份文件夹

	/**
	 * 初始化
	 *
	 * @param string $host
	 * @param string $username
	 * @param string $password
	 * @param string $database
	 * @param string $charset
	 */
	function __construct($host = 'localhost', $username = 'root', $password = '', $database = 'test', $charset = 'utf8') {
		$this->host = $host;
		$this->username = $username;
		$this->password = $password;
		$this->database = $database;
		$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 );
	}

	/**
	 * 导入备份数据
	 * 说明:分卷文件格式20120516211738_all_v1.sql
	 *
	 * @param string $sqlfile
	 * @param bool $single
	 */
	function restore($sqlfile, $single = FALSE) {
		// 检测文件是否存在
		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 (empty ( $volume [1] ) || $single) {
			echo "正在导入sql:<span style='color:#f00;'>" . $sqlfile . '</span><br />';
			// 没有分卷
			if ($this->_import ( $sqlfile )) {
				echo "数据库导入成功!";
			} 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 )) {
					// 执行导入方法
					echo "正在导入分卷$volume_id:<span style='color:#f00;'>" . $tmpfile . '</span><br />';
					if ($this->_import ( $tmpfile )) {

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

	/**
	 * 将sql导入到数据库(普通导入)
	 *
	 * @param string $sqlfile
	 * @return boolean
	 */
	private function _import($sqlfile) {
		$name = basename ( $sqlfile );
		$sqls = file ( $sqlfile );
		foreach ( $sqls as $sql ) {
			str_replace ( "\r", "", $sql );
			str_replace ( "\n", "", $sql );
			if (! mysql_query ( trim ( $sql ), $this->db ))
				return false;
		}
		return true;
	}

	// 关闭数据库连接
	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 );
	}
}