在做运营后台的时候,不免会收到运营导出数据的需求,我们一般先从db里发数据取出来,做一些简单的数据。让后echo到server,再到浏览器。
所以一般可能会这样
class ConcreteController {
public function actionDownload() {
$request = new Request();
$model = new ConcreteModel();
$data = $model->getAllData($request); //取出所有数据
$this->exportData($dataReader, 'txt.csv');
exit;
}
private function exportData($data, $filename) {
$file = '/tmp/Concrete.exportData.'.time().'csv';
$this->putToFile($data, $file);
$this->download($filename, $file, true);
}
private function putToFile($data,$file) {
$isFirst = true;
$fp = fopen($file, 'wb');
foreach($dataReader as $key => $row) {
if($isFirst) {
$isFirst = false;
fputcsv($fp, array_keys($row));
}
fputcsv($fp, $row);
}
fclose($fp);
}
private function download($filename, $file, $rmFile=false) {
$this->downloadHeader($filename);
$d = file_get_contents($file);
echo mb_convert_encoding($d, 'gbk', 'utf-8');
if($rmFile) {
unlink($file);
}
}
protected function downloadHeader($filename) {
header("Content-Type: application/force-download");
header("Content-Disposition: attachment; filename={$filename}");
}
}
但是到取出的数据量很大的时候可能会有问题,优化的方法有很多,比如用
PDO::query查询数据后只获取一个部分结果集,迭代获取全部结果,把数据输出到浏览器的时候输出一部分,就flush一部分,或者先把数据写到磁盘里,然后用X-sendfile技术结合sever来下载一个静态的文件.
我这里不想把文件写到磁盘上,就不用x-sendfile了,结合前面把数组变成csv字符串的方法,可以这么弄
class ConcreteController {
public function actionDownload() {
$request = new Request();
$model = new ConcreteModel();
$dataReader = $model->exportData($request); //dataReader是一个 实现了 Traversable 接口,可用foreach 遍历的数据集对象
$this->exportData($dataReader, 'test.csv');
exit;
}
private function exportData($dataReader, $filename) {
$this->downloadHeader($filename);
list($isFirst, $i, $limit) = array(true, 0, 1000);
foreach($dataReader as $key => $row) {
if($isFirst) {
$isFirst = false;
++$i;
echo CsvHelper::strPutCsv(array_keys($row));
}
echo CsvHelper::strPutCsv($row);
if(++$i > $limit) {
$i = 0;
ob_flush();
flush();
}
}
}
protected function downloadHeader($filename) {
header("Content-Type: application/force-download");
header("Content-Disposition: attachment; filename={$filename}");
}
}