MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-04-20 21:41:24 +07:00
Code Issues Packages Projects Releases Wiki Activity

ML in SE: Use coroutines for increasing use count (IDEA-325662)

Browse Source 
GitOrigin-RevId: e9d1f7a803eecfb76882b9e3a9f985eb56190657
This commit is contained in:
Adam Malek
2023-09-19 15:26:50 +02:00
committed by intellij-monorepo-bot
parent fbd8b70ca6
commit 4c3313ce55
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/search-everywhere-ml/ranking/src/com/intellij/searchEverywhereMl/ranking/features/statistician/SearchEverywhereStatisticianService.kt
View File

@@ -1,17 +1,29 @@
package com.intellij.searchEverywhereMl.ranking.features.statistician
import com.intellij.openapi.application.writeAction
import com.intellij.openapi.components.Service
import com.intellij.openapi.util.Key
import com.intellij.psi.statistics.StatisticsManager
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.launch
@Service(Service.Level.APP)
class SearchEverywhereStatisticianService {
class SearchEverywhereStatisticianService(private val coroutineScope: CoroutineScope) {
companion object {
val KEY: Key<SearchEverywhereStatistician<in Any>> = Key.create("searchEverywhere")
}
fun increaseUseCount(element: Any) = getSerializedInfo(element)?.let { StatisticsManager.getInstance().incUseCount(it) }
fun increaseUseCount(element: Any) {
coroutineScope.launch {
val info = getSerializedInfo(element) ?: return@launch
val manager = StatisticsManager.getInstance()
writeAction {
manager.incUseCount(info)
}
}
}
fun getCombinedStats(element: Any): SearchEverywhereStatisticianStats? {
val statisticsManager = StatisticsManager.getInstance()