Skip to content

Commit 51d7a05

Browse files
committed
fixes after rebase
1 parent 086e942 commit 51d7a05

File tree

2 files changed

+24
-7
lines changed

2 files changed

+24
-7
lines changed

iroh-dns/src/discovery.rs

Lines changed: 23 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
11
use std::sync::Arc;
22

33
use anyhow::Result;
4-
use futures::future::{BoxFuture, FutureExt};
5-
use iroh_net::key::SecretKey;
6-
use iroh_net::magicsock::Discovery;
7-
use iroh_net::{AddrInfo, NodeId};
4+
use futures::{future::FutureExt, stream::BoxStream, StreamExt};
5+
use iroh_net::{discovery::{Discovery, DiscoveryItem}, key::SecretKey, AddrInfo, MagicEndpoint, NodeId};
86
use tracing::warn;
97

108
use crate::publish::{self, Publisher};
@@ -49,7 +47,26 @@ impl Discovery for DnsDiscovery {
4947
}
5048
}
5149

52-
fn resolve<'a>(&'a self, node_id: &'a NodeId) -> BoxFuture<'a, Result<AddrInfo>> {
53-
self.resolver.resolve_node_by_id(*node_id).boxed()
50+
fn resolve(
51+
&self,
52+
_ep: MagicEndpoint,
53+
node_id: NodeId,
54+
) -> Option<BoxStream<'static, Result<DiscoveryItem>>> {
55+
let resolver = self.resolver.clone();
56+
let fut = async move {
57+
let addr_info = resolver.resolve_node_by_id(node_id).await?;
58+
Ok(DiscoveryItem {
59+
provenance: "iroh-dns",
60+
last_updated: None,
61+
addr_info,
62+
})
63+
};
64+
// let fut = fut.map_ok(|addr_info| DiscoveryItem {
65+
// provenance: "iroh-dns",
66+
// last_updated: None,
67+
// addr_info,
68+
// });
69+
let stream = fut.into_stream();
70+
Some(stream.boxed())
5471
}
5572
}

iroh/src/node.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ use iroh_gossip::net::{Gossip, GOSSIP_ALPN};
3535
use iroh_io::AsyncSliceReader;
3636
use iroh_net::derp::DerpUrl;
3737
use iroh_net::magic_endpoint::get_alpn;
38-
use iroh_net::magicsock::Discovery;
38+
use iroh_net::discovery::Discovery;
3939
use iroh_net::magicsock::LocalEndpointsStream;
4040
use iroh_net::util::AbortingJoinHandle;
4141
use iroh_net::{

0 commit comments

Comments
 (0)