-
Notifications
You must be signed in to change notification settings - Fork 125
/
Copy pathchrome_remote.rs
66 lines (55 loc) · 1.89 KB
/
chrome_remote.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
//! cargo run --example chrome_remote --features="chrome chrome_intercept"
extern crate spider;
use crate::spider::tokio::io::AsyncWriteExt;
use spider::features::chrome_common::RequestInterceptConfiguration;
use spider::tokio;
use spider::website::Website;
use std::io::Result;
async fn crawl_website(url: &str) -> Result<()> {
let mut website: Website = Website::new(url)
.with_limit(500)
.with_chrome_intercept(RequestInterceptConfiguration::new(true))
.with_stealth(true)
.with_fingerprint(true)
// .with_proxies(Some(vec!["http://localhost:8888".into()]))
.with_chrome_connection(Some("http://127.0.0.1:9222/json/version".into()))
.build()
.unwrap();
let mut rx2 = website.subscribe(16).unwrap();
let mut stdout = tokio::io::stdout();
tokio::spawn(async move {
while let Ok(page) = rx2.recv().await {
let _ = stdout
.write_all(
format!(
"- {} -- Bytes transferred {:?} -- HTML Size {:?}\n",
page.get_url(),
page.bytes_transferred.unwrap_or_default(),
page.get_html_bytes_u8().len()
)
.as_bytes(),
)
.await;
}
});
let start = crate::tokio::time::Instant::now();
website.crawl().await;
let duration = start.elapsed();
let links = website.get_all_links_visited().await;
println!(
"Time elapsed in website.crawl({}) is: {:?} for total pages: {:?}",
website.get_url(),
duration,
links.len()
);
Ok(())
}
#[tokio::main]
async fn main() -> Result<()> {
let _ = tokio::join!(
crawl_website("https://choosealicense.com"),
crawl_website("https://jeffmendez.com"),
crawl_website("https://example.com"),
);
Ok(())
}