MAGETWO-55299: [Customer] Fast Save of Product Variations

     - MAGETWO-55787: Fast saving of product with high number of variations generated

Author: kandy

app/code/Magento/Eav/Model/ResourceModel/CreateHandler.php

@@ -19,6 +19,9 @@
  */
 class CreateHandler implements AttributeInterface
 {
+    /** Name of ATTRIBUTE_SET_ID field */
+    const ATTRIBUTE_SET_ID = 'attribute_set_id';
+
     /**
      * @var AttributeRepository
      */
@@ -88,21 +91,35 @@ private function getAttributeCache()
      * @return \Magento\Eav\Api\Data\AttributeInterface[]
      * @throws \Exception
      */
-    protected function getAttributes($entityType)
+    protected function getAttributes($entityType, $attributeSetId = null)
     {
         /** @var AttributeCache $cache */
         $cache = $this->getAttributeCache();
-        if ($attributes = $cache->getAttributes($entityType)) {
+        $suffix = 'attribute_set_id-' . ($attributeSetId ?: 'all');
+        if ($attributes = $cache->getAttributes($entityType, $suffix)) {
             return $attributes;
         }
 
         $metadata = $this->metadataPool->getMetadata($entityType);
 
+        if ($attributeSetId === null) {
+            $criteria = $this->searchCriteriaBuilder->addFilter(self::ATTRIBUTE_SET_ID . '', null, 'neq')->create();
+        } else {
+            $criteria = $this->searchCriteriaBuilder->addFilter(self::ATTRIBUTE_SET_ID, $attributeSetId)->create();
+        }
+
         $searchResult = $this->attributeRepository->getList(
             $metadata->getEavEntityType(),
-            $this->searchCriteriaBuilder->addFilter('attribute_set_id', null, 'neq')->create()
+            $criteria
+        );
+        $attributes = $searchResult->getItems();
+
+        $this->attributeCache->saveAttributes(
+            $entityType,
+            $attributes,
+            $suffix
         );
-        return $searchResult->getItems();
+        return $attributes;
     }
 
     /**
@@ -120,8 +137,9 @@ public function execute($entityType, $entityData, $arguments = [])
         if ($metadata->getEavEntityType()) {
             $processed = [];
             $entityLinkField = $metadata->getLinkField();
+            $attributeSetId = isset($entityData[self::ATTRIBUTE_SET_ID]) ? $entityData[self::ATTRIBUTE_SET_ID] : null;
             /** @var \Magento\Eav\Model\Entity\Attribute\AbstractAttribute $attribute */
-            foreach ($this->getAttributes($entityType) as $attribute) {
+            foreach ($this->getAttributes($entityType, $attributeSetId) as $attribute) {
                 if ($attribute->isStatic()) {
                     continue;
                 }