/
Collection.php
114 lines (106 loc) · 3.41 KB
/
Collection.php
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
<?php
/**
* Copyright © 2016 Magento. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\AdminNotification\Model\ResourceModel\System\Message;
class Collection extends \Magento\Framework\Model\ResourceModel\Db\Collection\AbstractCollection
{
/**
* System message list
*
* @var \Magento\Framework\Notification\MessageList
*/
protected $_messageList;
/**
* Number of messages by severity
*
* @var array
*/
protected $_countBySeverity = [];
/**
* @param \Magento\Framework\Data\Collection\EntityFactory $entityFactory
* @param \Psr\Log\LoggerInterface $logger
* @param \Magento\Framework\Data\Collection\Db\FetchStrategyInterface $fetchStrategy
* @param \Magento\Framework\Event\ManagerInterface $eventManager
* @param \Magento\Framework\Notification\MessageList $messageList
* @param mixed $connection
* @param \Magento\Framework\Model\ResourceModel\Db\AbstractDb $resource
*/
public function __construct(
\Magento\Framework\Data\Collection\EntityFactory $entityFactory,
\Psr\Log\LoggerInterface $logger,
\Magento\Framework\Data\Collection\Db\FetchStrategyInterface $fetchStrategy,
\Magento\Framework\Event\ManagerInterface $eventManager,
\Magento\Framework\Notification\MessageList $messageList,
\Magento\Framework\DB\Adapter\AdapterInterface $connection = null,
\Magento\Framework\Model\ResourceModel\Db\AbstractDb $resource = null
) {
$this->_messageList = $messageList;
parent::__construct($entityFactory, $logger, $fetchStrategy, $eventManager, $connection, $resource);
}
/**
* Resource collection initialization
*
* @return void
*/
protected function _construct()
{
$this->_init(
'Magento\AdminNotification\Model\System\Message',
'Magento\AdminNotification\Model\ResourceModel\System\Message'
);
}
/**
* Initialize db query
*
* @return void
*/
protected function _initSelect()
{
parent::_initSelect();
$this->addOrder('severity', self::SORT_ORDER_ASC)->addOrder('created_at');
}
/**
* Initialize system messages after load
*
* @return void
*/
protected function _afterLoad()
{
foreach ($this->_items as $key => $item) {
$message = $this->_messageList->getMessageByIdentity($item->getIdentity());
if ($message) {
$item->setText($message->getText());
if (array_key_exists($message->getSeverity(), $this->_countBySeverity)) {
$this->_countBySeverity[$message->getSeverity()]++;
} else {
$this->_countBySeverity[$message->getSeverity()] = 1;
}
} else {
unset($this->_items[$key]);
}
}
}
/**
* Set message severity filter
*
* @param int $severity
* @return $this
*/
public function setSeverity($severity)
{
$this->addFieldToFilter('severity', ['eq' => $severity * 1]);
return $this;
}
/**
* Retrieve number of messages by severity
*
* @param int $severity
* @return int
*/
public function getCountBySeverity($severity)
{
return isset($this->_countBySeverity[$severity]) ? $this->_countBySeverity[$severity] : 0;
}
}