Fix duration deprecated usage (#70)

hfhbd · Jul 21, 2021 · 3bc470d · 3bc470d
1 parent 5287d04
commit 3bc470d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/kotlin/app/softwork/ratelimit/RateLimit.kt b/src/main/kotlin/app/softwork/ratelimit/RateLimit.kt
@@ -164,7 +164,7 @@ public class RateLimit(public val configuration: Configuration) {
                 is Block -> {
                     finish()
                     if (feature.configuration.sendRetryAfterHeader) {
-                        call.response.header(HttpHeaders.RetryAfter, isAllowed.retryAfter.inSeconds.toLong())
+                        call.response.header(HttpHeaders.RetryAfter, isAllowed.retryAfter.inWholeSeconds)
                     }
                     call.respond(HttpStatusCode.TooManyRequests)
                 }

diff --git a/src/test/kotlin/app/softwork/ratelimit/RateLimitTest.kt b/src/test/kotlin/app/softwork/ratelimit/RateLimitTest.kt
@@ -122,7 +122,7 @@ suspend fun RateLimit.test(limit: Int, timeout: Duration) {
     repeat(limit) {
         assertTrue(isAllowed("a") is Block)
     }
-    delay(timeMillis = timeout.toLongMilliseconds())
+    delay(timeMillis = timeout.inWholeMilliseconds)
     assertEquals(Allow, isAllowed("a"))
 
     repeat(limit) {
@@ -136,7 +136,7 @@ suspend fun RateLimit.test(limit: Int, timeout: Duration) {
     }
 
     repeat(limit * 2) {
-        delay(timeMillis = timeout.toLongMilliseconds())
+        delay(timeMillis = timeout.inWholeMilliseconds)
         assertEquals(Allow, isAllowed("a"))
     }
 }