Fix multiple shop id functionality.

Resources/config/extractor.yml

@@ -9,6 +9,7 @@ services:
             - [ setRelationsCollection, [ @ongr_connections.sync.relations_collection ] ]
             - [ setConnection, [ @database_connection ] ]
             - [ setStorageFacility, [ @ongr_connections.sync.sync_storage ] ]
+            - [ setContainer, [ @service_container ]]
 
     ongr_connections.sync.extractor.passthrough_extractor:
         class: %ongr_connections.extractor.passthrough_extractor.class%

Sync/Extractor/AbstractExtractor.php

@@ -0,0 +1,58 @@
+<?php
+
+/*
+ * This file is part of the ONGR package.
+ *
+ * (c) NFQ Technologies UAB <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ONGR\ConnectionsBundle\Sync\Extractor;
+
+use Symfony\Component\DependencyInjection\ContainerInterface;
+use Symfony\Component\DependencyInjection\Exception\InvalidArgumentException;
+
+/**
+ * Common actions required for all extractors.
+ */
+class AbstractExtractor
+{
+    /**
+     * @var ContainerInterface
+     */
+    private $container;
+
+    /**
+     * @param ContainerInterface $container
+     */
+    public function setContainer(ContainerInterface $container)
+    {
+        $this->container = $container;
+    }
+
+    /**
+     * Gets ids of shops from the configuration.
+     *
+     * @return array
+     */
+    protected function getShopIds()
+    {
+        try {
+            $shops = $this->getContainer()->getParameter('shop_ids');
+        } catch (InvalidArgumentException $e) {
+            $shops = [];
+        }
+
+        return $shops;
+    }
+
+    /**
+     * @return ContainerInterface
+     */
+    private function getContainer()
+    {
+        return $this->container;
+    }
+}

Sync/Extractor/DoctrineExtractor.php

@@ -27,7 +27,7 @@
 /**
  * Extractor that joins entities for insertion to SyncStorage.
  */
-class DoctrineExtractor implements ExtractorInterface
+class DoctrineExtractor extends AbstractExtractor implements ExtractorInterface
 {
     /**
      * @var SyncStorageInterface
@@ -69,7 +69,13 @@ public function extract(BaseDiffItem $item)
                 $itemId = $itemRow[$idFieldName];
 
                 $storage = $this->getStorageFacility();
-                $storage->save($action, $insertList[JobTableFields::TYPE]['value'], $itemId, $item->getTimestamp());
+                $storage->save(
+                    $action,
+                    $insertList[JobTableFields::TYPE]['value'],
+                    $itemId,
+                    $item->getTimestamp(),
+                    $this->getShopIds()
+                );
 
                 $statements = $relation->getStatements();
                 foreach ($statements as $statement) {
@@ -201,7 +207,8 @@ protected function saveResult(BaseDiffItem $item, Statement $results, $action =
                 $action,
                 $row[JobTableFields::TYPE],
                 $row[JobTableFields::ID],
-                $item->getTimestamp()
+                $item->getTimestamp(),
+                $this->getShopIds()
             );
         }
     }

Sync/StorageManager/MysqlStorageManager.php

@@ -30,12 +30,13 @@ public function createStorage($shopId = null, $connection = null)
     {
         $connection = $connection ? : $this->getConnection();
         $schemaManager = $connection->getSchemaManager();
+        $tableName = $this->getTableName($shopId);
 
-        if ($schemaManager->tablesExist([$this->getTableName($shopId)])) {
+        if ($schemaManager->tablesExist([$tableName])) {
             return true;
         }
 
-        $table = new Table($this->getTableName($shopId));
+        $table = new Table($tableName);
         $this->buildTable($table);
         $schemaManager->createTable($table);
 
@@ -88,12 +89,11 @@ public function getTableName($shopId = null)
             throw new InvalidArgumentException("Invalid table name specified: \"$tableName\"");
         }
 
-        $suffix = null;
         if ($shopId !== null) {
-            $suffix = '_' . $shopId;
+            $tableName .= '_' . $shopId;
         }
 
-        return $tableName . $suffix;
+        return $tableName;
     }
 
     /**
@@ -104,6 +104,7 @@ public function addRecord($operationType, $documentType, $documentId, DateTime $
         if (empty($shopIds)) {
             $shopIds = [null];
         }
+
         $connection = $this->getConnection();
 
         foreach ($shopIds as $shopId) {