commit

go.mod

@@ -0,0 +1,5 @@
+module code.yun.ink/pkg/limitx
+
+go 1.20
+
+require golang.org/x/time v0.5.0

go.sum

@@ -0,0 +1,2 @@
+golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
+golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=

ip_rate.go

@@ -0,0 +1,58 @@
+package limiter
+
+import (
+	"sync"
+
+	"golang.org/x/time/rate"
+)
+
+func init() {
+	SetupIpRateLimiter()
+}
+
+type IpRateLimiter struct {
+	ips map[string]*rate.Limiter
+	mu  *sync.RWMutex
+	r   rate.Limit // float64 速率 1秒
+	b   int        // 并发 1次请求
+}
+
+var RateLimiter *IpRateLimiter
+
+// 创建一个RateLimiter
+func SetupIpRateLimiter() error {
+	var r rate.Limit = 10
+	b := 1
+	RateLimiter = &IpRateLimiter{
+		ips: make(map[string]*rate.Limiter),
+		mu:  &sync.RWMutex{},
+		r:   r, // 速率
+		b:   b, // 并发
+	}
+	return nil
+}
+
+// 添加一个ip到map
+func (i *IpRateLimiter) AddIp(ip string) *rate.Limiter {
+	i.mu.Lock()
+	defer i.mu.Unlock()
+
+	limiter := rate.NewLimiter(i.r, i.b)
+	i.ips[ip] = limiter
+	return limiter
+}
+
+// 通过ip得到limiter
+func (i *IpRateLimiter) GetLimiter(ip string) *rate.Limiter {
+	i.mu.Lock()
+
+	limiter, exists := i.ips[ip]
+	if !exists {
+		i.mu.Unlock()
+		// 注意：这里面也上了锁，所以当前的锁要先释放，不然死锁
+		return i.AddIp(ip)
+	}
+	// 这里不能忘
+	i.mu.Unlock()
+	return limiter
+}

readme.md

@@ -0,0 +1,13 @@
+
+# 目标
+
+1. 滑动窗口
+2. 漏桶算法
+3. 令牌桶算法
+
+
+
+
+
+
+