Subscriber.php

Go to the documentation of this file.

<?php
namespace Magento\Newsletter\Model\ResourceModel;

class Subscriber extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
{
    protected $connection;

    protected $_subscriberLinkTable;

    protected $_messagesScope = 'newsletter/session';

    protected $_date;

    protected $mathRandom;

    public function __construct(
        \Magento\Framework\Model\ResourceModel\Db\Context $context,
        \Magento\Framework\Stdlib\DateTime\DateTime $date,
        \Magento\Framework\Math\Random $mathRandom,
        $connectionName = null
    ) {
        $this->_date = $date;
        $this->mathRandom = $mathRandom;
        parent::__construct($context, $connectionName);
    }

    protected function _construct()
    {
        $this->_init('newsletter_subscriber', 'subscriber_id');
        $this->_subscriberLinkTable = $this->getTable('newsletter_queue_link');
        $this->connection = $this->getConnection();
    }

    public function setMessagesScope($scope)
    {
        $this->_messagesScope = $scope;
    }

    public function loadByEmail($subscriberEmail)
    {
        $select = $this->connection->select()->from($this->getMainTable())->where('subscriber_email=:subscriber_email');

        $result = $this->connection->fetchRow($select, ['subscriber_email' => $subscriberEmail]);

        if (!$result) {
            return [];
        }

        return $result;
    }

    public function loadByCustomerData(\Magento\Customer\Api\Data\CustomerInterface $customer)
    {
        $select = $this->connection
            ->select()
            ->from($this->getMainTable())
            ->where('customer_id=:customer_id and store_id=:store_id');

        $result = $this->connection
            ->fetchRow(
                $select,
                [
                    'customer_id' => $customer->getId(),
                    'store_id' => $customer->getStoreId()
                ]
            );

        if ($result) {
            return $result;
        }

        $select = $this->connection
            ->select()
            ->from($this->getMainTable())
            ->where('subscriber_email=:subscriber_email and store_id=:store_id');

        $result = $this->connection
            ->fetchRow(
                $select,
                [
                    'subscriber_email' => $customer->getEmail(),
                    'store_id' => $customer->getStoreId()
                ]
            );

        if ($result) {
            return $result;
        }

        return [];
    }

    protected function _generateRandomCode()
    {
        return $this->mathRandom->getUniqueHash();
    }

    public function received(\Magento\Newsletter\Model\Subscriber $subscriber, \Magento\Newsletter\Model\Queue $queue)
    {
        $this->connection->beginTransaction();
        try {
            $data['letter_sent_at'] = $this->_date->gmtDate();
            $this->connection->update(
                $this->_subscriberLinkTable,
                $data,
                ['subscriber_id = ?' => $subscriber->getId(), 'queue_id = ?' => $queue->getId()]
            );
            $this->connection->commit();
        } catch (\Exception $e) {
            $this->connection->rollBack();
            throw new \Magento\Framework\Exception\LocalizedException(__('We cannot mark as received subscriber.'));
        }
        return $this;
    }
}