rate_limit.go

package main

import (
	"fmt"
	"time"

	"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
	// "github.com/juju/ratelimit"
)

func main() {
	limiter := util.NewTokenBucketRateLimiter(150, 300)
	// limiter := ratelimit.NewBucketWithRate(150, 300)

	// Keep track of our initial offset
	startNano := time.Now().Nanosecond()

	// Track how requests are spread across a 1 second interval
	spread := [100]int{}
	accepted := func() {
		offsetNano := (time.Now().Nanosecond() + int(time.Second) - startNano) % int(time.Second)
		bucket := offsetNano / 10000000
		spread[bucket] = spread[bucket] + 1
	}

	go func() {
		for {
			limiter.Accept()
			// limiter.Wait(1)
			accepted()
		}
	}()

	for range time.Tick(time.Second) {
		fmt.Printf("%v\n", spread)
		for i := range spread {
			spread[i] = 0
		}
	}
}