koa原理简易版实现

最新推荐文章于 2021-12-16 17:28:37 发布

转载最新推荐文章于 2021-12-16 17:28:37 发布 · 234 阅读

CC 4.0 BY-SA版权

原文链接：https://juejin.im/post/5cb97bbf51882532497395bb

本文介绍了一个基于Node.js的自定义服务器框架实现，通过继承EventEmitter类创建了一个名为Loa的类，该类用于处理HTTP请求。文章详细展示了如何使用中间件模式来组织和执行请求处理逻辑，包括中间件注册、请求响应流程以及异步处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

const http = require('http')
const Emitter = require('events')

class Loa extends Emitter {
  constructor() {
    super()
    this.middleware = []
  }
  use(fn) {
    this.middleware.push(fn)
  }

  listen(...args) {
    const server = http.createServer(this.callback())
    server.listen(...args)
  }
  callback() {
    const fn = this.compose(this.middleware)
    return (req, res) => {
        const ctx = this.createContext(req, res)
       return this.handlerRequest(ctx, fn)
    }
  }

  handlerRequest(ctx, fn) {
    return fn(ctx).then(() => this.respond(ctx))
  }

  compose(middleware) {
    return (context, next) => {
       let index = -1
       return dispatch(0)
       function dispatch(i) {
         let fn = middleware[i]
         if (i === middleware.length) fn = next
         if (!fn) return Promise.resolve()
         return Promise.resolve(fn(context, dispatch.bind(null, i + 1)))
       }
    }
  }

  createContext(req, res) {
    const context = Object.create({})
    context.app = this;
    context.req = req;
    context.res = res;
    context.originalUrl = req.url;
    context.state = {};
    return context
  }

  respond (ctx) {
    const {res, body} = ctx
    if (body) {
      return res.end(String(ctx.body))
    }
      return res.end()
  }
}

const loa = new Loa()

loa.use(async (ctx, next) => {
  const aa = await next()
  ctx.body = 'Hello ' + aa
  // ctx.res.end('hello ' + aa)
})
loa.use(async ({req, res}, next) => {
  return new Promise((resolve, reject) => {
    setTimeout(() => {
      resolve('Brolly!')
    }, 2000)
  })
})

loa.listen(8000)
复制代码