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