阅读 2751

使用Web Worker优化代码

前段时间有个需求,需要前端导出excel。一般来说,对于导出大量数据的功能,最好还是交给后端来做,然而后端老哥并不想做(撕逼失败),只能自力更生。

前端导出excel本身已经有很成熟的库了,比如js-xlsx, js-export-excel,所以实现起来并不难。但是,当导出的数据达到几万条时,就会发现页面产生了明显的卡顿。原因也很简单: 一般我们都是基于后端返回的json数据来生成excel,但是后端返回的数据一般都不能直接用来生成数据,我们还需要进行一些格式化:

const list = await request('/api/getExcelData');

const format = list.map((item) => {
  // 对返回的json数据进行格式化
  item.time = moment(item.time).format('YYYY-MM-DD HH:mm');
  // ... 省略其他各种操作
});

// 根据json生成excel
const toExcel = new ExportJsonExcel(format).saveExcel();
复制代码

卡顿就发生在对大量数据进行map操作。由于JS是单线程的,所以在进行大量复杂运算时会独占主线程,导致页面的其他事件无法及时响应,造成页面假死的现象。

那我们能不能把复杂的循环操作单独放在一个线程里呢?这时就要请出web worker

Web Worker

首先看个简单的例子

<button id="btn1">js</button>
<button id="btn2">worker</button>
<input type="text">
复制代码

index.js

const btn1 = document.getElementById('btn1');

btn1.addEventListener('click', function () {
    let total = 1;

    for (let i = 0; i < 5000000000; i++) {
      total += i;
    }
    console.log(total);
})
复制代码

点击btn1时,js会进行大量计算,你会发现页面卡死了,点击input不会有任何反应

我们使用web worker优化代码:

worker.js

onmessage = function(e) {
  if (e.data === 'total') {
    let total = 1;

    for (let i = 0; i < 5000000000; i++) {
      total += i;
    }
    postMessage(total);
  }
}
复制代码

index.js

if (window.Worker) {
  const myWorker = new Worker('worker.js');

  myWorker.onmessage = function (e) {
    console.log('total', e.data);
  };

  const btn1 = document.getElementById('btn1');
  const btn2 = document.getElementById('btn2');

  btn1.addEventListener('click', function () {
    let total = 1;

    for (let i = 0; i < 5000000000; i++) {
      total += i;
    }

    console.log('total', total);
  })

  btn2.addEventListener('click', function () {
    myWorker.postMessage('total');
  });

}
复制代码

点击btn2时,页面并不会卡死,你可以正常的对input进行输入操作

我们开启了一个单独的worker线程来进行复杂操作,通过postMessageonmessage来进行两个线程间的通信。

优化导出excel表格

看过前面的例子,我们可以同理使用web worker进行复杂的map操作

worker.js

onmessage = function(e) {
  const format = e.data.map((item) => {
  // 对返回的json数据进行格式化
  item.time = moment(item.time).format('YYYY-MM-DD HH:mm');
  // ... 省略其他各种操作
});

postMessage(format);
}
复制代码
const myWorker = new Worker('worker.js');

myWorker.onmessage = function (e) {
  // 根据json生成excel
  const toExcel = new ExportJsonExcel(e.data).saveExcel();
};
const list = await request('/api/getExcelData');
myWorker.postMessage(list);
复制代码

当然实际项目,我们一般都是用webpack打包的,这时就要进行一些特别处理,需要使用worker-loader,可以参考《怎么在 ES6+Webpack 下使用 Web Worker》文章学习。

进一步优化

在上面的代码修改中,我们只是优化了业务逻辑里面的map操作。因为我使用的js库是js-export-excel,从它的源码里可以看见,对于我们传进来的数据,它还会再一次forEach循环操作,进行数据的二进制转换。因此,这一步的forEach循环,理论上也可以在web worker里面进行操作。

最简单想到的方法是:

worker.js

onmessage = function(e) {
  const format = e.data.map((item) => {
    // 对返回的json数据进行格式化
    item.time = moment(item.time).format('YYYY-MM-DD HH:mm');
    // ... 省略其他各种操作
  });

  // 直接在worker里面生成excel
  const toExcel = new ExportJsonExcel(format).saveExcel();
}
复制代码

直接在worker.js里面生成excel。然而,saveExcel这个方法需要用到document对象,但是在worker里,我们不能访问类似window document的全局对象。

因此,只能魔改源码了。。。

真正用到document对象的是源码这一句:

// saveAs和Blob用到了document
saveAs(
  new Blob([s2ab(wbout)], {
    type: "application/octet-stream"
  }),
  _options.fileName + ".xlsx"
);
复制代码

saveExcel方法只需改成:

// 不生成excel,只返回数据
return s2ab(wbout);
复制代码

worker.js

onmessage = function(e) {
  const format = e.data.map((item) => {
    // 对返回的json数据进行格式化
    item.time = moment(item.time).format('YYYY-MM-DD HH:mm');
    // ... 省略其他各种操作
  });

  // saveExcel只返回blob数据
  const blob = new ExportJsonExcel(format).saveExcel();
  postMessage(blob);
}
复制代码

index.js

myWorker.onmessage = function (e) {
  // 在主线程生成excel
  saveAs(
    new Blob([e.data], {
      type: "application/octet-stream"
    }),
   "test.xlsx"
  );
};
复制代码

原理就是:我们只把数据转换放在worker里,最后生成excel仍然在主线程里完成。

至此,优化完成了!

总结

我们可以把一些耗性能的操作放在worker线程里(比如大文件上传),这样主线程就能及时响应用户操作而不会造成卡顿现象。需要注意的是,在worker里进行的复杂计算,运行时间并不会变短,有时耗费时间甚至更长,毕竟开启worker也需要消耗一定的性能。

关注下面的标签,发现更多相似文章
评论