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;
|
||
}
|
||
}
|