119 lines
4.4 KiB
PHP
119 lines
4.4 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006-2016 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: yunwuxin <448901948@qq.com>
|
|
// +----------------------------------------------------------------------
|
|
namespace think\console\command\optimize;
|
|
|
|
use think\console\Command;
|
|
use think\console\Input;
|
|
use think\console\input\Option;
|
|
use think\console\Output;
|
|
use think\Db;
|
|
use think\facade\App;
|
|
|
|
class Schema extends Command
|
|
{
|
|
protected function configure()
|
|
{
|
|
$this->setName('optimize:schema')
|
|
->addOption('db', null, Option::VALUE_REQUIRED, 'db name .')
|
|
->addOption('table', null, Option::VALUE_REQUIRED, 'table name .')
|
|
->addOption('module', null, Option::VALUE_REQUIRED, 'module name .')
|
|
->setDescription('Build database schema cache.');
|
|
}
|
|
|
|
protected function execute(Input $input, Output $output)
|
|
{
|
|
if (!is_dir(App::getRuntimePath() . 'schema')) {
|
|
@mkdir(App::getRuntimePath() . 'schema', 0755, true);
|
|
}
|
|
|
|
if ($input->hasOption('module')) {
|
|
$module = $input->getOption('module');
|
|
// 读取模型
|
|
$path = App::getAppPath() . $module . DIRECTORY_SEPARATOR . 'model';
|
|
$list = is_dir($path) ? scandir($path) : [];
|
|
$namespace = App::getNamespace();
|
|
|
|
foreach ($list as $file) {
|
|
if (0 === strpos($file, '.')) {
|
|
continue;
|
|
}
|
|
$class = '\\' . $namespace . '\\' . $module . '\\model\\' . pathinfo($file, PATHINFO_FILENAME);
|
|
$this->buildModelSchema($class);
|
|
}
|
|
|
|
$output->writeln('<info>Succeed!</info>');
|
|
return;
|
|
} elseif ($input->hasOption('table')) {
|
|
$table = $input->getOption('table');
|
|
if (false === strpos($table, '.')) {
|
|
$dbName = Db::getConfig('database');
|
|
}
|
|
|
|
$tables[] = $table;
|
|
} elseif ($input->hasOption('db')) {
|
|
$dbName = $input->getOption('db');
|
|
$tables = Db::getConnection()->getTables($dbName);
|
|
} elseif (!\think\facade\Config::get('app_multi_module')) {
|
|
$namespace = App::getNamespace();
|
|
$path = App::getAppPath() . 'model';
|
|
$list = is_dir($path) ? scandir($path) : [];
|
|
|
|
foreach ($list as $file) {
|
|
if (0 === strpos($file, '.')) {
|
|
continue;
|
|
}
|
|
$class = '\\' . $namespace . '\\model\\' . pathinfo($file, PATHINFO_FILENAME);
|
|
$this->buildModelSchema($class);
|
|
}
|
|
|
|
$output->writeln('<info>Succeed!</info>');
|
|
return;
|
|
} else {
|
|
$tables = Db::getConnection()->getTables();
|
|
}
|
|
|
|
$db = isset($dbName) ? $dbName . '.' : '';
|
|
$this->buildDataBaseSchema($tables, $db);
|
|
|
|
$output->writeln('<info>Succeed!</info>');
|
|
}
|
|
|
|
protected function buildModelSchema($class)
|
|
{
|
|
$reflect = new \ReflectionClass($class);
|
|
if (!$reflect->isAbstract() && $reflect->isSubclassOf('\think\Model')) {
|
|
$table = $class::getTable();
|
|
$dbName = $class::getConfig('database');
|
|
$content = '<?php ' . PHP_EOL . 'return ';
|
|
$info = $class::getConnection()->getFields($table);
|
|
$content .= var_export($info, true) . ';';
|
|
|
|
file_put_contents(App::getRuntimePath() . 'schema' . DIRECTORY_SEPARATOR . $dbName . '.' . $table . '.php', $content);
|
|
}
|
|
}
|
|
|
|
protected function buildDataBaseSchema($tables, $db)
|
|
{
|
|
if ('' == $db) {
|
|
$dbName = Db::getConfig('database') . '.';
|
|
} else {
|
|
$dbName = $db;
|
|
}
|
|
|
|
foreach ($tables as $table) {
|
|
$content = '<?php ' . PHP_EOL . 'return ';
|
|
$info = Db::getConnection()->getFields($db . $table);
|
|
$content .= var_export($info, true) . ';';
|
|
file_put_contents(App::getRuntimePath() . 'schema' . DIRECTORY_SEPARATOR . $dbName . $table . '.php', $content);
|
|
}
|
|
}
|
|
}
|