Project Icon

wasm_nvim

Neovim插件开发的WebAssembly接口库

wasm_nvim是Neovim的WebAssembly接口库,连接Lua和WebAssembly。可使用任何支持WebAssembly的语言开发插件,维持单进程运行。测试显示性能远超LuaJIT。库提供内存管理、函数导出和API交互功能,有利于开发高效Neovim插件。

:fire:WASM_NVIM:zap:

目标:

编写一个库来实现Lua和wasm之间的接口,以实现插件和Neovim API之间的通信。该库使用Rust语言编写,通过Neovim API使用Lua动态加载,而不是通过RPC路由,这种方式会在两个或多个不同进程之间造成网络瓶颈。这种方法允许单一进程**(NEOVIM)**,同时还允许插件生态系统使用任何可以编译为wasm的编程语言。

性能:

  • 第一个测试是一个简单的性能测试实现,模仿了Bram的实现,仅在for循环中对数字进行求和。wasm比LuaJIT快99%,这主要是由于编程语言(在这种情况下是zig)的编译器优化。这主要展示了编译为wasm可能带来的编译优化优势,相比于LuaJIT的即时编译。

  • 第二个测试尝试获取0-10000之间的素数,并计算在5秒内可以完成的次数。这里没有常量,因此即使是编译成wasm也不应该像第一个测试那样优化掉几乎所有的操作。在这里,wasm比LuaJIT快99%。我没有为每种特定语言优化算法,而是使用了基本结构,并避免在zig/wasm端使用位操作或栈预分配。

  • 这张图片展示了测试结果:1690628636000

  • 要运行这个测试,只需执行以下命令:

    cargo make perf
    

    要求:

  • 总的来说,wasm在不使用任何特殊优化技术的情况下,平均应该比luajit更快。对于更广泛的luajit vs wasm比较,可以查看这里

注意:当前显示的性能测试结果来自Windows 10 PC。这是我拥有的机器。

阅读:

理论

我们需要在每个wasm模块上实现内存分配和释放函数,这是模块开发者需要自己创建的,因为主机**(这个rust库)**和wasm插件之间目前只能直接共享i32或f32类型的数据,但其他对象如缓冲区、结构体等需要超过32位的数据,因此我们使用内存指针与模块通信,以在主机端访问和操作数据,在适用的情况下,通常涉及JSON。

从发布包安装和设置

-- 从发布包下载共享库,如果你不想从源代码构建
-- 而只想直接使用,可以这样做。警告:仅支持Windows、Linux和macOS
require("wasm_nvim_dl").download("windows")
-- 将"windows"替换为"linux"或"macos"以适应相应的操作系统。

-- 加载wasm_nvim
local wasm = require("wasm_nvim")

-- 下一行扫描并将.wasm文件加载到空间中
-- 应该只调用一次,最好在init.lua中
wasm.setup {
  --debug = true -- 取消注释以查看打印的调试信息,适用于调试问题。
}

-- 这里的tests是一个wasm文件,./wasm文件夹中的test.wasm
-- 位于neovim运行时路径中。
-- luaExecExample是test.wasm模块暴露的函数示例。
wasm.tests.luaExecExample();

从源代码构建

cargo make build

然后将/target/release/中的共享库(.so或.dll)移动到neovim运行时路径的./lua文件夹中以使用它。

要求:

  • Cargo Make cargo install cargo-make

模块要求

  • 一个名为functionality的函数,需要暴露/导出,以便可以从wasm_nvim库调用。该函数返回一个JSON,定义了导出的函数、它们接受的参数以及返回值。

    示例:

    export fn functionality() u32 {
        var functions = ArrayList(Functionality).init(aloc);
        _ = functions.append(CreateFunctionality("hi", "void", "void")) catch undefined;
        var stringified = ArrayList(u8).init(aloc);
        json.stringify(functions.items, .{}, stringified.writer()) catch undefined;
        var unmanaged = stringified.moveToUnmanaged();
        // 获取用于设置值的ID
        const id = get_id();
        const addr = get_addr(&unmanaged.items[0]);
        //设置要作为此函数返回类型使用的值
        set_value(id, addr, unmanaged.items.len);
        return id;
    }
    

    它返回一个ID,映射到由set_value()函数创建的JSON,该JSON指向一个定义模块导出的函数和参数的JSON字符串。在这种情况下,JSON看起来像这样:

    [{"name":"hi", "params": "void", "returns": "void"}]
    
    • Params字段:可以是"void"或其他任何内容,实际上并不重要。当为"void"时,表示由"name"标记的函数不消耗/接受任何参数。

    • Returns字段:与Params字段情况相同,除"void"以外的任何内容都表示函数返回某些内容,否则返回void。

      当wasm模块导出函数时,这些字段是必要的,以便可能调用它们的Lua端能够在函数返回值时获取值,或在函数期望某些值时传递值。

使用自定义接口实现Neovim API以获得更多wasm性能

这部分存在的原因是,我们目前无法直接与需要Lua函数作为引用或回调参数的Neovim API交互,因此用户可能需要将他们的wasm回调包装到Lua函数中,并从那一侧作为解决方案调用neovim api。 注意:如果你遇到需要Lua函数或回调作为参数的neovim api,请向./API.MD文件提交PR,将其添加到列表中,或者你也可以创建一个issue

Wasm_Nvim暴露的可被模块使用的有用函数

以下示例使用zig

  • extern "host" get_id() u32;
    

    获取一个唯一的id,用于在wasm模块和外部世界之间共享数据。

  • extern "host" get_addr(*u8) u32;
    

    从主机获取模块内存中的地址

  • extern "host" get_value_addr(id: u32) [*]u8;
    

    用于获取从模块外部世界创建的值的位置。这里指向的值由模块管理和释放。 注意:一旦调用,该值就会从外部世界的内存中清除,确保在调用get_value_size之前调用此函数,以便在调用此函数之前知道长度

  • extern "host" get_value_size(id: u32) u32;
    

    给定一个值的id,它返回位于给定id的值的大小。应该在get_value_addr之前调用,因为那会从内存中清除id。

  • extern "host" set_value(id: u32, loc: u32, size: u32) void;
    

    用于从wasm模块返回/设置值到wasm_nvim库或外部世界。使用此方法的用户应确保放弃指针所指向的任何内容的内存控制权。

  • extern "host" lua_exec(id: u32) void;
    

    执行通过get_idset_value组合创建并传递给主机的Lua脚本字符串,然后将字符串的id传递给此函数执行。

    注意:在某些neovim版本(例如Windows上的0.9.1)上,这会将加载的字符串打印到输出。

  • extern "host"lua_eval(id: u32) u32;
    

    类似于lua_exec,但这个函数评估表达式并返回映射到id的值。如果评估的表达式没有返回任何内容,返回的值将显示为"null",否则该值将包含结果的内容。

    注意:这是用于表达式的,传递给此函数的任何其他内容可能导致未定义的行为

教程

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号