From e68db75ceefac0aacfd17ffa44e5d68b772518e1 Mon Sep 17 00:00:00 2001
From: Thibault Duplessis <thibault.duplessis@gmail.com>
Date: Mon, 7 Mar 2016 12:03:40 +0700
Subject: [PATCH] rename token bucket

---
 app/controllers/LilaController.scala                        | 4 ++--
 app/controllers/LilaSocket.scala                            | 4 ++--
 app/controllers/Lobby.scala                                 | 2 +-
 .../src/main/{TokenBucketGroup.scala => TokenBucket.scala}  | 6 +++---
 4 files changed, 8 insertions(+), 8 deletions(-)
 rename modules/api/src/main/{TokenBucketGroup.scala => TokenBucket.scala} (94%)

diff --git a/app/controllers/LilaController.scala b/app/controllers/LilaController.scala
index cd61a14437..ff8529fcac 100644
--- a/app/controllers/LilaController.scala
+++ b/app/controllers/LilaController.scala
@@ -10,7 +10,7 @@ import play.api.mvc.WebSocket.FrameFormatter
 import play.twirl.api.Html
 import scalaz.Monoid
 
-import lila.api.{ PageData, Context, HeaderContext, BodyContext, TokenBucketGroup }
+import lila.api.{ PageData, Context, HeaderContext, BodyContext, TokenBucket }
 import lila.app._
 import lila.common.{ LilaCookie, HTTPRequest }
 import lila.security.{ Permission, Granter, FingerprintedUser }
@@ -59,7 +59,7 @@ private[controllers] trait LilaController
       }
     }
 
-  protected def SocketOptionLimited[A: FrameFormatter](consumer: TokenBucketGroup.Consumer)(f: Context => Fu[Option[(Iteratee[A, _], Enumerator[A])]]) =
+  protected def SocketOptionLimited[A: FrameFormatter](consumer: TokenBucket.Consumer)(f: Context => Fu[Option[(Iteratee[A, _], Enumerator[A])]]) =
     LilaSocket.rateLimited[A](consumer) { req =>
       reqToCtx(req) flatMap f map {
         case None       => Left(NotFound(jsonError("socket resource not found")))
diff --git a/app/controllers/LilaSocket.scala b/app/controllers/LilaSocket.scala
index dd3811d242..63241f21a2 100644
--- a/app/controllers/LilaSocket.scala
+++ b/app/controllers/LilaSocket.scala
@@ -6,7 +6,7 @@ import play.api.libs.json._
 import play.api.mvc._, Results._
 import play.api.mvc.WebSocket.FrameFormatter
 
-import lila.api.TokenBucketGroup
+import lila.api.TokenBucket
 import lila.app._
 import lila.common.HTTPRequest
 
@@ -16,7 +16,7 @@ object LilaSocket {
 
   private val logger = play.api.Logger("ratelimit")
 
-  def rateLimited[A: FrameFormatter](consumer: TokenBucketGroup.Consumer)(f: AcceptType[A]): WebSocket[A, A] =
+  def rateLimited[A: FrameFormatter](consumer: TokenBucket.Consumer)(f: AcceptType[A]): WebSocket[A, A] =
     WebSocket[A, A] { req =>
       val ip = HTTPRequest lastRemoteAddress req
       f(req).map { resultOrSocket =>
diff --git a/app/controllers/Lobby.scala b/app/controllers/Lobby.scala
index 51b8dd1cd7..6e6001d4e4 100644
--- a/app/controllers/Lobby.scala
+++ b/app/controllers/Lobby.scala
@@ -41,7 +41,7 @@ object Lobby extends LilaController {
     )
   }
 
-  private val socketConsumer = lila.api.TokenBucketGroup.create(
+  private val socketConsumer = lila.api.TokenBucket.create(
     system = lila.common.PlayApp.system,
     size = 10,
     rate = 5)
diff --git a/modules/api/src/main/TokenBucketGroup.scala b/modules/api/src/main/TokenBucket.scala
similarity index 94%
rename from modules/api/src/main/TokenBucketGroup.scala
rename to modules/api/src/main/TokenBucket.scala
index 7132704bf1..40af3dbbc2 100644
--- a/modules/api/src/main/TokenBucketGroup.scala
+++ b/modules/api/src/main/TokenBucket.scala
@@ -38,7 +38,7 @@ private case class TokenRequest(key: Any)
  * @param rate refill rate in tokens per second
  * @param clock for mocking the current time.
  */
-private class TokenBucketGroup(size: Int, rate: Float, clock: Clock) extends Actor {
+private class TokenBucket(size: Int, rate: Float, clock: Clock) extends Actor {
 
   private val intervalMillis: Int = (1000 / rate).toInt
 
@@ -75,7 +75,7 @@ private class TokenBucketGroup(size: Int, rate: Float, clock: Clock) extends Act
   }
 }
 
-object TokenBucketGroup {
+object TokenBucket {
 
   private val defaultTimeout = Timeout(100, TimeUnit.MILLISECONDS)
 
@@ -101,7 +101,7 @@ object TokenBucketGroup {
     require(size <= 1000)
     require(rate >= 0.000001f)
     require(rate <= 1000)
-    new Consumer(system.actorOf(Props(new TokenBucketGroup(size, rate, clock))))
+    new Consumer(system.actorOf(Props(new TokenBucket(size, rate, clock))))
   }
 
   /**