Add rate limiting

ratelimiting/ratelimiting.go

@@ -0,0 +1,55 @@
+package ratelimiting
+
+import (
+	"fmt"
+	"time"
+)
+
+func basicRateLimiting() {
+	reqs := make(chan int, 5)
+	limiter := time.Tick(200 * time.Millisecond)
+
+	for i := range 5 {
+		reqs <- i
+	}
+	close(reqs)
+
+	for i := range reqs {
+		<-limiter
+		fmt.Printf("request %d at %v\n", i, time.Now())
+	}
+}
+
+func burstyRateLimiting() {
+	burstyLimiter := make(chan time.Time, 3)
+
+	// Fill burstyLimiter to brim first
+	for range 3 {
+		burstyLimiter <- time.Now()
+	}
+
+	go func() {
+		for t := range time.Tick(200 * time.Millisecond) {
+			burstyLimiter <- t
+		}
+	}()
+
+	reqs := make(chan int, 5)
+
+	for i := range 5 {
+		reqs <- i
+	}
+	close(reqs)
+
+	for i := range reqs {
+		t := <-burstyLimiter
+		fmt.Printf("request %d at %v\n", i, t)
+	}
+}
+
+func RateLimiting() {
+	fmt.Println("Demo-ing basic rate limiting")
+	basicRateLimiting()
+	fmt.Println("Demo-ing bursty rate limiting")
+	burstyRateLimiting()
+}