caiping/app/admin/controller/Database.php

<?php
/**
 * @copyright Copyright (c) 2021 勾股工作室
 * @license https://opensource.org/licenses/Apache-2.0
 * @link https://www.gougucms.com
 */

declare (strict_types = 1);

namespace app\admin\controller;

use app\admin\BaseController;
use backup\Backup;
use think\facade\Session;
use think\facade\Db;
use think\facade\View;

class Database extends BaseController
{
	//数据表列表
	public function database()
    {
		if (request()->isAjax()) {
			// 数据信息
			$db = new Backup();
			$list = $db->dataList();
			// 计算总大小
			$total = 0;
			foreach ($list as $k => $v) {
				$total += $v['data_length'];
				$list[$k]['data_size'] = $v['data_length'];
				$list[$k]['data_length'] = format_bytes($v['data_length']);
			}
			// 提示信息
			$dataTips = '数据库中共有<strong> ' . count($list) . '</strong> 张表，共计 <strong>' . format_bytes($total) . '</strong>大小。';
			$data['data'] = $list;
			return table_assign(0, $dataTips, $data);
		}
		return view();
    }	
	
	//备份数据
	public function backup()
	{
		$db= new Backup();
		if(request()->isPost()){
			$tables=get_params('tables');           
			$fileinfo  =$db->getFile();
			//检查是否有正在执行的任务
			$lock = "{$fileinfo['filepath']}backup.lock";
			if(is_file($lock)){
			    return to_assign(2, '检测到有一个备份任务未完成');
			} else {
               //创建锁文件
               file_put_contents($lock,time());
            }
            // 检查备份目录是否可写
            if(!is_writeable($fileinfo['filepath'])){
				return to_assign(1, '备份目录不存在或不可写，请检查后重试！');
            }
 
			//缓存锁文件
			Session::set('lock', $lock);
			//缓存备份文件信息
			Session::set('backup_file', $fileinfo['file']);
			//缓存要备份的表
			Session::set('backup_tables', $tables);
			//创建备份文件
			if(false !== $db->Backup_Init()){
				return to_assign(0, '初始化成功，开始备份...',['tab'=>['id' => 0, 'start' => 0,'table'=>$tables[0]]]);
			}else{
				return to_assign(1, '初始化失败，备份文件创建失败');
			}
		}else if(request()->isGet()){
			$tables = Session::get('backup_tables');
			$file=Session::get('backup_file');
			$id=get_params('id'); 
			$start=get_params('start');
			$start= $db->setFile($file)->backup($tables[$id], $start);
			if(false === $start){
				return to_assign(1, '备份出错');
			}else if(0 === $start){
				if(isset($tables[++$id])){
				   return to_assign(0, '备份完成',['tab'=>['id' => $id, 'start' => 0,'table'=>$tables[$id-1]]]);
				} else { //备份完成，清空缓存
                   unlink(Session::get('lock'));
                   Session::delete('backup_tables');
                   Session::delete('backup_file');
				   add_log('bak');
				   return to_assign(0, '备份完成',['tab'=>['start' => 'ok','table'=>$tables[$id-1]]]);
               }
           }
		}else{
		   return to_assign(1, '参数错误！');
		}
	} 

	//优化表
	public function optimize($tables= null)
	{
        $db= new Backup();
		//return to_assign(0, $db->optimize($tables));
        if($db->optimize($tables)){
			add_log('optimize');
			return to_assign(0, '数据表优化完成');
        }else{
			return to_assign(1, '数据表优化出错请重试');
        }
	}
	
	//修复表
	public function repair($tables= null)
	{
		$db= new Backup();
		//return to_assign(0, $db->repair($tables));
		if($db->repair($tables)){
			add_log('repair');
			return to_assign(0, '数据表修复完成');
		}else{
			return to_assign(1, '数据表修复出错请重试');
		}
	}
	
	
	//备份文件列表
	public function backuplist()
	{
		$db= new Backup();
		$list = $db->fileList();
		$fileinfo  =$db->getFile();
		$lock = "{$fileinfo['filepath']}backup.lock";
		$lock_time = 0;	
		if(is_file($lock)){
			$lock_time = file_get_contents($lock);
		}
		$listNew = [];
		$indx = 0;
        foreach ($list as $k => $v) {
            $listNew[$indx]['time'] = $k;
            $listNew[$indx]['timespan'] = $v['time'];
            $listNew[$indx]['data'] = $v;
            $indx++;
        }
        $list = $listNew;
        array_multisort(array_column($list, 'time'), SORT_DESC, $list);
		return view('',['list'=>$list,'lock_time' => $lock_time]);
	}
	
	//数据还原
	public function import($time = 0, $part = null, $start = null)
	{
		$db= new Backup();
		$time =(int)$time;
		if(is_numeric($time) && is_null($part) && is_null($start)){
			$list  = $db->getFile('timeverif',$time);
			if(is_array($list)){
				Session::set('backup_list', $list);
				return to_assign(0, '初始化完成，开始还原...',array('part' => 1, 'start' => 0,'time' => $time));
			}else{
				return to_assign(1, '备份文件可能已经损坏,请检查');
			}
		}else if(is_numeric($part) && is_numeric($start)){ 
            $list=Session::get('backup_list');
			$part =(int)$part;
			$start =(int)$start;
            $start= $db->setFile($list)->import($start,$time,$part);               
            if(false===$start){
				return to_assign(1, '还原数据出错,请重试');
            }elseif(0 === $start){
                if(isset($list[++$part])){
                    $data = array('part' => $part, 'start' => 0,'time' => $time);
					return to_assign(0, "正在还原...卷{$part}，请勿关闭当前页面",$data);
                } else {
                    Session::delete('backup_list');
					return to_assign(0, '还原数据成功');
                }
            }else{
                $data = array('part' => $part, 'start' => $start[0],'time' => $time);
                if($start[1]){
                    $rate = floor(100 * ($start[0] / $start[1]));
					return to_assign(0, "正在还原...卷{$part} ({$rate}%)，请勿关闭当前页面",$data);
                } else {
                    $data['gz'] = 1;
                    return to_assign(0, "正在还原...卷{$part}，请勿关闭当前页面",$data);
                }
                return to_assign(0, "正在还原...卷{$part}，请勿关闭当前页面");                   
            }             
		}else{
		   return to_assign(1, "参数错误");
		}       
	}
	/**
    * 删除备份文件
    */
	public function del($time = 0,$lock=0){
		$db= new Backup();
		if($lock==1){
			$fileinfo  =$db->getFile();
			$lock = "{$fileinfo['filepath']}backup.lock";
			if(is_file($lock)){
				$time = file_get_contents($lock);
				unlink($lock);
			}
		}	
		if($db->delFile((int)$time)){
			add_log('delete');
			return to_assign(0, '删除成功');
		}else{
			return to_assign(0, '删除失败，请检查权限');
		}
	}
    /**
    * 下载备份文件
    */
	public function downfile($time = 0,$part=0){
		$db= new Backup();
		add_log('down');
		$db->downloadFile((int)$time,$part-1);
	}
	
}