导出业务数据为表格(csv)文件的一个简单通用的方法

jerry thinkphp 2015年11月19日收藏

一个通用的下载导出方法

/*
* 通用方法
* 导出大数据为CSV 
* 参数依次传入 查询对象,CSV文件列头(键是数据表中的列名,值是csv的列名),文件名.对数据二次处理的函数;
*/
private function DownloadCSV( $selectObject, $head, $fileName = time(), $callBack = '' ){
    
 if ( !is_object( $selectObject ) || !is_array( $head ) ) {
    exit('参数错误!');
 }

 set_time_limit(0);
 //下载头.
 header ('Content-Type: application/vnd.ms-excel;charset=gbk');
 header ('Content-Disposition: attachment;filename="'.$fileName.'.csv"');
 header ('Cache-Control: max-age=0');
    
 //输出流;
 $file = 'php://output';
 $fp = fopen ( $file, 'a' );

 $changCode = function( $changArr ) {
     // 破Excel2003中文只支持GBK编码;
     foreach ( $changArr as $k => $v ) {
         $changArr [$k] = iconv ( 'utf-8', 'gbk', $v );
     }
     //返回一个 索引数组;
     return array_values( $changArr );
 };

 //写入头部;
 fputcsv ( $fp, $changCode( $head ) );
    
 //写入数据;
 $pageSize = 100//每次查询一百条;
 $page  = 1;//起始页码;
 $list = array();

 //查库;
 $cloneObj = clone $selectObject;//因为thinkphp内部执行完select方法后会清空对象属性,所以clone;
    
 while ( $list = $cloneObj ->limit( $pageSize*( ($page++)-1 ), $pageSize )->select()  ) {

    $cloneObj = clone $selectObject;
            
    //对查询结果二次处理
    is_callable( $callBack ) && $list = call_user_func( $callBack, $list );

    foreach ( $list as $key => $value ) {
            
         $value = array_intersect_key( $value, $head );//返回需要写入CSV的数据;
         $value = array_merge( $head, $value );//利用此函数返回需要的顺序;
         $value = $changCode( $value );
         fputcsv ( $fp, $value );//写入数据;
         flush();
            
    }

    ob_flush();
                
 }

 exit();
}

//比如有一个方法 demo 在展示数据表时 还要有导出功能.

public function demo(){

/*

一堆判断搜索,之类的代码.

*/

$list = $table->field($x)->join($l)->where($m);

//比如$var是表单发送过来的 开启导出的"开关" 
if ( $var == 1 )

    $headerArr = array('数据表字段'=>'导出时CSV的列名');

    $callBack = function( $list ) {
        //对数据的二次处理;
        return $list;
    };

    $this->DownloadCSV( $list, $headerArr, 'XXX'.date('YmdHi',time()), $callBack );

}

$list = $list->select();

/* 分配到模板等等 */