test_lgq/niucloud/app/service/admin/upgrade/ExecuteSqlTrait.php

69 lines
2.5 KiB
PHP
Raw Normal View History

2024-01-24 17:36:08 +08:00
<?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;
}
}