From 9219e0b364f3548c5ac253e5ceadcad1b2bf51bf Mon Sep 17 00:00:00 2001
From: Christine Dodrill <me@christine.website>
Date: Wed, 18 Dec 2019 12:39:47 +0000
Subject: [PATCH] full cache lifecycle

---
 src/main.rs | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 3a5b688..263ddcc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -40,9 +40,10 @@ fn begin(cache: State<Cache>, msg: Json<battlesnake::SnakeRequest>) -> Json<batt
         path: path,
     };
 
-    if let Some(x) = cache.lock().expect("wanted cache to be unlockable").get_mut(&msg.game.id) {
-        *x = gs;
-    };
+    cache
+        .lock()
+        .expect("wanted to lock cache")
+        .insert(msg.game.id.clone(), gs);
 
     Json(battlesnake::StartResponse{
         color: "#5ce8c3".to_string(),
@@ -71,6 +72,19 @@ fn find_path(
     }
 }
 
+#[post("/end", format = "json", data = "<msg>")]
+fn end(
+    cache_state: State<Cache>,
+    msg: Json<battlesnake::SnakeRequest>,
+) -> String {
+    cache_state
+        .lock()
+        .expect("wanted cache to be lockable")
+        .remove(&msg.game.id);
+
+    "OK".to_string()
+}
+
 #[post("/move", format = "json", data = "<msg>")]
 fn make_move(
     cache_state: State<Cache>,