1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- <?php
- /**
- * Copyright © Magento, Inc. All rights reserved.
- * See COPYING.txt for license details.
- */
- /**
- * Active record implementation
- */
- namespace Magento\Framework\Model\ResourceModel;
- use Magento\Framework\DB\Adapter\AdapterInterface;
- use Magento\Framework\Exception\LocalizedException;
- use Magento\Framework\Phrase;
- class Iterator extends \Magento\Framework\DataObject
- {
- /**
- * Walk over records fetched from query one by one using callback function
- *
- * @param \Zend_Db_Statement_Interface|\Magento\Framework\DB\Select|string $query
- * @param array|string $callbacks
- * @param array $args
- * @param AdapterInterface $connection
- * @return \Magento\Framework\Model\ResourceModel\Iterator
- */
- public function walk($query, array $callbacks, array $args = [], $connection = null)
- {
- $stmt = $this->_getStatement($query, $connection);
- $args['idx'] = 0;
- while ($row = $stmt->fetch()) {
- $args['row'] = $row;
- foreach ($callbacks as $callback) {
- $result = call_user_func($callback, $args);
- if (!empty($result)) {
- $args = array_merge($args, (array)$result);
- }
- }
- $args['idx']++;
- }
- return $this;
- }
- /**
- * Fetch Zend statement instance
- *
- * @param \Zend_Db_Statement_Interface|\Magento\Framework\DB\Select|string $query
- * @param AdapterInterface $connection
- * @return \Zend_Db_Statement_Interface
- * @throws LocalizedException
- */
- protected function _getStatement($query, AdapterInterface $connection = null)
- {
- if ($query instanceof \Zend_Db_Statement_Interface) {
- return $query;
- }
- if ($query instanceof \Zend_Db_Select) {
- return $query->query();
- }
- if (is_string($query)) {
- if (!$connection instanceof AdapterInterface) {
- throw new LocalizedException(
- new Phrase('The connection is invalid. Verify the connection and try again.')
- );
- }
- return $connection->query($query);
- }
- throw new LocalizedException(new Phrase('The query is invalid. Verify the query and try again.'));
- }
- }
|