95 lines
2.4 KiB
Rust
95 lines
2.4 KiB
Rust
|
use http::{header, HeaderMap, StatusCode};
|
||
|
|
use reqwest::Client;
|
||
|
|
use std::process::Command;
|
||
|
|
use std::time::{Duration, SystemTime};
|
||
|
|
use tokio::time::sleep;
|
||
|
|
use tracing::{debug, error, info, warn};
|
||
|
|
|
||
|
|
pub struct DockerContainer {
|
||
|
|
id: String,
|
||
|
|
running: bool,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub const DOCKER_EXPOSED_PORT: usize = 8000;
|
||
|
|
|
||
|
|
impl DockerContainer {
|
||
|
|
pub fn start(version: &str, file_path: &str, user: &str, pass: &str) -> Self {
|
||
|
|
let docker_image = format!("surrealdb/surrealdb:{version}");
|
||
|
|
info!("Start Docker image {docker_image} with file {file_path}");
|
||
|
|
let mut args =
|
||
|
|
Arguments::new(["run", "-p", &format!("127.0.0.1:8000:{DOCKER_EXPOSED_PORT}"), "-d"]);
|
||
|
|
args.add([docker_image]);
|
||
|
|
args.add(["start", "--auth", "--user", user, "--pass", pass]);
|
||
|
|
args.add([format!("file:{file_path}")]);
|
||
|
|
let id = Self::docker(args);
|
||
|
|
Self {
|
||
|
|
id,
|
||
|
|
running: true,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn logs(&self) {
|
||
|
|
info!("Logging Docker container {}", self.id);
|
||
|
|
Self::docker(Arguments::new(["logs", &self.id]));
|
||
|
|
}
|
||
|
|
pub fn stop(&mut self) {
|
||
|
|
if self.running {
|
||
|
|
info!("Stopping Docker container {}", self.id);
|
||
|
|
Self::docker(Arguments::new(["stop", &self.id]));
|
||
|
|
self.running = false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn extract_data_dir(&self, file_path: &str) {
|
||
|
|
let container_src_path = format!("{}:{file_path}", self.id);
|
||
|
|
info!("Extract directory from Docker container {}", container_src_path);
|
||
|
|
Self::docker(Arguments::new(["cp", &container_src_path, file_path]));
|
||
|
|
}
|
||
|
|
|
||
|
|
fn docker(args: Arguments) -> String {
|
||
|
|
let mut command = Command::new("docker");
|
||
|
|
|
||
|
|
let output = command.args(args.0).output().unwrap();
|
||
|
|
let std_out = String::from_utf8(output.stdout).unwrap().trim().to_string();
|
||
|
|
if !output.stderr.is_empty() {
|
||
|
|
error!("{}", String::from_utf8(output.stderr).unwrap());
|
||
|
|
}
|
||
|
|
assert_eq!(output.status.code(), Some(0), "Docker command failure: {:?}", command);
|
||
|
|
std_out
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Drop for DockerContainer {
|
||
|
|
fn drop(&mut self) {
|
||
|
|
// Be sure the container is stopped
|
||
|
|
self.stop();
|
||
|
|
// Delete the container
|
||
|
|
info!("Delete Docker container {}", self.id);
|
||
|
|
Self::docker(Arguments::new(["rm", &self.id]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
struct Arguments(Vec<String>);
|
||
|
|
|
||
|
|
impl Arguments {
|
||
|
|
fn new<I, S>(args: I) -> Self
|
||
|
|
where
|
||
|
|
I: IntoIterator<Item = S>,
|
||
|
|
S: Into<String>,
|
||
|
|
{
|
||
|
|
let mut a = Self(vec![]);
|
||
|
|
a.add(args);
|
||
|
|
a
|
||
|
|
}
|
||
|
|
|
||
|
|
fn add<I, S>(&mut self, args: I)
|
||
|
|
where
|
||
|
|
I: IntoIterator<Item = S>,
|
||
|
|
S: Into<String>,
|
||
|
|
{
|
||
|
|
for arg in args {
|
||
|
|
self.0.push(arg.into());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|