Golang의 흐름 제어 응용 기술

package main

import (
    "fmt"
    "golang.org/x/time/rate"
    "time"
)

func main() {
    limiter := rate.NewLimiter(rate.Limit(10), 1)
    for {
        if limiter.Allow() {
            // 处理数据传输
            fmt.Println("处理数据")
        } else {
            // 等待1秒
            time.Sleep(1 * time.Second)
        }
    }
}

package main

import (
    "fmt"
    "time"
)

type SlidingWindow struct {
    Data     []int
    Interval time.Duration
    MaxCount int
}

func (sw *SlidingWindow) Allow() bool {
    now := time.Now().Unix()
    if len(sw.Data) < sw.MaxCount {
        sw.Data = append(sw.Data, int(now))
        return true
    }
    if now-int64(sw.Data[0]) < sw.Interval.Nanoseconds() {
        return false
    }
    sw.Data = append(sw.Data[1:], int(now))
    return true
}

func main() {
    sw := SlidingWindow{
        Interval: 1 * time.Second,
        MaxCount: 10,
    }
    for {
        if sw.Allow() {
            // 处理数据传输
            fmt.Println("处理数据")
        } else {
            // 等待1秒
            time.Sleep(1 * time.Second)
        }
    }
}