src/Wallabag/CoreBundle/Event/Subscriber/TablePrefixSubscriber.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Event\Subscriber;
   4
   5 use Doctrine\Common\EventSubscriber;
   6 use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
   7 use Doctrine\ORM\Mapping\ClassMetadataInfo;
   8
   9 /**
  10  * Puts a prefix to each table.
  11  * This way were used instead of using the built-in strategy from Doctrine, using `naming_strategy`
  12  * Because it conflicts with the DefaultQuoteStrategy (that espace table name, like user for Postgres)
  13  * see #1498 for more detail.
  14  *
  15  * Solution from :
  16  *      - http://stackoverflow.com/a/23860613/569101
  17  *      - http://doctrine-orm.readthedocs.org/en/latest/reference/namingstrategy.html
  18  */
  19 class TablePrefixSubscriber implements EventSubscriber
  20 {
  21     protected $prefix = '';
  22
  23     public function __construct($prefix)
  24     {
  25         $this->prefix = (string) $prefix;
  26     }
  27
  28     public function getSubscribedEvents()
  29     {
  30         return ['loadClassMetadata'];
  31     }
  32
  33     public function loadClassMetadata(LoadClassMetadataEventArgs $args)
  34     {
  35         $classMetadata = $args->getClassMetadata();
  36
  37         // if we are in an inheritance hierarchy, only apply this once
  38         if ($classMetadata->isInheritanceTypeSingleTable() && !$classMetadata->isRootEntity()) {
  39             return;
  40         }
  41
  42         $classMetadata->setTableName($this->prefix.$classMetadata->getTableName());
  43
  44         foreach ($classMetadata->getAssociationMappings() as $fieldName => $mapping) {
  45             if ($mapping['type'] === ClassMetadataInfo::MANY_TO_MANY && isset($classMetadata->associationMappings[$fieldName]['joinTable']['name'])) {
  46                 $mappedTableName = $classMetadata->associationMappings[$fieldName]['joinTable']['name'];
  47                 $classMetadata->associationMappings[$fieldName]['joinTable']['name'] = $this->prefix.$mappedTableName;
  48             }
  49         }
  50     }
  51 }