feat: upgrade assemblyscript faas demo(#256) (#486)

Author: nanjingboy

demo/faas/code/assemblyscript/README.md

@@ -0,0 +1,8 @@
+# Build
+
+The Node.js version should >=16.14.0.
+
+* npm i
+* npm run asbuild
+* Copy build/function_1.wasm and build/function_2.wasm to your actual directory.
+* Test the demo follow https://mosn.io/layotto/#/zh/start/faas/start.

demo/faas/code/assemblyscript/asconfig.json

@@ -1,20 +1,23 @@
 {
   "targets": {
-    "debug": {
-      "binaryFile": "build/untouched.wasm",
-      "textFile": "build/untouched.wat",
-      "sourceMap": true,
-      "debug": true
+    "client": {
+      "outFile": "build/function_1.wasm",
+      "sourceMap": false,
+      "optimizeLevel": 3,
+      "shrinkLevel": 0,
+      "converge": false,
+      "noAssert": false
     },
-    "release": {
-      "binaryFile": "build/optimized.wasm",
-      "textFile": "build/optimized.wat",
-      "sourceMap": true,
+    "server": {
+      "outFile": "build/function_2.wasm",
+      "sourceMap": false,
       "optimizeLevel": 3,
       "shrinkLevel": 0,
       "converge": false,
       "noAssert": false
     }
   },
-  "options": {}
+  "options": {
+    "bindings": "esm"
+  }
 }

demo/faas/code/assemblyscript/assembly/client.ts

@@ -0,0 +1,59 @@
+import {
+  BufferTypeValues,
+  Context,
+  FilterDataStatusValues,
+  get_buffer_bytes,
+  log,
+  LogLevelValues,
+  registerRootContext,
+  RootContext,
+  set_http_response_body,
+} from "@nobodyiam/proxy-runtime/assembly";
+import { invokeService, registerId } from "./proxy";
+
+export * from "@nobodyiam/proxy-runtime/assembly/proxy";
+
+function getBookName(body: string): string | null {
+  const parts = body.split("&");
+  for (let index = 0; index < parts.length; ++index) {
+    const item = parts[index];
+    if (item.startsWith("name=")) {
+      return item.slice("name=".length);
+    }
+  }
+  return null;
+}
+
+class ClientRootHttpContext extends RootContext {
+  createContext(context_id: u32): Context {
+    return new ClientHttpContext(context_id, this);
+  }
+}
+
+class ClientHttpContext extends Context {
+  constructor(context_id: u32, root_context: ClientRootHttpContext) {
+    super(context_id, root_context);
+  }
+
+  onRequestBody(body_buffer_length: usize, _end_of_stream: bool): FilterDataStatusValues {
+    const name = getBookName(
+      String.UTF8.decode(get_buffer_bytes(BufferTypeValues.HttpRequestBody, 0, body_buffer_length as u32))
+    );
+    if (name === null) {
+      log(LogLevelValues.error, "Param 'name' not found");
+    } else {
+      set_http_response_body(`There are ${String.UTF8.decode(invokeService("id_2", "", name))} inventories for ${name}.\n`);
+    }
+    return FilterDataStatusValues.Continue
+  }
+}
+
+export function _start(): void {
+  registerRootContext((context_id: u32) => {
+    return new ClientRootHttpContext(context_id);
+  }, "");
+}
+
+export function proxy_get_id(): void {
+  registerId("id_1");
+}

demo/faas/code/assemblyscript/assembly/index.ts

@@ -0,0 +1,98 @@
+import * as imports from '@nobodyiam/proxy-runtime/assembly/imports';
+import { free } from '@nobodyiam/proxy-runtime/assembly/malloc';
+import { WasmResultValues, BufferTypeValues, LogLevelValues, log } from "@nobodyiam/proxy-runtime/assembly";
+
+class ArrayBufferReference {
+  private buffer: usize;
+  private size: usize;
+  constructor() {
+  }
+  sizePtr(): usize {
+    return changetype<usize>(this) + offsetof<ArrayBufferReference>("size");
+  }
+  bufferPtr(): usize {
+    return changetype<usize>(this) + offsetof<ArrayBufferReference>("buffer");
+  }
+  // Before calling toArrayBuffer below, you must call out to the host to fill in the values.
+  // toArrayBuffer below **must** be called once and only once.
+  toArrayBuffer(): ArrayBuffer {
+    if (this.size == 0) {
+      return new ArrayBuffer(0);
+    }
+    let array = changetype<ArrayBuffer>(this.buffer);
+    // host code used malloc to allocate this buffer.
+    // release the allocated ptr. array variable will retain it, so it won't be actually free (as it is ref counted).
+    free(this.buffer);
+    // should we return a this sliced up to size?
+    return array;
+  }
+}
+var globalArrayBufferReference = new ArrayBufferReference();
+type ptr<T> = usize;
+
+export function registerId(id: string): WasmResultValues {
+  const idBuffer = String.UTF8.encode(id);
+  const result = imports.proxy_set_buffer_bytes(BufferTypeValues.CallData, 0, id.length,
+      changetype<usize>(idBuffer), idBuffer.byteLength);
+  if (result != WasmResultValues.Ok) {
+    // @ts-ignore
+    log(LogLevelValues.critical, `Unable to set http response body: ${id} with result: ${result}`);
+  }
+  return result;
+}
+
+// @ts-ignore: decorator
+@external("env", "proxy_invoke_service")
+declare function proxy_invoke_service(
+  idPtr: ptr<u8>,
+  idSize: usize,
+  methodPtr: ptr<u8>,
+  messageSize: ptr<usize>,
+  paramPtr: ptr<u8>,
+  paramSize: usize,
+  resultPtr: ptr<ptr<u8>>,
+  resultSize: ptr<usize>,
+): u32;
+
+export function invokeService(id: string, method: string, param: string): ArrayBuffer {
+  const idBuffer = String.UTF8.encode(id);
+  const methodBuffer = String.UTF8.encode(method);
+  const paramBuffer = String.UTF8.encode(param);
+  let result = proxy_invoke_service(
+    changetype<usize>(idBuffer), idBuffer.byteLength,
+    changetype<usize>(methodBuffer), methodBuffer.byteLength,
+    changetype<usize>(paramBuffer), paramBuffer.byteLength,
+    globalArrayBufferReference.bufferPtr(),
+    globalArrayBufferReference.sizePtr(),
+  );
+  if (result == WasmResultValues.Ok) {
+    return globalArrayBufferReference.toArrayBuffer();
+  }
+  return new ArrayBuffer(0);
+}
+
+// @ts-ignore: decorator
+@external("env", "proxy_get_state")
+declare function proxy_get_state(
+  storeNamePtr: ptr<u8>,
+  storeNameSize: usize,
+  keyPtr: ptr<u8>,
+  keySize: ptr<usize>,
+  resultPtr: ptr<ptr<u8>>,
+  resultSize: ptr<usize>,
+): u32;
+
+export function getState(storeName: string, key: string): ArrayBuffer {
+  const storeNameBuffer = String.UTF8.encode(storeName);
+  const keyBuffer = String.UTF8.encode(key);
+  let result = proxy_get_state(
+    changetype<usize>(storeNameBuffer), storeNameBuffer.byteLength,
+    changetype<usize>(keyBuffer), keyBuffer.byteLength,
+    globalArrayBufferReference.bufferPtr(),
+    globalArrayBufferReference.sizePtr(),
+  );
+  if (result == WasmResultValues.Ok) {
+    return globalArrayBufferReference.toArrayBuffer();
+  }
+  return new ArrayBuffer(0);
+}

demo/faas/code/assemblyscript/assembly/proxy.ts

@@ -0,0 +1,40 @@
+import {
+  BufferTypeValues,
+  Context,
+  FilterDataStatusValues,
+  get_buffer_bytes,
+  registerRootContext,
+  RootContext,
+  set_http_response_body,
+} from "@nobodyiam/proxy-runtime/assembly";
+import { getState, registerId } from "./proxy";
+
+export * from "@nobodyiam/proxy-runtime/assembly/proxy";
+
+class ServerRootHttpContext extends RootContext {
+  createContext(context_id: u32): Context {
+    return new ServerHttpContext(context_id, this);
+  }
+}
+
+class ServerHttpContext extends Context {
+  constructor(context_id: u32, root_context: ServerRootHttpContext) {
+    super(context_id, root_context);
+  }
+
+  onRequestBody(body_buffer_length: usize, _end_of_stream: bool): FilterDataStatusValues {
+    let name = String.UTF8.decode(get_buffer_bytes(BufferTypeValues.HttpRequestBody, 0, body_buffer_length as u32));
+    set_http_response_body(String.UTF8.decode(getState("redis", name)));
+    return FilterDataStatusValues.Continue
+  }
+}
+
+export function _start(): void {
+  registerRootContext((context_id: u32) => {
+    return new ServerRootHttpContext(context_id);
+  }, "");
+}
+
+export function proxy_get_id(): void {
+  registerId("id_2");
+}

demo/faas/code/assemblyscript/assembly/server.ts

@@ -0,0 +1,2 @@
+*
+!.gitignore