Skip to content

Commit f2fc230

Browse files
committed
Update to the latest transport lib
- Rename gateways to portals - Pass idle connection timeout from env vars
1 parent d132786 commit f2fc230

9 files changed

Lines changed: 171 additions & 1928 deletions

File tree

Cargo.lock

Lines changed: 150 additions & 1899 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,6 @@ members = ["crates/*"]
33
resolver = "2"
44

55
[workspace.dependencies]
6-
sqd-contract-client = { git = "https://github.com/subsquid/sqd-network.git", rev = "5747a3d", version = "1.2.0" }
7-
sqd-messages = { git = "https://github.com/subsquid/sqd-network.git", rev = "5747a3d", version = "2.0.1", features = ["bitstring"] }
8-
sqd-network-transport = { git = "https://github.com/subsquid/sqd-network.git", rev = "5747a3d", version = "3.0.0" }
6+
sqd-contract-client = { git = "https://github.com/subsquid/sqd-network.git", rev = "c57f520", version = "1.2.0" }
7+
sqd-messages = { git = "https://github.com/subsquid/sqd-network.git", rev = "c57f520", version = "2.0.1", features = ["bitstring"] }
8+
sqd-network-transport = { git = "https://github.com/subsquid/sqd-network.git", rev = "c57f520", version = "3.0.0" }

crates/logs-collector/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[package]
22
name = "logs-collector"
3-
version = "2.1.2"
3+
version = "2.1.3"
44
edition = "2021"
55

66
[dependencies]

crates/pings-collector/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[package]
22
name = "pings-collector"
3-
version = "2.1.2"
3+
version = "2.1.3"
44
edition = "2021"
55

66
[dependencies]

crates/pings-collector/src/cli.rs

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,6 @@ pub struct Cli {
3838
)]
3939
pub buffer_dir: PathBuf,
4040

41-
#[arg(long, env, default_value_t = true)]
42-
pub use_gossipsub: bool,
43-
44-
#[arg(long, env, default_value_t = false)]
45-
pub use_polling: bool,
46-
4741
#[arg(long, env, default_value_t = 15)]
4842
pub request_timeout_sec: u32,
4943

crates/pings-collector/src/main.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,6 @@ async fn main() -> anyhow::Result<()> {
4545
transport_builder.contract_client().into();
4646
let (incoming_pings, transport_handle) =
4747
transport_builder.build_pings_collector(PingsCollectorConfig {
48-
worker_status_via_gossipsub: args.use_gossipsub,
49-
worker_status_via_requests: args.use_polling,
5048
..Default::default()
5149
})?;
5250

crates/portal-logs-collector/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[package]
22
name = "portal-logs-collector"
3-
version = "1.0.2"
3+
version = "1.0.3"
44
edition = "2021"
55

66
[dependencies]

crates/portal-logs-collector/src/main.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ async fn main() -> anyhow::Result<()> {
7676
let agent_info = get_agent_info!();
7777
let transport_builder = P2PTransportBuilder::from_cli(args.transport, agent_info).await?;
7878
let contract_client: Arc<_> = transport_builder.contract_client().into();
79-
let config = PortalLogsCollectorConfig::new();
79+
let config = PortalLogsCollectorConfig::default();
8080

8181
let transport = transport_builder.build_portal_logs_collector(config)?;
8282

crates/portal-logs-collector/src/server.rs

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ where
1919
{
2020
_transport_handle: PortalLogsCollectorTransportHandle,
2121
logs_collector: Arc<PortalLogsCollector<T>>,
22-
registered_gateways: Arc<Mutex<HashSet<PeerId>>>,
22+
registered_portals: Arc<Mutex<HashSet<PeerId>>>,
2323
task_manager: TaskManager,
2424
event_stream: Box<dyn Stream<Item = PortalLogsCollectorEvent> + Send + Unpin + 'static>,
2525
collector_index: usize,
@@ -41,7 +41,7 @@ where
4141
Self {
4242
_transport_handle: transport_handle,
4343
logs_collector: Arc::new(logs_collector),
44-
registered_gateways: Default::default(),
44+
registered_portals: Default::default(),
4545
task_manager: Default::default(),
4646
event_stream: Box::new(event_stream),
4747
collector_index,
@@ -59,13 +59,13 @@ where
5959
) -> anyhow::Result<()> {
6060
log::info!("Starting logs collector server");
6161

62-
// Get registered gateways from chain
63-
let gateways = contract_client
64-
.active_gateways()
62+
// Get registered portals from chain
63+
let portals = contract_client
64+
.active_portals()
6565
.await?
6666
.into_iter()
6767
.collect();
68-
*self.registered_gateways.lock() = gateways;
68+
*self.registered_portals.lock() = portals;
6969

7070
self.spawn_portal_update_task(contract_client, portal_update_interval);
7171

@@ -94,7 +94,7 @@ where
9494
if !self.should_process(&peer_id) {
9595
continue
9696
}
97-
if self.registered_gateways.lock().contains(&peer_id) {
97+
if self.registered_portals.lock().contains(&peer_id) {
9898
log::debug!("Got log from {peer_id:?}: {log:?}");
9999
self.logs_collector.buffer_logs(peer_id, vec![log]);
100100
} else {
@@ -126,19 +126,19 @@ where
126126
contract_client: Arc<dyn ContractClient>,
127127
interval: Duration,
128128
) {
129-
log::info!("Starting gateway update task");
129+
log::info!("Starting portal update task");
130130

131-
let registered_gateways = self.registered_gateways.clone();
131+
let registered_portals = self.registered_portals.clone();
132132
let contract_client: Arc<dyn ContractClient> = contract_client;
133133
let task = move |_| {
134-
let registered_gateways = registered_gateways.clone();
134+
let registered_portals = registered_portals.clone();
135135
let contract_client = contract_client.clone();
136136
async move {
137-
let gateways = match contract_client.active_gateways().await {
138-
Ok(gateways) => gateways,
139-
Err(e) => return log::error!("Error getting registered gateways: {e:?}"),
137+
let portals = match contract_client.active_portals().await {
138+
Ok(portals) => portals,
139+
Err(e) => return log::error!("Error getting registered portals: {e:?}"),
140140
};
141-
*registered_gateways.lock() = gateways.into_iter().collect::<HashSet<PeerId>>();
141+
*registered_portals.lock() = portals.into_iter().collect::<HashSet<PeerId>>();
142142
}
143143
};
144144
self.task_manager.spawn_periodic(task, interval);

0 commit comments

Comments
 (0)