|
@@ -12,15 +12,22 @@
|
|
|
namespace Symfony\Bundle\FrameworkBundle\Command;
|
|
|
|
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
|
+use Symfony\Component\Console\Input\InputOption;
|
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
|
+use Symfony\Component\DependencyInjection\ContainerInterface;
|
|
|
+use Symfony\Component\Finder\Finder;
|
|
|
|
|
|
/**
|
|
|
* Warmup the cache.
|
|
|
*
|
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
|
|
+ * @author Francis Besset <francis.besset@gmail.com>
|
|
|
*/
|
|
|
class CacheWarmupCommand extends Command
|
|
|
{
|
|
|
+ protected $cacheDir;
|
|
|
+ protected $kernelTmp;
|
|
|
+
|
|
|
/**
|
|
|
* @see Command
|
|
|
*/
|
|
@@ -29,15 +36,27 @@ class CacheWarmupCommand extends Command
|
|
|
$this
|
|
|
->setName('cache:warmup')
|
|
|
->setDescription('Warms up an empty cache')
|
|
|
+ ->setDefinition(array(
|
|
|
+ new InputOption('warmup-dir', '', InputOption::VALUE_OPTIONAL, 'Warms up the cache in a specific directory')
|
|
|
+ ))
|
|
|
->setHelp(<<<EOF
|
|
|
-The <info>cache:warmup</info> command warms up the cache.
|
|
|
+The <info>cache:warmup --warmup-dir=new_cache</info> command warms up the cache.
|
|
|
|
|
|
-Before running this command, the cache must be empty.
|
|
|
+Before running this command, the cache must be empty if not use warmup-dir option.
|
|
|
EOF
|
|
|
)
|
|
|
;
|
|
|
}
|
|
|
|
|
|
+ protected function initialize(InputInterface $input, OutputInterface $output)
|
|
|
+ {
|
|
|
+ parent::initialize($input, $output);
|
|
|
+
|
|
|
+ if ($input->hasOption('warmup-dir')) {
|
|
|
+ $this->cacheDir = $input->getOption('warmup-dir');
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* {@inheritdoc}
|
|
|
*/
|
|
@@ -45,8 +64,63 @@ EOF
|
|
|
{
|
|
|
$output->writeln('Warming up the cache');
|
|
|
|
|
|
- $warmer = $this->container->get('cache_warmer');
|
|
|
+ if (!$this->cacheDir) {
|
|
|
+ $this->warmUp($this->container);
|
|
|
+ } else {
|
|
|
+ $this->kernelTmp = new \AppKernel(
|
|
|
+ $this->container->getParameter('kernel.environment'),
|
|
|
+ $this->container->getParameter('kernel.debug'),
|
|
|
+ $this->cacheDir
|
|
|
+ );
|
|
|
+
|
|
|
+ $this->clearDir($this->kernelTmp->getCacheDir());
|
|
|
+
|
|
|
+ $this->kernelTmp->boot();
|
|
|
+ unlink($this->kernelTmp->getCacheDir().DIRECTORY_SEPARATOR.$this->kernelTmp->getContainerClass().'.php');
|
|
|
+
|
|
|
+ $this->warmUp($this->kernelTmp->getContainer());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ protected function warmUp(ContainerInterface $container)
|
|
|
+ {
|
|
|
+ $warmer = $container->get('cache_warmer');
|
|
|
$warmer->enableOptionalWarmers();
|
|
|
- $warmer->warmUp($this->container->getParameter('kernel.cache_dir'));
|
|
|
+ $warmer->warmUp($container->getParameter('kernel.cache_dir'));
|
|
|
+ }
|
|
|
+
|
|
|
+ protected function clearDir($dir)
|
|
|
+ {
|
|
|
+ if (is_dir($dir)) {
|
|
|
+ $finder = new Finder();
|
|
|
+ $files = $finder
|
|
|
+ ->in($dir)
|
|
|
+ ->getIterator()
|
|
|
+ ;
|
|
|
+
|
|
|
+ $array = iterator_to_array($files);
|
|
|
+
|
|
|
+ foreach (array_reverse($array) as $file) {
|
|
|
+ if ($file->isFile()) {
|
|
|
+ if (!is_writable($file->getPathname())) {
|
|
|
+ throw new \RuntimeException(sprintf('Unable to delete %s file', $file->getPathname()));
|
|
|
+ }
|
|
|
+
|
|
|
+ unlink($file->getPathname());
|
|
|
+ } else {
|
|
|
+ if (!is_writable($file->getPathname())) {
|
|
|
+ throw new \RuntimeException(sprintf('Unable to delete %s directory', $file->getPathname()));
|
|
|
+ }
|
|
|
+
|
|
|
+ rmdir($file->getPathname());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!is_writable($dir)) {
|
|
|
+ throw new \RuntimeException(sprintf('Unable to delete %s directory', $dir));
|
|
|
+ }
|
|
|
+
|
|
|
+ rmdir($dir);
|
|
|
+ }
|
|
|
}
|
|
|
}
|