IJPL-198927 Search Everywhere: release the first throttled batch w/o delay if all essential providers sent >= 15 results

GitOrigin-RevId: 0e31f23a58f6156124db92fc3d85a3a780463084

Author: Aydar Mukhametzyanov

platform/searchEverywhere/frontend/src/vm/SeTabVm.kt

@@ -12,6 +12,7 @@ import com.intellij.lang.Language
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.DataContext
 import com.intellij.openapi.application.EDT
+import com.intellij.openapi.options.advanced.AdvancedSettings
 import com.intellij.openapi.project.DumbService
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Disposer
@@ -190,25 +191,43 @@ class SeTabVm(
 
 private const val ESSENTIALS_WAITING_TIMEOUT: Long = 2000
 private const val ESSENTIALS_THROTTLE_DELAY: Long = 100
+private const val ESSENTIALS_ENOUGH_COUNT: Int = 15
 
 private fun Flow<SeResultEvent>.throttleUntilEssentialsArrive(essentialProviderIds: Set<SeProviderId>): Flow<ThrottledItems<SeResultEvent>> {
-  val nonArrivedEssentialProviders = essentialProviderIds.toMutableSet()
+  val essentialProvidersCounts = essentialProviderIds.associateWith { 0 }.toMutableMap()
 
   SeLog.log(SeLog.THROTTLING) { "Will start throttle with essential providers: $essentialProviderIds"}
   return throttledWithAccumulation(ESSENTIALS_WAITING_TIMEOUT, { it !is SeResultEndEvent }) { event, size ->
-    val idToRemove = when (event) {
-      is SeResultAddedEvent -> event.itemData.providerId
-      is SeResultReplacedEvent -> event.newItemData.providerId
-      is SeResultEndEvent -> event.providerId
-    }
+    val providerId = event.providerId()
 
-    if (nonArrivedEssentialProviders.remove(idToRemove)) {
-      SeLog.log(SeLog.THROTTLING) { "Arrived: $idToRemove" }
+    when (event) {
+      is SeResultEndEvent -> {
+        if (essentialProvidersCounts.remove(providerId) != null) {
+          SeLog.log(SeLog.THROTTLING) { "Ended: $providerId" }
+        }
+      }
+      else -> {
+        essentialProvidersCounts[providerId]?.let {
+          SeLog.log(SeLog.THROTTLING) { "Arrived: $providerId ($it)" }
+          essentialProvidersCounts[providerId] = it + 1
+        }
+      }
     }
 
-    return@throttledWithAccumulation if (nonArrivedEssentialProviders.isEmpty()) ESSENTIALS_THROTTLE_DELAY else null
+    return@throttledWithAccumulation (
+      if (essentialProvidersCounts.isEmpty()) 0
+      else if (essentialProvidersCounts.values.all { it >= ESSENTIALS_ENOUGH_COUNT }) 0
+      else if (essentialProvidersCounts.values.all { it > 0 }) ESSENTIALS_THROTTLE_DELAY
+      else null
+    )
   }
 }
 
+private fun SeResultEvent.providerId() = when (this) {
+  is SeResultAddedEvent -> itemData.providerId
+  is SeResultReplacedEvent -> newItemData.providerId
+  is SeResultEndEvent -> providerId
+}
+
 @ApiStatus.Internal
 class SeSearchContext(val searchId: String, val tabId: String, val searchPattern: String, val resultsFlow: Flow<ThrottledItems<SeResultEvent>>)