1 <?php2 /**3 * @author yanue4 * @copyright Copyright (c) 2012 yanue.net5 * @link http://yanue.net/6 * @version 1.07 * 创建时间: 2012年5月21日8 *9 * 说明:分卷文件是以_v1.sql为结尾(20120522021241_all_v1.sql)10 * 功能:实现mysql数据库分卷备份,选择表进行备份,实现单个sql文件及分卷sql导入11 * 使用方法:12 *13 * ------1. 数据库备份(导出)------------------------------------------------------------14 //分别是主机,用户名,密码,数据库名,数据库编码15 $db = new DBManage ( 'localhost', 'root', 'root', 'test', 'utf8' );16 // 参数:备份哪个表(可选),备份目录(可选,默认为backup),分卷大小(可选,默认2000,即2M)17 $db->backup ();18 * ------2. 数据库恢复(导入)------------------------------------------------------------19 //分别是主机,用户名,密码,数据库名,数据库编码20 $db = new DBManage ( 'localhost', 'root', 'root', 'test', 'utf8' );21 //参数:sql文件22 $db->restore ( './backup/20120516211738_all_v1.sql');23 *----------------------------------------------------------------------24 */
25 classDBManage26 {27 var $db; //数据库连接
28 var $database; //所用数据库
29 var $sqldir; //数据库备份文件夹
30 var $record;31 //换行符
32 private $ds = "\n";33 //存储SQL的变量
34 public $sqlContent = "";35 //每条sql语句的结尾符
36 public $sqlEnd = ";";37 /**38 * 初始化39 *40 * @param string $host41 * @param string $username42 * @param string $password43 * @param string $thisatabase44 * @param string $charset45 */
46 function __construct($host = 'localhost', $username = 'root', $password = '', $thisatabase = 'test', $charset = 'utf8')47 {48 $this->host = $host;49 $this->username = $username;50 $this->password = $password;51 $this->database = $thisatabase;52 $this->charset = $charset;53 //连接数据库
54 $this->db = mysql_connect ( $this->host, $this->username, $this->password ) or die ( "数据库连接失败.");55 //选择使用哪个数据库
56 mysql_select_db ( $this->database, $this->db ) or die ( "无法打开数据库");57 //数据库编码方式
58 mysql_query ( 'SET NAMES ' . $this->charset, $this->db );59 }60
61 /*
62 * ------------------------------------------数据库备份start----------------------------------------------------------63 */
64
65 /**66 * 数据库备份67 * 参数:备份哪个表(可选),备份目录(可选,默认为backup),分卷大小(可选,默认2000,即2M)68 *69 * @param $string $dir70 * @param int $size71 * @param $string $tablename72 */
73 function backup($tablename = '', $dir, $size)74 {75 $dir = $dir ? $dir : 'backup/';76 $size = $size ? $size : 2000;77 $sql = '';78 //只备份某个表
79 if (! empty ( $tablename))80 {81 echo '正在备份表' . $tablename . '
';82 //插入dump信息
83 $sql = $this->_retrieve();84 //插入表结构信息
85 $sql .= $this->_insert_table_structure ( $tablename);86 //插入数据
87 $data = mysql_query ( "select * from " . $tablename);88 //文件名前面部分
89 $filename = date ( 'YmdHis' ) . "_" . $tablename;90 //字段数量
91 $num_fields = mysql_num_fields ( $data);92 //第几分卷
93 $p = 1;94 //循环每条记录
95 while ( $record = mysql_fetch_array ( $data) )96 {97 //单条记录
98 $sql .= $this->_insert_record ( $tablename, $num_fields, $record);99 //如果大于分卷大小,则写入文件
100 if (strlen ( $sql ) >= $size * 1000)101 {102 $file = $filename . "_v" . $p . ".sql";103 if ($this->_write_file ( $sql, $file, $dir))104 {105 echo "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 $dir$filename
";106 }107 else
108 {109 echo "备份表-" . $tablename . "-失败
";110 }111 //下一个分卷
112 $p ++;113 //重置$sql变量为空,重新计算该变量大小
114 $sql = "";115 }116 }117 //sql大小不够分卷大小
118 if ($sql != "")119 {120 $filename .= "_v" . $p . ".sql";121 if ($this->_write_file ( $sql, $filename, $dir))122 {123 echo "表-" . $tablename . "-卷-" . $p . "-数据备份完成,生成备份文件 $dir$filename
";124 }125 else
126 {127 echo "备份卷-" . $p . "-失败
";128 }129 }130 }131 else
132 { //备份全部表
133 if ($tables = mysql_query ( "show table status from " . $this->database ))134 {135 echo "读取数据库结构成功!
";136 }137 else
138 {139 exit ( "读取数据库结构成功!
");140 }141 //插入dump信息
142 $sql .= $this->_retrieve();143 //文件名前面部分
144 $filename = date ( 'YmdHis' ) . "_all";145 //查出所有表
146 $tables = mysql_query ( 'SHOW TABLES');147 //第几分卷
148 $p = 1;149 //循环所有表
150 while ( $table = mysql_fetch_array ( $tables) )151 {152 //获取表名
153 $tablename = $table [0];154 //获取表结构
155 $sql .= $this->_insert_table_structure ( $tablename);156 $data = mysql_query ( "select * from " . $tablename);157 $num_fields = mysql_num_fields ( $data);158
159 //循环每条记录
160 while ( $record = mysql_fetch_array ( $data) )161 {162 //单条记录
163 $sql .= $this->_insert_record ( $tablename, $num_fields, $record);164 //如果大于分卷大小,则写入文件
165 if (strlen ( $sql ) >= $size * 1000)166 {167
168 $file = $filename . "_v" . $p . ".sql";169 //写入文件
170 if ($this->_write_file ( $sql, $file, $dir))171 {172 echo "-卷-" . $p . "-数据备份完成,生成备份文件$dir$file
";173 }174 else
175 {176 echo "备份卷-" . $p . "-失败
";177 }178 //下一个分卷
179 $p ++;180 //重置$sql变量为空,重新计算该变量大小
181 $sql = "";182 }183 }184 }185 //sql大小不够分卷大小
186 if ($sql != "")187 {188 $filename .= "_v" . $p . ".sql";189 if ($this->_write_file ( $sql, $filename, $dir))190 {191 echo "-卷-" . $p . "-数据备份完成,生成备份文件 $dir$filename
";192 }193 else
194 {195 echo "备份卷-" . $p . "-失败
";196 }197 }198 }199 }200
201 /**202 * 插入数据库备份基础信息203 *204 * @return string205 */
206 private function_retrieve() {207 $value = '';208 $value .= '--' . $this->ds;209 $value .= '-- MySQL database dump' . $this->ds;210 $value .= '-- Created by DBManage class, Power By yanue. ' . $this->ds;211 $value .= '-- http://yanue.net ' . $this->ds;212 $value .= '--' . $this->ds;213 $value .= '-- 主机: ' . $this->host . $this->ds;214 $value .= '-- 生成日期: ' . date ( 'Y' ) . ' 年 ' . date ( 'm' ) . ' 月 ' . date ( 'd' ) . ' 日 ' . date ( 'H:i' ) . $this->ds;215 $value .= '-- MySQL版本: ' . mysql_get_server_info () . $this->ds;216 $value .= '-- PHP 版本: ' . phpversion () . $this->ds;217 $value .= $this->ds;218 $value .= '--' . $this->ds;219 $value .= '-- 数据库: `' . $this->database . '`' . $this->ds;220 $value .= '--' . $this->ds . $this->ds;221 $value .= '-- -------------------------------------------------------';222 $value .= $this->ds . $this->ds;223 return $value;224 }225
226 /**227 * 插入表结构228 *229 * @param unknown_type $table230 * @return string231 */
232 private function _insert_table_structure($table) {233 $sql = '';234 $sql .= "--" . $this->ds;235 $sql .= "-- 表的结构" . $table . $this->ds;236 $sql .= "--" . $this->ds . $this->ds;237
238 //如果存在则删除表
239 $sql .= "DROP TABLE IF EXISTS `" . $table . '`' . $this->sqlEnd . $this->ds;240 //获取详细表信息
241 $res = mysql_query ( 'SHOW CREATE TABLE `' . $table . '`');242 $row = mysql_fetch_array ( $res);243 $sql .= $row [1];244 $sql .= $this->sqlEnd . $this->ds;245 //加上
246 $sql .= $this->ds;247 $sql .= "--" . $this->ds;248 $sql .= "-- 转存表中的数据 " . $table . $this->ds;249 $sql .= "--" . $this->ds;250 $sql .= $this->ds;251 return $sql;252 }253
254 /**255 * 插入单条记录256 *257 * @param string $table258 * @param int $num_fields259 * @param array $record260 * @return string261 */
262 private function _insert_record($table, $num_fields, $record) {263 //sql字段逗号分割
264 $comma = "";265 $insert .= "INSERT INTO `" . $table . "` VALUES(";266 //循环每个子段下面的内容
267 for($i = 0; $i < $num_fields; $i ++) {268 $insert .= ($comma . "'" . mysql_escape_string ( $record [$i] ) . "'");269 $comma = ",";270 }271 $insert .= ");" . $this->ds;272 return $insert;273 }274
275 /**276 * 写入文件277 *278 * @param string $sql279 * @param string $filename280 * @param string $dir281 * @return boolean282 */
283 private function _write_file($sql, $filename, $dir) {284 $dir = $dir ? $dir : './backup/';285 //不存在文件夹则创建
286 if (! file_exists ( $dir)) {287 mkdir ( $dir);288 }289 $re = true;290 if (! @$fp = fopen ( $dir . $filename, "w+")) {291 $re = false;292 echo "打开文件失败!";293 }294 if (! @fwrite ( $fp, $sql)) {295 $re = false;296 echo "写入文件失败,请文件是否可写";297 }298 if (! @fclose ( $fp)) {299 $re = false;300 echo "关闭文件失败!";301 }302 return $re;303 }304
305 /*
306 *307 * -------------------------------上:数据库导出-----------分割线----------下:数据库导入--------------------------------308 */
309
310 /**311 * 导入备份数据312 * 说明:分卷文件格式20120516211738_all_v1.sql313 * 参数:文件路径(必填)314 *315 * @param string $sqlfile316 */
317 function restore($sqlfile)318 {319 //检测文件是否存在
320 if (! file_exists ( $sqlfile))321 {322 exit ( "文件不存在!请检查");323 }324 $this->lock ( $this->database );325 //获取数据库存储位置
326 $sqlpath = pathinfo ( $sqlfile);327 $this->sqldir = $sqlpath ['dirname'];328 //检测是否包含分卷,将类似20120516211738_all_v1.sql从_v分开,有则说明有分卷
329 $volume = explode ( "_v", $sqlfile);330 $volume_path = $volume [0];331 echo "请勿刷新及关闭浏览器以防止程序被中止,如有不慎!将导致数据库结构受损
";332 echo "正在导入备份数据,请稍等!
";333 if (empty ( $volume [1] ))334 {335 echo "正在导入sql:" . $sqlfile . '
';336 //没有分卷
337 if ($this->_import ( $sqlfile)) {338 echo "数据库导入成功!";339 }340 else
341 {342 exit ( '数据库导入失败!');343 }344 }345 else
346 {347 //$volume_id = array();348 // 存在分卷,则获取当前是第几分卷,循环执行余下分卷
349 $volume_id = explode ( ".sq", $volume [1] );350 //当前分卷为$volume_id
351 $volume_id = intval ( $volume_id [0] );352 while ( $volume_id)353 {354 $tmpfile = $volume_path . "_v" . $volume_id . ".sql";355 //存在其他分卷,继续执行
356 if (file_exists ( $tmpfile)) {357 //执行导入方法
358 echo "正在导入分卷$volume_id:" . $tmpfile . '
';359 if ($this->_import ( $tmpfile))360 {361
362 }363 else
364 {365 exit ( "导入分卷$volume_id:" . $tmpfile . '失败!可能是数据库结构已损坏!请尝试从分卷1开始导入');366 }367 }368 else
369 {370 echo "此分卷备份全部导入成功!
";371 return;372 }373 $volume_id++;374 }375 }376 }377
378 /**379 * 将sql导入到数据库(普通导入)380 *381 * @param string $sqlfile382 * @return boolean383 */
384 private function _import($sqlfile) {385 //sql文件包含的sql语句数组
386 $sqls = array();387 $f = fopen ( $sqlfile, "rb");388 //创建表缓冲变量
389 $create = '';390 while ( ! feof ( $f) ) {391 //读取每一行sql
392 $line = fgets ( $f);393 //如果包含'-- '等注释,或为空白行,则跳过
394 if (trim ( $line ) == '' || preg_match ( '/--*?/', $line, $match)) {395 continue;396 }397 //如果结尾包含';'(即为一个完整的sql语句,这里是插入语句),并且不包含'ENGINE='(即创建表的最后一句),
398 if (! preg_match ( '/;/', $line, $match ) || preg_match ( '/ENGINE=/', $line, $match)) {399 //将本次sql语句与创建表sql连接存起来
400 $create .= $line;401 //如果包含了创建表的最后一句
402 if (preg_match ( '/ENGINE=/', $create, $match)) {403 //则将其合并到sql数组
404 $sqls [] = $create;405 //清空当前,准备下一个表的创建
406 $create = '';407 }408 //跳过本次
409 continue;410 }411 $sqls [] = $line;412 }413 fclose ( $f);414 //循环sql语句数组,分别执行
415 foreach ( $sqls as $sql) {416 str_replace ( "\n", "", $sql);417 if (! mysql_query ( trim ( $sql) )) {418 echo mysql_error();419 return false;420 }421 }422 return true;423 }424
425 /*
426 * -------------------------------数据库导入end---------------------------------427 */
428
429 //关闭数据库连接
430 private functionclose() {431 mysql_close ( $this->db );432 }433
434 //锁定数据库,以免备份或导入时出错
435 private function lock($tablename, $op = "WRITE") {436 if (mysql_query ( "lock tables " . $tablename . " " . $op))437 return true;438 else
439 return false;440 }441
442 //解锁
443 private functionunlock() {444 if (mysql_query ( "unlock tables"))445 return true;446 else
447 return false;448 }449
450 //析构
451 function__destruct() {452 mysql_query ( "unlock tables", $this->db );453 mysql_close ( $this->db );454 }455 }456 ?>