Fix store_session impl

Fixes #3

Author: yoshuawuyts

Diff for: Cargo.toml

@@ -22,6 +22,7 @@ mongodb = { git = "https://github.com/saghm/mongo-rust-driver", default-features
 async-session = "2.0.0"
 async-trait = "0.1.36"
 serde_json = "1.0.56"
+# chrono = "0.4.13"
 
 [dev-dependencies]
 async-std = { version = "1.6.2", features = ["attributes"] }

Diff for: src/lib.rs

@@ -12,8 +12,11 @@
 
 use async_session::{Result, Session, SessionStore};
 use async_trait::async_trait;
+// use mongodb::bson::{doc, DateTime};
 use mongodb::bson::doc;
+use mongodb::options::ReplaceOptions;
 use mongodb::Client;
+// use chrono::offset::Utc;
 
 /// A MongoDB session store.
 #[derive(Debug, Clone)]
@@ -48,19 +51,19 @@ impl SessionStore for MongodbSessionStore {
         // TODO: mongodb supports TTL for auto-expiry somehow, need to figure out how!
         let value = serde_json::to_string(&session)?;
         let id = session.id();
-        let doc = doc! { "session_id": id, "session": value };
-        coll.insert_one(doc, None).await?;
+        let query = doc! { "session_id": id };
+        let replacement = doc! { "session_id": id, "session": value };
+        let opts = ReplaceOptions::builder().upsert(true).build();
+        coll.replace_one(query, replacement, Some(opts)).await?;
 
         Ok(session.into_cookie_value())
     }
 
     async fn load_session(&self, cookie_value: String) -> Result<Option<Session>> {
         let id = Session::id_from_cookie_value(&cookie_value)?;
         let coll = self.client.database(&self.db).collection(&self.coll_name);
-
         let filter = doc! { "session_id": id };
-        let result = coll.find_one(filter, None).await?;
-        match result {
+        match coll.find_one(filter, None).await? {
             None => Ok(None),
             Some(doc) => Ok(Some(serde_json::from_str(doc.get_str("session")?)?)),
         }