ReportCounts.php

Go to the documentation of this file.

<?php
namespace Magento\NewRelicReporting\Model\Cron;

use Magento\NewRelicReporting\Model\Config;
use Magento\Catalog\Api\ProductManagementInterface;
use Magento\Catalog\Model\Product\Attribute\Source\Status;
use Magento\ConfigurableProduct\Api\ConfigurableProductManagementInterface;
use Magento\Catalog\Api\CategoryManagementInterface;

class ReportCounts
{
    protected $config;

    protected $productManagement;

    protected $configurableManagement;

    protected $categoryManagement;

    protected $countsFactory;

    protected $countsCollectionFactory;

    public function __construct(
        Config $config,
        ProductManagementInterface $productManagement,
        ConfigurableProductManagementInterface $configurableManagement,
        CategoryManagementInterface $categoryManagement,
        \Magento\NewRelicReporting\Model\CountsFactory $countsFactory,
        \Magento\NewRelicReporting\Model\ResourceModel\Counts\CollectionFactory $countsCollectionFactory
    ) {
        $this->config = $config;
        $this->productManagement = $productManagement;
        $this->configurableManagement = $configurableManagement;
        $this->categoryManagement = $categoryManagement;
        $this->countsFactory = $countsFactory;
        $this->countsCollectionFactory = $countsCollectionFactory;
    }

    protected function updateCount($count, \Magento\NewRelicReporting\Model\Counts $model, $type)
    {
        $collection = $this->countsCollectionFactory->create()
            ->addFieldToFilter(
                'type',
                ['eq' => $type]
            )->addOrder(
                'updated_at',
                'DESC'
            )->setPageSize(1);
        $latestUpdate = $collection->getFirstItem();

        if ((!$latestUpdate) || ($count != $latestUpdate->getCount())) {
            $model->setEntityId(null);
            $model->setType($type);
            $model->setCount($count);
            $model->save();
        }
    }

    protected function reportProductsSize()
    {
        $productCount = $this->productManagement->getCount();
        $model = $this->countsFactory->create()->load(Config::PRODUCT_COUNT, 'type');
        $this->updateCount($productCount, $model, Config::PRODUCT_COUNT);
    }

    protected function reportConfigurableProductsSize()
    {
        $configurableCount = $this->configurableManagement->getCount();
        $model = $this->countsFactory->create()->load(Config::CONFIGURABLE_COUNT, 'type');
        $this->updateCount($configurableCount, $model, Config::CONFIGURABLE_COUNT);
    }

    protected function reportProductsActive()
    {
        $productsActiveCount = $this->productManagement->getCount(Status::STATUS_ENABLED);
        $model = $this->countsFactory->create()->load(Config::ACTIVE_COUNT, 'type');
        $this->updateCount($productsActiveCount, $model, Config::ACTIVE_COUNT);
    }

    protected function reportCategorySize()
    {
        $categoryCount = $this->categoryManagement->getCount();
        $model = $this->countsFactory->create()->load(Config::CATEGORY_SIZE, 'type');
        $this->updateCount($categoryCount, $model, Config::CATEGORY_SIZE);
    }

    public function report()
    {
        if ($this->config->isNewRelicEnabled()) {
            $this->reportProductsSize();
            $this->reportConfigurableProductsSize();
            $this->reportProductsActive();
            $this->reportCategorySize();
        }

        return $this;
    }
}