mirrors/relay
1
0
Fork 1
mirror of https://git.asonix.dog/asonix/relay.git synced 2024-11-25 02:51:12 +00:00
Code Issues Projects Releases Packages Wiki Activity

Allow sub-actors of connected server-actors

Browse source
This commit is contained in:
asonix 2021-02-10 09:23:55 -06:00
parent 3074823dc9
commit b9f9fd4a12
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/db.rs
View file

@ -420,11 +420,19 @@ impl Db {
.await
}
pub(crate) async fn is_connected(&self, id: Url) -> Result<bool, MyError> {
pub(crate) async fn is_connected(&self, mut id: Url) -> Result<bool, MyError> {
id.set_path("");
id.set_query(None);
id.set_fragment(None);
self.unblock(move |inner| {
let connected = inner
.connected_actor_ids
.contains_key(id.as_str().as_bytes())?;
.scan_prefix(id.as_str().as_bytes())
.values()
.filter_map(|res| res.ok())
.next()
.is_some();
Ok(connected)
})
@ -634,8 +642,9 @@ mod tests {
fn connect_and_verify() {
run(|db| async move {
let example_actor: Url = "http://example.com/actor".parse().unwrap();
let example_sub_actor: Url = "http://example.com/users/fake".parse().unwrap();
db.add_connection(example_actor.clone()).await.unwrap();
assert!(db.is_connected(example_actor).await.unwrap());
assert!(db.is_connected(example_sub_actor).await.unwrap());
})
}