test_lgq/niucloud/core/util/DbBackup.php
2024-01-24 17:36:08 +08:00

526 lines
16 KiB
PHP
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

<?php
namespace core\util;
use think\facade\Db;
class DbBackup
{
/**
* 文件指针
* @var resource
*/
private $fp;
/**
* 备份文件信息 part - 卷号name - 文件名
* @var array
*/
private $file;
/**
* 当前打开文件大小
* @var integer
*/
private $size = 0;
/**
* 数据库配置
* @var integer
*/
private $dbconfig = array();
/**
* 备份配置
* @var integer
*/
private $config = array(
// 数据库备份路径
'path' => './backup/',
// 数据库备份卷大小
'part' => 20971520,
// 数据库备份文件是否启用压缩 0不压缩 1 压缩
'compress' => 0,
// 数据库备份文件压缩级别 1普通 4 一般 9最高
'level' => 9,
);
/**
* 数据库备份构造方法
* @param array $file 备份或还原的文件信息
* @param array $config 备份配置信息
*/
public function __construct($config = [])
{
$this->config = is_array($config) && !empty($config) ? array_merge($this->config, $config) : $this->config;
//初始化文件名
$this->setFile();
//初始化数据库连接参数
$this->setDbConn();
//检查文件是否可写
if (!$this->checkPath($this->config['path'])) {
throw new \Exception("The current directory is not writable");
}
}
/**
* 设置脚本运行超时时间
* 0表示不限制支持连贯操作
*/
public function setTimeout($time = null)
{
if (!is_null($time)) {
set_time_limit($time) || ini_set("max_execution_time", $time);
}
return $this;
}
/**
* 设置数据库连接必备参数
* @param array $dbconfig 数据库连接配置信息
* @return object
*/
public function setDbConn($dbconfig = [])
{
if (empty($dbconfig)) {
$this->dbconfig = config('database.connections.'.config('database.default'));
} else {
$this->dbconfig = $dbconfig;
}
return $this;
}
/**
* 设置备份文件名
*
* @param Array $file 文件名字
* @return object
*/
public function setFile($file = null)
{
if (is_null($file)) {
$this->file = ['name' => date('Ymd-His'), 'part' => 1];
} else {
if (!array_key_exists("name", $file) && !array_key_exists("part", $file)) {
$this->file = $file['1'];
} else {
$this->file = $file;
}
}
return $this;
}
/**
* 数据库表列表
*
* @param null $table
* @param int $type
* @return array
*/
public function dataList($table = null, $type = 1)
{
if (is_null($table)) {
$list = Db::query("SHOW TABLE STATUS");
} else {
if ($type) {
$list = Db::query("SHOW FULL COLUMNS FROM {$table}");
} else {
$list = Db::query("show columns from {$table}");
}
}
return array_map('array_change_key_case', $list);
}
/**
* 数据库备份文件列表
*
* @return array
*/
public function fileList()
{
if (!is_dir($this->config['path'])) {
mkdir($this->config['path'], 0755, true);
}
$path = realpath($this->config['path']);
$flag = \FilesystemIterator::KEY_AS_FILENAME;
$glob = new \FilesystemIterator($path, $flag);
$list = array();
foreach ($glob as $name => $file) {
if (preg_match('/^\\d{8,8}-\\d{6,6}-\\d+\\.sql(?:\\.gz)?$/', $name)) {
$name1 = $name;
$name = sscanf($name, '%4s%2s%2s-%2s%2s%2s-%d');
$date = "{$name[0]}-{$name[1]}-{$name[2]}";
$time = "{$name[3]}:{$name[4]}:{$name[5]}";
$part = $name[6];
if (isset($list["{$date} {$time}"])) {
$info = $list["{$date} {$time}"];
$info['part'] = max($info['part'], $part);
$info['size'] = $info['size'] + $file->getSize();
} else {
$info['part'] = $part;
$info['size'] = $file->getSize();
}
$extension = strtoupper(pathinfo($file->getFilename(), PATHINFO_EXTENSION));
$info['name'] = $name1;
$info['compress'] = $extension === 'SQL' ? '-' : $extension;
$info['time'] = strtotime("{$date} {$time}");
$list["{$date} {$time}"] = $info;
}
}
return $list;
}
/**
* 获取文件名称
*
* @param string $type
* @param int $time
* @return array|false|mixed|string
* @throws \Exception
*/
public function getFile($type = '', $time = 0)
{
//
if (!is_numeric($time)) {
throw new \Exception("{$time} Illegal data type");
}
switch ($type) {
case 'time':
$name = date('Ymd-His', $time).'-*.sql*';
$path = realpath($this->config['path']).DIRECTORY_SEPARATOR.$name;
return glob($path);
break;
case 'timeverif':
$name = date('Ymd-His', $time).'-*.sql*';
$path = realpath($this->config['path']).DIRECTORY_SEPARATOR.$name;
$files = glob($path);
$list = array();
foreach ($files as $name) {
$basename = basename($name);
$match = sscanf($basename, '%4s%2s%2s-%2s%2s%2s-%d');
$gz = preg_match('/^\\d{8,8}-\\d{6,6}-\\d+\\.sql.gz$/', $basename);
$list[$match[6]] = array($match[6], $name, $gz);
}
$last = end($list);
if (count($list) === $last[0]) {
return $list;
} else {
throw new \Exception("File {$files['0']} may be damaged, please check again");
}
break;
case 'pathname':
return "{$this->config['path']}{$this->file['name']}-{$this->file['part']}.sql";
break;
case 'filename':
return "{$this->file['name']}-{$this->file['part']}.sql";
break;
case 'filepath':
return $this->config['path'];
break;
default:
$arr = array(
'pathname' => "{$this->config['path']}{$this->file['name']}-{$this->file['part']}.sql",
'filename' => "{$this->file['name']}-{$this->file['part']}.sql",
'filepath' => $this->config['path'], 'file' => $this->file
);
return $arr;
}
}
/**
* 删除备份文件
*
* @param $time
* @return mixed
* @throws \Exception
*/
public function delFile($time)
{
if ($time) {
$file = $this->getFile('time', $time);
array_map("unlink", $file);
$file = $this->getFile('time', $time);
if (count($file)) {
throw new \Exception("File ".implode('##', $file)." deleted failed");
} else {
return $time;
}
} else {
throw new \Exception("{$time} Time parameter is incorrect");
}
}
/**
* 下载备份
*
* @param string $time
* @param integer $part
* @return array|mixed|string
*/
public function downloadFile($time, $part = 0)
{
$file = $this->getFile('time', $time);
$fileName = $file[$part];
if (file_exists($fileName)) {
ob_end_clean();
header("Cache-Control: must-revalidate, post-check=0, pre-check=0");
header('Content-Description: File Transfer');
header('Content-Type: application/octet-stream');
header('Content-Length: '.filesize($fileName));
header('Content-Disposition: attachment; filename='.basename($fileName));
readfile($fileName);
} else {
throw new \Exception("{$time} File is abnormal");
}
}
/**
* 导入表
*
* @param $start
* @param $time
* @return array|false|int
* @throws Exception
*/
public function import($start, $time)
{
//还原数据
$this->file = $this->getFile('time', $time);
if ($this->config['compress']) {
$gz = gzopen($this->file[0], 'r');
$size = 0;
} else {
$size = filesize($this->file[0]);
$gz = fopen($this->file[0], 'r');
}
$sql = '';
if ($start) {
$this->config['compress'] ? gzseek($gz, $start) : fseek($gz, $start);
}
for ($i = 0; $i < 1000; $i++) {
$sql .= $this->config['compress'] ? gzgets($gz) : fgets($gz);
if (preg_match('/.*;$/', trim($sql))) {
if (false !== Db::query($sql)) {
$start += strlen($sql);
} else {
return false;
}
$sql = '';
} elseif ($this->config['compress'] ? gzeof($gz) : feof($gz)) {
return 0;
}
}
return array($start, $size);
}
/**
* 写入初始数据
*
* @return boolean true - 写入成功false - 写入失败
*/
public function backupInit()
{
$sql = "-- -----------------------------\n";
$sql .= "-- Think MySQL Data Transfer \n";
$sql .= "-- \n";
$sql .= "-- Host : ".$this->dbconfig['hostname']."\n";
$sql .= "-- Port : ".$this->dbconfig['hostport']."\n";
$sql .= "-- Database : ".$this->dbconfig['database']."\n";
$sql .= "-- \n";
$sql .= "-- Part : #{$this->file['part']}\n";
$sql .= "-- Date : ".date("Y-m-d H:i:s")."\n";
$sql .= "-- -----------------------------\n\n";
$sql .= "SET FOREIGN_KEY_CHECKS = 0;\n\n";
return $this->write($sql);
}
/**
* 查询单条
* @param $sql
* @return array|mixed
*/
public function selectOne($sql) {
$result = Db::query($sql);
return $result[0] ?? [];
}
/**
* 备份表结构
*
* @param string $table 表名
* @param integer $start 起始行数
* @return boolean false - 备份失败
*/
public function backup($table, $start = 0)
{
// 备份表结构
if (0 == $start) {
$result = $this->selectOne("SHOW CREATE TABLE `{$table}`");
$sql = "\n";
$sql .= "-- -----------------------------\n";
$sql .= "-- Table structure for `{$table}`\n";
$sql .= "-- -----------------------------\n";
$sql .= "DROP TABLE IF EXISTS `{$table}`;\n";
$sql .= trim($result['Create Table']).";\n\n";
if (false === $this->write($sql)) {
return false;
}
}
//数据总数
$result = $this->selectOne("SELECT COUNT(*) AS count FROM `{$table}`");
$count = $result['count'];
//备份表数据
if ($count) {
//写入数据注释
if (0 == $start) {
$sql = "-- -----------------------------\n";
$sql .= "-- Records of `{$table}`\n";
$sql .= "-- -----------------------------\n";
$this->write($sql);
}
//备份数据记录
$result = Db::query("SELECT * FROM `{$table}` LIMIT {$start}, 1000");
$sql = "INSERT INTO `{$table}` VALUES\n";
foreach ($result as $index => $row) {
$row = array_map('addslashes', $row);
$sql .= "('".str_replace(array("\r", "\n"), array('\\r', '\\n'),
implode("', '", $row))."')";
$sql .= $index < (count($result) - 1) ? ",\n" : ";\n";
}
if (false === $this->write($sql)) {
return false;
}
//还有更多数据
if ($count > $start + 1000) {
return $this->backup($table, $start + 1000);
}
}
//备份下一表
return true;
}
/**
* 优化表
*
* @param String $tables 表名
* @return String $tables
*/
public function optimize($tables = null)
{
if ($tables) {
if (is_array($tables)) {
$tables = implode('`,`', $tables);
$list = db ::select("OPTIMIZE TABLE `{$tables}`");
} else {
$list = Db::query("OPTIMIZE TABLE `{$tables}`");
}
if ($list) {
return $tables;
} else {
throw new \Exception("data sheet'{$tables}'Repair mistakes please try again!");
}
} else {
throw new \Exception("Please specify the table to be repaired!");
}
}
/**
* 修复表
*
* @param String $tables 表名
* @return String $tables
*/
public function repair($tables = null)
{
if ($tables) {
if (is_array($tables)) {
$tables = implode('`,`', $tables);
$list = Db::query("REPAIR TABLE `{$tables}`");
} else {
$list = Db::query("REPAIR TABLE `{$tables}`");
}
if ($list) {
return $list;
} else {
throw new \Exception("data sheet'{$tables}'Repair mistakes please try again!");
}
} else {
throw new \Exception("Please specify the table to be repaired!");
}
}
/**
* 写入SQL语句
*
* @param string $sql 要写入的SQL语句
* @return boolean true - 写入成功false - 写入失败!
*/
private function write($sql)
{
$size = strlen($sql);
//由于压缩原因无法计算出压缩后的长度这里假设压缩率为50%
//一般情况压缩率都会高于50%
$size = $this->config['compress'] ? $size / 2 : $size;
$this->open($size);
return $this->config['compress'] ? @gzwrite($this->fp, $sql) : @fwrite($this->fp, $sql);
}
/**
* 打开一个卷,用于写入数据
*
* @param integer $size 写入数据的大小
*/
private function open($size)
{
if ($this->fp) {
$this->size += $size;
if ($this->size > $this->config['part']) {
$this->config['compress'] ? @gzclose($this->fp) : @fclose($this->fp);
$this->fp = null;
$this->file['part']++;
session('backup_file', $this->file);
$this->backupInit();
}
} else {
$backuppath = $this->config['path'];
$filename = "{$backuppath}{$this->file['name']}-{$this->file['part']}.sql";
if ($this->config['compress']) {
$filename = "{$filename}.gz";
$this->fp = @gzopen($filename, "a{$this->config['level']}");
} else {
$this->fp = @fopen($filename, 'a');
}
$this->size = filesize($filename) + $size;
}
}
/**
* 检查目录是否可写
*
* @param string $path 目录
* @return boolean
*/
protected function checkPath($path)
{
if (is_dir($path)) {
return true;
}
if (mkdir($path, 0755, true)) {
return true;
} else {
return false;
}
}
/**
* 析构方法,用于关闭文件资源
*/
public function __destruct()
{
if ($this->fp) {
$this->config['compress'] ? @gzclose($this->fp) : @fclose($this->fp);
}
}
}