导出业务数据为表格(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();


/* 分配到模板等等 */