mmucklo / DtcQueueBundle

DependencyInjection/Compiler/BeanstalkdCompilerPass.php

<?php

namespace Dtc\QueueBundle\DependencyInjection\Compiler;

use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;

The type Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface was not found. Maybe you did not declare it correctly or list all dependencies?

use Symfony\Component\DependencyInjection\ContainerBuilder;

The type Symfony\Component\DependencyInjection\ContainerBuilder was not found. Maybe you did not declare it correctly or list all dependencies?

use Symfony\Component\DependencyInjection\Definition;

The type Symfony\Component\DependencyInjection\Definition was not found. Maybe you did not declare it correctly or list all dependencies?

use Symfony\Component\DependencyInjection\Reference;

The type Symfony\Component\DependencyInjection\Reference was not found. Maybe you did not declare it correctly or list all dependencies?

class BeanstalkdCompilerPass implements CompilerPassInterface
{
    public function process(ContainerBuilder $container)
    {
        if ($container->hasParameter('dtc_queue.beanstalkd.host') &&
            $container->getParameter('dtc_queue.beanstalkd.host')) {
            $definition = new Definition(
                'Pheanstalk\\Pheanstalk',
                [
                    new Definition(
                        'Pheanstalk\\Connection',
                        [
                            new Definition(
                                'Pheanstalk\\SocketFactory',
                                [$container->getParameter('dtc_queue.beanstalkd.host'), $container->getParameter('dtc_queue.beanstalkd.port')]
                            ),
                        ]
                    ),
                ]
            );
            $container->setDefinition('dtc_queue.beanstalkd', $definition);
            $definition = $container->getDefinition('dtc_queue.manager.job.beanstalkd');
            $definition->addMethodCall('setBeanstalkd', [new Reference('dtc_queue.beanstalkd')]);
            if ($container->hasParameter('dtc_queue.beanstalkd.tube')) {
                $definition->addMethodCall('setTube', [$container->getParameter('dtc_queue.beanstalkd.tube')]);
            }
        }
    }
}