1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\ProductAlert\Model\ResourceModel;
/**
* Product alert for back in stock resource model
*
* @api
* @since 100.0.2
*/
class Stock extends \Magento\ProductAlert\Model\ResourceModel\AbstractResource
{
/**
* @var \Magento\Framework\Stdlib\DateTime\DateTimeFactory
*/
protected $_dateFactory;
/**
* @param \Magento\Framework\Model\ResourceModel\Db\Context $context
* @param \Magento\Framework\Stdlib\DateTime\DateTimeFactory $dateFactory
* @param string $connectionName
*/
public function __construct(
\Magento\Framework\Model\ResourceModel\Db\Context $context,
\Magento\Framework\Stdlib\DateTime\DateTimeFactory $dateFactory,
$connectionName = null
) {
$this->_dateFactory = $dateFactory;
parent::__construct($context, $connectionName);
}
/**
* Initialize connection
*
* @return void
*/
protected function _construct()
{
$this->_init('product_alert_stock', 'alert_stock_id');
}
/**
* Before save action
*
* @param \Magento\Framework\Model\AbstractModel $object
* @return $this
*/
protected function _beforeSave(\Magento\Framework\Model\AbstractModel $object)
{
if ($object->getId() === null
&& $object->getCustomerId()
&& $object->getProductId()
&& $object->getWebsiteId()
) {
if ($row = $this->_getAlertRow($object)) {
$object->addData($row);
$object->setStatus(0);
}
}
if ($object->getAddDate() === null) {
$object->setAddDate($this->_dateFactory->create()->gmtDate());
$object->setStatus(0);
}
return parent::_beforeSave($object);
}
}