我是 GoLang 的新手，正在開發(fā)我的第一個(gè) API。我有兩個(gè)端點(diǎn)，我只想對(duì)其中一個(gè)端點(diǎn)進(jìn)行速率限制。我找到了一個(gè)有用的教程來幫助我入門，我的方法基于教程，認(rèn)識(shí)到這種方法將限制我的兩個(gè)端點(diǎn)：var limiter = rate.NewLimiter(rate.Every((1*time.Hour)/3), 1)func limit(next http.Handler) http.Handler { return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) { if limiter.Allow() == false { http.Error(res, http.StatusText(429), http.StatusTooManyRequests) return } next.ServeHTTP(res, req) })}func main() { mux := http.NewServeMux() mux.HandleFunc("/", createNewToken) mux.HandleFunc("/notify", sendPushNotificationToAllTokens) log.Fatal(http.ListenAndServeTLS(":5050", "localhost.crt", "localhost.key", limit(mux)))}我研究了http.Handle 和 http.HandleFunc之間的區(qū)別，天真地認(rèn)為我可以http.HandleFunc 替代http.Handle. 這種方法是完全有缺陷的，因?yàn)橛肋h(yuǎn)不會(huì)執(zhí)行中包含的邏輯HandlerFunc：var limiter = rate.NewLimiter(rate.Every(1*time.Hour/3), 1)func limit(next http.HandlerFunc) http.HandlerFunc { return func(res http.ResponseWriter, req *http.Request) { if limiter.Allow() == false { http.Error(res, http.StatusText(429), http.StatusTooManyRequests) return } next.ServeHTTP(res, req) }}func main() { //mux := http.NewServeMux() http.HandleFunc("/", createNewToken) http.HandleFunc("/notify", sendPushNotificationToAllTokens) // attempt to only rate limit the /notify endpoint log.Fatal(http.ListenAndServeTLS(":5050", "localhost.crt", "localhost.key", limit(sendPushNotificationToAllTokens)))誰(shuí)能解釋為什么這不起作用，以及我如何解決這個(gè)問題以僅對(duì)特定端點(diǎn)進(jìn)行速率限制？
1 回答
溫溫醬
TA貢獻(xiàn)1752條經(jīng)驗(yàn) 獲得超4個(gè)贊
使用 plainhttp.Handler和 a之間的區(qū)別http.HanlderFunc在這里并不重要。http.HandleFunc只是一種將常規(guī)函數(shù)轉(zhuǎn)換為 a 的方法http.Handler- 它本質(zhì)上與原始版本的limit.
您對(duì)limit兩者的實(shí)現(xiàn)看起來都不錯(cuò)；可能第二個(gè)更好，因?yàn)樗?jiǎn)單。相反，問題出在main. 當(dāng)您調(diào)用http.ListenAndServeTLS并為最終參數(shù)提供值時(shí)，它會(huì)請(qǐng)求僅將您作為最終參數(shù)傳入的處理程序用作根請(qǐng)求處理程序。除非您作為最后一個(gè)參數(shù)傳入，否則對(duì)http.Handle()或的任何調(diào)用都會(huì)被忽略。http.HandleFunc()nil
您要做的是應(yīng)用于limit您要限制的特定處理程序。為此，您有兩種選擇。ServeMux首先，您可以在第一個(gè)代碼片段中使用like：
func main() {
mux := http.NewServeMux()
mux.HandleFunc("/", createNewToken)
// Limit only the handler for "/notify".
mux.HandleFunc("/notify", limit(sendPushNotificationToAllTokens))
// Don't limit the whole mux.
log.Fatal(http.ListenAndServeTLS(":5050", "localhost.crt", "localhost.key", mux))
}
nil或者，您可以做一些更像您的第二個(gè)代碼片段的事情，但將最后一個(gè)參數(shù)傳遞給http.ListenAndServeTLS以便http.ServeMux使用默認(rèn)值，這意味著http.HandleFunc()將尊重對(duì)的調(diào)用：
func main() {
http.HandleFunc("/", createNewToken)
// Limit only the handler for "/notify".
http.HandleFunc("/notify", limit(sendPushNotificationToAllTokens))
// Pass in nil here so that http.DefaultServeMux is used.
log.Fatal(http.ListenAndServeTLS(":5050", "localhost.crt", "localhost.key", nil))
}
對(duì)于一個(gè)簡(jiǎn)單的應(yīng)用程序，第一種方法可能沒問題。對(duì)于更復(fù)雜的事情，我推薦后一種方法，因?yàn)槿绻蜷_多個(gè)服務(wù)器或做其他更復(fù)雜的事情，它就會(huì)起作用。
- 1 回答
- 0 關(guān)注
- 94 瀏覽
添加回答
舉報(bào)
0/150
提交
取消