fix(amazonq): Replacing message bus with direct communication #5949
Conversation
Qodana Community for JVM1 new problem were found
💡 Qodana analysis was run in the pull request mode: only the changed files were checked Contact Qodana teamContact us at [email protected]
|
LokeshDogga13
changed the title
LokeshDogga13
changed the title
| @@ -134,6 +136,11 @@ class AmazonQPanel(val project: Project, private val scope: CoroutineScope) : Di | |||
| browser.complete( | |||
| Browser(this@AmazonQPanel, webUri, project).also { browserInstance -> | |||
| wrapper.setContent(browserInstance.component()) | |||
|
|
|||
| // Register direct callback instead of using message bus | |||
| ChatCommunicationManager.getInstance(project).setChatUpdateCallback { message -> | |||
There was a problem hiding this comment.
we should probably just associate the browser with the communication manager
There was a problem hiding this comment.
Browser is not in shared package, and cannot import it. I am able to do it using Any, but not sure if it is a right way to do it.
…kit-jetbrains into lodogga/memoryUsage
| try { | ||
| chatAsyncResultManager.createRequestId(partialResultToken) | ||
| chatAsyncResultManager.getResult(partialResultToken) | ||
| handleCancellation(tabId, browser) |
There was a problem hiding this comment.
Yes, this code is introduced when stop conversation needed to be handled on JB side. Now it is not required. So reverting this change.
There was a problem hiding this comment.
we also maintain records for partial results cause they are received separately and we mark them done when the result is completed, are we sure that line 599-600 can be deleted?
| try { | ||
| chatAsyncResultManager.createRequestId(partialResultToken) | ||
| chatAsyncResultManager.getResult(partialResultToken) | ||
| handleCancellation(tabId, browser) |
There was a problem hiding this comment.
we also maintain records for partial results cause they are received separately and we mark them done when the result is completed, are we sure that line 599-600 can be deleted?
| @@ -40,7 +40,8 @@ class ActionRegistrar { | |||
| val params = SendToPromptParams(selection = codeSelection, triggerType = TriggerType.CONTEXT_MENU) | |||
| uiMessage = FlareUiMessage(command = SEND_TO_PROMPT, params = params) | |||
| } | |||
| AsyncChatUiListener.notifyPartialMessageUpdate(project, uiMessage) | |||
| ChatCommunicationManager.getInstance(project).notifyUi(uiMessage) | |||
| // AsyncChatUiListener.notifyPartialMessageUpdate(project, uiMessage) | |||
Optimize AmazonQ Plugin Memory Usage [DO NOT MERGE]
Problem
AmazonQ plugin memory usage has increased due to:
Solution
Changes
ChatCommunicationManagerAmazonQPanelto register callback instead of subscribing to message busAmazonQLanguageClientImplandActionRegistrarto use direct communicationChecklist
Before
After
system is idle from 21:00 to 6:00 in second image
License
I confirm that my contribution is made under the terms of the Apache 2.0 license.