paul/autarco-scraper
paul/autarco-scraper
Archived
1
0
Fork 0
Code Issues 2 Pull requests Releases Activity

Pass the gecko driver port as spawn argument

Browse source
This commit is contained in:
Paul van Tilburg 2020-10-09 17:44:03 +02:00
parent 6b60c064ce
commit 0866da3b57
Signed by: paul
GPG key ID: C6DE073EDA9EEC4D
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/main.rs
View file

@ -3,6 +3,7 @@ use lazy_static::lazy_static;
use rocket::{get, launch, routes, Rocket}; use rocket::{get, launch, routes, Rocket};
use rocket_contrib::json::Json; use rocket_contrib::json::Json;
use serde::Serialize; use serde::Serialize;
use std::process::{Child, Command, Stdio};
use std::sync::Mutex; use std::sync::Mutex;
use std::thread; use std::thread;
use std::time::{Duration, SystemTime}; use std::time::{Duration, SystemTime};
@ -15,18 +16,16 @@ const POLL_INTERVAL: u64 = 300;
const GECKO_DRIVER_PORT: u16 = 18019; const GECKO_DRIVER_PORT: u16 = 18019;
use std::process::{Child, Command, Stdio};
struct GeckoDriver(Child); struct GeckoDriver(Child);
impl GeckoDriver { impl GeckoDriver {
pub fn spawn() -> Result<Self> { pub fn spawn(port: u16) -> Result<Self> {
// This is taken from the webdriver-client crate. // This is taken from the webdriver-client crate.
let child = Command::new("geckodriver") let child = Command::new("geckodriver")
.arg("-b") .arg("-b")
.arg("firefox") .arg("firefox")
.arg("--port") .arg("--port")
.arg(format!("{}", GECKO_DRIVER_PORT)) .arg(format!("{}", port))
.stdin(Stdio::null()) .stdin(Stdio::null())
.stderr(Stdio::null()) .stderr(Stdio::null())
.stdout(Stdio::null()) .stdout(Stdio::null())
@ -78,7 +77,7 @@ lazy_static! {
async fn update_loop() -> Result<()> { async fn update_loop() -> Result<()> {
color_eyre::install()?; color_eyre::install()?;
let _gecko_driver = GeckoDriver::spawn()?; let _gecko_driver = GeckoDriver::spawn(GECKO_DRIVER_PORT)?;
let mut caps = DesiredCapabilities::firefox(); let mut caps = DesiredCapabilities::firefox();
caps.set_headless()?; caps.set_headless()?;
let driver = WebDriver::new(&format!("http://localhost:{}", GECKO_DRIVER_PORT), &caps).await?; let driver = WebDriver::new(&format!("http://localhost:{}", GECKO_DRIVER_PORT), &caps).await?;