PHP快速按行讀取CSV大文件的封裝類分享（也適用于其它超大文本文件）

2024-05-04 23:22:47

字體：大中小

來源：轉載

供稿：網友

CSV大文件的讀取已經在前面講述過了（PHP按行讀取、處理較大CSV文件的代碼實例），但是如何快速完整的操作大文件仍然還存在一些問題。

1、如何快速獲取CSV大文件的總行數？

辦法一：直接獲取文件內容，使用換行符進行拆分得出總行數，這種辦法對小文件可行，處理大文件時不可行；
辦法二：使用fgets一行一行遍歷，得出總行數，這種辦法比辦法一好一些，但大文件仍有超時的可能；
辦法三：借助SplFileObject類，直接將指針定位到文件末尾，通過SplFileObject::key方法獲取總行數，這種辦法可行，且高效。

具體實現方法：

復制代碼代碼如下:

$csv_file = 'path/bigfile.csv';
$spl_object = new SplFileObject($csv_file, 'rb');
$spl_object->seek(filesize($csv_file));
echo $spl_object->key();

2、如何快速獲取CSV大文件的數據？

仍然使用PHP的SplFileObject類，通過seek方法實現快速定位。

復制代碼代碼如下:

$csv_file = 'path/bigfile.csv';
$start = 100000; // 從第100000行開始讀取
$num = 100; // 讀取100行
$data = array();
$spl_object = new SplFileObject($csv_file, 'rb');
$spl_object->seek($start);
while ($num-- && !$spl_object->eof()) {
$data[] = $spl_object->fgetcsv();
$spl_object->next();
}
print_r($data);

3、綜合上面兩點，整理成一個csv文件讀取的類：

復制代碼代碼如下:

class CsvReader {
private $csv_file;
private $spl_object = null;
private $error;

public function __construct($csv_file = '') {
  if($csv_file && file_exists($csv_file)) {
   $this->csv_file = $csv_file;
  }
}

public function set_csv_file($csv_file) {
  if(!$csv_file || !file_exists($csv_file)) {
   $this->error = 'File invalid';
   return false;
  }
  $this->csv_file = $csv_file;
  $this->spl_object = null;
}

public function get_csv_file() {
  return $this->csv_file;
}

private function _file_valid($file = '') {
  $file = $file ? $file : $this->csv_file;
  if(!$file || !file_exists($file)) {
   return false;
  }
  if(!is_readable($file)) {
   return false;
  }
  return true;
}

private function _open_file() {
  if(!$this->_file_valid()) {
   $this->error = 'File invalid';
   return false;
  }
  if($this->spl_object == null) {
   $this->spl_object = new SplFileObject($this->csv_file, 'rb');
  }
  return true;
}

public function get_data($length = 0, $start = 0) {
  if(!$this->_open_file()) {
   return false;
  }
  $length = $length ? $length : $this->get_lines();
  $start = $start - 1;
  $start = ($start < 0) ? 0 : $start;
  $data = array();
  $this->spl_object->seek($start);
  while ($length-- && !$this->spl_object->eof()) {
   $data[] = $this->spl_object->fgetcsv();
   $this->spl_object->next();
  }
  return $data;
}

public function get_lines() {
  if(!$this->_open_file()) {
   return false;
  }
  $this->spl_object->seek(filesize($this->csv_file));
  return $this->spl_object->key();
}

public function get_error() {
  return $this->error;
}
}