<?php
require_once(__DIR__.'/../vendor/autoload.php');

use Pxlswrite\DB\DB;
use Pxlswrite\Pxlswrite;
use Pxlswrite\WebSocket\WebSocketClient;

$time = time();


//实例化pxlswrite
$fileObj = new Pxlswrite(['path' => __DIR__ . '/uploads']);
//实例化WebSocketClient--需要推送进度才实例化
$pushHandle = new WebSocketClient('ws://192.168.18.192:9502',$_GET['fd']);
//创建excel文件
$fileObj->fileName('123.xlsx');

//定义样式
$leftStyle = $fileObj->styleFormat()
    ->bold()
    ->align(Pxlswrite::FORMAT_ALIGN_LEFT, Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)
    ->toResource();
$borderStyle = $fileObj->styleFormat()
    ->align(Pxlswrite::FORMAT_ALIGN_RIGHT, Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)
    ->border(Pxlswrite::BORDER_SLANT_DASH_DOT)
    ->toResource();
$colorStyle = $fileObj->styleFormat()
    ->fontColor(Pxlswrite::COLOR_BLUE)
    ->toResource();
$backgroundStyle  = $fileObj->styleFormat()
    ->background(Pxlswrite::COLOR_RED)
    ->toResource();
$numberStyle = $fileObj->styleFormat()
    ->number('#,##0')
    ->toResource();
$defaultStyle = $fileObj->styleFormat()
    ->fontColor(Pxlswrite::COLOR_ORANGE)
    ->border(Pxlswrite::BORDER_DASH_DOT)
    ->align(Pxlswrite::FORMAT_ALIGN_CENTER,Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)
    ->toResource();

$filePath = $fileObj->header(array('id', 'c1', 'c2', 'c3', 'c4','c5'))//设置表格头
    ->defaultFormat($defaultStyle)//全局默认样式
    ->setDataByGenerator('generateData',$pushHandle)//设置数据，$pushHandle 用于推送，可不传
    ->setRow('A1:A3', 80, $leftStyle)//设置范围行样式
    ->setRow('A2',50,$borderStyle)//设置指定某一行样式
    ->setRow('A3',50,$colorStyle)//设置文字颜色
    ->setRow('A4',40,$backgroundStyle)//设置背景色
    ->setColumn('B:B',50,$numberStyle)//设置列样式
    ->mergeCells('A1:C1', 'Merge cells',$fileObj->styleFormat()->align(Pxlswrite::FORMAT_ALIGN_CENTER,Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)->toResource())//合并单元格
    ->output();//输出excel文件到磁盘

//单元格插入文本
//for ($index = 0; $index < 10; $index++) {
//    $fileObj->insertText($index, 0, 'viest');
//    $fileObj->insertText($index, 1, 10000, '#,##0');
//}
//$filePath = $fileObj->output();

$memory = floor((memory_get_peak_usage()) / 1024 / 1024) . "MB";#10M 22S
$execute_time = time() - $time . 's';

//同步下载
//$fileObj->download($filePath);
//ajax请求返回下载地址
echo json_encode(['code' => 1, 'msg' => '导出完毕', 'url' => '/download.php?file=' . $filePath, 'data' => ['memory' => $memory, 'excute_time' => $execute_time]]);

//数据生成器--封装数据获取的方法
function generateData(){
    $db = DB::getInstance();
    $step = 10000;
    for ($i = 0; $i < 100000; $i = $i + $step) {
        yield $db->get_records_sql("select * from sheet1 limit {$i},{$step}", null, PDO::FETCH_ASSOC);
    }
}