Golang如何实现简单的API网关(2)-木庄网络博客

当前第2页返回上一页

在NewSingleHostReverseProxy中源码已经对传入的URLs进行解析并且完成了Director的修改，我们只需要调用NewSingleHostReverseProxy函数并且传入目标服务器的URL即可，一个简单的反向代理就完成了啦。

代码

实例代码只涉及微服务中 user 与 auth模块，可以根据实际需求自行修改部分

package mainimport (
    "fmt"
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
    "strings")type handle struct {
    host string
    port string}type Service struct {
    auth *handle
    user *handle}func (this *Service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    var remote *url.URL if strings.Contains(r.RequestURI, "api/auth") {
        remote, _ = url.Parse("http://" + this.auth.host + ":" + this.auth.port)
    } else if strings.Contains(r.RequestURI, "api/user") {
        remote, _ = url.Parse("http://" + this.user.host + ":" + this.user.port)
    } else {
        fmt.Fprintf(w, "404 Not Found")
        return
    }
    proxy := httputil.NewSingleHostReverseProxy(remote)
    proxy.ServeHTTP(w, r)}func startServer() {
    // 注册被代理的服务器 (host， port)
    service := &Service{
        auth: &handle{host: "127.0.0.1", port: "8081"},
        user: &handle{host: "127.0.0.1", port: "8082"},
    }
    err := http.ListenAndServe(":8888", service)
    if err != nil {
        log.Fatalln("ListenAndServe: ", err)
    }}func main() {
    startServer()}