mirror of
https://gitee.com/ulthon/ulthon_admin.git
synced 2026-07-02 16:02:48 +08:00
108 lines
3.3 KiB
PHP
108 lines
3.3 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | TopThink [ WE CAN DO IT JUST THINK IT ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2016 http://www.topthink.com All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Author: zhangyajun <448901948@qq.com>
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace think\migration\command\seed;
|
|
|
|
use Phinx\Seed\SeedInterface;
|
|
use think\console\Input;
|
|
use think\console\input\Option as InputOption;
|
|
use think\console\Output;
|
|
use think\migration\command\Seed;
|
|
|
|
class Run extends Seed
|
|
{
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function configure()
|
|
{
|
|
$this->setName('seed:run')
|
|
->setDescription('Run database seeders')
|
|
->addOption('--seed', '-s', InputOption::VALUE_REQUIRED, 'What is the name of the seeder?')
|
|
->setHelp(<<<EOT
|
|
The <info>seed:run</info> command runs all available or individual seeders
|
|
|
|
<info>php think seed:run</info>
|
|
<info>php think seed:run -s UserSeeder</info>
|
|
<info>php think seed:run -v</info>
|
|
|
|
EOT
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Run database seeders.
|
|
*
|
|
* @param Input $input
|
|
* @param Output $output
|
|
* @return void
|
|
*/
|
|
protected function execute(Input $input, Output $output)
|
|
{
|
|
$seed = $input->getOption('seed');
|
|
|
|
// run the seed(ers)
|
|
$start = microtime(true);
|
|
$this->seed($seed);
|
|
$end = microtime(true);
|
|
|
|
$output->writeln('');
|
|
$output->writeln('<comment>All Done. Took ' . sprintf('%.4fs', $end - $start) . '</comment>');
|
|
}
|
|
|
|
public function seed($seed = null)
|
|
{
|
|
$seeds = $this->getSeeds();
|
|
|
|
if (null === $seed) {
|
|
// run all seeders
|
|
foreach ($seeds as $seeder) {
|
|
if (array_key_exists($seeder->getName(), $seeds)) {
|
|
$this->executeSeed($seeder);
|
|
}
|
|
}
|
|
} else {
|
|
// run only one seeder
|
|
if (array_key_exists($seed, $seeds)) {
|
|
$this->executeSeed($seeds[$seed]);
|
|
} else {
|
|
throw new \InvalidArgumentException(sprintf('The seed class "%s" does not exist', $seed));
|
|
}
|
|
}
|
|
}
|
|
|
|
protected function executeSeed(SeedInterface $seed)
|
|
{
|
|
$this->output->writeln('');
|
|
$this->output->writeln(' ==' . ' <info>' . $seed->getName() . ':</info>' . ' <comment>seeding</comment>');
|
|
|
|
// Execute the seeder and log the time elapsed.
|
|
$start = microtime(true);
|
|
$seed->setAdapter($this->getAdapter());
|
|
|
|
// begin the transaction if the adapter supports it
|
|
if ($this->getAdapter()->hasTransactions()) {
|
|
$this->getAdapter()->beginTransaction();
|
|
}
|
|
|
|
// Run the seeder
|
|
if (method_exists($seed, SeedInterface::RUN)) {
|
|
$seed->run();
|
|
}
|
|
|
|
// commit the transaction if the adapter supports it
|
|
if ($this->getAdapter()->hasTransactions()) {
|
|
$this->getAdapter()->commitTransaction();
|
|
}
|
|
$end = microtime(true);
|
|
|
|
$this->output->writeln(' ==' . ' <info>' . $seed->getName() . ':</info>' . ' <comment>seeded' . ' ' . sprintf('%.4fs', $end - $start) . '</comment>');
|
|
}
|
|
}
|