Go实现接口访问速率限制

mingzhehaolove · · 4009 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

接口的访问限制,10分钟内,接口访问限制100次

基于go语言进行编写,抽离出统一配置。


func CheckRateLimit(ip, request, action string) bool {
	current := int(time.Now().Unix())
	currentStr := strconv.Itoa(current)
	//limit  100次
	//timeset 600秒
	//限制600秒最多访问100次
	limit, timeset := GetRateLimitConfig()
	allowanceStr, timestampStr := LoadAllowance(ip, request, action)
	allowance, _ := strconv.Atoi(allowanceStr)
	timestamp, _ := strconv.Atoi(timestampStr)
	allowance += int(current-timestamp) * limit / timeset
	if allowance > limit {
		allowance = limit
	}

	if allowance < 1 {
		SaveAllowance(ip, request, action, "0", currentStr)
		//返回true 代表速率超过,进行错误输出
		return true
	} else {
		allowanceStr = strconv.Itoa(allowance - 1)
		SaveAllowance(ip, request, action, allowanceStr, currentStr)
		//返回false 代表速率未超过
		return false
	}
}

func LoadAllowance(ip, request, action string) (allowance, timestamp string) {
	redisConfig := getRedisConfig()
	rs, err := cache.NewCache("redis", redisConfig)
	if err != nil {
		fmt.Println(err)
		return
	}
	res, _ := (redis.String(rs.Get(ip+"_"+request), err))
	if len(res) == 0 {
		currentStr := string(time.Now().Unix())
		defaultLimitInt, _ := GetRateLimitConfig()
		defaultLimitStr := strconv.Itoa(defaultLimitInt)
		allowance, timestamp = defaultLimitStr, currentStr
	} else {
		kv := strings.Split(res, "-")
		allowance, timestamp = kv[0], kv[1]
	}
	return

}

func GetRateLimitConfig() (limit, timeset int) {
	limit, _ = beego.AppConfig.Int("rateLimit")
	timeset, _ = beego.AppConfig.Int("rateTimeset")
	return
}

func SaveAllowance(ip, request, action, allowance, current string) {
	redisConfig := getRedisConfig()
	rs, err := cache.NewCache("redis", redisConfig)
	if err != nil {
		fmt.Println(err)
		return
	}
	rs.Put(ip+"_"+request, allowance+"-"+current, 600*time.Second)
}

func getRedisConfig() string {
	env := beego.AppConfig.String("runmode")
	conn := beego.AppConfig.String("redisConn")
	pass := beego.AppConfig.String("redisPass")
	name := beego.AppConfig.String("redisName")
	redisHash := make(map[string]interface{})
	redisHash["conn"] = conn
	redisHash["key"] = name
	if env == "prod" {
		redisHash["password"] = pass
	}
	redisConfig, _ := json.Marshal(redisHash)
	return string(redisConfig)
}

能够完全限制住接口的访问



有疑问加站长微信联系(非本文作者)

本文来自:CSDN博客

感谢作者:mingzhehaolove

查看原文:Go实现接口访问速率限制

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:701969077

4009 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传