69 lines
2.5 KiB
PHP
69 lines
2.5 KiB
PHP
|
<?php
|
|||
|
|
// +----------------------------------------------------------------------
|
|||
|
|
// | Niucloud-admin 企业快速开发的多应用管理平台
|
|||
|
|
// +----------------------------------------------------------------------
|
|||
|
|
// | 官方网址：https://www.niucloud.com
|
|||
|
|
// +----------------------------------------------------------------------
|
|||
|
|
// | niucloud团队 版权所有 开源版本可自由商用
|
|||
|
|
// +----------------------------------------------------------------------
|
|||
|
|
// | Author: Niucloud Team
|
|||
|
|
// +----------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
namespace app\service\admin\upgrade;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 执行sql文件
|
|||
|
|
*/
|
|||
|
|
trait ExecuteSqlTrait
|
|||
|
|
{
|
|||
|
|
public function getSqlQuery($sql_data)
|
|||
|
|
{
|
|||
|
|
$sql_data = preg_replace("/TYPE=(InnoDB|MyISAM|MEMORY)( DEFAULT CHARSET=[^; ]+)?/", "ENGINE=\\1 DEFAULT CHARSET=utf8", $sql_data);
|
|||
|
|
|
|||
|
|
$sql_data = str_replace("\r", "\n", $sql_data);
|
|||
|
|
$sql_query = [];
|
|||
|
|
$num = 0;
|
|||
|
|
$sql_arr = explode(";\n", trim($sql_data));
|
|||
|
|
unset($sql);
|
|||
|
|
foreach ($sql_arr as $sql) {
|
|||
|
|
$sql_query[ $num ] = '';
|
|||
|
|
$sqls = explode("\n", trim($sql));
|
|||
|
|
$sqls = array_filter($sqls);
|
|||
|
|
foreach ($sqls as $query) {
|
|||
|
|
$str1 = $query[0] ?? '';
|
|||
|
|
if ($str1 != '#' && $str1 != '-')
|
|||
|
|
$sql_query[ $num ] .= $query;
|
|||
|
|
}
|
|||
|
|
$num++;
|
|||
|
|
}
|
|||
|
|
return $sql_query;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理sql增加表前缀
|
|||
|
|
* @param $sql
|
|||
|
|
* @return void
|
|||
|
|
*/
|
|||
|
|
public function handleSqlPrefix($sql, $prefix) {
|
|||
|
|
if (str_contains($sql, 'CREATE TABLE')) {
|
|||
|
|
$match_item = preg_match('/CREATE TABLE [`]?(\\w+)[`]?/is', $sql, $match_data);
|
|||
|
|
} elseif (str_contains($sql, 'ALTER TABLE')) {
|
|||
|
|
$match_item = preg_match('/ALTER TABLE [`]?(\\w+)[`]?/is', $sql, $match_data);
|
|||
|
|
} elseif (str_contains($sql, 'INSERT INTO')) {
|
|||
|
|
$match_item = preg_match('/INSERT INTO [`]?(\\w+)[`]?/is', $sql, $match_data);
|
|||
|
|
} elseif (str_contains($sql, 'DELETE FROM')) {
|
|||
|
|
$match_item = preg_match('/DELETE FROM [`]?(\\w+)[`]?/is', $sql, $match_data);
|
|||
|
|
} elseif (str_contains($sql, 'UPDATE')) {
|
|||
|
|
$match_item = preg_match('/UPDATE [`]?(\\w+)[`]?/is', $sql, $match_data);
|
|||
|
|
} else {
|
|||
|
|
$match_item = 0;
|
|||
|
|
}
|
|||
|
|
if ($match_item > 0) {
|
|||
|
|
$table_name = $match_data[ 1 ];
|
|||
|
|
$new_table_name = $prefix . $table_name;
|
|||
|
|
$sql = implode($new_table_name, explode($table_name, $sql, 2));
|
|||
|
|
}
|
|||
|
|
return $sql;
|
|||
|
|
}
|
|||
|
|
}
|