Golang

Golang WebAssembly 前端开发 - 2. Gol

2019-01-02  本文已影响25人  Platanuses

函数调用

Golang 标准库中的 syscall/js 包提供了一系列接口。其中 js.Global() 返回一个 js.Value 类型的结构体,它指代 JS 中的全局对象,在浏览器环境中即为 window 对象。可以通过其 Get() 方法获取 window 对象中的字段,也是 js.Value 类型,包括其中的函数对象,并使用其 Invoke() 方法调用 JS 函数。

另一方面,可以使用 js.Value 类型的 Set() 方法向 JS 中注入字段,包括用 js.NewCallback() 封装的 Golang 函数,这样就能在 JS 中调用 Golang 的函数。Golang 函数必须是 func(args []js.Value) 形式的,使用 args 参数接收 JS 调用的参数,且没有返回值。

package main

import (
    "sync"
    "syscall/js"
)

func main() {
    jsGlobal := js.Global()
    goFuncs := jsGlobal.Get("goFuncs")
    goFuncs.Set("goFunc", js.NewCallback(func(args []js.Value) {
        i, s := args[0].Int(), args[1].String()
        i, s = i+2, s+"b"
        res := jsGlobal.Get("jsFunc").Invoke(i, s)
        i, s = res.Get("i").Int(), res.Get("s").String()
        ret := args[2]
        ret.Set("i", i)
        ret.Set("s", s)
    }))

    wg := &sync.WaitGroup{}
    wg.Add(1)
    wg.Wait()
}
<html>
  <head>
    <meta charset="utf-8">
    <script src="wasm_exec.js"></script>
    <script>
      window.goFuncs = {}
      window.jsFunc = (i, s) => {
        return {i: i + 3, s: s + "c"}
      }

      const go = new Go()
      WebAssembly.instantiateStreaming(fetch("go_main.wasm"), go.importObject).
        then(res => {
          go.run(res.instance)
        })

      window.onload = () => {
        document.getElementById("btn").addEventListener("click", event => {
          let ret = {}
          goFuncs.goFunc(1, "a", ret)
          console.dir(ret)
        })
      }
    </script>
  </head>
  <body>
    <input id="btn" type="button" value="go" />
  </body>
</html>

编译、部署、运行方式与上一节相同。从上述例子也能看出 Golang 和 JS 两端的代码运行的生命周期,JS 中的 go.run() 异步执行对应 Golang 模块中的 main()main() 作为 Golang 端的 main loop 在整个页面的生命周期中不能返回,因为后续在 JS 中对该模块中 Golang 函数的调用,会在 main loop 的子协程中执行。

内存访问

除了函数调用的交互,还可以通过内存直接共享数据。

Golang 端使用的内存空间,通过 instance.exports.mem 暴露给 JS 端,这里 instanceWebAssembly.instantiate* 函数实例化 wasm 模块得到的 instance。可以通过 mem 创建 TypedArray,以此在 JS 直接读写 Golang 使用的内存。

下面的例子会在 JS 端打开一个图片文件,显示在页面上,并将文件内容直接写入 Golang 使用的内存,在 Golang 中将图片的色调改变,再回调 JS 端来读取改变之后的图片,并显示在页面上。

package main

import (
    "bytes"
    "image"
    "reflect"
    "sync"
    "syscall/js"
    "unsafe"

    "github.com/anthonynsimon/bild/adjust"
    "github.com/anthonynsimon/bild/imgio"
)

type Ctx struct {
    SetFileArrCb    js.Value
    SetImageToHueCb js.Value
}

func setFile(ctx *Ctx, fileJsArr js.Value, length int) {
    bs := make([]byte, length)
    ptr := (*reflect.SliceHeader)(unsafe.Pointer(&bs)).Data
    ctx.SetFileArrCb.Invoke(fileJsArr, ptr)

    img, _, _ := image.Decode(bytes.NewReader(bs))
    buf := &bytes.Buffer{}
    imgio.JPEGEncoder(93)(buf, adjust.Hue(img, -150))

    bs = buf.Bytes()
    ptr = (*reflect.SliceHeader)(unsafe.Pointer(&bs)).Data
    ctx.SetImageToHueCb.Invoke(ptr, len(bs))
}

func main() {
    jsGlobal := js.Global()
    ctx := &Ctx{
        SetFileArrCb:    jsGlobal.Get("setFileArrCb"),
        SetImageToHueCb: jsGlobal.Get("setImageToHueCb"),
    }

    goFuncs := jsGlobal.Get("goFuncs")
    goFuncs.Set("setFile", js.NewCallback(func(args []js.Value) {
        setFile(ctx, args[0], args[1].Int())
    }))

    wg := &sync.WaitGroup{}
    wg.Add(1)
    wg.Wait()
}
<html>
  <head>
    <meta charset="utf-8">
    <script src="wasm_exec.js"></script>
    <script>
      let goMemArr, fileType

      let setImageToElem = (elemId, dateArr) => {
        document.getElementById(elemId).src = URL.createObjectURL(
          new Blob([dateArr], {"type": fileType}))
      }

      window.setFileArrCb = (fileArr, ptr) => {
        goMemArr.set(fileArr, ptr)
      }
      window.setImageToHueCb = (ptr, len) => {
        setImageToElem("img-hue", goMemArr.slice(ptr, ptr + len))
      }
      window.goFuncs = {}

      const go = new Go()
      WebAssembly.instantiateStreaming(fetch("go_main.wasm"), go.importObject).
        then(res => {
          goMemArr = new Uint8Array(res.instance.exports.mem.buffer)
          go.run(res.instance)
        }
      )

      let onFileSelected = event => {
        let reader = new FileReader()
        let file = event.target.files[0]
        fileType = file.type
        reader.onload = event => {
          let fileArr = new Uint8Array(event.target.result)
          setImageToElem("img-ori", fileArr)
          window.goFuncs.setFile(fileArr, fileArr.length)
        }
        reader.readAsArrayBuffer(file)
      }

      window.onload = () => {
        document.getElementById("file-input").addEventListener("change", onFileSelected)
      }
    </script>
  </head>

  <body>
    <input id="file-input" type="file" />
    <br />
    <image id="img-ori" />
    <br />
    <image id="img-hue" />
  </body>
</html>

放上一张浏览器截图,笔者选择了一张摄影名作《曼联球迷罗玉phone》,可以看到下方的图片色调被调整,球衣的颜色变为城城(队徽为笔者头像)的天蓝色。

screenshot.png
上一篇下一篇

猜你喜欢

热点阅读