Merge pull request #2 from asundukov/dev

fix maximum connections issue
asundukov · Jul 25, 2020 · f778472 · f778472
2 parents 28b6899 + c1569a9
commit f778472
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/kotlin-telegram-framework/src/main/kotlin/io/cutebot/telegram/client/TelegramApi.kt b/kotlin-telegram-framework/src/main/kotlin/io/cutebot/telegram/client/TelegramApi.kt
@@ -32,14 +32,24 @@ import org.apache.http.entity.mime.MultipartEntity
 import org.apache.http.entity.mime.content.FileBody
 import org.apache.http.entity.mime.content.StringBody
 import org.apache.http.impl.client.HttpClientBuilder
+import org.apache.http.impl.conn.PoolingHttpClientConnectionManager
 import org.slf4j.LoggerFactory
 
 
-class TelegramApi {
+class TelegramApi(
+        maxConnections: Int = 5000
+) {
+
+    private val connectionManager = PoolingHttpClientConnectionManager()
+    init {
+        connectionManager.maxTotal = maxConnections
+        connectionManager.defaultMaxPerRoute = maxConnections
+    }
 
     private val objectMapper = jacksonObjectMapper()
     private val httpClient = HttpClientBuilder
             .create()
+            .setConnectionManager(connectionManager)
             .build()
 
     fun getMe(token: String): TgUser {

diff --git a/...tateful-bot/src/main/kotlin/io/cutebot/telegram/samples/statefulbot/StatefulBotExample.kt b/...tateful-bot/src/main/kotlin/io/cutebot/telegram/samples/statefulbot/StatefulBotExample.kt
@@ -7,9 +7,7 @@ class StatefulBotExample(
         private val token: String,
         currentBlock: BotBlock
 ): StatefulBot(currentBlock) {
-
     override fun getToken(): String {
         return token
     }
-
 }