mirror of
https://gitee.com/zoujingli/ThinkAdmin.git
synced 2026-06-07 12:38:11 +08:00
85 lines
2.8 KiB
PHP
85 lines
2.8 KiB
PHP
<?php
|
||
|
||
// +----------------------------------------------------------------------
|
||
// | Developer Tools for ThinkAdmin
|
||
// +----------------------------------------------------------------------
|
||
// | 版权所有 2014~2025 Anyon <zoujingli@qq.com>
|
||
// +----------------------------------------------------------------------
|
||
// | 官方网站: https://thinkadmin.top
|
||
// +----------------------------------------------------------------------
|
||
// | 开源协议 ( https://mit-license.org )
|
||
// | 免责声明 ( https://thinkadmin.top/disclaimer )
|
||
// +----------------------------------------------------------------------
|
||
// | gitee 代码仓库：https://gitee.com/zoujingli/think-plugs-helper
|
||
// | github 代码仓库：https://github.com/zoujingli/think-plugs-helper
|
||
// +----------------------------------------------------------------------
|
||
|
||
declare (strict_types=1);
|
||
|
||
namespace plugin\helper;
|
||
|
||
use think\admin\service\SystemService;
|
||
use think\console\Command;
|
||
use think\facade\Db;
|
||
|
||
class DbIndexStruct extends Command
|
||
{
|
||
|
||
/**
|
||
* 配置命令参数。
|
||
*/
|
||
public function configure(): void
|
||
{
|
||
$this->setName("xadmin:helper:index");
|
||
$this->setDescription("刷新数据库的结构索引");
|
||
}
|
||
|
||
/**
|
||
* 检查是否允许执行任务
|
||
* @return bool 返回true表示允许执行
|
||
*/
|
||
public function isEnabled(): bool
|
||
{
|
||
return SystemService::isDebug();
|
||
}
|
||
|
||
/**
|
||
* 命令执行入口
|
||
* 遍历数据库表并重命名索引.
|
||
*/
|
||
public function handle(): void
|
||
{
|
||
[$tables, $total, $count] = SystemService::getTables();
|
||
foreach ($tables as $table) {
|
||
$this->output->writeln(sprintf("[%s/%s] 开始处理表 {$table}", $count++, $total));
|
||
foreach (Db::query(sprintf('SHOW INDEX FROM `%s`', $table)) as $index) {
|
||
$keyName = $index['Key_name'] ?? '';
|
||
if ($keyName === 'PRIMARY') {
|
||
continue;
|
||
}
|
||
$newName = $this->genIndexName($table, (array)$index);
|
||
if ($keyName === $newName) {
|
||
continue;
|
||
}
|
||
Db::execute(sprintf('ALTER TABLE `%s` RENAME INDEX `%s` TO `%s`', $table, $keyName, $newName));
|
||
++$count;
|
||
}
|
||
}
|
||
$this->output->writeln("✅ 完成 {$count} 个索引重命名");
|
||
}
|
||
|
||
/**
|
||
* 生成索引名称.
|
||
* @param string $table 表名
|
||
* @param array $index 索引信息
|
||
* @return string 生成的索引名称
|
||
*/
|
||
private function genIndexName(string $table, array $index): string
|
||
{
|
||
$abbr = implode('', array_map(function ($word) {
|
||
return $word[0];
|
||
}, explode('_', $table)));
|
||
return ($index['Non_unique'] ? 'idx_' : 'uni_') . $abbr . '_' . substr(md5($table), -4) . '_' . $index['Column_name'];
|
||
}
|
||
}
|