pomm-project /
pomm-bundle
| @@ 28-53 (lines=26) @@ | ||
| 25 | * @license X11 {@link http://opensource.org/licenses/mit-license.php} |
|
| 26 | * @see ClientPooler |
|
| 27 | */ |
|
| 28 | class ContainerModelLayerPooler extends ModelLayerPooler implements ContainerAwareInterface, ServiceMapInterface |
|
| 29 | { |
|
| 30 | use ContainerAwareTrait; |
|
| 31 | ||
| 32 | private $serviceMap = []; |
|
| 33 | ||
| 34 | /** |
|
| 35 | * {@inheritdoc} |
|
| 36 | */ |
|
| 37 | public function addModelToServiceMapping($class, $serviceId) |
|
| 38 | { |
|
| 39 | $this->serviceMap[$class] = $serviceId; |
|
| 40 | } |
|
| 41 | ||
| 42 | /** |
|
| 43 | * {@inheritdoc} |
|
| 44 | */ |
|
| 45 | protected function createClient($class) |
|
| 46 | { |
|
| 47 | if (array_key_exists($class, $this->serviceMap)) { |
|
| 48 | return $this->container->get($this->serviceMap[$class]); |
|
| 49 | } |
|
| 50 | ||
| 51 | return parent::createClient($class); |
|
| 52 | } |
|
| 53 | } |
|
| 54 | ||
| @@ 27-52 (lines=26) @@ | ||
| 24 | * @license X11 {@link http://opensource.org/licenses/mit-license.php} |
|
| 25 | * @see Extension |
|
| 26 | */ |
|
| 27 | class ContainerModelPooler extends ModelPooler implements ContainerAwareInterface, ServiceMapInterface |
|
| 28 | { |
|
| 29 | use ContainerAwareTrait; |
|
| 30 | ||
| 31 | private $serviceMap = []; |
|
| 32 | ||
| 33 | /** |
|
| 34 | * {@inheritdoc} |
|
| 35 | */ |
|
| 36 | public function addModelToServiceMapping($class, $serviceId) |
|
| 37 | { |
|
| 38 | $this->serviceMap[$class] = $serviceId; |
|
| 39 | } |
|
| 40 | ||
| 41 | /** |
|
| 42 | * {@inheritdoc} |
|
| 43 | */ |
|
| 44 | protected function createClient($class) |
|
| 45 | { |
|
| 46 | if (array_key_exists($class, $this->serviceMap)) { |
|
| 47 | return $this->container->get($this->serviceMap[$class]); |
|
| 48 | } |
|
| 49 | ||
| 50 | return parent::createClient($class); |
|
| 51 | } |
|
| 52 | } |
|
| 53 | ||
