Wrapper for Object Manager from Request

ObjectManagerRequest.php

<?php

namespace xxx\xxxAPIBundle\Services;

use Symfony\Bridge\Doctrine\RegistryInterface;
use Symfony\Component\HttpFoundation\RequestStack;

/**
 * Class ObjectManagerRequest
 * Wrapper for Object Manager from Request
 *
 * @package xxx\xxxAPIBundle\Services
 */
class ObjectManagerRequest
{
    protected $doctrine;
    protected $requestStack;
    protected $country;

    /**
     * @param RegistryInterface $doctrine
     * @param RequestStack      $requestStack
     */
    public function __construct(RegistryInterface $doctrine, RequestStack $requestStack)
    {
        $this->doctrine = $doctrine;
        $this->requestStack = $requestStack;
    }

    /**
     * Get Object Manager from Request
     *
     * @param null $country
     *
     * @return \Doctrine\Common\Persistence\ObjectManager
     * @throws \InvalidArgumentException
     */
    public function getObjectManager($country = null)
    {
        $this->country = $country;

        if (null === $this->country && null !== $this->requestStack->getMasterRequest()) {
            $this->country = $this->requestStack->getMasterRequest()->get('country');
        }

        return $this->doctrine->getManager($this->country);
    }
}