简化样式设置

examples/export_demo.php

@@ -1,5 +1,5 @@
 <?php
-require_once(__DIR__.'/../vendor/autoload.php');
+require_once(__DIR__ . '/../vendor/autoload.php');
 
 use Pxlswrite\DB\DB;
 use Pxlswrite\Pxlswrite;
@@ -11,52 +11,42 @@ $time = time();
 //实例化pxlswrite
 $fileObj = new Pxlswrite(['path' => __DIR__ . '/uploads']);
 //实例化WebSocketClient--需要推送进度才实例化
-$pushHandle = new WebSocketClient('ws://192.168.18.192:9502',$_GET['fd']);
+$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();
-
+$style = [
+    'align' => [Pxlswrite::FORMAT_ALIGN_CENTER, Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER],//对齐 [x,y]
+    'border' => Pxlswrite::BORDER_SLANT_DASH_DOT,//单元格边框
+    'background' => Pxlswrite::COLOR_RED,//单元格背景色
+    'fontColor' => Pxlswrite::COLOR_BLUE,//字体颜色
+    'fontSize' => 30,//字体大小
+    'font' => 'FontName',//设置字体 字体名称，字体必须存在于本机
+    'number' => '#,##0',//数字格式化
+    'bold' => true,//粗题
+    'strikeout' => false,//文本删除线
+    'wrap' => true,//文本换行
+    'italic' => true,//斜体
+];
 //定义字段
 $field = [
-    'id'=>['name'=>'title'],
-    'c1'=>['name'=>'age'],
-    'c2'=>['name'=>'year'],
-    'c3'=>['name'=>'kk'],
-    'c4'=>['name'=>'ll'],
-    'c5'=>['name'=>'aa','callback'=>'myFormat']//callback 回调处理格式化值 可以是函数/对象方法
+    'id' => ['name' => 'title'],
+    'c1' => ['name' => 'age'],
+    'c2' => ['name' => 'year'],
+    'c3' => ['name' => 'kk'],
+    'c4' => ['name' => 'll'],
+    'c5' => ['name' => 'aa', 'callback' => 'myFormat']//callback 回调处理格式化值 可以是函数/对象方法
 ];
+
+//注意:设置行与行/列与列样式 交集范围会覆盖；行样式优先于列样式
 $filePath = $fileObj->field($field)//设置字段&表格头
-    ->defaultFormat($defaultStyle)//全局默认样式
-    ->setDataByGenerator('generateData',$pushHandle)//设置数据 回调生成器方法获取数据，$pushHandle 用于推送，可不传
-    ->setRow('A1:A3', 80, $leftStyle)//设置范围行样式
-    ->setRow('A2',50,$borderStyle)//设置指定某一行样式
-    ->setRow('A3',50,$colorStyle)//设置文字颜色
-    ->setRow('A4',40,$backgroundStyle)//设置背景色
-    ->setColumn('F:F',40,$numberStyle)//设置列样式
-    ->mergeCells('A1:C1', 'Merge cells',$fileObj->styleFormat()->align(Pxlswrite::FORMAT_ALIGN_CENTER,Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER)->toResource())//合并单元格
+    ->setDataByGenerator('generateData', $pushHandle)//设置数据 回调生成器方法获取数据，$pushHandle 用于推送，可不传
+    ->setRow('A1:A3', 80, $style)//设置范围行样式 80行高
+    ->setColumn('A:F', 20, ['background' => Pxlswrite::COLOR_GRAY])//设置范围列样式 20列宽
+    ->setRow('A1', 50, ['background' => Pxlswrite::COLOR_PINK, 'align' => [Pxlswrite::FORMAT_ALIGN_CENTER, Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER]])//设置指定某一行样式
+    ->setColumn('F:F', 60, ['background' => Pxlswrite::COLOR_YELLOW])//指定某一列样式
+    ->defaultFormat(['background' => Pxlswrite::COLOR_GREEN])//全局默认样式
+    ->mergeCells('A1:C1', 'Merge cells', ['align' => [Pxlswrite::FORMAT_ALIGN_CENTER, Pxlswrite::FORMAT_ALIGN_VERTICAL_CENTER]])//合并单元格
     ->output();//输出excel文件到磁盘
 
 //单元格插入文本
@@ -75,14 +65,17 @@ $execute_time = time() - $time . 's';
 echo json_encode(['code' => 1, 'msg' => '导出完毕', 'url' => '/download.php?file=' . $filePath, 'data' => ['memory' => $memory, 'excute_time' => $execute_time]]);
 
 //数据生成器--封装模拟数据获取的方法
