change callback logic

src/main/kotlin/io.evest.simpleratelimiter/RateLimiter.kt

@@ -41,14 +41,18 @@ object RateLimiter {
     }
 
     /**
-     * Try to bucket the call, if the rate limit is exceeded, return an [Option] with the exception.
+     * Bucket the call, if the rate limit is exceeded, call the [error] function.
      */
     fun <T> T.tryBucket(
         key: String,
         maxCalls: Int = 1,
         per: Duration = Duration.ofSeconds(defaultMinWait),
-    ): Option<T> = when (BucketHandler.count(key, maxCalls, per)) {
-        true -> Option(this, RateLimitException("Rate limit exceeded for key: $key"))
-        false -> Option(this)
+        error: (exception: RateLimitException?) -> Unit = {},
+    ): T = also {
+        val exhausted = when (BucketHandler.count(key, maxCalls, per)) {
+            true -> RateLimitException("Rate limit exceeded for key: $key")
+            false -> null
+        }
+        error(exhausted)
     }
 }

src/test/kotlin/io/evest/simpleratelimiter/test/TestRateLimiter.kt

@@ -2,6 +2,7 @@ package io.evest.simpleratelimiter.test
 
 import io.evest.simpleratelimiter.RateLimitException
 import io.evest.simpleratelimiter.RateLimiter.bucket
+import io.evest.simpleratelimiter.RateLimiter.tryBucket
 import java.time.Duration
 import kotlin.test.Test
 
@@ -15,6 +16,7 @@ class TestRateLimiter {
             val startTime = System.currentTimeMillis()
             try {
                 any.bucket("1", 2, Duration.ofSeconds(1))
+
             } catch (e: RateLimitException) {
                 println("Rate limit exceeded for iteration $i")
             }