Fix @mention search performance and refactor search functionality #614

ZhangHanDong · 2025-10-22T03:34:31Z

Fixed issue #506

This is a remastered version of the PR #550 .

Add a lightweight cpu_worker thread that stays alive for the app lifetime and runs CPU-bound jobs (currently the room-member search); App::handle_startup initializes it and widgets enqueue work via cpu_worker::spawn_cpu_job.
Rework MentionableTextInput into an explicit state machine with search IDs, cancellation tokens, loading/empty UI states, and streaming updates that arrive over the background worker channel.
Replace the old MatrixRequest::SearchRoomMembers path: sliding_sync now fetches joined-member lists directly, emits RoomMembersListFetched, and RoomScreen caches those members plus precomputed sort data for downstream widgets.
Refactor room/member_search.rs to support batched streaming, cancellation, richer match heuristics, optional precomputed sort reuse, and add unit tests for the new helpers.

There are three related issues that still need to be fixed.

#615
#617

…reuse BouncingDots

kevinaboos · 2025-10-22T20:46:59Z

@alanpoon & @tyreseluo, can you please give this a review before I dig in? Thanks.

tyreseluo · 2025-10-23T04:42:51Z

@alanpoon & @tyreseluo, can you please give this a review before I dig in? Thanks.

yep, i will.

…tartup so CPU-bound work like member searches runs off the UI thread. - Reworked mention search to enqueue jobs on the CPU worker with cancellation tokens, unique search IDs, and better handling of loading state when the member list is still fetching. - Updated room loading and sync flow to fetch the full member list from the homeserver and deliver it through a new RoomMembersListFetched timeline update, removing the old MatrixRequest::SearchRoomMembers path. - Refactored the member search algorithm for batching, cancellation, richer match strategies, and added unit tests for the new helpers.

…tion bug after re-login

…ix into fixed-issue-452

ZhangHanDong · 2025-10-28T04:07:15Z

Review suggestions:
• Select a large room (thousands or tens of thousands of members), @ a user, and search.
• Continue testing in rooms with more than 50 members.
• Continue testing in rooms with fewer than 50 members.
• Close large/small rooms, reopen them, and continue testing.
• Logout, then login again, and rerun the above four steps.

alanpoon

Overall the performance of loading the mentionable list seems to be impacted.
There's seem too many fields added MentionableTextInput.
There's also bug when clicking the user from mentionable list, it does not apply into the text box.
The search does not work after "comma".

5. There should not be "Notify the Room" for direct message.

alanpoon · 2025-10-29T05:16:24Z

src/sliding_sync.rs

+                    match timeline.room().members(RoomMemberships::JOIN).await {
+                        Ok(members) => {
+                            let count = members.len();
+                            log!("Fetched {count} members for room {room_id} after sync.");
+                            if let Err(err) = sender.send(TimelineUpdate::RoomMembersListFetched { members }) {
+                                warning!("Failed to send RoomMembersListFetched update for room {room_id}: {err:?}");
+                            } else {
+                                SignalToUI::set_ui_signal();
+                            }
+                        }
+                        Err(err) => {
+                            warning!("Failed to fetch room members from server for room {room_id}: {err:?}");
+                        }
+                    }


There is already timeline.fetched_members above. This looks like a redundant call. There is extra matrix sqlite call. This could explain why this PR is slower when loading the mentionable list.
It looks like the mentionable list still loads even this change is reverted.

alanpoon · 2025-10-29T05:34:24Z

src/home/room_screen.rs

+                    // Note: This can be sent from either GetRoomMembers (local cache lookup)
+                    // or SyncRoomMemberList (full server sync). We only clear the sync pending
+                    // flag when we receive RoomMembersSynced (which is only sent after full sync).
+                    let sort_data = precompute_member_sort(&members);


Does it make sense to move this processing operation to async function in sliding_sync rather than in UI thread?

alanpoon · 2025-10-29T05:56:17Z

src/room/member_search.rs

