使用GO编译wasm文件并在nodejs中使用
使用GO编译wasm文件并在nodejs中使用
安装Go相关环境
# 安装GO
# mac使用homebrew安装
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
brew install go# vi ~/.bashrc, 添加如下内容
export GOPATH=$HOME/go
export PATH=$PATH:/usr/local/go/bin:$GOPATH/bin# 确认安装成功
source ~/.bashrc
go version// 取消联网下载库
export GO111MODULE=off # 安装tinygo
brew tap tinygo-org/tools
brew install tinygo
获取wasm_exec.js
wasm_exec.js是Go WebAssembly相关的JavaScript文件,用于加载和初始化WebAssembly运行时环境,创建一个全局的GO对象,已经JavaScript和WebAssembly之间进行通讯该文件在GO和TinyGO中使用的略有不同, 使用不同方式打的包需要使用对应的wasm_exec.js文件,安装对应的包后可以以下位置找到它们,将其复制到项目目录下 GO:$(go env GOROOT)/misc/wasm/wasm_exec.jsTinyGo:$(tinygo env TINYGOROOT)/targets/wasm_exec.js
斐波那契数计算的n种方法
1. 原生JS直接计算
// 执行方式: node fibonacci.js// 原生JS环境运行
async function calcByJS(calcNum) {console.log('-----------原生JS运行环境-----------');console.time("原生JS耗时");const resultJS = calcFibonacci(calcNum);// console.log('原生JS计算结果:', resultJS);console.timeEnd("原生JS耗时");console.log('-----------原生JS运行环境-----------\n');}
// JS计算斐波那契函数
function calcFibonacci(n) {if (n == 0) {return 0;} else if (n == 1) {return 1;} else {return calcFibonacci(n - 1) + calcFibonacci(n - 2);}
}async function main() {const calcNum = 40await calcByPureGo(calcNum)
}main()// 执行效率: 计算fibonacci(40)耗时: 约1350ms
2. 原生GO直接计算
// 运行方式: go run fibonacci.go
package mainimport ("fmt""time"
)func fibonacci(n uint) uint {if n == 0 {return 0} else if n == 1 {return 1} else {return fibonacci(n-1) + fibonacci(n-2)}
}func main() {start := time.Now()n := uint(40)fibonacci(n)fmt.Println("纯GO运算时间:", time.Since(start).Milliseconds(),"ms")
}// 执行效率: 计算fibonacci(40)耗时: 约470ms
3. GO打包成WASM包由JS进行调用
- GO文件
package mainimport ("syscall/js""fmt"// "time"
)//export fibonacci
func fibonacci(this js.Value, args []js.Value) interface{} {n := args[0].Int()result := calculateFibonacci(n)return js.ValueOf(result)
}func calculateFibonacci(n int) int {if n <= 1 {return n}return calculateFibonacci(n-1) + calculateFibonacci(n-2)
}func main() {// 注册fibonacci方法js.Global().Set("fibonacci", js.FuncOf(fibonacci))// 调用fibonacci方法// start := time.Now()// n := int(40)// fibonacci(js.Undefined(), []js.Value{js.ValueOf(n)})// fmt.Printf("纯GO耗时:%vms \n", time.Since(start).Milliseconds())fmt.Println("fibonacci方法注册成功")// 阻塞住 goroutinedone := make(chan struct{}, 0)<-done
}
-
打包命令:
GOOS=js GOARCH=wasm go build -o ./fibonacci.wasm fibonacci.go
-
调用was_exec.js文件完成WebAssembly运行时环境初始化
// Copyright 2021 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file."use strict";globalThis.require = require; globalThis.fs = require("fs"); globalThis.TextEncoder = require("util").TextEncoder; globalThis.TextDecoder = require("util").TextDecoder;globalThis.performance = {now() {const [sec, nsec] = process.hrtime();return sec * 1000 + nsec / 1000000;}, };const crypto = require("crypto"); const path = require("path") globalThis.crypto = {getRandomValues(b) {crypto.randomFillSync(b);}, };require("./wasm_exec");// 加载Go编写的wasm文件 async function getWasmOfGo(wasmPath) {const go = new Go();go.argv = [wasmPath];go.env = Object.assign({ TMPDIR: require("os").tmpdir() }, process.env);go.exit = process.exit;const { instance } = await WebAssembly.instantiate(fs.readFileSync(path.join(__dirname, wasmPath)), go.importObject);// 执行 Go WebAssembly 实例go.run(instance) }module.exports = getWasmOfGo
-
JS调用wasm文件执行
// node fibonacci.jsconst getWasmOfGo = require('./go_init')// GO计算斐波那契函数 async function calcByGo(num) {console.log('-----------Go打包成wasm包运行-----------');const wasmPath = 'fibonacci.wasm';console.time('GO-wasm总耗时')console.time('wasm初始化总耗时')await getWasmOfGo(wasmPath);console.timeEnd('wasm初始化总耗时')// 调用导出的函数并传递参数console.time('GO-wasm计算耗时')const result = fibonacci(num);// console.log('GO计算结果1', result);console.timeEnd('GO-wasm计算耗时');console.timeEnd('GO-wasm总耗时')console.log('-----------Go打包成wasm包运行-----------\n'); } calcByGo(40)// module.exports = calcByGo-----------Go打包成wasm包运行----------- fibonacci方法注册成功 wasm初始化总耗时: 436.300ms GO-wasm计算耗时: 2341.956ms GO-wasm总耗时: 2778.486ms -----------Go打包成wasm包运行-----------
4. TinyGo打包成wasm包运行
-
Go文件
// tinygo_fibonacci.gopackage mainimport "syscall/js"func main() {// 包装导出的函数作为闭包,并传递给js.FuncOffibFunc := js.FuncOf(func(this js.Value, args []js.Value) interface{} {n := args[0].Int()result := fibonacci(n)return result})// 将包装后的函数赋值给全局对象的某个属性(例如window)js.Global().Set("fibonacci", fibFunc)select {} }// Fibonacci 函数 func fibonacci(n int) int {if n <= 1 {return n}return fibonacci(n-1) + fibonacci(n-2) }
-
打包命令
tinygo build -o tinygo_fibonacci.wasm -target wasm tinygo_fibonacci.go
-
调用was_exec.js文件完成WebAssembly运行时环境初始化
// Copyright 2021 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file."use strict";globalThis.require = require; globalThis.fs = require("fs"); globalThis.TextEncoder = require("util").TextEncoder; globalThis.TextDecoder = require("util").TextDecoder;globalThis.performance = {now() {const [sec, nsec] = process.hrtime();return sec * 1000 + nsec / 1000000;}, };const crypto = require("crypto"); const path = require("path") globalThis.crypto = {getRandomValues(b) {crypto.randomFillSync(b);}, };require("./wasm_exec_tiny");// 加载Go编写的wasm文件 async function getWasmOfGo(wasmPath) {const go = new Go();go.argv = [wasmPath];go.env = Object.assign({ TMPDIR: require("os").tmpdir() }, process.env);go.exit = process.exit;const { instance } = await WebAssembly.instantiate(fs.readFileSync(path.join(__dirname, wasmPath)), go.importObject);// 执行 Go WebAssembly 实例go.run(instance)}module.exports = getWasmOfGo
-
JS文件
const getWasmOfGo = require('./tinygo_init')// GO计算斐波那契函数 async function calcByTinyGo(num) {const wasmPath = 'tinygo_fibonacci.wasm';console.log('-----------TinyGo打包成wasm包运行-----------');console.time('GO总耗时')console.time('wasm初始化总耗时')await getWasmOfGo(wasmPath);console.timeEnd('wasm初始化总耗时')// 调用导出的函数并传递参数console.time('GO计算耗时')const result = fibonacci(num);// console.log('GO计算结果', result);console.timeEnd('GO计算耗时');console.timeEnd('GO总耗时')console.log('-----------TinyGo打包成wasm包运行-----------'); }module.exports = calcByTinyGo-----------TinyGo打包成wasm包运行----------- wasm初始化总耗时: 19.252ms GO计算耗时: 633.676ms GO总耗时: 653.138ms -----------TinyGo打包成wasm包运行-----------
5. html调用GO打的wasm包
需启动http服务打开才能调用, 可使用vscode 的LiveServer插件
<!DOCTYPE html> <html><head><meta charset="utf-8"><title>TinyGo WebAssembly Demo</title><script src="wasm_exec.js"></script><!-- <script src="wasm_exec_tiny.js"></script> --><script src="//aeu.alicdn.com/waf/interfaceacting220819.js"></script><script src="//aeu.alicdn.com/waf/antidomxss_v640.js"></script><script>const go = new Go();WebAssembly.instantiateStreaming(fetch('fibonacci.wasm'), go.importObject).then(result => {go.run(result.instance);}).catch(err => {console.error(err);});// 在页面上定义fibonacci函数,并调用WebAssembly模块中的fibonacci函数function fibonacci(n) {const wasmFibonacci = window.fibonacci;if (wasmFibonacci === undefined || typeof wasmFibonacci !== 'function') {console.error('fibonacci function not found');return;}return wasmFibonacci(n);}async function test() {console.time('Web计算耗时');const result = await fibonacci(40)console.log(result); console.timeEnd('Web计算耗时');}// JS计算斐波那契函数async function calcByJS(calcNum) {console.time("原生JS耗时");const resultJS = calcFibonacci(calcNum);// console.log('原生JS计算结果:', resultJS);console.timeEnd("原生JS耗时");}function calcFibonacci(n) {if (n == 0) {return 0;} else if (n == 1) {return 1;} else {return calcFibonacci(n - 1) + calcFibonacci(n - 2);}}</script> </head><body><h1>TinyGo WebAssembly Demo</h1><button onclick="test()">Calculate Fibonacci</button><button onclick="calcByJS(40)">Calculate Fibonacci By JS</button> </body></html>GO的wasm:Web计算耗时: 3429.2451171875 ms TinyGO的wasm: Web计算耗时: 963.423095703125 ms 浏览器JS耗时: 原生JS耗时: 1429.368896484375 ms