improvement: handle limit and set limited in /user_directory/search

next
timokoesters 2020-07-30 11:12:49 +02:00
parent 18dcf44aa4
commit 52b1df6fd2
No known key found for this signature in database
GPG Key ID: 24DA7517711A2BA4
1 changed files with 33 additions and 30 deletions

View File

@ -2262,11 +2262,13 @@ pub fn search_users_route(
db: State<'_, Database>, db: State<'_, Database>,
body: Ruma<search_users::Request>, body: Ruma<search_users::Request>,
) -> ConduitResult<search_users::Response> { ) -> ConduitResult<search_users::Response> {
Ok(search_users::Response { let limit = if let Some(limit) = body.limit {
results: db u64::from(limit)
.users } else {
.iter() 10
.filter_map(|user_id| { } as usize;
let mut users = db.users.iter().filter_map(|user_id| {
// Filter out buggy users (they should not exist, but you never know...) // Filter out buggy users (they should not exist, but you never know...)
let user_id = user_id.ok()?; let user_id = user_id.ok()?;
if db.users.is_deactivated(&user_id).ok()? { if db.users.is_deactivated(&user_id).ok()? {
@ -2290,11 +2292,12 @@ pub fn search_users_route(
} }
Some(user) Some(user)
}) });
.collect(),
limited: false, let results = users.by_ref().take(limit).collect();
} let limited = users.next().is_some();
.into())
Ok(search_users::Response { results, limited }.into())
} }
#[cfg_attr( #[cfg_attr(