速率限制特定端点
今天golang学习网给大家带来了《速率限制特定端点》,其中涉及到的知识点包括等等,无论你是小白还是老手,都适合看一看哦~有好的建议也欢迎大家在评论留言,若是看完有所收获,也希望大家能多多点赞支持呀!一起加油学习~
我是 golang 新手,正在开发我的第一个 api。我有两个端点,我只想对其中之一进行速率限制。我找到了一个有用的教程来帮助我入门,并且我的方法基于该教程,认识到这种方法将限制我的两个端点的速率:
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之间的区别,并天真地相信我可以用http.handlefunc
代替http.handle
。这种方法是完全有缺陷的,因为 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)))
任何人都可以解释为什么这不起作用,以及我如何解决这个问题以仅对特定端点进行速率限制?
正确答案
使用普通 http.handler
和 http.hanlderfunc
之间的区别在这里并不重要。 http.handlefunc
只是将常规函数转换为 http.handler
的一种方法 - 它本质上与 limit
的原始版本执行相同的操作。
您的 limit
实现看起来都很好;可能第二个更好,因为它更简单。相反,问题出在 main
中。当您调用 http.listenandservetls
并为最终参数提供值时,它请求仅将您作为最终参数传入的处理程序用作根请求处理程序。除非您传入 nil
作为最后一个参数,否则对 http.handle()
或 http.handlefunc()
的任何调用都将被忽略。
您想要做的是将 limit
应用于您想要限制的特定处理程序。为此,您有两种选择。首先,您可以像第一个代码片段中一样使用 servemux
:
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
作为最后一个参数传递给 http.listenandservetls
,以便使用默认的 http.servemux
,这意味着对 http.handlefunc()
的调用将受到尊重:
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)) }
对于简单的应用程序,第一种方法可能就很好。对于更复杂的事情,我建议使用后面的方法,因为如果您打开多个服务器或执行其他更复杂的事情,它会起作用。
今天关于《速率限制特定端点》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

- 上一篇
- 如何获取特定修订版的所有修改用户

- 下一篇
- 使用 FindAndModify 在官方 Mongo Go 驱动程序中运行查询
-
- Golang · Go问答 | 1年前 |
- 在读取缓冲通道中的内容之前退出
- 139浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 戈兰岛的全球 GOPRIVATE 设置
- 204浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何将结构作为参数传递给 xml-rpc
- 325浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何用golang获得小数点以下两位长度?
- 478浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何通过 client-go 和 golang 检索 Kubernetes 指标
- 486浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 将多个“参数”映射到单个可变参数的习惯用法
- 439浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 将 HTTP 响应正文写入文件后出现 EOF 错误
- 357浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 结构中映射的匿名列表的“复合文字中缺少类型”
- 352浏览 收藏
-
- Golang · Go问答 | 1年前 |
- NATS Jetstream 的性能
- 101浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何将复杂的字符串输入转换为mapstring?
- 440浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 相当于GoLang中Java将Object作为方法参数传递
- 212浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何确保所有 goroutine 在没有 time.Sleep 的情况下终止?
- 143浏览 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 511次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 498次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 484次学习
-
- 千音漫语
- 千音漫语,北京熠声科技倾力打造的智能声音创作助手,提供AI配音、音视频翻译、语音识别、声音克隆等强大功能,助力有声书制作、视频创作、教育培训等领域,官网:https://qianyin123.com
- 190次使用
-
- MiniWork
- MiniWork是一款智能高效的AI工具平台,专为提升工作与学习效率而设计。整合文本处理、图像生成、营销策划及运营管理等多元AI工具,提供精准智能解决方案,让复杂工作简单高效。
- 190次使用
-
- NoCode
- NoCode (nocode.cn)是领先的无代码开发平台,通过拖放、AI对话等简单操作,助您快速创建各类应用、网站与管理系统。无需编程知识,轻松实现个人生活、商业经营、企业管理多场景需求,大幅降低开发门槛,高效低成本。
- 189次使用
-
- 达医智影
- 达医智影,阿里巴巴达摩院医疗AI创新力作。全球率先利用平扫CT实现“一扫多筛”,仅一次CT扫描即可高效识别多种癌症、急症及慢病,为疾病早期发现提供智能、精准的AI影像早筛解决方案。
- 195次使用
-
- 智慧芽Eureka
- 智慧芽Eureka,专为技术创新打造的AI Agent平台。深度理解专利、研发、生物医药、材料、科创等复杂场景,通过专家级AI Agent精准执行任务,智能化工作流解放70%生产力,让您专注核心创新。
- 210次使用
-
- GoLand调式动态执行代码
- 2023-01-13 502浏览
-
- 用Nginx反向代理部署go写的网站。
- 2023-01-17 502浏览
-
- Golang取得代码运行时间的问题
- 2023-02-24 501浏览
-
- 请问 go 代码如何实现在代码改动后不需要Ctrl+c,然后重新 go run *.go 文件?
- 2023-01-08 501浏览
-
- 如何从同一个 io.Reader 读取多次
- 2023-04-11 501浏览