让你的函数运行在web worker中

前言

我们都知道JavaScript在浏览器中执行采用的是单线程模型,也就是说,在同一时间下所有的任务都只能在一个线程上完成,只有上一件任务完成才能开始下一件任务。如果javascript的代码计算量太大,执行会耗费很长的时间,会影响了其他任务的执行,严重的还可能阻塞UI线程的渲染,导致页面出现卡顿等情况。而且现在很多的cpu都是多核的,单线程执行会浪费很多cpu的性能。所以浏览器厂商提供的web worker的接口,就是为了提供让javascript代码运行在多个线程的环境。web worker的工作线程和主线程是分开的,两者互不干扰,相互间通过事件接口进行通信。不过web worker提供的接口太原始了,不是很方便我们使用,每次实例化worker之后都要预定义单独的javascript脚本文件,而且还要单独维护一套通信的方法。
所以我们得想一个办法让worker用起来优雅顺手一点,最好是提供一个接口可以像函数一样调用,比如像这样:

  const work = someWorker()
  work.add(function count(n){
    return n + 1
  })
  work.count(1).then(res => {
    console.log(res) // 2
  })

这样看上去是不是比较直观一点,配合asyncawait用起来就非常优雅了,完全屏蔽了js主线程和worker之间通信的细节。

实现方法

其实具体通信的解决方法很简单,我们只需要在函数调用的时候,postMessage要调用的函数和参数到worker里面去,再监听worker返回的结果。worker内部也是一样的道理,监听主线程传过来的消息,再执行相应的函数,用postMessage返回执行结果。

// js主线程
function invoke (method = '', params = []) {
  const promise = new Promise((resolve, reject)=> {
    worker.onmessage = (e) => {
      resolve(JSON.parse(e.data))
    }
    worker.onerror = (e) => {
      reject(e)
    }
  })
  worker.postMessage(JSON.stringify({
    method,
    params
  }))
  return promise
}
// worker
self.onmessage = (e) => {
  const {method, params} = JSON.parse(e).data
  const result = self[method].apply(null, params)
  postMessage(JSON.stringify(result))
}
// self是worker全局环境的引用,和window差不多
// 所以调用self的方法就是调用全局环境下注册的方法

这样我们就实现了一个简单的worker通信模型,只需要在传入worker的js脚本中提前定义好函数,就可以在主线程通过invoke调用函数了。但这和我们的想法还是有点不一样,我们的模型的可以动态地往worker中添加函数,而且函数可以定义在主线程中,这样可以获得更好的灵活性和可维护性。
那么问题来了,实例化worker需要传入js文件的地址,而且这个地址不能是file://开头的,意味着不能访问本地的文件,所以worker的脚本必须加载至网络。那么有没有一种好的方法可以动态生成js代码片段,而且能够包装成worker可以接受的类型呢?
其实是有的,浏览器厂商提供了一个URL的对象,这个对象有一个create​ObjectURL方法,这个方法可以接受一个二进制对象生成URL,所以我们还需要Blob类来生成二进制数据,我们的问题就可以完美解决了。

let funcStr = ''
// 我们把函数名和函数引用以key-value的方式用Map储存起来
for (const [name, func] of methods.entries()) {
  let str = ''
  if (isArrowFunc(func)) {
    str = `;var ${name} = ${Function.prototype.toString.call(func)}`
  } else {
    str = `;${Function.prototype.toString.call(func)}`
  }
  funcStr += str
}

const code = `${code};\n${worker_scheduler}`
const url = URL.createObjectURL(new Blob([code]))
const worker = new Worker(url)

其实原理也很简单,我们通过Function.toString这个方法得到函数的定义,相当于把函数定义复制到了worker脚本。这里需要提醒一下的是,es6箭头函数的函数定义和普通的函数有一定的区别,我们需要分别处理。

const fn1 = () => {}
function fn2 () {}
Function.prototype.toString.call(fn1)
// () => {}
Function.prototype.toString.call(fn2)
// function fn2 () {}

我们可以看到箭头函数的定义没有定义的名字,不过我们可以通过Function.name获取到函数定义名。

fn1.name //  "fn1"
fn2.name //   "fn2"

接下来的东西都很简单了,我们自己在内部维护这样的一套机制,只需要对外暴露addinvoke两个接口就可以让主线程定义的函数跑在worker当中了。
所以我顺手实现了一个库funcwork,内部实现代码只要一百多行,对外暴露了3个方法,用起来很方便。

import funcwork form 'funcwork'

const { add, invoke, terminate } = funcwork()

function sayName (name) {
  return `Hello ${name}!`
}

const sayHi () {
  return 'Hi!'
}

async function requestInfo (url, id) {
    return fetch(url, {id})
}

add(sayName, sayHi)

await invoke('sayName', ['naeco'])    // Hello naeco!
await invoke('sayHi')                // Hi!
await invoke('requestInfo', ['api/getUserInfo', 'xxx123456']) //  user info...

// 不用的时候记得销毁
terminate()

大家觉得不错的可以顺手给个star😂😂😂

后续

其实web worker这个东西出现了也挺久了,现在浏览器支持度已经很不错了,但是我发现实际项目还是很少人用到。个人认为主要原因有两个:

  1. 接口不友好
  2. 使用场景有限

针对第一点我们可以自己进行封装,可以让web worker用起来像promise一样顺手。第二点要看我们具体的业务场景了,一些计算量比较大的工作可以尝试交给web worker,比如canvas和图片的计算,服务器轮询和上传文件等等场景。


🤔🤔🤔
我是naecoo,前端打杂工程师,偶尔写写灌水文章...
Github
博客

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,736评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,167评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,442评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,902评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,302评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,573评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,847评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,562评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,260评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,531评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,021评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,367评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,016评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,068评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,827评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,610评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,514评论 2 269

推荐阅读更多精彩内容

  • 概要 64学时 3.5学分 章节安排 电子商务网站概况 HTML5+CSS3 JavaScript Node 电子...
    阿啊阿吖丁阅读 8,618评论 0 3
  • 一、概述 JavaScript 语言采用的是单线程模型,也就是说,所有任务只能在一个线程上完成,一次只能做一件事。...
    零星小雨_c84a阅读 2,373评论 0 2
  • 作者:阮一峰www.ruanyifeng.com/blog/2018/07/web-worker.html 概述 ...
    grain先森阅读 1,033评论 0 1
  • 清晨是闯入我梦的怪兽,灯笼一样的眼睛,告诉我他的名字。长长的舌头舔着我,让我知晓清晨的风,自雪山上来,融化在这惬意...
    应心阅读 134评论 1 0
  • 问君可有归期 未语先有泪滴 心中千言万语 悄然藏于梦里
    兰亭舟阳阅读 176评论 0 2