HTTP限流控制：Go语言中的精细把关

package?main

import?(

"fmt"?

"sync"?

"time"?

)

type?RateLimiter struct?{

limit int?// 限流阈值

counter int?// 当前计数

lastUpdate time.Time // 上次更新时间

mutex sync.Mutex // 互斥锁，确保并发安全

}

func?NewRateLimiter(limit int)?*RateLimiter {

return?&RateLimiter{

limit: limit,

counter: 0,

lastUpdate: time.Now(),

}

}

func?(rl *RateLimiter)?Allow() bool?{

rl.mutex.Lock()

defer?rl.mutex.Unlock()

now := time.Now()

elapsed := now.Sub(rl.lastUpdate) // 计算时间间隔

rl.lastUpdate = now

// 更新计数器

rl.counter = rl.counter + 1?

if?rl.counter >= rl.limit {

return?false?// 超过阈值，直接返回false

} else?if?elapsed > time.Second { // 时间间隔超过1秒，重置计数器为1，确保在限流周期内只允许一定数量的请求通过

rl.counter = 1?

} else?{ // 时间间隔未超过1秒，继续累加计数器

rl.counter++

}

return?true?// 通过限流检查，返回true

}