nginx unit WebAssembly 试用

nginx unit 已经支持WebAssembly ，刚好体验下

环境准备

基于docker 运行unit，对于wasm 的开发基于rust，实际上测试直接试用了官方的示例代码

  • docker-compose
version: "3"
services:
  app:
     image: unit:1.31.1-wasm
     ports:
       - 8080:8080
       - 8000:8000
     volumes:
       - ./app:/www

rust wasm 开发

基于了官方的示例

  • 安装工具
rustup target add wasm32-wasi
  • 初始化项目
cargo init --lib wasm_on_unit
  • 添加依赖
cargo add unit-wasm
  • 核心代码
/* SPDX-License-Identifier: Apache-2.0 */
 
/*
 * Copyright (C) Andrew Clayton
 * Copyright (C) Timo Stark
 * Copyright (C) F5, Inc.
 */
 
use unit_wasm::rusty::*;
 
use std::ffi::CStr;
use std::os::raw::c_char;
use std::os::raw::c_void;
use std::ptr::null_mut;
 
// Buffer of some size to store the copy of the request
static mut REQUEST_BUF: *mut u8 = null_mut();
 
#[no_mangle]
pub unsafe extern "C" fn uwr_module_end_handler() {
    uwr_free(REQUEST_BUF);
}
 
#[no_mangle]
pub unsafe extern "C" fn uwr_module_init_handler() {
    REQUEST_BUF = uwr_malloc(uwr_mem_get_init_size());
}
 
pub extern "C" fn hdr_iter_func(
    ctx: *mut luw_ctx_t,
    name: *const c_char,
    value: *const c_char,
    _data: *mut c_void,
) -> bool {
    uwr_write_str!(ctx, "{} = {}\n", C2S!(name), C2S!(value));
 
    return true;
}
 
#[no_mangle]
pub extern "C" fn uwr_request_handler(addr: *mut u8) -> i32 {
    // Declare a 0-initialised context structure
    let ctx = &mut UWR_CTX_INITIALIZER();
    // Initialise the context structure.
    //
    // addr is the address of the previously allocated memory shared
    // between the module and unit.
    //
    // The response data will be stored @ addr + offset (of 4096 bytes).
    // This will leave some space for the response headers.
    uwr_init_ctx(ctx, addr, 4096);
 
    // Set where we will copy the request into
    uwr_set_req_buf(ctx, unsafe { &mut REQUEST_BUF }, LUW_SRB_NONE);
 
    // Define the Response Body Text.
 
    uwr_write_str!(
        ctx,
        " * Welcome to WebAssembly in Rust on Unit! \
            [libunit-wasm ({}.{}.{}/{:#010x})] *\n\n",
        LUW_VERSION_MAJOR,
        LUW_VERSION_MINOR,
        LUW_VERSION_PATCH,
        LUW_VERSION_NUMBER,
    );
 
    uwr_write_str!(ctx, "[Request Info]\n");
 
    uwr_write_str!(ctx, "REQUEST_PATH = {}\n", uwr_get_http_path(ctx));
    uwr_write_str!(ctx, "METHOD       = {}\n", uwr_get_http_method(ctx));
    uwr_write_str!(ctx, "VERSION      = {}\n", uwr_get_http_version(ctx));
    uwr_write_str!(ctx, "QUERY        = {}\n", uwr_get_http_query(ctx));
    uwr_write_str!(ctx, "REMOTE       = {}\n", uwr_get_http_remote(ctx));
    uwr_write_str!(ctx, "LOCAL_ADDR   = {}\n", uwr_get_http_local_addr(ctx));
    uwr_write_str!(ctx, "LOCAL_PORT   = {}\n", uwr_get_http_local_port(ctx));
    uwr_write_str!(ctx, "SERVER_NAME  = {}\n", uwr_get_http_server_name(ctx));
 
    uwr_write_str!(ctx, "\n[Request Headers]\n");
 
    uwr_http_hdr_iter(ctx, Some(hdr_iter_func), null_mut());
 
    let method = uwr_get_http_method(ctx);
    if method == "POST" || method == "PUT" {
        uwr_write_str!(ctx, "\n[{} data]\n", method);
        uwr_mem_write_buf(
            ctx,
            uwr_get_http_content(ctx),
            uwr_get_http_content_len(ctx),
        );
        uwr_write_str!(ctx, "\n");
    }
 
    // Init Response Headers
    //
    // Needs the context, number of headers about to add as well as
    // the offset where to store the headers. In this case we are
    // storing the response headers at the beginning of our shared
    // memory at offset 0.
    uwr_http_init_headers(ctx, 2, 0);
    uwr_http_add_header_content_type(ctx, "text/plain");
    uwr_http_add_header_content_len(ctx);
 
    // This calls nxt_wasm_send_headers() in Unit
    uwr_http_send_headers(ctx);
 
    // This calls nxt_wasm_send_response() in Unit
    uwr_http_send_response(ctx);
 
    // This calls nxt_wasm_response_end() in Unit
    uwr_http_response_end();
 
    return 0;
}

启动&效果

  • 启动
docker-compose up -d
  • 配置
    在容器内部运行
 
curl -X PUT --data-binary '{
      "listeners": {
          "0.0.0.0:8080": {
              "pass": "applications/wasm"
          }
      },
 
      "applications": {
          "wasm": {
              "type": "wasm",
              "module": "/www/wasm_on_unit.wasm",
              "request_handler": "uwr_request_handler",
              "malloc_handler": "luw_malloc_handler",
              "free_handler": "luw_free_handler",
              "module_init_handler": "uwr_module_init_handler",
              "module_end_handler": "uwr_module_end_handler"
          }
      }
  }' --unix-socket /var/run/control.unit.sock http://localhost/config/
  • 效果

 

说明

目前api7也提供了一个nginx 的扩展应该也是可以运行wasm的，但是还没测试，社区也有一个修改版本的nginx 可以支持（但是已经不维护了）

参考资料

https://unit.nginx.org/configuration/#configuration-wasm
https://unit.nginx.org/howto/samples/#sample-wasm
https://github.com/WebAssembly/wasi-sdk
https://github.com/rongfengliang/nginx-unit-wasm-learning
https://github.com/api7/wasm-nginx-module
https://syrusakbary.medium.com/running-nginx-with-webassembly-6353c02c08ac
https://github.com/nginx/unit/blob/fb33ec86a3b6ca6a844dfa6980bb9e083094abec/pkg/docker/Dockerfile.wasm

posted on 2023-10-24 11:46  荣锋亮  阅读(69)  评论(0)  编辑  收藏  举报

导航