MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-01-07 22:09:38 +07:00
Code Issues Packages Projects Releases Wiki Activity

[remote-driver] fix JcefComponentWrapper.JsExecutor

Browse Source 
GitOrigin-RevId: c79cae0fd32ca9a2c97a34eee0f3fee36e1acbe7
This commit is contained in:
vladislav.shishov
2024-10-05 23:37:38 +03:00
committed by intellij-monorepo-bot
parent 2c056ddc9a
commit 91c82cd5cf
1 changed files with 18 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
plugins/performanceTesting/remote-driver/src/com/jetbrains/performancePlugin/remotedriver/jcef/JcefComponentWrapper.kt
View File

@@ -2,11 +2,11 @@ package com.jetbrains.performancePlugin.remotedriver.jcef
import com.intellij.ui.jcef.JBCefBrowserBase
import com.intellij.ui.jcef.JBCefJSQuery
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import kotlinx.coroutines.runBlocking
import kotlinx.coroutines.suspendCancellableCoroutine
import kotlinx.coroutines.withTimeout
import java.awt.Component
import java.util.function.Consumer
import java.util.function.Function
import javax.swing.JComponent
import kotlin.coroutines.resumeWithException
@@ -60,37 +60,34 @@ internal class JcefComponentWrapper(private val component: Component) {
private class JsExecutor(jbCefBrowser: JBCefBrowserBase) {
private val cefBrowser = jbCefBrowser.cefBrowser
private var callback: Consumer<String>? = null
private val jsResultQuery = JBCefJSQuery.create(jbCefBrowser).apply {
addHandler {
callback?.accept(it)
null
}
}
private val mutex = Mutex()
private val jsResultQuery = JBCefJSQuery.create(jbCefBrowser)
fun runJs(js: String) {
cefBrowser.executeJavaScript(js, cefBrowser.url, 0)
}
suspend fun callJs(js: String, executeTimeoutMs: Long = 3000): String = mutex.withLock {
suspend fun callJs(js: String, executeTimeoutMs: Long = 3000): String =
withTimeout(executeTimeoutMs) {
suspendCancellableCoroutine { continuation ->
val handler = object : Function<String, JBCefJSQuery.Response?> {
override fun apply(result: String): JBCefJSQuery.Response? {
jsResultQuery.removeHandler(this)
continuation.resumeWith(Result.success(result))
return null
}
}
jsResultQuery.addHandler(handler)
continuation.invokeOnCancellation {
callback = null
jsResultQuery.removeHandler(handler)
continuation.resumeWithException(IllegalStateException("""
|No result from script '$js' in embedded browser in ${executeTimeoutMs}ms.
|Check logs in the browsers devTools(`ide.browser.jcef.contextMenu.devTools.enabled` key in the Registry...)""".trimMargin()))
}
callback = Consumer<String> {
continuation.resumeWith(Result.success(it))
callback = null
}
cefBrowser.executeJavaScript(jsResultQuery.inject(js), cefBrowser.url, 0)
}
}
}
}
}