Extractor: 一个MySQL数据库备份工具

因为要在Ant中导出MySQL数据库，写了mysql_extractor类，供大家参考。

 

extractor.php： 

if (!empty($_SERVER['argv'])) {

    for ($i = 1; $i < $_SERVER['argc']; $i++) {

        list($k, $v) = explode('=', $_SERVER['argv'][$i]);

        $_GET[$k] = $v;

    }

}



if (empty($_GET['table_prefix'])) {

    $_GET['table_prefix'] = '';

}



if (empty($_GET['output_file'])) {

    $_GET['output_file'] = '';

}



include_once(dirname(__FILE__) . '/../config.php');



$db = mysql_connect(DB_HOST . ':' . DB_PORT, DB_USER, DB_PASS);

if (!$db) {

    die('[' . mysql_errno() . '] ' . mysql_error());

}



if (!mysql_select_db(DB_NAME, $db)) {

    die('[' . mysql_errno() . '] ' . mysql_error());

}

        

mysql_query("SET NAMES 'utf8'", $db);

mysql_query("SET NAMES 'utf8'", $db);



$extractor = new mysql_extractor($db, $_GET['output_file']);



$extractor->write_start(DB_PREFIX);



$result = mysql_query("SHOW TABLE STATUS WHERE name like '" . DB_PREFIX . $_GET['table_prefix'] . "%'", $db);

while (false !== ($row = mysql_fetch_array($result, MYSQL_NUM))) {

    $extractor->write_table($row[0]);

    $extractor->write_data($row[0]);

}

mysql_free_result($result);



$extractor->write_end();



class mysql_extractor

{

    var $db;

    var $fp;



    public function __construct($db, $output_file)

    {

        $this->db = $db;

        

        if (!empty($output_file)) {

            $this->fp = fopen($output_file, 'w');



            if (!$this->fp) {

                trigger_error('FILE_WRITE_FAIL', E_USER_ERROR);

            }

        }

    }



    public function write_start($table_prefix)

    {

        $sql_data = "#\n";

        $sql_data .= "# Omnitrix Backup Script\n";

        $sql_data .= "# Dump of tables for $table_prefix\n";

        $sql_data .= "# DATE : " . gmdate("d-m-Y H:i:s", time()) . " GMT\n";

        $sql_data .= "#\n\n";

        $this->flush($sql_data);

    }



    public function write_table($table_name)

    {

        $sql = 'SHOW CREATE TABLE ' . $table_name;

        $result = mysql_query($sql, $this->db);

        $row = mysql_fetch_array($result, MYSQL_ASSOC);

        mysql_free_result($result);



        $sql_data = "#\n";

        $sql_data .= "# Source for table $table_name \n";

        $sql_data .= "#\n";

        $sql_data .= "DROP TABLE IF EXISTS `$table_name`;\n";

        $this->flush($sql_data . $row['Create Table'] . ";\n\n");

    }



    public function write_data($table_name)

    {

        $sql_data = "#\n";

        $sql_data .= "# Dumping data for table $table_name \n";

        $sql_data .= "#\n";

        $this->flush($sql_data);

        

        $sql = "SELECT * FROM $table_name";

        $result = mysql_query($sql, $this->db);

        if ($result != false) {

            $fields_cnt = mysql_num_fields($result);



            // Get field information

            $field = $field_set = array();

            for ($j = 0; $j < $fields_cnt; $j++) {

                $info = new stdClass();

                $info->name = mysql_field_name($result, $j);

                $info->type = mysql_field_type($result, $j);

                $info->flags = mysql_field_flags($result, $j);

                $field[$j] = $info;

                

                $field_set[] = $info->name;

            }



            $search            = array("\\", "'", "\x00", "\x0a", "\x0d", "\x1a", '"');

            $replace        = array("\\\\", "\\'", '\0', '\n', '\r', '\Z', '\\"');



            while (($row = mysql_fetch_row($result)) !== false) {

                $values    = array();



                for ($j = 0; $j < $fields_cnt; $j++) {

                    if (!isset($row[$j]) || is_null($row[$j])) {

                        $values[$j] = 'NULL';

                    } else if (($field[$j]->flags & 32768) && !($field[$j]->flags & 1024))

                    {

                        $values[$j] = $row[$j];

                    } else {

                        $values[$j] = "'" . str_replace($search, $replace, $row[$j]) . "'";

                    }

                }

                

                $query = 'INSERT INTO `' . $table_name . '` VALUES (' . implode(', ', $values) . ');';

                $this->flush($query . "\n");

            }

            mysql_free_result($result);

            

            $this->flush("\n");

        }

    }

    

    public function write_end()

    {

        if ($this->fp) {

            fclose($this->fp);

        }    

    }

    

    public function flush($data)

    {

        if ($this->fp) {

            fwrite($this->fp, $data);

        } else {

            echo $data;            

        }

    }

}

 

在命令行执行：

php -f extractor.php #直接输出

php -f extractor.php output_file=omnitrix.sql #输出到文件

php -f extractor.php output_file=omnitrix.sql table_prefix=system_user #限定指定开头的表

 

在浏览器执行：

http://localhost/omnitrix/utilities/extractor.php #直接输出

http://localhost/omnitrix/utilities/extractor.php?output_file=omnitrix.sql #输出到文件

http://localhost/omnitrix/utilities/extractor.php?output_file=omnitrix.sql&table_prefix=system_user #限定指定开头的表