<?php
/**
* <方法描述>导出excel或csv文件
* 需求描述：在项目开发中遇到了需要导出大数据量excel的问题,发现用框架在现有的基础上无法实现,特此总结
* 使用该方法我已成功导出100W的数据分别到excel和csv文件中,
* 下载文件大概1分钟左右,读取文件,excel时间长点大概需要4、5分钟才能打开,在2003版本中打开只能显示到65536行
* 打开csv文件则快多了，而且不区分2003和2007,都可以显示100W的数据
* excel 2003版本最多允许65536条数据
* 2007版本以上允许1048576条数据
* 采用数据库直连方式,否则可能会内存溢出,无法导出excel,由于PHP版本不同可能会报mysql连接函数已废弃,建议使用PDO
* 我使用的是PHP5.3版本，因为我们正式环境的版本就是5.3,尽量保证开发环境与生产环境的一致
* PHP可以从数据库中取出上百万的数据，但是取出的数据放哪里呢，数组？对象？此时就会有内存溢出的问题
* php.ini中修改memory_limit 的值可以将内存设置为很大，为-1则不限制，但是数据量大时会特别卡，
* 而且有时你不一定有权限修改php.ini
* 可在程序中临时修改php.ini中的值
* ini_set('memory_limit', '128M');//设置脚本允许分配的字节的最大内存大小
* @param [int] limit 导出条数
* @param [int] isexcel 1导出excel,2导出csv
* @return 直接输出excel或csv文件到浏览器下载
* @author guangzhengren@sina.com
* @date 2017-07-21
*/
//设置超时时间,PHP默认30秒超时时间，当数据量大时若超时写入文件的内容则不全
set_time_limit(1800);
//默认导出20条数据
$limit = $_GET['limit'] ? intval($_GET['limit']) : 20;
//默认导出excel文件
$isexcel = $_GET['isexcel'] ? intval($_GET['isexcel']) : 1;
//连接数据库
$con = mysql_connect('localhost','root','123456') or die('DB connect failed!'."\n");
//选库
mysql_select_db('db_ljlj',$con);
//sql语句
$wechatSql = 'SELECT id,appid,mch_appid,wx_appid,openid,consume_id,contract_number,out_trade_no,transaction_id,pass_trade_no,value,charge,pay_result,out_refund_no,refund_id,pass_refund_no,is_subscribe,bank_type,refund_channel,add_time,modify_time FROM `wechat_annal` ORDER BY id DESC LIMIT '.$limit;
//执行查询，返回类型为 resource 的结果集
//mysql_unbuffered_query() 向 MySQL 发送一条 SQL 查询 query ，但不像 mysql_query() 那样自动获取并缓存结果集。一方面，这在处理很大的结果集时会节省可观的内存。另一方面，可以在获取第一行后立即对结果集进行操作，而不用等到整个 SQL 语句都执行完毕
$resource = mysql_unbuffered_query($wechatSql);
//文件名称，不含后缀
$filename = 'test';
//数据，表头
$tableHeader = array('ID','商户appid','商户号','微信号','openid','收款表主键','合同号','商户订单号','微信订单号','商户单号','金额','手续费','支付结果','商户退款单号','微信退款单号','通道退款单号','是否关注公众号','付款银行','退款手续费','订单生成时间','订单修改时间');
//判断导出excel还是csv
if($isexcel == 2){
//导出csv文件
exportCsv($resource,$tableHeader,$filename);
}else{
//导出excel
exportExcel($resource,$tableHeader,$filename);
}
/**
* 导出excel文件
* @param $resource Mysql结果集
* @param $tableHeader 文件表头
* @param $filename 文件名称，不含后缀
*/
function exportExcel($resource,$tableHeader,$filename)
{
// 输出excel文件头
header("Content-Type:application/vnd.ms-excel");
header("Content-Disposition:attachment;filename=".iconv("utf-8", "GB2312", $filename).".xls");
header("Content-type: text/html; charset=utf-8");
echo "<!DOCTYPE html PUBLIC '-//W3C//DTD XHTML 1.0 Transitional//EN' 'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'><html xmlns='http://www.w3.org/1999/xhtml'><head><meta http-equiv='Content-Type' content='text/html; charset=UTF-8' /></head><title>".$filename."</title><body><table width='90%'><tr>";
foreach ($tableHeader as $tab)
{
echo '<td >'.$tab.'</td>';
}
echo '</tr>';
while($rowCon = mysql_fetch_assoc($resource))
{
echo '<tr>';
foreach ($rowCon as $v)
{
echo '<td>'.$v.'</td>';
}
echo '</tr>';
}
echo '</table>';
exit;
}
/**
* 导出csv文件
* @param $resource Mysql结果集
* @param $tableHeader 文件表头
* @param $filename 文件名称，不含后缀
*/
function exportCsv($resource,$tableHeader,$filename)
{
// 输出csv文件头
header('Content-Type: application/vnd.ms-excel;charset=gbk');
header('Content-Disposition: attachment;filename="'.$filename.'.csv"');
header('Cache-Control: max-age=0');
// PHP文件句柄，php://output 表示直接输出到浏览器
$fp = fopen('php://output', 'a');
// 输出csv列头信息
foreach ($tableHeader as $i => $v)
{
// CSV的Excel支持GBK编码，一定要转换，否则乱码
$tableHeader[$i] = iconv('utf-8', 'gbk', $v);
}
// 写入列头
fputcsv($fp, $tableHeader);
// 计数器
$cnt = 0;
// 每隔$limit行，刷新一下输出buffer，节约资源
$limit = 10000;
while($rowCon = mysql_fetch_assoc($resource))
{
if ($limit == $cnt)
{
//刷新一下输出buffer，防止由于数据过多造成问题
ob_flush();
flush();
$cnt = 0;
}
foreach ($rowCon as $j=>$val)
{
$row[$j] = iconv('utf-8', 'gbk', $val);
}
$cnt++;
fputcsv($fp, $row);
}
exit;
}
?>
