go语言实现 滑动窗口 流量控制

实现滑动窗口流量控制的方法在不同编程语言中有所差异。下面是一个使用 Go 语言实现滑动窗口流量控制的简单示例：

package main

import (
	"fmt"
	"sync"
	"time"
)

type SlidingWindow struct {
	mu           sync.Mutex
	windowSize   int
	requests     []time.Time
	rateLimit    int
	rateInterval time.Duration
}

func NewSlidingWindow(windowSize, rateLimit int, rateInterval time.Duration) *SlidingWindow {
	return &SlidingWindow{
		windowSize:   windowSize,
		requests:     make([]time.Time, 0, windowSize),
		rateLimit:    rateLimit,
		rateInterval: rateInterval,
	}
}

func (s *SlidingWindow) AllowRequest() bool {
	s.mu.Lock()
	defer s.mu.Unlock()

	now := time.Now()

	// 移除过期的请求
	for len(s.requests) > 0 && now.Sub(s.requests[0]) > s.rateInterval {
		s.requests = s.requests[1:]
	}

	// 判断请求数是否超过限制
	if len(s.requests) >= s.rateLimit {
		return false
	}

	s.requests = append(s.requests, now)
	return true
}

func main() {
	windowSize := 5         // 窗口大小
	rateLimit := 2          // 限流数
	rateInterval := 5 * time.Second // 限流时间间隔

	sw := NewSlidingWindow(windowSize, rateLimit, rateInterval)

	for i := 1; i <= 10; i++ {
		if sw.AllowRequest() {
			fmt.Println("请求通过")
		} else {
			fmt.Println("请求被限流")
		}
		time.Sleep(1 * time.Second)
	}
}

上面的代码使用了互斥锁 `sync.Mutex` 来保护共享数据的读写。`SlidingWindow` 结构体表示滑动窗口，其中的 `requests` 数组用于存储请求的时间戳。`AllowRequest` 方法用于判断是否允许请求通过。在该方法中，我们首先移除过期的请求，然后检查当前窗口中请求数是否超过限制。如果请求通过，则将当前请求的时间戳加入到 `requests` 数组中。

在 `main` 函数中，我们创建了一个 `SlidingWindow` 实例，并模拟了 10 次请求。根据设置的窗口大小和限流数，如果请求数超过限制，该请求将被限流。

这只是一个简单的示例，实际应用中可能需要更复杂的逻辑和数据结构来实现滑动窗口流量控制。希望这个示例对你有所帮助！如有任何问题，请随时提问。