friendica/src/Util/ReversedFileReader.php

116 lines
2.5 KiB
PHP
Raw Normal View History

<?php
/**
* @copyright Copyright (C) 2021, Friendica
*
* @license GNU AGPL version 3 or any later version
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*
*/
namespace Friendica\Util;
/**
* An iterator which returns lines from file in reversed order
*
* original code https://stackoverflow.com/a/10494801
*/
class ReversedFileReader implements \Iterator
{
2021-05-24 16:05:02 -04:00
const BUFFER_SIZE = 4096;
const SEPARATOR = "\n";
2021-05-24 16:05:02 -04:00
/** @var int */
private $filesize;
2021-05-24 15:47:10 -04:00
2021-05-24 16:05:02 -04:00
/** @var int */
private $pos;
2021-05-24 15:47:10 -04:00
2021-05-24 16:05:02 -04:00
/** @var array */
private $buffer;
2021-05-24 15:47:10 -04:00
2021-05-24 16:05:02 -04:00
/** @var int */
private $key;
2021-05-24 15:47:10 -04:00
2021-05-24 16:05:02 -04:00
/** @var string */
private $value;
2021-05-24 15:47:10 -04:00
2021-05-24 16:05:02 -04:00
public function __construct($filename)
{
$this->_fh = fopen($filename, 'r');
if (!$this->_fh) {
// this should use a custom exception.
throw \Exception("Unable to open $filename");
}
2021-05-24 16:05:02 -04:00
$this->filesize = filesize($filename);
$this->pos = -1;
$this->buffer = null;
$this->key = -1;
$this->value = null;
}
2021-05-24 16:05:02 -04:00
public function _read($size)
{
$this->pos -= $size;
fseek($this->_fh, $this->pos);
return fread($this->_fh, $size);
}
2021-05-24 16:05:02 -04:00
public function _readline()
{
$buffer = & $this->buffer;
while (true) {
if ($this->pos == 0) {
return array_pop($buffer);
}
if (count($buffer) > 1) {
return array_pop($buffer);
}
$buffer = explode(self::SEPARATOR, $this->_read(self::BUFFER_SIZE) . $buffer[0]);
}
}
2021-05-24 16:05:02 -04:00
public function next()
{
++$this->key;
$this->value = $this->_readline();
}
2021-05-24 16:05:02 -04:00
public function rewind()
{
if ($this->filesize > 0) {
$this->pos = $this->filesize;
$this->value = null;
$this->key = -1;
$this->buffer = explode(self::SEPARATOR, $this->_read($this->filesize % self::BUFFER_SIZE ?: self::BUFFER_SIZE));
$this->next();
}
}
2021-05-24 16:05:02 -04:00
public function key()
{
2021-05-24 15:47:10 -04:00
return $this->key;
}
2021-05-24 16:05:02 -04:00
public function current()
{
2021-05-24 15:47:10 -04:00
return $this->value;
}
2021-05-24 16:05:02 -04:00
public function valid()
{
2021-05-24 15:47:10 -04:00
return ! is_null($this->value);
}
}