Split server into own file

signal-handling
Jef Roosens 2021-09-13 12:56:41 +02:00
parent f8a65ba2f3
commit 6961c064c0
Signed by: Jef Roosens
GPG Key ID: B580B976584B5F30
3 changed files with 69 additions and 40 deletions

View File

@ -10,3 +10,10 @@ edition = "2018"
tar = "0.4.37"
# Used to compress said tarballs using gzip
flate2 = "1.0.21"
# Used for backup filenames
chrono = "0.4.19"
[profile.release]
lto = "fat"
codegen-units = 1
panic = "abort"

View File

@ -1,47 +1,9 @@
use std::io;
use std::io::Write;
use std::process::{Child, Command, Stdio};
struct Server {
child: Child,
}
impl Server {
fn spawn() -> Server {
let child = Command::new("/usr/lib/jvm/java-16-openjdk/bin/java")
.arg("-jar")
.arg("paper-1.17.1-259.jar")
.arg("--universe")
.arg("worlds")
.arg("--nogui")
.stdin(Stdio::piped())
.spawn()
.expect("Couldn't start child process");
Server { child }
}
pub fn send_command(&mut self, cmd: &str) {
match cmd.trim() {
"stop" | "exit" => self.stop(),
s => self.custom(s),
}
}
fn custom(&mut self, cmd: &str) {
let mut stdin = self.child.stdin.as_ref().unwrap();
stdin.write_all(cmd.as_bytes()).unwrap();
stdin.flush().unwrap();
}
pub fn stop(&mut self) {
self.custom("stop");
self.child.wait();
}
}
mod server;
fn main() {
let mut server = Server::spawn();
let mut server = server::Server::spawn();
let stdin = io::stdin();
let input = &mut String::new();

60
src/server.rs 100644
View File

@ -0,0 +1,60 @@
use std::io;
use std::io::Write;
use std::process::{Child, Command, Stdio};
use std::fs::File;
use flate2::Compression;
use flate2::write::GzEncoder;
pub struct Server {
child: Child,
}
impl Server {
pub fn spawn() -> Server {
let child = Command::new("/usr/lib/jvm/java-16-openjdk/bin/java")
.arg("-jar")
.arg("paper-1.17.1-259.jar")
.arg("--universe")
.arg("worlds")
.arg("--nogui")
.stdin(Stdio::piped())
.spawn()
.expect("Couldn't start child process");
Server { child }
}
pub fn send_command(&mut self, cmd: &str) {
match cmd.trim() {
"stop" | "exit" => self.stop(),
"backup" => self.backup(),
s => self.custom(s),
}
}
fn custom(&mut self, cmd: &str) {
let mut stdin = self.child.stdin.as_ref().unwrap();
stdin.write_all(format!("{}\n", cmd.trim()).as_bytes()).unwrap();
stdin.flush().unwrap();
}
pub fn stop(&mut self) {
self.custom("stop");
self.child.wait();
}
pub fn backup(&mut self) {
// Make sure the server isn't modifying the files during the backup
self.custom("save-off");
self.custom("save-all");
// Create a gzip-compressed tarball of the worlds folder
let filename = format!("{}", chrono::offset::Local::now().format("backups/%Y-%m-%d_%H-%M-%S.tar.gz"));
let tar_gz = std::fs::File::create(filename).unwrap();
let enc = GzEncoder::new(tar_gz, Compression::default());
let mut tar = tar::Builder::new(enc);
tar.append_dir_all("worlds", "worlds").unwrap();
self.custom("save-on");
}
}