From 142e423c84a7fef60ddcd1f7df51447ffa90ef81 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Mon, 12 Feb 2024 11:56:15 +0100
Subject: [PATCH] [ML4SE-372] Added timeout for the sendFile function.

---
 .../research/tasktracker/requests/FileRequests.kt         | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/requests/FileRequests.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/requests/FileRequests.kt
index 824f0767..51af2126 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/requests/FileRequests.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/requests/FileRequests.kt
@@ -3,8 +3,8 @@ package org.jetbrains.research.tasktracker.requests
 import com.intellij.openapi.diagnostic.Logger
 import io.ktor.client.*
 import io.ktor.client.engine.cio.*
+import io.ktor.client.plugins.*
 import io.ktor.client.request.forms.*
-import io.ktor.client.statement.*
 import io.ktor.http.*
 import org.apache.http.client.utils.URIBuilder
 import org.jetbrains.research.tasktracker.config.MainTaskTrackerConfig.Companion.getRoute
@@ -15,6 +15,7 @@ object FileRequests {
 
     private val client = HttpClient(CIO)
     private val logger: Logger = Logger.getInstance(FileRequests::class.java)
+    private const val TIMEOUT: Long = 15_000 // 15 seconds
 
     @Suppress("TooGenericExceptionCaught")
     suspend fun sendFile(file: File, logFileType: String): Boolean {
@@ -35,6 +36,11 @@ object FileRequests {
                             )
                         }
                     )
+                },
+                block = {
+                    timeout {
+                        requestTimeoutMillis = TIMEOUT
+                    }
                 }
             )
             return response.status.isSuccess()