Merge pull request #3 from xtuc/feat-flock

Adds flock for concurrent access

Author: ashleygwilliams

Cargo.lock

@@ -18,6 +18,7 @@ is_executable = "0.1.2"
 siphasher = "0.2.3"
 tar = "0.4.16"
 zip = "0.5.0"
+fs2 = "0.4.3"
 
 [dev-dependencies]
 tempfile = "3.0.5"

Cargo.toml

@@ -5,18 +5,21 @@ extern crate curl;
 extern crate failure;
 extern crate dirs;
 extern crate flate2;
+extern crate fs2;
 extern crate hex;
 extern crate is_executable;
 extern crate siphasher;
 extern crate tar;
 extern crate zip;
 
 use failure::{Error, ResultExt};
+use fs2::FileExt;
 use siphasher::sip::SipHasher13;
 use std::collections::HashSet;
 use std::env;
 use std::ffi;
 use std::fs;
+use std::fs::File;
 use std::hash::{Hash, Hasher};
 use std::io;
 use std::path::{Path, PathBuf};
@@ -48,6 +51,9 @@ impl Cache {
                 Some(home.join(&cache_name))
             })
             .ok_or_else(|| format_err!("couldn't find your home directory, is $HOME not set?"))?;
+        if !destination.exists() {
+            fs::create_dir_all(&destination)?;
+        }
         Ok(Cache::at(&destination))
     }
 
@@ -87,6 +93,9 @@ impl Cache {
 
         let destination = self.destination.join(&dirname);
 
+        let flock = File::create(self.destination.join(&format!(".{}.lock", dirname)))?;
+        flock.lock_exclusive()?;
+
         if destination.exists() {
             return Ok(Some(Download { root: destination }));
         }
@@ -119,6 +128,8 @@ impl Cache {
         // Now that everything is ready move this over to our destination and
         // we're good to go.
         fs::rename(&temp, &destination)?;
+
+        flock.unlock()?;
         Ok(Some(Download { root: destination }))
     }