Skip to content

Instantly share code, notes, and snippets.

@ilhamarrouf
Created March 29, 2019 11:53
Show Gist options
  • Save ilhamarrouf/9f4109ecd27e45f02121df5a47eed1d5 to your computer and use it in GitHub Desktop.
Save ilhamarrouf/9f4109ecd27e45f02121df5a47eed1d5 to your computer and use it in GitHub Desktop.
Golang rate limiting
package main
import (
"fmt"
"time"
)
func main() {
requests := make(chan int, 5)
for i := 1; i <= 5; i++ {
requests <- i
}
close(requests)
limiter := time.Tick(200 * time.Millisecond)
for req := range requests {
<-limiter
fmt.Println("request", req, time.Now())
}
burstyLimiter := make(chan time.Time, 3)
for i := 0; i < 3; i++ {
burstyLimiter <- time.Now()
}
go func() {
for t := range time.Tick(200 * time.Millisecond) {
burstyLimiter <- t
}
}()
burstyRequests := make(chan int, 5)
for i := 1; i <= 5; i++ {
burstyRequests <- i
}
close(burstyRequests)
for req := range burstyRequests {
<- burstyLimiter
fmt.Println("request", req, time.Now())
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment