PHP读取CSV大文件导入数据库的示例

文章详细介绍了csv文件在php中快速导入到mysql数据库中的例子,虽然从最简单的几百MB的到最后使用插件实现几个GB数据导入中间有一些嗑碰了,但结果还是好的.

对于数百万条数据量的CSV文件,文件大小可能达到数百M,如果简单读取的话很可能出现超时或者卡死的现象.

为了成功将CSV文件里的数据导入数据库,分批处理是非常必要的,下面这个函数是读取CSV文件中指定的某几行数据,代码如下:

1. /** 
2.  * csv_get_lines 读取CSV文件中的某几行数据 
3.  * @param $csvfile csv文件路径 
4.  * @param $lines 读取行数 
5.  * @param $offset 起始行数 
6.  * @return array 
7.  * */ 
8. function csv_get_lines($csvfile, $lines, $offset = 0) { 
9.     if(!$fp = fopen($csvfile, 'r')) { 
10.     //开源代码phpfensi.com 
11.      return false; 
12.     } 
13.     $i = $j = 0; 
14.  while (false !== ($line = fgets($fp))) { 
15.   if($i++ < $offset) { 
16.    continue;  
17.   } 
18.   break; 
19.  } 
20.  $data = array(); 
21.  while(($j++ < $lines) && !feof($fp)) { 
22.   $data[] = fgetcsv($fp); 
23.  } 
24.  fclose($fp); 
25.     return $data; 
26. } 
27.  
28. //调用方法： 
29.  
30. $data = csv_get_lines('path/bigfile.csv', 10, 2000000); 
31.  
32. print_r($data); 

函数主要采用行定位的思路,通过跳过起始行数来实现文件指针定位,至于数据如何入库本文不再详细讲述.

上述函数对500M以内的文件进行过测试,运行通畅,对于1GB的文件发现有点慢了,于是再接着找方法.

如何快速完整的操作大文件仍然还存在一些问题.

1、如何快速获取CSV大文件的总行数？

办法一:直接获取文件内容，使用换行符进行拆分得出总行数，这种办法对小文件可行，处理大文件时不可行；

办法二:使用fgets一行一行遍历，得出总行数，这种办法比办法一好一些，但大文件仍有超时的可能；

办法三:借助SplFileObject类，直接将指针定位到文件末尾，通过SplFileObject::key方法获取总行数，这种办法可行，且高效。

具体实现方法,代码如下:

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

2、如何快速获取CSV大文件的数据？

仍然使用PHP的SplFileObject类,通过seek方法实现快速定位,代码如下:

1. $csv_file = 'path/bigfile.csv'; 
2. $start = 100000;  // 从第100000行开始 www.phpfensi.com读取 
3. $num = 100;    // 读取100行 
4. $data = array(); 
5. $spl_object = new SplFileObject($csv_file, 'rb'); 
6. $spl_object->seek($start); 
7. while ($num-- && !$spl_object->eof()) { 
8.  $data[] = $spl_object->fgetcsv(); 
9.  $spl_object->next(); 
10. } 
11. print_r($data); 

综合上面两点,整理成一个csv文件读取的类,代码如下:

1. class CsvReader { 
2.  private $csv_file; 
3.  private $spl_object = null; 
4.  private $error; 
5.  
6.  public function __construct($csv_file = '') { 
7.   if($csv_file && file_exists($csv_file)) { 
8.    $this->csv_file = $csv_file; 
9.   } 
10.  } 
11.  
12.  public function set_csv_file($csv_file) { 
13.   if(!$csv_file || !file_exists($csv_file)) { 
14.    $this->error = 'File invalid'; 
15.    return false; 
16.   } 
17.   $this->csv_file = $csv_file; 
18.   $this->spl_object = null; 
19.  } 
20.  
21.  public function get_csv_file() { 
22.   return $this->csv_file; 
23.  } 
24.  
25.  private function _file_valid($file = '') { 
26.   $file = $file ? $file : $this->csv_file; 
27.   if(!$file || !file_exists($file)) { 
28.    return false; 
29.   } 
30.   if(!is_readable($file)) { 
31.    return false; 
32.   } 
33.   return true; 
34.  } 
35.  
36.  private function _open_file() { 
37.   if(!$this->_file_valid()) { 
38.    $this->error = 'File invalid'; 
39.    return false; 
40.   } 
41.   if($this->spl_object == null) { 
42.    $this->spl_object = new SplFileObject($this->csv_file, 'rb'); 
43.   } 
44.   return true; 
45.  } 
46.  
47.  public function get_data($length = 0, $start = 0) { 
48.   if(!$this->_open_file()) { 
49.    return false; 
50.   } 
51.   $length = $length ? $length : $this->get_lines(); 
52.   $start = $start - 1; 
53.   $start = ($start < 0) ? 0 : $start; 
54.   $data = array(); 
55.   $this->spl_object->seek($start); 
56.   while ($length-- && !$this->spl_object->eof()) { 
57.    $data[] = $this->spl_object->fgetcsv(); 
58.    $this->spl_object->next(); 
59.   } 
60.   return $data; 
61.  } 
62.  
63.  public function get_lines() { 
64.   if(!$this->_open_file()) { 
65.    return false; 
66.   } 
67.   $this->spl_object->seek(filesize($this->csv_file)); 
68.   return $this->spl_object->key(); 
69.  } 
70.  
71.  public function get_error() { 
72.   return $this->error; 
73.  } 
74. } 

调用方法如下:

1. include('CsvReader.class.php'); 
2.  
3. $csv_file = 'path/bigfile.csv'; 
4.  
5. $csvreader = new CsvReader($csv_file); 
6.  
7. $line_number = $csvreader->get_lines(); 
8.  
9. $data = $csvreader->get_data(10); 
10.  
11. echo $line_number, chr(10); 
12.  
13. print_r($data); 

其实,上述CsvReader类并不只针对CSV大文件,对于其他文本类型的大文件或超大文件同样可用,前提是将类中fgetcsv方法稍加改动为current即可.