MAGETWO-66727: Update performance profiles with swatches

- fix memory limit

Author: mslabko

setup/src/Magento/Setup/Model/FixtureGenerator/EntityGenerator.php

@@ -6,6 +6,7 @@
 
 namespace Magento\Setup\Model\FixtureGenerator;
 
+use Magento\Framework\Exception\LocalizedException;
 use Magento\Framework\Exception\ValidatorException;
 
 /**
@@ -138,14 +139,25 @@ public function __construct(
      * @param TemplateEntityGeneratorInterface $entityGenerator
      * @param int $entitiesAmount
      * @param callable $fixture
+     * @throws LocalizedException
      * @return void
      */
     public function generate(TemplateEntityGeneratorInterface $entityGenerator, $entitiesAmount, callable $fixture)
     {
-        $this->sqlCollector->enable();
-        $entity = $entityGenerator->generateEntity();
-        $this->sqlCollector->disable();
-        $entity->delete();
+        $this->getConnection()->beginTransaction();
+        try {
+            $this->sqlCollector->enable();
+            $entity = $entityGenerator->generateEntity();
+            $this->sqlCollector->disable();
+            $entity->delete();
+            $this->getConnection()->commit();
+        } catch (\Exception $e) {
+            $this->getConnection()->rollBack();
+            throw new LocalizedException(
+                __('Cannot generate entities - error occurred during template creation: %1', $e->getMessage()),
+                $e
+            );
+        }
 
         $map = [];
         $processed = 0;
@@ -239,11 +251,22 @@ private function bindWithCustomHandler($table, $entityId, $entityNumber, $fixtur
      *
      * @param array $map
      * @return void
+     * @throws LocalizedException
      */
     private function saveEntities(array &$map)
     {
-        foreach ($map as $table => $data) {
-            $this->getConnection()->insertMultiple($table, $data);
+        $this->getConnection()->beginTransaction();
+        try {
+            foreach ($map as $table => $data) {
+                $this->getConnection()->insertMultiple($table, $data);
+            }
+            $this->getConnection()->commit();
+        } catch (\Exception $e) {
+            $this->getConnection()->rollBack();
+            throw new LocalizedException(
+                __('Cannot save entity. Error occurred: %1', $e->getMessage()),
+                $e
+            );
         }
 
         $map = [];