net: add WriteHeader to flatend microservice api

lithdew · Jun 15, 2020 · 210581c · 210581c
1 parent 7e98524
commit 210581c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/net.go b/net.go
@@ -39,6 +39,10 @@ type Context struct {
 	written bool              // written before?
 }
 
+func (c *Context) WriteHeader(key, val string) {
+	c.headers[key] = val
+}
+
 func (c *Context) Write(data []byte) (int, error) {
 	if !c.written {
 		packet := ServiceResponsePacket{
@@ -97,6 +101,9 @@ func acquireContext(headers map[string]string, body io.ReadCloser, id uint32, co
 
 func releaseContext(ctx *Context) {
 	ctx.written = false
+	for key := range ctx.headers {
+		delete(ctx.headers, key)
+	}
 	contextPool.Put(ctx)
 }
 

diff --git a/nodejs/src/index.ts b/nodejs/src/index.ts
@@ -5,7 +5,7 @@ async function main() {
     const node = new Node();
 
     node.register("get_todos", (ctx: Context) => ctx.json(ctx.headers));
-    node.register("all_todos", (ctx: Context) => fs.createReadStream("tsconfig.json").pipe(ctx));
+    node.register("all_todos", (ctx: Context) => fs.createReadStream("package.json").pipe(ctx.header('content-type', 'application/json')));
 
     node.register('pipe', async (ctx: Context) => {
         const body = await ctx.body({limit: 65536});