nebula\component\filesystem\ReadLineIterator

Complexity

Total Complexity

9

Size/Duplication

Total Lines	49
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
B	getIterator()	0	24	8

<?php
namespace nebula\component\filesystem;

use Iterator;
use IteratorAggregate;

/**
 * 逐行读文件迭代器
 *
 */
class ReadLineIterator implements IteratorAggregate
{
    protected $path;
    protected $bufferSize;
    protected $offset;

    /**
     * 构建迭代器
     *
     * @param string $path 路径
     * @param integer|null $offset 起始偏移
     * @param integer $bufferSize
     */
    public function __construct(string $path, ?int $offset = 0, int $bufferSize = 1024)
    {
        $this->path = $path;
        $this->offset = $offset;
        $this->bufferSize = $bufferSize;
    }

    /**
     * 获取迭代器
     *
     * @return Iterator
     */
    public function getIterator():Iterator
    {
        if (file_exists($this->path) && ($fp = fopen($this->path, 'rb')) !== false) {
            $offset = $this->offset ?: 0;
            while (!feof($fp)) {
                fseek($fp, $offset);
                $content = fread($fp, $this->bufferSize);
                $content.=(feof($fp))? "\n":'';
                $size = strpos($content, "\n");
                $offset += $size;
                if ($content[$size - 1] === "\r") {
                    $content = substr($content, 0, $size -1);
                } else {
                    $content = substr($content, 0, $size);
                }
                if (is_null($this->offset)) {
                    yield $content;
                } else {
                    yield $offset => $content;
                }
            }
            fclose($fp);
        } else {
            throw new \Exception(sprintf('readline iterator: file note exists %s', $this->path));
        }
    }
}