单主机业务限流实现

实现背景：

      最近在实现一个服务，给外部企业提供接口，为了防止频繁访问导致资源占用问题。要对API接口限流，所以封装一个业务接口限流的功能。

优点：

      内存限流，相对于分布式少一次网络请求，性能更优。

缺点：

      只能单主机接口限流，没有实现多点部署分布式限流。

 

      以下是源码，GOLANG做了一个比较简单的功能实现。仅做交流学习，希望对老铁们有所帮助。

import (
	"sync/atomic"
	"time"

	"github.com/gpmgo/gopm/modules/log"
)

type RequestLimitServer struct {
	QPSLimit int32
	ReqCount int32
	Interval time.Duration
}

func New(qps_limit int32, interval time.Duration) *RequestLimitServer {
	reqLimit := &RequestLimitServer{QPSLimit: qps_limit, Interval: interval}
	go func() {
		ticker := time.NewTicker(interval)
		for {
			<-ticker.C
			atomic.SwapInt32(&reqLimit.ReqCount, 0)
		}
	}()
	return reqLimit
}

func (q *RequestLimitServer) Increase() {
	atomic.AddInt32(&q.ReqCount, 1)
}

func (q *RequestLimitServer) IsAvailable() bool {
	return !atomic.CompareAndSwapInt32(&q.ReqCount, q.QPSLimit, q.QPSLimit)
}

func (q *RequestLimitServer) UpdateLimiteCfg(newLimit int32) {
	atomic.SwapInt32(&q.QPSLimit, newLimit)
}

func (q *RequestLimitServer) Check() bool {
	if q.IsAvailable() {
		q.Increase()
		return true
	}

	log.Warn("QueryExchangeCurrency QPS[%d] get limit[%d]", q.ReqCount, q.QPSLimit)
	return false
}