YGGverse/btracker
1
0
Fork 0
mirror of https://github.com/YGGverse/btracker.git synced 2026-03-31 17:15:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update UDP scrapper config api

Browse source
This commit is contained in:
yggverse 2025-08-07 15:49:48 +03:00
parent 3b0dd1942f
commit 1150f93817
2 changed files with 32 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/main.rs
View file

@ -121,8 +121,31 @@ fn rocket() -> _ {
);
let scraper = Scraper::init(
config
.scrape_udp_server
.map(|s| (config.scrape_udp_client, s)),
.scrape
.map(|u| {
u.into_iter()
.map(|url| {
use std::str::FromStr;
if url.scheme() == "tcp" {
todo!("TCP scrape is not implemented")
}
if url.scheme() != "udp" {
todo!("Scheme `{}` is not supported", url.scheme())
}
std::net::SocketAddr::new(
std::net::IpAddr::from_str(
url.host_str()
.expect("Required valid host value")
.trim_start_matches('[')
.trim_end_matches(']'),
)
.unwrap(),
url.port().expect("Required valid port value"),
)
})
.collect()
})
.map(|a| (config.udp, a)),
);
let storage = Storage::init(config.preload, config.list_limit, config.capacity).unwrap(); // @TODO handle
rocket::build()