README

README.md

@@ -1 +1,20 @@
-# xlswrite
+# xlswriter+yield+websocket
+这是一个基于xlswriter+yield高性能操作excel的库，集成了websocket用于excel操作时的进度推送。
+##Installing
+```
+composer require yzh0325/xlswrite
+```
+##安装xlswriter扩展 
+* php>=7.0
+* windows环境下 php>=7.2 ,xlswriter版本号大于等于 1.3.4.1
+```
+pecl install xlswriter
+```
+##WebSocket 
+* 服务端采用swoole搭建，需要安装swoole扩展 swoole>=4.4.*,需在cli模式下运行
+* 客户端采用textalk/websocket的client作为websocket客户端,可在php-fpm模式下运行
+##examples
+见 examples/ 
+* xlswrite_demo.html 前端demo
+* export_demo.php excel导出demo
+* import_demo.php excel导入demo

examples/export_demo.php

@@ -41,12 +41,12 @@ $defaultStyle = $fileObj->styleFormat()
 
 //定义字段
 $field = [
-    'id'=>['name'=>'title','callback'=>'myFormat'],//callback 回调处理格式化值 可以是函数/对象方法
+    'id'=>['name'=>'title'],
     'c1'=>['name'=>'age'],
     'c2'=>['name'=>'year'],
     'c3'=>['name'=>'kk'],
     'c4'=>['name'=>'ll'],
-    'c5'=>['name'=>'aa']
+    'c5'=>['name'=>'aa','callback'=>'myFormat']//callback 回调处理格式化值 可以是函数/对象方法
 ];
 $filePath = $fileObj->field($field)//设置字段&表格头
     ->defaultFormat($defaultStyle)//全局默认样式
@@ -55,7 +55,7 @@ $filePath = $fileObj->field($field)//设置字段&表格头
     ->setRow('A2',50,$borderStyle)//设置指定某一行样式
     ->setRow('A3',50,$colorStyle)//设置文字颜色
     ->setRow('A4',40,$backgroundStyle)//设置背景色
-    ->setColumn('B:B',50,$numberStyle)//设置列样式
+    ->setColumn('F:F',40,$numberStyle)//设置列样式
     ->mergeCells('A1:C1', 'Merge cells',$fileObj->styleFormat()->align(Pxlswrite::FORMAT_ALIGN_CENTER,Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)->toResource())//合并单元格
     ->output();//输出excel文件到磁盘
 
@@ -74,15 +74,15 @@ $execute_time = time() - $time . 's';
 //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) {
+    for ($i = 0; $i < 1000000; $i = $i + $step) {
         yield $db->get_records_sql("select * from sheet1 limit {$i},{$step}", null, PDO::FETCH_ASSOC);
     }
 }
 //格式化字段值
-function myFormat($v,$valuse){
-    return $v.$valuse['c1'];
+function myFormat($v,$values){
+    return $v.'自定义格式化-'.$values['id'];
 }

src/Pxlswrite.php

@@ -1,12 +1,11 @@
 <?php
 /**
- * xlsxwrite简单封装
+ * xlsxwriter简单封装
  */
 
 namespace Pxlswrite;
 set_time_limit(0);
 
-use http\Header;
 use Pxlswrite\WebSocket\WebSocketClient;
 use \Vtiful\Kernel\Format;
 use \Vtiful\Kernel\Excel;
@@ -151,25 +150,35 @@ class Pxlswrite extends Excel
     public function setDataByGenerator($_generator, WebSocketClient $_pushHandle = null)
     {
         $count = 0;
-        foreach (call_user_func($_generator) as $item) {
-            //循环逐行写入excel
-            foreach ($item as $key => $value) {
-                if(!empty($this->fieldsCallback)){
+        //判断是否有定义字段
+        if(!empty($this->fieldsCallback)){
+            foreach (call_user_func($_generator) as $item){
+                foreach ($item as $value){
                     $temp = [];
-                    foreach ($this->fieldsCallback as $k => $v) {
+                    //字段过滤
+                    foreach ($this->fieldsCallback as $k=>$v){
                         $temp[$k] = isset($value[$k]) && !empty($value[$k]) ? $value[$k] : '';
+                        //回调字段处理方法
                         if (isset($v['callback'])) {
                             $temp[$k] = call_user_func($v['callback'], $temp[$k], $value);
                         }
                     }
                     $this->data([array_values($temp)]);//二维索引数组
-                }else{
+                }
+                //推送消息
+                $count += count($item);
+                $this->push($_pushHandle,$count);
+            }
+        }else{
+            foreach (call_user_func($_generator) as $item){
+                //循环逐行写入excel
+                foreach ($item as $value){
                     $this->data([array_values($value)]);//二维索引数组
                 }
+                //推送消息
+                $count += count($item);
+                $this->push($_pushHandle,$count);
             }
-            //推送消息
-            $count += count($item);
-            $this->push($_pushHandle,$count);
         }
         return $this;
     }
@@ -198,6 +207,12 @@ class Pxlswrite extends Excel
             $this->push($_pushHandle,$count);
         }
     }
+
+    /**
+     * 消息推送
+     * @param $_pushHandle
+     * @param $count
+     */
     public function push($_pushHandle,$count){
         if ($_pushHandle && $_pushHandle->m_receiverFd) {
             $_pushHandle->send(['status' => 'processing', 'process' => $count]);