vendor/php-task/task-bundle/src/DependencyInjection/TaskExtension.php line 151

  1. <?php
  3. /*
  4.  * This file is part of php-task library.
  5.  *
  6.  * (c) php-task
  7.  *
  8.  * This source file is subject to the MIT license that is bundled
  9.  * with this source code in the file LICENSE.
  10.  */
  12. namespace Task\TaskBundle\DependencyInjection;
  14. use Symfony\Component\Config\Definition\Exception\InvalidConfigurationException;
  15. use Symfony\Component\Config\FileLocator;
  16. use Symfony\Component\Config\Loader\LoaderInterface;
  17. use Symfony\Component\DependencyInjection\ContainerBuilder;
  18. use Symfony\Component\DependencyInjection\ContainerInterface;
  19. use Symfony\Component\DependencyInjection\Definition;
  20. use Symfony\Component\DependencyInjection\Loader;
  21. use Symfony\Component\DependencyInjection\Reference;
  22. use Symfony\Component\HttpKernel\DependencyInjection\Extension;
  23. use Task\Event\Events;
  24. use Task\TaskBundle\EventListener\DoctrineTaskExecutionListener;
  26. /**
  27.  * Container extension for php-task library.
  28.  */
  29. class TaskExtension extends Extension
  30. {
  31.     /**
  32.      * {@inheritdoc}
  33.      */
  34.     public function load(array $configs, ContainerBuilder $container)
  35.     {
  36.         $configuration = $this->getConfiguration($configs, $container);
  37.         $config = $this->processConfiguration($configuration, $configs);
  39.         $container->setParameter('task.system_tasks', $config['system_tasks']);
  41.         $container->setAlias('task.lock.storage', $configuration->getLockingStorageId($config['locking']['storage']));
  42.         foreach (array_keys($config['locking']['storages']) as $key) {
  43.             $container->setParameter('task.lock.storages.' . $key, $config['locking']['storages'][$key]);
  44.         }
  46.         $loader = new Loader\XmlFileLoader($container, new FileLocator(__DIR__ . '/../Resources/config'));
  47.         $loader->load(sprintf('storage/%s.xml', $config['storage']));
  48.         $loader->load('task_event_listener.xml');
  49.         $loader->load('scheduler.xml');
  50.         $loader->load('command.xml');
  51.         $loader->load('locking/services.xml');
  53.         if ($config['run']['mode'] === 'listener') {
  54.             $loader->load('listener.xml');
  55.         }
  57.         $this->loadDoctrineAdapter($config['adapters']['doctrine'], $container);
  58.         $this->loadLockingComponent($config['locking'], $container, $loader);
  59.         $this->loadExecutorComponent($config['executor'], $container, $loader);
  60.     }
  62.     /**
  63.      * Load doctrine adapter.
  64.      *
  65.      * @param array $config
  66.      * @param ContainerBuilder $container
  67.      */
  68.     private function loadDoctrineAdapter(array $config, ContainerBuilder $container)
  69.     {
  70.         if ($config['clear']) {
  71.             $definition = new Definition(
  72.                 DoctrineTaskExecutionListener::class,
  73.                 [new Reference('doctrine.orm.entity_manager', ContainerInterface::IGNORE_ON_INVALID_REFERENCE)]
  74.             );
  75.             $definition->addTag(
  76.                 'kernel.event_listener',
  77.                 ['event' => Events::TASK_AFTER, 'method' => 'clearEntityManagerAfterTask']
  78.             );
  79.             $container->setDefinition('task.adapter.doctrine.execution_listener', $definition);
  80.         }
  81.     }
  83.     /**
  84.      * Load services for locking component.
  85.      *
  86.      * @param array $config
  87.      * @param LoaderInterface $loader
  88.      * @param ContainerBuilder $container
  89.      */
  90.     private function loadLockingComponent(array $config, ContainerBuilder $container, LoaderInterface $loader)
  91.     {
  92.         if (!$config['enabled'] || 'null' === $config['storage']) {
  93.             return $loader->load('locking/null.xml');
  94.         }
  96.         $loader->load('locking/services.xml');
  97.         $container->setParameter('task.lock.ttl', $config['ttl']);
  98.     }
  100.     /**
  101.      * Load services for executor component.
  102.      *
  103.      * @param array $config
  104.      * @param LoaderInterface $loader
  105.      * @param ContainerBuilder $container
  106.      */
  107.     private function loadExecutorComponent(array $config, ContainerBuilder $container, LoaderInterface $loader)
  108.     {
  109.         $loader->load('executor/' . $config['type'] . '.xml');
  110.         $container->setAlias('task.executor', 'task.executor.' . $config['type']);
  112.         if (!array_key_exists($config['type'], $config)) {
  113.             return;
  114.         }
  116.         foreach ($config[$config['type']] as $key => $value) {
  117.             $container->setParameter('task.executor.' . $key, $value);
  118.         }
  120.         if (!file_exists($container->getParameter('task.executor.console_path'))) {
  121.             throw new InvalidConfigurationException(
  122.                 'Console file does not exists! Given in "task.executor.seperate.console".'
  123.             );
  124.         }
  125.     }
  127.     /**
  128.      * Find storage aliases and related ids.
  129.      *
  130.      * @param ContainerBuilder $container
  131.      *
  132.      * @return array
  133.      */
  134.     private function getLockingStorageAliases(ContainerBuilder $container)
  135.     {
  136.         $taggedServices = $container->findTaggedServiceIds('task.lock.storage');
  138.         $result = ['null'];
  139.         foreach ($taggedServices as $id => $tags) {
  140.             foreach ($tags as $tag) {
  141.                 $result[$tag['alias']] = $id;
  142.             }
  143.         }
  145.         return $result;
  146.     }
  148.     /**
  149.      * {@inheritdoc}
  150.      */
  151.     public function getConfiguration(array $config, ContainerBuilder $container)
  152.     {
  153.         $loader = new Loader\XmlFileLoader($container, new FileLocator(__DIR__ . '/../Resources/config'));
  154.         $loader->load('locking/storages.xml');
  156.         return new Configuration($this->getLockingStorageAliases($container));
  157.     }
  158. }