1行Go代碼實(shí)現(xiàn)反向代理的示例

第1行Go代碼實(shí)現(xiàn)反向代理的示例//Logthetypeformpayloadandredirecturl

funclogRequestPayload(requestionPayloadrequestPayloadStruct,proxyUrlstring){

log.Printf("proxy_condition:%s,proxy_url:%s\n",requestionPayload.ProxyCondition,proxyUrl)

//Gettheurlforagivenproxycondition

funcgetProxyUrl(proxyConditionRawstring)string{

proxyCondition:=strings.ToUpper(proxyConditionRaw)

a_condtion_url:=os.Getenv("A_CONDITION_URL")

b_condtion_url:=os.Getenv("B_CONDITION_URL")

default_condtion_url:=os.Getenv("DEFAULT_CONDITION_URL")

ifproxyCondition=="A"{

returna_condtion_url

ifproxyCondition=="B"{

returnb_condtion_url

returndefault_condtion_url

//Givenarequestsendittotheappropriateurl

funchandleRequestAndRedirect(reshttp.ResponseWriter,req*http.Request){

requestPayload:=parseRequestBody(req)

url:=getProxyUrl(requestPayload.ProxyCondition)

logRequestPayload(requestPayload,url)

//morestilltocome...

}

反向代理到URL

最終我們來到了實(shí)際的反向代理部分。在如此多的語言中要編寫一個(gè)反向代理需要考慮很多東西，寫大段的代碼。或者至少引入一個(gè)復(fù)雜的外部庫。

然而Go的標(biāo)準(zhǔn)庫使得創(chuàng)建一個(gè)反向代理非常簡單以至于你都不敢相信。下面就是你所需要的最關(guān)鍵的一行代碼：

httputil.NewSingleHostReverseProxy(url).ServeHTTP(res,req)

注意下面代碼中我們做了些許修改來讓它能完整地支持SSL重定向（雖然不是必須的）。

//Serveareverseproxyforagivenurl

funcserveReverseProxy(targetstring,reshttp.ResponseWriter,req*http.Request){

//parsetheurl

url,_:=url.Parse(target)

//createthereverseproxy

proxy:=httputil.NewSingleHostReverseProxy(url)

//UpdatetheheaderstoallowforSSLredirection

req.URL.Host=url.Host

req.URL.Scheme=url.Scheme

req.Header.Set("X-Forwarded-Host",req.Header.Get("Host"))

req.Host=url.Host

//NotethatServeHttpisnonblockingandusesagoroutineunderthehood

proxy.ServeHTTP(res,req)

//Givenarequestsendittotheappropriateurl

funchandleRequestAndRedirect(reshttp.ResponseWriter,req*http.Request){

requestPayload:=parseRequestBody(req)

url:=getProxyUrl(requestPayload.ProxyCondition)

logRequestPayload(requestPayload,url)

serveReverseProxy(url,res,req)

}

全部啟動(dòng)

好了，現(xiàn)在啟動(dòng)我們的反向代理程序讓其監(jiān)聽1330端口。讓其他的3個(gè)簡單的服務(wù)分別監(jiān)聽1331–1333端口（在各自的終端中）。

source.envgoinstall$GOPATH/bin/reverse-proxy-demo

http-server-p1331

http-server-p1332

http-server-p1333

這些服務(wù)都啟動(dòng)之后，我們就可以在另一個(gè)終端中像下面這樣開始發(fā)送帶有JSON體的請求了：

curl--requestGET\

--urlhttp://localhost:1330/\

--header'content-type:application/json'\

--data'{

"proxy_condition":"a"

}'

如果你在找一個(gè)好用的HTTP請求客戶端，我極力推薦Insomnia。

然后我們就會(huì)看到我們的反向代理將流量轉(zhuǎn)發(fā)給了我們根據(jù)proxy_condition字段配置的3臺(tái)服務(wù)中的其中一臺(tái)。

總結(jié)

Go為此提供了很多，但真正支撐起它的在于這些低級(jí)的網(wǎng)絡(luò)管道任務(wù)，沒有更好的語言了。我們寫的這個(gè)程序簡單，高性能，可靠并且隨時(shí)可用于生產(chǎn)環(huán)境。

我能看