context support set & get data

zhmushan · Aug 6, 2020 · 09ca677 · 09ca677
1 parent 3859efe
commit 09ca677
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 0 deletions.
diff --git a/context.ts b/context.ts
@@ -34,6 +34,8 @@ export class Context {
   params: Record<string, string> = {};
   customContext: any;
 
+  #store?: Map<string | symbol, unknown>;
+
   get cookies(): Cookies {
     return getCookies(this.request);
   }
@@ -54,6 +56,18 @@ export class Context {
     return params;
   }
 
+  get(key: string | symbol): unknown {
+    return this.#store?.get(key);
+  }
+
+  set(key: string | symbol, val: unknown): void {
+    if (this.#store === undefined) {
+      this.#store = new Map();
+    }
+
+    this.#store.set(key, val);
+  }
+
   constructor(opts: ContextOptions);
   constructor(c: Context);
   constructor(optionsOrContext: ContextOptions | Context) {

diff --git a/context_test.ts b/context_test.ts
@@ -188,3 +188,16 @@ test("context custom", function (): void {
 
   assertEquals(cc.hello(), "hello");
 });
+
+test("context get set", function (): void {
+  const c = new Context({ app: undefined!, r: createMockRequest() });
+
+  c.set("Hello", "World");
+  assertEquals(c.get("hello"), undefined);
+  assertEquals(c.get("Hello"), "World");
+
+  const key = Symbol("Hello");
+  c.set(key, "World");
+  assertEquals(c.get(Symbol("Hello")), undefined);
+  assertEquals(c.get(key), "World");
+});
diff --git a/docs/context.md b/docs/context.md
@@ -1,5 +1,18 @@
 ## Context
 
+### Data sharing
+
+```ts
+app.use((next) => (c) => {
+  c.set("Name", "Mu Shan");
+  return next(c);
+});
+
+app.get("/", (c) => {
+  return `Hello ${c.get("Name")}`;
+});
+```
+
 ### Use a custom context
 
 ```ts