1行Go代碼實(shí)現(xiàn)反向代理的示例
暫且放下你的編程語(yǔ)言來(lái)瞻仰下我所見(jiàn)過(guò)的最棒的標(biāo)準(zhǔn)庫(kù)。
為項(xiàng)目選擇編程語(yǔ)言和挑選你最愛(ài)的球隊(duì)不一樣。應(yīng)該從實(shí)用主義出發(fā),根據(jù)特定的工作選擇合適的工具。
在這篇文章中我會(huì)告訴你從何時(shí)開(kāi)始并且為什么我認(rèn)為 Go 語(yǔ)言如此閃耀,具體來(lái)說(shuō)是它的標(biāo)準(zhǔn)庫(kù)對(duì)于基本的網(wǎng)絡(luò)編程來(lái)說(shuō)顯得非常穩(wěn)固。更具體一點(diǎn),我們將要編寫(xiě)一個(gè)反向代理程序。
Go 為此提供了很多,但真正支撐起它的在于這些低級(jí)的網(wǎng)絡(luò)管道任務(wù),沒(méi)有更好的語(yǔ)言了。
反向代理是什么? 有個(gè)很棒的說(shuō)法是流量轉(zhuǎn)發(fā) 。我獲取到客戶端來(lái)的請(qǐng)求,將它發(fā)往另一個(gè)服務(wù)器,從服務(wù)器獲取到響應(yīng)再回給原先的客戶端。反向的意義簡(jiǎn)單來(lái)說(shuō)在于這個(gè)代理自身決定了何時(shí)將流量發(fā)往何處。
為什么這很有用?因?yàn)榉聪虼淼母拍钍侨绱撕?jiǎn)單以至于它可以被應(yīng)用于許多不同的場(chǎng)景:負(fù)載均衡,A/B 測(cè)試,高速緩存,驗(yàn)證等等。
當(dāng)讀完這篇文章之后,你會(huì)學(xué)到:
- 如何響應(yīng) HTTP 請(qǐng)求
- 如何解析請(qǐng)求體
- 如何通過(guò)反向代理將流量轉(zhuǎn)發(fā)到另一臺(tái)服務(wù)器
我們的反向代理項(xiàng)目
我們來(lái)實(shí)際寫(xiě)一下項(xiàng)目。我們需要一個(gè) Web 服務(wù)器能夠提供以下功能:
- 獲取到請(qǐng)求
- 讀取請(qǐng)求體,特別是 proxy_condition 字段
- 如果代理域?yàn)?A,則轉(zhuǎn)發(fā)到 URL 1
- 如果代理域?yàn)?B,則轉(zhuǎn)發(fā)到 URL 2
- 如果代理域都不是以上,則轉(zhuǎn)發(fā)到默認(rèn)的 URL
準(zhǔn)備工作
- Go 語(yǔ)言環(huán)境。
- http-server 用來(lái)創(chuàng)建簡(jiǎn)單的服務(wù)。
環(huán)境配置
我們要做的第一件事是將我們的配置信息寫(xiě)入環(huán)境變量,如此就可以使用它們而不必寫(xiě)死在我們的源代碼中。
我發(fā)現(xiàn)最好的方式是創(chuàng)建一個(gè)包含所需環(huán)境變量的 .env
文件。
以下就是我為特定項(xiàng)目編寫(xiě)的文件內(nèi)容:
export PORT=1330 export A_CONDITION_URL="http://localhost:1331" export B_CONDITION_URL="http://localhost:1332" export DEFAULT_CONDITION_URL=http://localhost:1333
這是我從 12 Factor App 項(xiàng)目中獲得的技巧。
保存完 .env
文件之后就可以運(yùn)行:
source .env
在任何時(shí)候都可以運(yùn)行該指令來(lái)將配置加載進(jìn)環(huán)境變量。
項(xiàng)目基礎(chǔ)工作
接著我們創(chuàng)建 main.go
文件做如下事情:
- 將
PORT
,A_CONDITION_URL
,B_CONDITION_URL
和DEFAULT_CONDITION_URL
變量通過(guò)日志打印到控制臺(tái)。 - 在
/
路徑上監(jiān)聽(tīng)請(qǐng)求:
package main import ( "bytes" "encoding/json" "io/ioutil" "log" "net/http" "net/http/httputil" "net/url" "os" "strings" ) // Get env var or default func getEnv(key, fallback string) string { if value, ok := os.LookupEnv(key); ok { return value } return fallback } // Get the port to listen on func getListenAddress() string { port := getEnv("PORT", "1338") return ":" + port } // Log the env variables required for a reverse proxy func logSetup() { a_condtion_url := os.Getenv("A_CONDITION_URL") b_condtion_url := os.Getenv("B_CONDITION_URL") default_condtion_url := os.Getenv("DEFAULT_CONDITION_URL") log.Printf("Server will run on: %s\n", getListenAddress()) log.Printf("Redirecting to A url: %s\n", a_condtion_url) log.Printf("Redirecting to B url: %s\n", b_condtion_url) log.Printf("Redirecting to Default url: %s\n", default_condtion_url) } // Given a request send it to the appropriate url func handleRequestAndRedirect(res http.ResponseWriter, req *http.Request) { // We will get to this... } func main() { // Log setup values logSetup() // start server http.HandleFunc("/", handleRequestAndRedirect) if err := http.ListenAndServe(getListenAddress(), nil); err != nil { panic(err) } }
現(xiàn)在你就可以運(yùn)行代碼了。
解析請(qǐng)求體
有了項(xiàng)目的基本骨架之后,我們需要添加邏輯來(lái)處理解析請(qǐng)求的請(qǐng)求體部分。更新 handleRequestAndRedirect
函數(shù)來(lái)從請(qǐng)求體中解析出 proxy_condition
字段。
type requestPayloadStruct struct { ProxyCondition string `json:"proxy_condition"` } // Get a json decoder for a given requests body func requestBodyDecoder(request *http.Request) *json.Decoder { // Read body to buffer body, err := ioutil.ReadAll(request.Body) if err != nil { log.Printf("Error reading body: %v", err) panic(err) } // Because go lang is a pain in the ass if you read the body then any susequent calls // are unable to read the body again.... request.Body = ioutil.NopCloser(bytes.NewBuffer(body)) return json.NewDecoder(ioutil.NopCloser(bytes.NewBuffer(body))) } // Parse the requests body func parseRequestBody(request *http.Request) requestPayloadStruct { decoder := requestBodyDecoder(request) var requestPayload requestPayloadStruct err := decoder.Decode(&requestPayload) if err != nil { panic(err) } return requestPayload } // Given a request send it to the appropriate url func handleRequestAndRedirect(res http.ResponseWriter, req *http.Request) { requestPayload := parseRequestBody(req) // ... more to come }
通過(guò) proxy_condition 判斷將流量發(fā)往何處
現(xiàn)在我們從請(qǐng)求中取得了 proxy_condition
的值,可以根據(jù)它來(lái)判斷我們要反向代理到何處。記住上文我們提到的三種情形:
- 如果
proxy_condition
值為A
,我們將流量發(fā)送到A_CONDITION_URL
- 如果
proxy_condition
值為B
,我們將流量發(fā)送到B_CONDITION_URL
- 其他情況將流量發(fā)送到
DEFAULT_CONDITION_URL
// Log the typeform payload and redirect url func logRequestPayload(requestionPayload requestPayloadStruct, proxyUrl string) { log.Printf("proxy_condition: %s, proxy_url: %s\n", requestionPayload.ProxyCondition, proxyUrl) } // Get the url for a given proxy condition func getProxyUrl(proxyConditionRaw string) 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") if proxyCondition == "A" { return a_condtion_url } if proxyCondition == "B" { return b_condtion_url } return default_condtion_url } // Given a request send it to the appropriate url func handleRequestAndRedirect(res http.ResponseWriter, req *http.Request) { requestPayload := parseRequestBody(req) url := getProxyUrl(requestPayload.ProxyCondition) logRequestPayload(requestPayload, url) // more still to come... }
反向代理到 URL
最終我們來(lái)到了實(shí)際的反向代理部分。在如此多的語(yǔ)言中要編寫(xiě)一個(gè)反向代理需要考慮很多東西,寫(xiě)大段的代碼?;蛘咧辽僖胍粋€(gè)復(fù)雜的外部庫(kù)。
然而 Go 的標(biāo)準(zhǔn)庫(kù)使得創(chuàng)建一個(gè)反向代理非常簡(jiǎn)單以至于你都不敢相信。下面就是你所需要的最關(guān)鍵的一行代碼:
httputil.NewSingleHostReverseProxy(url).ServeHTTP(res, req)
注意下面代碼中我們做了些許修改來(lái)讓它能完整地支持 SSL 重定向(雖然不是必須的)。
// Serve a reverse proxy for a given url func serveReverseProxy(target string, res http.ResponseWriter, req *http.Request) { // parse the url url, _ := url.Parse(target) // create the reverse proxy proxy := httputil.NewSingleHostReverseProxy(url) // Update the headers to allow for SSL redirection req.URL.Host = url.Host req.URL.Scheme = url.Scheme req.Header.Set("X-Forwarded-Host", req.Header.Get("Host")) req.Host = url.Host // Note that ServeHttp is non blocking and uses a go routine under the hood proxy.ServeHTTP(res, req) } // Given a request send it to the appropriate url func handleRequestAndRedirect(res http.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)聽(tīng) 1330
端口。讓其他的 3 個(gè)簡(jiǎn)單的服務(wù)分別監(jiān)聽(tīng) 1331–1333
端口(在各自的終端中)。
- source .env && go install && $GOPATH/bin/reverse-proxy-demo
- http-server -p 1331
- http-server -p 1332
- http-server -p 1333
這些服務(wù)都啟動(dòng)之后,我們就可以在另一個(gè)終端中像下面這樣開(kāi)始發(fā)送帶有 JSON 體的請(qǐng)求了:
curl --request GET \ --url http://localhost:1330/ \ --header 'content-type: application/json' \ --data '{ "proxy_condition": "a" }'
如果你在找一個(gè)好用的 HTTP 請(qǐng)求客戶端,我極力推薦 Insomnia 。
然后我們就會(huì)看到我們的反向代理將流量轉(zhuǎn)發(fā)給了我們根據(jù) proxy_condition
字段配置的 3 臺(tái)服務(wù)中的其中一臺(tái)。
總結(jié)
Go 為此提供了很多,但真正支撐起它的在于這些低級(jí)的網(wǎng)絡(luò)管道任務(wù),沒(méi)有更好的語(yǔ)言了。我們寫(xiě)的這個(gè)程序簡(jiǎn)單,高性能,可靠并且隨時(shí)可用于生產(chǎn)環(huán)境。
我能看到在以后我會(huì)經(jīng)常使用 Go 來(lái)編寫(xiě)簡(jiǎn)單的服務(wù)。
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
代碼是開(kāi)源的,你可以在 Github 上找到。 :heart: 在 Twitter 上我只聊關(guān)于編程和遠(yuǎn)程工作相關(guān)的東西。如果關(guān)注我,你不會(huì)后悔的。
相關(guān)文章
Go語(yǔ)言利用Unmarshal解析json字符串的實(shí)現(xiàn)
本文主要介紹了Go語(yǔ)言利用Unmarshal解析json字符串的實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-05-05Golang設(shè)計(jì)模式之工廠方法模式講解和代碼示例
工廠方法是一種創(chuàng)建型設(shè)計(jì)模式, 解決了在不指定具體類的情況下創(chuàng)建產(chǎn)品對(duì)象的問(wèn)題,本文將通過(guò)代碼示例詳細(xì)給大家介紹一下Golang工廠方法模式,感興趣的同學(xué)可以參考一下2023-06-06Go語(yǔ)言題解LeetCode1266訪問(wèn)所有點(diǎn)的最小時(shí)間示例
這篇文章主要為大家介紹了Go語(yǔ)言題解LeetCode1266訪問(wèn)所有點(diǎn)的最小時(shí)間示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-01-01Linux環(huán)境下編譯并運(yùn)行g(shù)o項(xiàng)目的全過(guò)程
Go語(yǔ)言是Google的開(kāi)源編程語(yǔ)言,廣泛應(yīng)用于云計(jì)算、分布式系統(tǒng)開(kāi)發(fā)等領(lǐng)域,在Linux上也有大量的應(yīng)用場(chǎng)景,這篇文章主要給大家介紹了關(guān)于Linux環(huán)境下編譯并運(yùn)行g(shù)o項(xiàng)目的相關(guān)資料,需要的朋友可以參考下2023-11-11淺析Go設(shè)計(jì)模式之Facade(外觀)模式
本文將介紹外觀模式的概念、結(jié)構(gòu)和工作原理,并提供一些在Go中實(shí)現(xiàn)外觀模式的示例代碼,通過(guò)使用外觀模式,可以降低代碼的耦合度,提高代碼的可維護(hù)性和可讀性,需要的朋友可以參考下2023-05-05