koa 利用 node-fetch 寫個自己的代理

2501207950 發(fā)布于2019-08-21 10:47 / 663人閱讀

摘要：在公司的項(xiàng)目中用了向前端還是我提供數(shù)據(jù)接口和渲染頁面。有一些操作是和端交互，所以需要寫一些代理轉(zhuǎn)發(fā)請求，從網(wǎng)上找了一些的代理庫，要不就是橫生要不就是功能不完全，只能代理請求，于是用寫了個簡單的，代碼挺簡單的，寫篇博文記錄下。

在公司的項(xiàng)目中用了 koa 向前端（還是我）提供數(shù)據(jù)接口和渲染頁面。有一些操作是和 Java 端交互，所以需要寫一些代理轉(zhuǎn)發(fā)請求，從網(wǎng)上找了一些koa的代理庫，要不就是bug橫生；要不就是功能不完全，只能代理 get 請求，于是用 node-fetch 寫了個簡單的 proxy ，代碼挺簡單的，寫篇博文記錄下。

用到了 fetch api，可以看 node-fetch

// proxy.js
import fetch from "node-fetch"

export default (...args) => {
  return fetch.apply(null, args).then(function (res) {
    return res.text()
  })
}
}

// 沒錯，就是這么簡單，稍微把fetch封裝下就可以了


// app.js
import koa from "koa"
import proxy from "./proxy"

const app         = koa()
const proxyServer = koa()

app.use(function* (next) {
  if ("/users" === this.path && "GET" === this.method)
    this.body = yield proxy("http://localhost:8087/users")
  if ("/user" === this.path)
    this.body = yield proxy("http://localhost:8087/user", {
      method: "POST",
      headers: {
        "Accept": "application/json",
        "Content-Type": "application/json"
      },
      body: JSON.stringify({
        name: "Hubot",
        login: "hubot",
      })
    })
  yield next
})

proxyServer.use(function* (next) {
  if ("/users" === this.path && "GET" === this.method)
    this.body = {
      "test": 1
    }
  if ("/user" === this.path && "POST" === this.method) {
    this.body= {
      "data" : `yes, it"s right`
    }
  }
  yield next
})

app.listen(8086)
proxyServer.listen(8087)
console.log("server start 8086")
console.log("server start 8087")

上面 app.js 中創(chuàng)建了兩個 server，8086端口為代理 server, 8087端口為被代理的 server，訪問 localhost:8086/users 會返回 {"test": 1}，說明get請求代理成功，同理訪問 localhost:8086/user，會返回
{ "data": "yes, it"s right"}，說明成功代理了post請求并返回了被代理server的結(jié)果。