go做反向代理

后端服务器代码

package main
import (
    "log"
    "net/http"
    "os"
    "os/signal"
)
type webHandler struct {

}
func (webHandler) ServeHTTP (w http.ResponseWriter, request *http.Request ){
    w.Write([]byte("this is web1"))
}
func main(){
    c:=make(chan os.Signal)//信号
    go func() {
        http.ListenAndServe(":9091",webHandler{})
    }()
    signal.Notify(c,os.Interrupt)
    s:=<-c
    log.Println(s)
}

前端代码

package main

import (
    "io/ioutil"
    "log"
    "net/http"
)

type ProxyHandle struct {

}

func (p ProxyHandle) ServeHTTP(w http.ResponseWriter,r *http.Request){
    defer func(){
        if err:= recover();err != nil {
            w.WriteHeader(500)
            log.Println(err)
        }
    }()
    if r.URL.Path == "/a" {
        newreq,_:=http.NewRequest(r.Method,"http://localhost:9091",r.Body)
        req,_:=http.DefaultClient.Do(newreq)
        defer req.Body.Close()
        ss,_:=ioutil.ReadAll(req.Body)
        w.Write(ss)
        return
    }

    w.Write([]byte("<h1>defualt index</h1>"))
}
func main(){
    http.ListenAndServe(":8080",&ProxyHandle{})
}