美文网首页
rust webAssembly 尝试

rust webAssembly 尝试

作者: songkl | 来源:发表于2020-09-07 10:02 被阅读0次

    最近看到web版的图片处理,想起了以前看到过的关于web端 二维码扫描识别率提升的文章。里面提到了使用webassembly 达到了最终97%以上的准确识别率和很短的时间消耗。于是来尝试一下

    why webassembly?

    webassembly简单来说就是,c/c++/rust等一些可以最大限度发挥性能的语言,编译为字节码,称为wasm格式,可以被浏览器中的wasm虚拟机执行。
    JIT是浏览器端对于js解释执行的优化,将频繁使用的js代码,进行优化和存储为字节码,不用再次解释,减少了运行时间。但是仍依赖于对js代码hot的判断,对于低频高性能需求,就无奈了。

    主流浏览器可支持,目前小程序官方不支持。因此我们得出:

    webassembly == web端极致性能
    

    文档:https://www.wasm.com.cn/

    why rust?

    rust 出来有一段时间了,几年前就听同事推荐过,一直没有机会尝试。既然要性能,那么连gc都没有的rust就很有战斗力了。其次,webassembly是rust官方主推的四大场景之一,支持度自然要更好点。最后就是真的挺感兴趣,slogen也很带感:in rust, we trust.
    文档:https://kaisery.github.io/trpl-zh-cn/

    talk is cheap, show me the code

    图片处理就是调包侠,处理完抛出去就行。值得注意的是,js无法直接传给ws所获得的图片data,要转为Uint8Array。有经验同学们一眼就看出了问题,大量的js-ws交互,会带来很大的序列化成本,做需求时要优化方案,尽量避免这种情况。

    • 生成项目、将rust打包为wasm和包装为可直接引用的js:wasm-pack
    • rust端使用的crate:wasm-bindgen、js-sys、web-sys、image、base64
    • 打包通过webpack:ws不允许直接放入js首包,需要通过import()函数导入,从而拆分为另一个js包。

    我们要实现的demo是将一张彩色图片变为灰度图,使用image这个crate,这个库有灰度函数:grayscale。
    wasm-bindgen 提供 WebAssembly(wasm) 模块和 JavaScript 之间的高级别交互,有很多使用功能,我们这边只是将rust函数抛出。

    #[wasm_bindgen]
    pub fn grayscale(_array: &[u8]) -> Result<(), JsValue> {
        let mut img = load_image_from_array(_array);
        img = img.grayscale();
        let base64_str = get_image_as_base64(img);
        return append_img(base64_str);
    }
    

    web_sys可以让我们获取到dom对象,直接通过rust来操作dom,这远比来回序列化要方便。

    pub fn append_img(image_src: String) -> Result<(), JsValue> {
        let window = web_sys::window().expect("no global `window` exists");
        let document = window.document().expect("should have a document on window");
        let body = document.body().expect("document should have a body");
        let val = document.create_element("img")?;
        val.set_attribute("src", &image_src)?;
        val.set_attribute("style", "height: 200px")?;
        body.append_child(&val)?;
        Ok(())
    }
    

    最后我们在js文件中的引入,需要通过import()经过webpack编译后会拆分为两个js包

    window.fileImport = function () {
      //获取读取我文件的 File 对象
      var selectedFile = document.getElementById("files").files[0];
      var reader = new FileReader(); // 这是核心, 读取操作就是由它完成.
      reader.readAsArrayBuffer(selectedFile); // 读取文件的内容,也可以读取文件的URL
      reader.onload = function () {
        var uint8Array = new Uint8Array(this.result);
        import('./pkg/rw').then(res => {
            res.grayscale(uint8Array);
        })
      };
    };
    

    wasm-pack 为我们准备好了打包的一切工具,具体步骤有:

    • 确保我们有 Rust 1.30 或打上版本,通过rustup安装wasm32-unknown-unknown目标(target),
    • 用cargo将我们的 Rust 的源 编译为 WebAssembly 的 .wasm二进制文件,
    • 为 Rust 生成的 WebAssembly 使用wasm-bindgen,生成 JavaScript API.

    wasm-pack build后,生成物为:

    pkg/
    ├── package.json
    ├── README.md
    ├── xxx_bg.wasm
    ├── xxx_bg.d.ts
    ├── xxx.d.ts
    ├── xxx_bg.js
    └── xxx.js
    

    我们引入的是xxx.js,其中有引入了xxx_bg.js,由此看来,有点类似头文件和实现文件。

    最后

    现在有了create project的cli:create-wasm-app
    可以很方便创建工程。

    代码详见:https://github.com/silencesdg/rust_webAssembly_demo

    相关文章

      网友评论

          本文标题:rust webAssembly 尝试

          本文链接:https://www.haomeiwen.com/subject/aylkektx.html