From 56e74239d6b34df8f30ef046f0b0ff4ff0866a71 Mon Sep 17 00:00:00 2001
From: =Corey Hulen <corey@hulen.com>
Date: Sun, 14 Jun 2015 23:53:32 -0800
Subject: first commit

---
 .../_workspace/src/gopkg.in/redis.v2/rate_limit.go | 53 ++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 Godeps/_workspace/src/gopkg.in/redis.v2/rate_limit.go

(limited to 'Godeps/_workspace/src/gopkg.in/redis.v2/rate_limit.go')

diff --git a/Godeps/_workspace/src/gopkg.in/redis.v2/rate_limit.go b/Godeps/_workspace/src/gopkg.in/redis.v2/rate_limit.go
new file mode 100644
index 000000000..20d851270
--- /dev/null
+++ b/Godeps/_workspace/src/gopkg.in/redis.v2/rate_limit.go
@@ -0,0 +1,53 @@
+package redis
+
+import (
+	"sync/atomic"
+	"time"
+)
+
+type rateLimiter struct {
+	v int64
+
+	_closed int64
+}
+
+func newRateLimiter(limit time.Duration, bucketSize int) *rateLimiter {
+	rl := &rateLimiter{
+		v: int64(bucketSize),
+	}
+	go rl.loop(limit, int64(bucketSize))
+	return rl
+}
+
+func (rl *rateLimiter) loop(limit time.Duration, bucketSize int64) {
+	for {
+		if rl.closed() {
+			break
+		}
+		if v := atomic.LoadInt64(&rl.v); v < bucketSize {
+			atomic.AddInt64(&rl.v, 1)
+		}
+		time.Sleep(limit)
+	}
+}
+
+func (rl *rateLimiter) Check() bool {
+	for {
+		if v := atomic.LoadInt64(&rl.v); v > 0 {
+			if atomic.CompareAndSwapInt64(&rl.v, v, v-1) {
+				return true
+			}
+		} else {
+			return false
+		}
+	}
+}
+
+func (rl *rateLimiter) Close() error {
+	atomic.StoreInt64(&rl._closed, 1)
+	return nil
+}
+
+func (rl *rateLimiter) closed() bool {
+	return atomic.LoadInt64(&rl._closed) == 1
+}
-- 
cgit v1.2.3-1-g7c22