Fixing resolvers.

scandipwa · May 24, 2019 · c0b443c · c0b443c
1 parent 7efb4e0
commit c0b443c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/Model/Resolver/AttributesWithValue.php b/src/Model/Resolver/AttributesWithValue.php
@@ -19,21 +19,27 @@
 use Magento\Framework\GraphQl\Query\ResolverInterface;
 use Magento\Swatches\Helper\Data;
 
+use Magento\Catalog\Model\ProductRepository;
+
 class AttributesWithValue implements ResolverInterface
 {
     /**
      * @var Data
      */
     private $swatchHelper;
 
+    private $productRepository;
+
     /**
      * CustomAttributes constructor.
      * @param Data $swatchHelper
      */
     public function __construct(
-        Data $swatchHelper
+        Data $swatchHelper,
+        ProductRepository $productRepository
     ) {
         $this->swatchHelper = $swatchHelper;
+        $this->productRepository = $productRepository;
     }
 
     /**
@@ -54,7 +60,8 @@ public function resolve(
         array $value = null,
         array $args = null
     ) {
-        $product = isset($value['product']) ? $value['product']['model'] : $value['model'];
+        $product = $this->productRepository->getById($value['entity_id']);
+//        $product = isset($value['product']) ? $value['product']['model'] : $value['model'];
         $attributes = $product->getAttributes();
         $attributesToSelect = array_unique($args['attributes']);
         $attributesToReturn = [];

diff --git a/src/etc/schema.graphqls b/src/etc/schema.graphqls
@@ -51,7 +51,7 @@ type AttributeWithValue {
     attribute_type: String
     attribute_value: String
     attribute_label: String
-    attribute_id: String
+    attribute_id: Int
     attribute_options: [AttributeWithValueOption]
 }