代码生成器，截取数据表名称方法不恰当

### 问题描述

`$table_name = str_replace($tablePrefix, '', $table_info['Name']);`  表名称中含有前缀，会被一并替换，例如表名：nc_test_nu_test , 其中表前缀是nc_, 会得到表名test_test，不恰当

```
 /**
     * 添加代码生成
     * @param array $data
     * @return GenerateTable|Model
     * @throws Exception
     */
    public function add(array $data)
    {
        Db::startTrans();
        try {
            $sql = 'SHOW TABLE STATUS WHERE 1=1 ';
            $tablePrefix = config('database.connections.mysql.prefix');
            if (!empty($data['table_name'])) {
                $sql .= "AND name='" . $data['table_name']."'";
            }

$tables = Db::query($sql);
            $table_info = $tables[0] ?? [];
            if(empty($table_info)) throw new AdminException('DATA_NOT_EXIST');

$table_name = str_replace($tablePrefix, '', $table_info['Name']);
            $fields = Db::name($table_name)->getFields();

$add_table_data = [
                'table_name' => $table_name,
                'table_content' => $table_info['Comment'],
                'class_name' => $table_name,
                'create_time' => time(),
                'module_name' => $table_name
            ];

$res = (new GenerateTable())->create($add_table_data);

$table_id = $res->id;
            $add_column_data = [];
            $default_column = ['id', 'create_time', 'update_time'];
            foreach ($fields as $k => $v){
                $required = 0;
                if ($v['notnull'] && !$v['primary'] && !in_array($v['name'], $default_column)) {
                    $required = 1;
                }

$add_column_data[] = [
                    'table_id' => $table_id,
                    'column_name' => $v['name'],
                    'column_comment' => $v['comment'],
                    'column_type' => self::getDbFieldType($v['type']),
                    'is_required' => $required,
                    'is_pk' => $v['primary'] ? 1 : 0,
                    'is_insert' => !in_array($v['name'], $default_column) ? 1 : 0,
                    'is_update' => !in_array($v['name'], $default_column) ? 1 : 0,
                    'is_lists' => !in_array($v['name'], $default_column) ? 1 : 0,
                    'is_delete' => 0,
//                    'is_query' => !in_array($v['name'], $default_column) ? 1 : 0,
                    'query_type' => '=',
                    'view_type' => 'input',
                    'dict_type' => $v['dict_type'] ?? '',
                    'addon' => $v['addon'] ?? '',
                    'model' => $v['model'] ?? '',
                    'label_key' => $v['label_key'] ?? '',
                    'value_key' => $v['value_key'] ?? '',
                    'create_time' => time(),
                    'update_time' => time()
                ];
            }

(new GenerateColumn())->saveAll($add_column_data);

Db::commit();
            return $table_id;
        } catch ( Exception $e) {
            Db::rollback();
            throw new AdminException($e->getMessage());
        }

}
``

牛之云科技/niucloud-Admin-SaaS 后台管理框架

内容风险标识

评论 (1)

牛之云科技/niucloud-Admin-SaaS 后台管理框架 .gitee-modal { width: 500px !important; }

内容风险标识