fasthttp 的性能是标准库的10+ 倍，用来做简单网关、微服务是个很好的选择。
这里是用 fasthttp 来做一个简单的网关，使用场景是应用通过网关访问多个微服务，使微服务的接口不暴露，也可以在网关做更多的认证操作。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package main
import (
"flag"
"log"
"github.com/valyala/fasthttp"
)
var (
proxyAddr string
proxyClient = &fasthttp.HostClient{
IsTLS: false,
Addr: "",
// set other options here if required - most notably timeouts.
// ReadTimeout: 60, // 如果在生产环境启用会出现多次请求现象
}
)
func ReverseProxyHandler(ctx *fasthttp.RequestCtx) {
req := &ctx.Request
resp := &ctx.Response
prepareRequest(req)
if err := proxyClient.Do(req, resp); err != nil {
ctx.Logger().Printf("error when proxying the request: %s", err)
}
postprocessResponse(resp)
}
func prepareRequest(req *fasthttp.Request) {
// do not proxy "Connection" header.
req.Header.Del("Connection")
// strip other unneeded headers.
// alter other request params before sending them to upstream host
req.Header.SetHost(proxyAddr)
}
func postprocessResponse(resp *fasthttp.Response) {
// do not proxy "Connection" header
resp.Header.Del("Connection")
// strip other unneeded headers
// alter other response data if needed
// resp.Header.Set("Access-Control-Allow-Origin", "*")
// resp.Header.Set("Access-Control-Request-Method", "OPTIONS,HEAD,POST")
// resp.Header.Set("Content-Type", "application/json; charset=utf-8")
}
func main() {
port := flag.String("port", "8082", "listen port")
targetAddr := flag.String("target", "api.yourdomain.com", "your server domain")
flag.Parse()
proxyClient.Addr = *targetAddr
log.Println("port:", *port)
log.Println("target:", *targetAddr)
if err := fasthttp.ListenAndServe("localhost:"+*port, ReverseProxyHandler); err != nil {
log.Fatalf("error in fasthttp server: %s", err)
}
}
fasthttp https://github.com/valyala/fasthttp
本文网址: https://golangnote.com/topic/206.html 转摘请注明来源