+}
+
+/// Search room members in background thread with streaming support (backward compatible)
+pub fn search_room_members_streaming(


This function is not used.

alanpoon · 2025-10-29T07:20:09Z

src/cpu_worker.rs

+    pub sender: Sender<SearchResult>,
+    pub search_id: u64,
+    pub precomputed_sort: Option<Arc<PrecomputedMemberSort>>,
+    pub cancel_token: Option<Arc<AtomicBool>>,


The PR is significantly slow enough when loading the mentionable list to test the cancel token when there is a new search input. The cancel token does not seem to work.

Go to room with large number of members. (mentionable list is loading)

While it is loading, type "k". I am expecting the mentionable list to load faster for shorter query.

After it takes several seconds to load "k" result, remove "k" to go to empty string queries.

I am expecting mentionable list to display "loading". But it shows the full results.

alanpoon · 2025-10-29T08:19:42Z

src/cpu_worker.rs

+    cx.spawn_thread(move || match job {
+        CpuJob::SearchRoomMembers(params) => run_member_search(params),
+    });


Is there a way to close this CpuJob when closing the room screen tab?

alanpoon · 2025-10-29T09:01:39Z

src/room/member_search.rs

+    sender: &Sender<SearchResult>,
+    cancel_token: &Option<Arc<AtomicBool>>,
+    search_id: u64,
+    search_text: &Arc<String>,


No need for Arc

alanpoon · 2025-10-29T09:18:03Z

src/shared/mentionable_text_input.rs

+        if self.is_searching() {
+            if let Event::KeyUp(key_event) = event {
+                if key_event.key_code == KeyCode::Escape {
+                    self.cancel_active_search();


After pressing KeyCode::Escaope, it is not cancelled. is_cancelled does not return true,

alanpoon · 2025-10-29T09:18:49Z

src/shared/mentionable_text_input.rs

+            // ESC key is now handled in the main event handler using KeyUp event
+            // This avoids conflicts with escaped() method being consumed by other components
+


can be removed

alanpoon · 2025-10-29T09:25:47Z

src/shared/mentionable_text_input.rs

+                                // Force a fresh search now that members are available
+                                let search_text = self.cmd_text_input.search_text();
+                                self.last_search_text = None;
+                                self.update_user_list(cx, &search_text, scope);


There is a cyclic dependencies. MentionableTextInputAction::RoomMembersLoaded -> "update_user_list" -> submit_async_request(MatrixRequest::GetRoomMembers) -> TimelineUpdate::RoomMembersListFetched -> MentionableTextInputAction::RoomMembersLoaded

alanpoon · 2025-10-29T09:32:58Z

src/shared/mentionable_text_input.rs

+        if is_complete {
+            self.search_results_pending = false;
+            // Search is complete - get results for final UI update
+            let final_results = if let MentionSearchState::Searching {
+                accumulated_results,
+                ..
+            } = &self.search_state
+            {
+                accumulated_results.clone()
+            } else {
+                Vec::new()
+            };
+
+            if final_results.is_empty() {


For some reason, when it is loading, the search result is complete but is final results is empty. Hence result the overall code has several checks.

ZhangHanDong added 3 commits October 22, 2025 11:31

Streamlined @mention member search: precompute sort, stream results, …

2fcd9ed

…reuse BouncingDots

Merge branch 'main' into fixed-issue-452

c6df225

fixed fmt

3c9ec1a

ZhangHanDong requested a review from kevinaboos October 22, 2025 18:17

ZhangHanDong added the waiting-on-review This issue is waiting to be reviewed label Oct 22, 2025

kevinaboos requested review from alanpoon and tyreseluo October 22, 2025 20:46

ZhangHanDong added 2 commits October 23, 2025 14:54

Merge branch 'main' into fixed-issue-452

a009763

fixed Conflict

e93fd1e

kevinaboos mentioned this pull request Oct 23, 2025

Tighten mention search loading-state detection #616

Closed

ZhangHanDong added 2 commits October 25, 2025 00:51

Merge branch 'main' into fixed-issue-452

22bec53

ZhangHanDong added waiting-on-author This issue is waiting on the original author for a response and removed waiting-on-review This issue is waiting to be reviewed labels Oct 25, 2025

ZhangHanDong added 2 commits October 26, 2025 20:39

Introduced a CPU worker for member search and fixed the loading anima…

777967d

…tion bug after re-login

Merge branch 'fixed-issue-452' of github.com-robrix:ZhangHanDong/robr…

5938646

…ix into fixed-issue-452

ZhangHanDong added waiting-on-review This issue is waiting to be reviewed and removed waiting-on-author This issue is waiting on the original author for a response labels Oct 26, 2025

Merge branch 'main' into fixed-issue-452

2d0abd0

alanpoon reviewed Oct 29, 2025

View reviewed changes

ZhangHanDong added waiting-on-author This issue is waiting on the original author for a response and removed waiting-on-review This issue is waiting to be reviewed labels Oct 30, 2025

		// ESC key is now handled in the main event handler using KeyUp event
		// This avoids conflicts with escaped() method being consumed by other components

Fix @mention search performance and refactor search functionality #614

Are you sure you want to change the base?

Fix @mention search performance and refactor search functionality #614

Conversation

ZhangHanDong commented Oct 22, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

kevinaboos commented Oct 22, 2025

Uh oh!

tyreseluo commented Oct 23, 2025

Uh oh!

ZhangHanDong commented Oct 28, 2025

Uh oh!

alanpoon left a comment • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

ZhangHanDong commented Oct 22, 2025 •

edited

Loading

alanpoon left a comment •

edited

Loading