-function generateData(){
+function generateData()
+{
     $db = DB::getInstance();
     $step = 10000;
-    for ($i = 0; $i < 1000000; $i = $i + $step) {
+    for ($i = 0; $i < 100000; $i = $i + $step) {
         yield $db->get_records_sql("select * from sheet1 limit {$i},{$step}", null, PDO::FETCH_ASSOC);
     }
 }
+
 //格式化字段值
-function myFormat($v,$values){
-    return $v.'自定义格式化-'.$values['id'];
+function myFormat($v, $values)
+{
+    return $v . '自定义格式化-' . $values['id'];
 }

src/Pxlswrite.php

@@ -70,6 +70,9 @@ class Pxlswrite extends Excel
      * @var array
      */
     public $fieldsCallback = [];
+    /**
+     * @var array
+     */
     public $header = [];
 
     /**
@@ -82,7 +85,6 @@ class Pxlswrite extends Excel
             $this->m_config[$k] = $v;
         }
         parent::__construct($this->m_config);
-//        $this->m_instance = new \Vtiful\Kernel\Excel($this->m_config);
     }
 
     /**
@@ -100,6 +102,7 @@ class Pxlswrite extends Excel
      * 设置字段
      * @param $field
      * @return $this
+     * @throws \Exception
      */
     public function field($field)
     {
@@ -117,11 +120,12 @@ class Pxlswrite extends Excel
      * @param $header
      * @param null $format_handle
      * @return mixed
+     * @throws \Exception
      */
     public function header($header, $format_handle = NULL)
     {
-        if(count($header) !== count($header, 1)){
-            echo '数据格式错误,必须是一位数索引数组';exit();
+        if (count($header) !== count($header, 1)) {
+            throw new \Exception('header数据格式错误,必须是一位数索引数组');
         }
         $this->header = $header;
         if ($format_handle) {
@@ -151,12 +155,12 @@ class Pxlswrite extends Excel
     {
         $count = 0;
         //判断是否有定义字段
-        if(!empty($this->fieldsCallback)){
-            foreach (call_user_func($_generator) as $item){
-                foreach ($item as $value){
+        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'])) {
@@ -167,28 +171,29 @@ class Pxlswrite extends Excel
                 }
                 //推送消息
                 $count += count($item);
-                $this->push($_pushHandle,$count);
+                $this->push($_pushHandle, $count);
             }
-        }else{
-            foreach (call_user_func($_generator) as $item){
+        } else {
+            foreach (call_user_func($_generator) as $item) {
                 //循环逐行写入excel
-                foreach ($item as $value){
+                foreach ($item as $value) {
                     $this->data([array_values($value)]);//二维索引数组
                 }
                 //推送消息
                 $count += count($item);
-                $this->push($_pushHandle,$count);
+                $this->push($_pushHandle, $count);
             }
         }
         return $this;
     }
 
     /**
+     * 导入数据
      * @param $_func string 方法名 回调数据插入的方法
      * @param WebSocketClient|null $_pushHandle
      * @param array $_dataType 可指定每个单元格数据类型进行读取
      */
-    public function importData($_func, WebSocketClient $_pushHandle = null,array $_dataType = [])
+    public function importData($_func, WebSocketClient $_pushHandle = null, array $_dataType = [])
     {
         $count = 0;
         //游标读取excel数据 每一万条数据执行一次插入数据库 防止数据装载在内存过大
@@ -197,15 +202,15 @@ class Pxlswrite extends Excel
             $count++;
             if ($count % 10000 == 0) {
                 //回调数据插入的方法
-                call_user_func($_func,$data);
+                call_user_func($_func, $data);
                 //消息推送
-                $this->push($_pushHandle,$count);
+                $this->push($_pushHandle, $count);
                 unset($data);
             }
         }
         if (!empty($data)) {
-            call_user_func($_func,$data);
-            $this->push($_pushHandle,$count);
+            call_user_func($_func, $data);
+            $this->push($_pushHandle, $count);
         }
     }
 
@@ -214,23 +219,29 @@ class Pxlswrite extends Excel
      * @param $_pushHandle
      * @param $count
      */
-    public function push($_pushHandle,$count){
-        if ($_pushHandle && $_pushHandle->m_receiverFd) {
-            $_pushHandle->send(['status' => 'processing', 'process' => $count]);
+    public function push($_pushHandle, $count)
+    {
+        try {
+            if ($_pushHandle && $_pushHandle->m_receiverFd) {
+                $_pushHandle->send(['status' => 'processing', 'process' => $count]);
+            }
+        } catch (\Exception $exception) {
+            $this->writeLog($exception->getMessage(), [$exception->getTraceAsString()]);
         }
+
     }
+
     /**
      * 文件下载
      * @param $_filePath 文件绝对路径
      * @param bool $_isDelete 下载后是否删除原文件
-     * @return string
+     * @throws \Exception
      */
     public function download($_filePath, $_isDelete = true)
     {
 //        setcookie("loadingFlag",1);
         if (dirname($_filePath) != $this->m_config['path']) {
-            echo '未知文件路径';
-            exit();
+            throw new \Exception('未知文件路径');
         }
         header("Content-Type: application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
         header('Content-Disposition: attachment;filename="' . end(explode('/', $_filePath)) . '"');
@@ -275,12 +286,110 @@ class Pxlswrite extends Excel
         return $data;
     }
 
+    public function writeLog($_message, array $_arr)
+    {
+        $dir = rtrim($this->m_config['path'], '/') . '/log/';
+        if (!is_dir($dir)) {
+            mkdir($dir);
+        }
+        $time = date('Y-m-d H:i:s');
+        file_put_contents($dir . date("Y-m-d") . "_error.log", "[{$time}] " . $_message . PHP_EOL . serialize($_arr) . PHP_EOL, FILE_APPEND);
+    }
+
     /**
-     * 样式容器
-     * @return Format
+     * 格式化样式
+     * @param array $_style 样式列表数组
+     * @return Format resource
+     * @throws \Exception
      */
-    public function styleFormat()
+    public function styleFormat($_style)
     {
-        return new Format($this->getHandle());
+        $format = new Format($this->getHandle());
+        $_style = empty($_style) ? [] : $_style;
+        foreach ($_style as $key => $value) {
+            switch ($key) {
+                case 'align':
+                    if (!is_array($value) || count($value) != 2) {
+                        throw new \Exception('align 数据格式错误');
+                    }
+                    $format->align($value[0], $value[1]);
+                    break;
+                default:
+                    if (is_bool($value)) {
+                        if ($value === true) {
+                            $format->$key();
+                        }
+                    } else {
+                        $format->$key($value);
+                    }
+            }
+        }
+        return $format->toResource();
+    }
+
+    /**
+     * 行单元格样式
+     * @param $range string 单元格范围
+     * @param $height double 单元格高度
+     * @param null $formatHandler resource|array 单元格样式
+     * @return $this
+     * @throws \Exception
+     */
+    public function setRow($range, $height, $formatHandler = null)
+    {
+        if (!is_resource($formatHandler)) {
+            $formatHandler = $this->styleFormat($formatHandler);
+        }
+        parent::setRow($range, $height, $formatHandler);
+        return $this;
+    }
+
+    /**
+     * 列单元格样式
+     * @param $range string 单元格范围
+     * @param $width double 单元格宽度
+     * @param null $formatHandler resource|array 单元格样式
+     * @return $this
+     * @throws \Exception
+     */
+    public function setColumn($range, $width, $formatHandler = null)
+    {
+        if (!is_resource($formatHandler)) {
+            $formatHandler = $this->styleFormat($formatHandler);
+        }
+        parent::setColumn($range, $width, $formatHandler);
+        return $this;
+    }
+
+    /**
+     * 合并单元格
+     * @param $scope  string 单元格范围
+     * @param $data   string data
+     * @param null $formatHandler resource|array style
+     * @return $this
+     * @throws \Exception
+     */
+    public function mergeCells($scope, $data, $formatHandler = null)
+    {
+        if (!is_resource($formatHandler)) {
+            $formatHandler = $this->styleFormat($formatHandler);
+        }
+        parent::mergeCells($scope, $data, $formatHandler);
+        return $this;
+    }
+
+    /**
+     * 全局默认样式
+     * @param $formatHandler resource|array style
+     * @return $this
+     * @throws \Exception
+     */
+    public function defaultFormat($formatHandler)
+    {
+        if (!is_resource($formatHandler)) {
+            $formatHandler = $this->styleFormat($formatHandler);
+        }
+        parent::defaultFormat($formatHandler);
+        return $this;
     }
 }