ratelimit 걸리지 않아서 걸리도록 수정

wafflestudio · Sep 10, 2023 · 3745634 · 3745634
1 parent 9cabb5d
commit 3745634
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/batch/src/main/kotlin/timetables/job/AutoPrimaryJobConfig.kt b/batch/src/main/kotlin/timetables/job/AutoPrimaryJobConfig.kt
@@ -2,6 +2,7 @@ package com.wafflestudio.snu4t.timetables.job
 
 import com.wafflestudio.snu4t.common.enum.Semester
 import com.wafflestudio.snu4t.timetables.repository.TimetableRepository
+import io.github.resilience4j.kotlin.ratelimiter.executeSuspendFunction
 import io.github.resilience4j.kotlin.ratelimiter.rateLimiter
 import io.github.resilience4j.ratelimiter.RateLimiter
 import io.github.resilience4j.ratelimiter.RateLimiterConfig
@@ -59,9 +60,7 @@ class AutoPrimaryJobConfig(
         val agg = Aggregation.newAggregation(Key::class.java, Aggregation.group("user_id", "semester", "year"))
         reactiveMongoTemplate.aggregate(agg, "timetables", AggResult::class.java)
             .asFlow()
-            .map { autoSetPrimary(it.id) }
-            .rateLimiter(rateLimiter)
-            .collect { }
+            .collect { rateLimiter.executeSuspendFunction { autoSetPrimary(it.id) } }
     }
 
     private suspend fun autoSetPrimary(key: Key) {