vendor/codeages/biz-framework/src/Provider/DoctrineServiceProvider.php line 120

Open in your IDE?
  1. <?php
  3. /*
  4.  * 此文件来自 Silex 项目(https://github.com/silexphp/Silex).
  5.  *
  6.  * 版权信息请看 LICENSE.SILEX
  7.  */
  9. namespace Codeages\Biz\Framework\Provider;
  11. use Codeages\Biz\Framework\Util\Lock;
  12. use Pimple\Container;
  13. use Pimple\ServiceProviderInterface;
  14. use Doctrine\DBAL\DriverManager;
  15. use Doctrine\DBAL\Configuration;
  16. use Doctrine\Common\EventManager;
  17. use Symfony\Bridge\Doctrine\Logger\DbalLogger;
  19. /**
  20.  * Doctrine DBAL Provider.
  21.  *
  22.  * @author Fabien Potencier <fabien@symfony.com>
  23.  */
  24. class DoctrineServiceProvider implements ServiceProviderInterface
  25. {
  26.     /**
  27.      * @SuppressWarnings(PHPMD.NPathComplexity)
  28.      * @SuppressWarnings(PHPMD.CyclomaticComplexity)
  29.      */
  30.     public function register(Container $app)
  31.     {
  32.         $app['db.default_options'] = array(
  33.             'driver' => 'pdo_mysql',
  34.             'dbname' => null,
  35.             'host' => 'localhost',
  36.             'user' => 'root',
  37.             'password' => null,
  38.             'wrapperClass' => 'Codeages\Biz\Framework\Dao\Connection',
  39.         );
  41.         $app['dbs.options.initializer'] = $app->protect(function () use ($app) {
  42.             static $initialized = false;
  44.             if ($initialized) {
  45.                 return;
  46.             }
  48.             $initialized = true;
  50.             if (!isset($app['dbs.options'])) {
  51.                 $app['dbs.options'] = array('default' => isset($app['db.options']) ? $app['db.options'] : array());
  52.             }
  54.             $tmp = $app['dbs.options'];
  55.             foreach ($tmp as $name => &$options) {
  56.                 $options = array_replace($app['db.default_options'], $options);
  58.                 if (!isset($app['dbs.default'])) {
  59.                     $app['dbs.default'] = $name;
  60.                 }
  61.             }
  62.             $app['dbs.options'] = $tmp;
  63.         });
  65.         $app['dbs'] = function ($app) {
  66.             $app['dbs.options.initializer']();
  68.             $dbs = new Container();
  69.             foreach ($app['dbs.options'] as $name => $options) {
  70.                 if ($app['dbs.default'] === $name) {
  71.                     // we use shortcuts here in case the default has been overridden
  72.                     $config = $app['db.config'];
  73.                     $manager = $app['db.event_manager'];
  74.                 } else {
  75.                     $config = $app['dbs.config'][$name];
  76.                     $manager = $app['dbs.event_manager'][$name];
  77.                 }
  79.                 $dbs[$name] = function () use ($options, $config, $manager) {
  80.                     return DriverManager::getConnection($options, $config, $manager);
  81.                 };
  82.             }
  84.             return $dbs;
  85.         };
  87.         $app['dbs.config'] = function ($app) {
  88.             $app['dbs.options.initializer']();
  90.             $configs = new Container();
  91.             $addLogger = isset($app['logger']) && null !== $app['logger'] && class_exists('Symfony\Bridge\Doctrine\Logger\DbalLogger');
  92.             foreach ($app['dbs.options'] as $name => $options) {
  93.                 $configs[$name] = new Configuration();
  94.                 if ($addLogger) {
  95.                     $configs[$name]->setSQLLogger(new DbalLogger($app['logger'], isset($app['stopwatch']) ? $app['stopwatch'] : null));
  96.                 }
  97.             }
  99.             return $configs;
  100.         };
  102.         $app['dbs.event_manager'] = function ($app) {
  103.             $app['dbs.options.initializer']();
  105.             $managers = new Container();
  106.             foreach ($app['dbs.options'] as $name => $options) {
  107.                 $managers[$name] = new EventManager();
  108.             }
  110.             return $managers;
  111.         };
  113.         $this->registerShortcutForFirstDb($app);
  114.         $this->registerLock($app);
  115.     }
  117.     private function registerShortcutForFirstDb($app)
  118.     {
  119.         $app['db'] = function ($app) {
  120.             $dbs = $app['dbs'];
  122.             return $dbs[$app['dbs.default']];
  123.         };
  125.         $app['db.config'] = function ($app) {
  126.             $dbs = $app['dbs.config'];
  128.             return $dbs[$app['dbs.default']];
  129.         };
  131.         $app['db.event_manager'] = function ($app) {
  132.             $dbs = $app['dbs.event_manager'];
  134.             return $dbs[$app['dbs.default']];
  135.         };
  136.     }
  138.     private function registerLock($app)
  139.     {
  140.         $app['lock'] = function ($app) {
  141.             return new Lock($app);
  142.         };
  143.     }
  144. }