Add Docker Support

Author: SIMULATAN

Cargo.lock

@@ -8,3 +8,4 @@ repository = "https://github.com/SIMULATAN/ErrorServer"
 
 [dependencies]
 regex = "1.7.0"
+signal-hook = "0.3.14"

Cargo.toml

@@ -0,0 +1,19 @@
+FROM rust:alpine as builder
+
+WORKDIR /app
+
+RUN rustup toolchain install nightly
+
+COPY Cargo.toml Cargo.lock ./
+COPY src ./src
+
+RUN cargo +nightly build --release -Z sparse-registry
+
+FROM alpine
+
+WORKDIR /app
+
+COPY error.html .
+COPY --from=builder /app/target/release/ErrorServer .
+
+ENTRYPOINT [ "/app/ErrorServer" ]

Dockerfile

@@ -5,8 +5,10 @@ use regex::Regex;
 use crate::http_codes::get_code;
 
 mod http_codes;
+mod signals;
 
 fn main() {
+    signals::setup();
     let listener = TcpListener::bind("0.0.0.0:7878").unwrap();
     println!("Listening on :7878");
     let regex: Regex = Regex::new(r"GET /(.*) HTTP/\d.\d").unwrap();

src/main.rs

@@ -0,0 +1,14 @@
+use std::thread;
+use signal_hook::consts::{SIGINT, SIGTERM};
+use signal_hook::iterator::Signals;
+
+pub fn setup() {
+    let mut signals = Signals::new(&[SIGINT, SIGTERM]).unwrap();
+
+    thread::spawn(move || {
+        for sig in signals.forever() {
+            println!("Received signal {:?}", sig);
+            std::process::exit(0);
+        }
+    });
+}