#139 Add rate limiting

vityaman-edu · Jun 14, 2024 · 33328fa · 33328fa
1 parent 3051d43
commit 33328fa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/...n/kotlin/ru/vityaman/lms/botalka/core/publication/telegram/TelegramPublicationConsumer.kt b/...n/kotlin/ru/vityaman/lms/botalka/core/publication/telegram/TelegramPublicationConsumer.kt
@@ -1,10 +1,12 @@
 package ru.vityaman.lms.botalka.core.publication.telegram
 
+import kotlinx.coroutines.delay
 import ru.vityaman.lms.botalka.core.external.telegram.TelegramBot
 import ru.vityaman.lms.botalka.core.external.telegram.TelegramChat
 import ru.vityaman.lms.botalka.core.external.telegram.TelegramMessage
 import ru.vityaman.lms.botalka.core.model.Homework
 import ru.vityaman.lms.botalka.core.publication.PublicationConsumer
+import kotlin.time.Duration.Companion.seconds
 
 class TelegramPublicationConsumer(
     private val telegram: TelegramBot,
@@ -21,5 +23,6 @@ class TelegramPublicationConsumer(
             append("Id: ${value.id.number}\n")
         }
         telegram.send(adminChat, TelegramMessage(text))
+        delay(1.seconds)
     }
 }
diff --git a/infra/fuzzing/custom/homeworks.sh b/infra/fuzzing/custom/homeworks.sh
@@ -8,7 +8,7 @@ MOMENT="$2"
 echo "[fuzz] Got access token: '$TOKEN'"
 echo "[fuzz] Got moment: '$MOMENT'"
 
-for _ in $(seq 1 32); do
+for _ in $(seq 1 128); do
     curl -X 'POST' \
       'http://localhost:8080/api/v1/homework' \
       -H 'accept: application/json' \
@@ -20,5 +20,5 @@ for _ in $(seq 1 32); do
       \"max_score\": 500,
       \"publication_moment\": \"$MOMENT\",
       \"deadline_moment\": \"2026-07-11T12:00:00Z\"
-    }"
+    }" &
 done
diff --git a/infra/kafka/setup.sh b/infra/kafka/setup.sh
@@ -21,7 +21,7 @@ topic() {
   $KAFKA_TOPICS \
     --bootstrap-server "$HOST:$PORT" \
     --create --if-not-exists --topic "$NAME" \
-    --replication-factor 1 --partitions 1
+    --replication-factor 1 --partitions 2
 }
 
 topic "publication"