使用PHP实现一个MySQL备份和恢复程序
站长
· 阅读数 9
背景
由于客户用的阿里云将要到期，现在要把项目交给政府，需要把项目和阿里云上的数据库都转到政府云，数据肯定要转过去，但是阿里云导出数据有数量限制，导出不全，所以就要程序执行下，生成sql文件，然后导入政府云数据库。这样我们就需要一个脚本执行下，下面分享下具体实现过程。
操作类的封装
为了程序的代码重用，我们首先封装个类来实现备份与备份恢复功能，为了节约时间，直接上代码，不懂的可以看下注释
<?php
class DbOperate
{
var $conn; // 数据库连接
var $database; // 所用数据库
var $sqldir; // 数据库备份文件夹
var $record;
// sql语句变量
public $sqlContent = "";
/**
* 实例化
* @param string $host $username $password $database $charset
*/
function __construct($host = 'localhost', $username = 'root', $password = '123456', $database = 'test', $charset = 'utf8')
{
$this->host = $host;
$this->username = $username;
$this->password = $password;
$this->database = $database;
$this->charset = $charset;
// 数据库连接
$this->conn = mysqli_connect ( $this->host, $this->username, $this->password ) or die ( "数据库连接失败!" );
// 数据库选择
mysql_select_db ( $this->database, $this->conn ) or die ( "无法打开数据库" );
// 设置数据库编码
mysql_query ( 'SET NAMES ' . $this->charset, $this->conn );
}
/**
* 数据库导出sql文件
* @param 备份表名称(为空导出全部表),备份目录(默认目录‘，./sql/’),文件大小(可选,默认5000，即5M左右)
*/
function export($tablename = '', $dir, $size)
{
$dir = $dir ? $dir : './sql/';
$size = $size ? $size : 5000;
$sql = '';
// 导出单张表
if (! empty ( $tablename ))
{
echo '正在导出表' . $tablename . PHP_EOL;
// 插入数据库信息
$sql = $this->get_mysql_info();
// 插入表结构信息-获取表的基本信息
$sql .= $this->get_table_structure ( $tablename );
// 插入数据-执行查询语句
$data = mysql_query ( "SELECT * FROM " . $tablename );
// 文件名前面部分
$filename = date ( 'YmdHis' ) . "_" . $tablename;
// 字段数量
$num_fields = mysql_num_fields ( $data );
// 第一个sql文件
$no = 1;
// 循环每条记录
while ( $record = mysql_fetch_array ( $data ) )
{
// 单条记录
$sql .= $this->insert_record ( $tablename, $num_fields, $record );
// 如果大于分卷大小，则写入文件
if (strlen ( $sql ) >= $size * 1000)
{
$file = $filename . "_num" . $no . ".sql";
if ($this->write_file ( $sql, $file, $dir ))
{
echo "表-" . $tablename . "-编号-" . $no . "-数据导出完成,生成SQL文件 $dir$filename".PHP_EOL;
}
else
{
echo "导出表-" . $tablename . "-失败".PHP_EOL;
}
// 下一个SQL文件
$no ++;
// 重置$sql变量为空，重新计算该变量大小
$sql = "";
}
}
// sql大小不够
if ($sql != "")
{
$filename .= "_num" . $no . ".sql";
if ($this->write_file ( $sql, $filename, $dir ))
{
echo "表-" . $tablename . "-编号-" . $no . "-数据导出完成,生成sql文件 $dir$filename".PHP_EOL;
}
else
{
echo "表-" . $tablename . "-编号-" . $no . "-数据写入失败".PHP_EOL;
}
}
}
else
{ // 备份全部表
if ($tables = mysql_query ( "show table status from " . $this->database ))
{
echo "读取数据库结构成功".PHP_EOL;
}
else
{
exit ( "读取数据库结构成功" ).PHP_EOL;
}
// 插入dump信息
$sql .= $this->get_mysql_info();
// 文件名前面部分
$filename = date ( 'YmdHis' ) . "_all";
// 查出所有表
$tables = mysql_query ( 'SHOW TABLES' );
// SQL文件编号
$no = 1;
// 循环所有表
while ( $table = mysql_fetch_array ( $tables ) )
{
echo '开始导出表：'.$table [0].PHP_EOL;
// 获取表名
$tablename = $table [0];
// 获取表结构
$sql .= $this->get_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 . "_num" . $no . ".sql";
// 写入文件
if ($this->write_file ( $sql, $file, $dir ))
{
echo "表-" . $tablename . "-编号-" . $no . "-数据写入成功-$dir$file".PHP_EOL;
}
else
{
echo "表-" . $tablename . "-编号-" . $no . "-数据写入失败".PHP_EOL;
}
// 下一个sql文件
$no ++;
// 重置$sql变量为空，重新计算写入sql大小
$sql = "";
}
}
}
// sql大小不够分卷大小
if ($sql != "")
{
$filename .= "_num" . $p . ".sql";
if ($this->write_file ( $sql, $filename, $dir ))
{
echo "-编号-" . $no . "-数据导出完成,生成SQL文件 $dir$filename".PHP_EOL;
}
else
{
echo "-编号-" . $no . "-写入失败".PHP_EOL;
}
}
}
}
/**
* 获取mysql基础信息
*
* @return string
*/
private function get_mysql_info() {
$value = '';
$value .= '--' . "\n";
$value .= '-- 主机: ' . $this->host . "\n";
$value .= '-- 生成日期: ' . date ( 'Y' ) . ' 年 ' . date ( 'm' ) . ' 月 ' . date ( 'd' ) . ' 日 ' . date ( 'H:i' ) . "\n";
$value .= '-- MySQL版本: ' . mysql_get_server_info () . "\n";
$value .= "\n";
$value .= '--' . "\n";
$value .= '-- 数据库: `' . $this->database . '`' . "\n";
$value .= '--' . "\n" . "\n";
$value .= '-- -------------------------------------------------------';
$value .= "\n" . "\n";
return $value;
}
/**
* 获取表结构
* @param $table
* @return string
*/
private function get_table_structure($table) {
$sql = '';
$sql .= "--" . "\n";
$sql .= "-- 表的结构" . $table . "\n";
$sql .= "--" . "\n" . "\n";
// 如果存在则删除表
$sql .= "DROP TABLE IF EXISTS `" . $table . '`' . ";" . "\n";
// 获取详细表信息
$res = mysql_query ( 'SHOW CREATE TABLE `' . $table . '`' );
$row = mysql_fetch_array ( $res );
$sql .= $row [1];
$sql .= ";" . "\n";
// 加上
$sql .= "\n";
$sql .= "--" . "\n";
$sql .= "-- 转存表中的数据 " . $table . "\n";
$sql .= "--" . "\n";
$sql .= "\n";
return $sql;
}
/**
* 插入单条记录
*
* @param string $table
* @param int $num_fields
* @param array $record
* @return string
*/
private function insert_record($table, $num_fields, $record) {
// sql字段逗号分割
$comma = "";
$insert .= "INSERT INTO `" . $table . "` VALUES(";
// 循环每个子段下面的内容
for($i = 0; $i < $num_fields; $i ++) {
$insert .= ($comma . "'" . mysql_escape_string ( $record [$i] ) . "'");
$comma = ",";
}
$insert .= ");" . "\n";
return $insert;
}
/**
* 写入文件
*
* @param string $sql
* @param string $filename
* @param string $dir
* @return boolean
*/
private function write_file($sql, $filename, $dir) {
$dir = $dir ? $dir : './sql/';
// 不存在文件夹则创建
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;
}
/**
* 导入sql文件数据
* 说明：sql文件格式20120516211738_all_no1.sql
* 参数：sql文件路径(必填)
*
* @param string $sqlfile
*/
function import($sqlfile)
{
// 检测文件是否存在
if (! file_exists ( $sqlfile ))
{
exit ( "文件不存在！请检查" );
}
$this->lock ( $this->database );
// 获取数据库存储位置
$sqlpath = pathinfo ( $sqlfile );
$this->sqldir = $sqlpath ['dirname'];
$volume = explode ( "_no", $sqlfile );
$volume_path = $volume [0];
echo "正在导入sql数据，请稍等！".PHP_EOL;
if (empty ( $volume [1] ))
{
echo "正在导入sql：" . $sqlfile . PHP_EOL;
if ($this->_import ( $sqlfile )) {
echo "数据导入成功！";
}
else
{
exit ( '数据导入失败！' );
}
}
else
{
$volume_id = explode ( ".sq", $volume [1] );
$volume_id = intval ( $volume_id [0] );
while ( $volume_id )
{
$tmpfile = $volume_path . "_no" . $volume_id . ".sql";
// 存在其他分卷，继续执行
if (file_exists ( $tmpfile )) {
// 执行导入方法
echo "正在导入分卷$volume_id：" . $tmpfile . PHP_EOL;
if ($this->_import ( $tmpfile ))
{
}
else
{
exit ( "导入$volume_id：" . $tmpfile . '失败！可能是数据库结构已损坏！请尝试从编号1开始导入' );
}
}
else
{
echo "此文件数据全部导入成功！".PHP_EOL;
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;
}
// 关闭数据库连接
private function close() {
mysql_close ( $this->conn );
}
// 锁定数据库，以免备份或导入时出错
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->conn );
mysql_close ( $this->conn );
}
}
?>
调用操作类，实现导出sql、导入数据
<?php
include 'DbOperate.php';
/**
* 数据库导出完整数据
* @param 主机名、数据库账号、数据库密码、数据库名、编码格式
*/
$db = new DbOperate ( '主机名','数据库账号','数据库密码','数据库名','编码格式' );
// 参数：备份哪个表(可选),备份目录(可选，默认为backup),分卷大小(可选,默认2000，即2M)
$db->export('',"/data/wwwtest/mysql_backup/sql/");
/**
* 数据库导入
* @param sql文件路径
*/
//$db->import ( 'sql文件路径');
$db->close();//关闭数据库连接
执行以上代码，就可以实现数据库全部数据导出了，如果想要导出某张表，第一个参数写上表名称即可。那么将导出的数据导入新的数据库又该怎么办呢？直接将上面的程序改下数据库连接配置，将调用导出export的方法注释掉，打开下面的import方法，运行程序，就可以实现数据的导入了。
总结
MySQL数据库导出，虽然也可以用数据库工具实现，例如navicat等，但是对于阿里云的数据库，连接需要设置白名单，而自己公司的公网ip又经常变，操作起来就会很麻烦，所以写个脚本还是有必要的。由于工作中有用到，记录下实现过程，把工作中的经验记录下来，加深下印象。同时也希望能够帮到有需要的同学。