123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270 |
- <?php
- namespace mall;
- use think\facade\Db;
- class Backup{
-
- private $fp;
-
- private $file;
-
- private $size = 0;
-
- private $config;
-
- public function __construct($file, $config, $type = 'export'){
- $this->file = $file;
- $this->config = $config;
- }
-
- 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->create();
- }
- } 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;
- }
- }
-
- public function create(){
- $sql = "-- -----------------------------\n";
- $sql .= "-- Think MySQL Data Transfer \n";
- $sql .= "-- \n";
- $sql .= "-- Host : " . config('database.hostname') . "\n";
- $sql .= "-- Port : " . config('database.hostport') . "\n";
- $sql .= "-- Database : " . config('database.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);
- }
-
- private function write($sql){
- $size = strlen($sql);
-
-
- $size = $this->config['compress'] ? $size / 2 : $size;
- $this->open($size);
- return $this->config['compress'] ? @gzwrite($this->fp, $sql) : @fwrite($this->fp, $sql);
- }
-
- public function backup($table, $start){
-
- if(0 == $start){
- $result = Db::name($table)->query("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[0]['Create Table']) . ";\n\n";
- if(false === $this->write($sql)){
- return false;
- }
- }
-
- $result = Db::name($table)->query("SELECT COUNT(*) AS count FROM `{$table}`");
- $count = $result['0']['count'];
-
- if($count){
-
- if(0 == $start){
- $sql = "-- -----------------------------\n";
- $sql .= "-- Records of `{$table}`\n";
- $sql .= "-- -----------------------------\n";
- $this->write($sql);
- }
-
- $result = Db::name($table)->query("SELECT * FROM `{$table}` LIMIT {$start}, 1000");
- foreach ($result as $row) {
- $row = array_map('addslashes', $row);
- $sql = "INSERT INTO `{$table}` VALUES ('" . str_replace(array("\r","\n"),array('\r','\n'),implode("', '", $row)) . "');\n";
- if(false === $this->write($sql)){
- return false;
- }
- }
-
- if($count > $start + 1000){
- return array($start + 1000, $count);
- }
- }
-
- return 0;
- }
- public function import($start){
-
- if($this->config['compress']){
- $gz = gzopen($this->file[1], 'r');
- $size = 0;
- } else {
- $size = filesize($this->file[1]);
- $gz = fopen($this->file[1], '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::name($start)->execute($sql)){
- $start += strlen($sql);
- } else {
- return false;
- }
- $sql = '';
- } elseif ($this->config['compress'] ? gzeof($gz) : feof($gz)) {
- return 0;
- }
- }
- return array($start, $size);
- }
-
- public function __destruct(){
- $this->config['compress'] ? @gzclose($this->fp) : @fclose($this->fp);
- }
-
- public static function parseSql($filename){
- $lines=file($filename);
- $lines[0]=str_replace(chr(239).chr(187).chr(191),"",$lines[0]);
- $flage = true;
- $sqls = array();
- $sql="";
- foreach($lines as $line)
- {
- $line=trim($line);
- $char=substr($line,0,1);
- if($char!='#' && strlen($line)>0)
- {
- $prefix=substr($line,0,2);
- switch($prefix)
- {
- case '/*':
- {
- $flage=(substr($line,-3)=='*/;'||substr($line,-2)=='*/')?true:false;
- break 1;
- }
- case '--': break 1;
- default :
- {
- if($flage)
- {
- $sql.=$line;
- if(substr($line,-1)==";")
- {
- $sqls[]=$sql;
- $sql="";
- }
- }
- if(!$flage)$flage=(substr($line,-3)=='*/;'||substr($line,-2)=='*/')?true:false;
- }
- }
- }
- }
- return $sqls;
- }
-
- public static function install($sqls){
- $flag = true;
- if(is_array($sqls))
- {
- foreach($sqls as $sql)
- {
- if(Db::name()->execute($sql) === false){ $flag = false;}
- }
- }
- return $flag;
- }
- }
|