Kaynağa Gözat

Merge pull request #7 from rust-lang/dockerize-2

More changes needed for Docker
Mark Rousskov 6 yıl önce
ebeveyn
işleme
31141b0dfb
2 değiştirilmiş dosya ile 10 ekleme ve 2 silme
  1. 2 1
      Dockerfile
  2. 8 1
      src/main.rs

+ 2 - 1
Dockerfile

@@ -52,4 +52,5 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
     ca-certificates
 
 COPY --from=build /tmp/source/target/release/triagebot /usr/local/bin/
-CMD ["triagebot"]
+ENV TRIAGEBOT_PORT=80
+CMD triagebot

+ 8 - 1
src/main.rs

@@ -117,7 +117,14 @@ fn main() {
     let username = Arc::new(User::current(&gh).unwrap().login);
     let mut registry = HandleRegistry::new();
     handlers::register_all(&mut registry, gh.clone(), username);
-    rocket::ignite()
+
+    let mut config = rocket::Config::active().unwrap();
+    config.set_port(
+        env::var("TRIAGEBOT_PORT")
+            .map(|port| port.parse().unwrap())
+            .unwrap_or(8000),
+    );
+    rocket::custom(config)
         .manage(gh)
         .manage(registry)
         .mount("/", routes![webhook])