tyamashi-oss commented on code in PR #12296:
URL: https://github.com/apache/kafka/pull/12296#discussion_r913695123
##########
core/src/test/scala/unit/kafka/utils/ThrottlerTest.scala:
##########
@@ -58,4 +58,50 @@ class ThrottlerTest {
val actualCountPerSec = 4 * desiredCountPerInterval * 1000 / elapsedTimeMs
assertTrue(actualCountPerSec <= desiredCountPerSec)
}
+
+ @Test
+ def testUpdateThrottleDesiredRate(): Unit = {
+ val throttleCheckIntervalMs = 100
+ val desiredCountPerSec = 1000.0
+ val desiredCountPerInterval = desiredCountPerSec * throttleCheckIntervalMs
/ 1000.0
+ val updatedDesiredCountPerSec = 1500.0;
+ val updatedDesiredCountPerInterval = updatedDesiredCountPerSec *
throttleCheckIntervalMs / 1000.0
+
+ val mockTime = new MockTime()
+ val throttler = new Throttler(desiredRatePerSec = desiredCountPerSec,
+ checkIntervalMs = throttleCheckIntervalMs,
+ time = mockTime)
+
+ // Observe desiredCountPerInterval at t1
+ val t1 = mockTime.milliseconds()
+ throttler.maybeThrottle(desiredCountPerInterval)
+ assertEquals(t1, mockTime.milliseconds())
+
+ // Observe desiredCountPerInterval at t1 + throttleCheckIntervalMs + 1,
+ mockTime.sleep(throttleCheckIntervalMs + 1)
+ throttler.maybeThrottle(desiredCountPerInterval)
+ val t2 = mockTime.milliseconds()
+ assertTrue(t2 >= t1 + 2 * throttleCheckIntervalMs)
+
+ val elapsedTimeMs = t2 - t1
+ val actualCountPerSec = 2 * desiredCountPerInterval * 1000 / elapsedTimeMs
+ assertTrue(actualCountPerSec <= desiredCountPerSec)
+
+ // Update ThrottleDesiredRate
+ throttler.updateDesiredRatePerSec(updatedDesiredCountPerSec);
+
+ // Observe desiredCountPerInterval at t2
+ throttler.maybeThrottle(updatedDesiredCountPerInterval)
Review Comment:
Thank you. The comments have been changed.
https://github.com/apache/kafka/pull/12296/commits/ec4c24ee6e315657f364d0953ddfee7d61712692
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
