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
67
68
|
use serde_derive::{ Serialize, Deserialize };
use crate::backends::{ Backend, RunStatus };
use std::process::{ Command };
use std::path::Path;
use log::error;
use wait_timeout::ChildExt;
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct PythonBackend {
template: Option<String>,
version: Option<String>,
timeout: Option<f32>,
}
impl PythonBackend {
fn get_interpreter(&self) -> String {
format!(
"python{}",
self.version
.as_ref()
.unwrap_or(&String::new())
)
}
}
impl Backend for PythonBackend {
fn get_template(&self) -> Option<&str> {
match self.template {
Some(ref t) => Some(t),
None => None
}
}
fn run(&self, fname: &Path) -> std::io::Result<RunStatus> {
let timer = std::time::SystemTime::now();
let mut child = Command::new(self.get_interpreter())
.arg(fname.as_os_str())
.stdin(self.get_stdin(fname))
.spawn()?;
let timeout = std::time::Duration::from_secs_f32(self.timeout.unwrap_or(1.0));
match child.wait_timeout(timeout) {
Ok(maybe_status) => match maybe_status {
Some(status) => {
if !status.success() {
return Ok(RunStatus::ErrorCode(status.code().unwrap_or(0)));
}
Ok(RunStatus::Success)
},
None => {
let elapsed = timer.elapsed().unwrap_or(Default::default());
child.kill()?;
child.wait()?;
Ok(RunStatus::TimedOut(elapsed))
}
},
Err(err) => {
error!("could not wait for child: {}", err);
child.kill()?;
child.wait()?; // Wait defunct
Err(err)
}
}
}
}
|
