hq-tools 0.5.0 → 0.6.0

data/lib/hq/tools/lock.rb

@@ -0,0 +1,80 @@
+module HQ
+module Tools
+module Lock
+
+	def self.lock_simple filename
+
+		# create a lock file, will raise EEXIST if it exists
+
+		mode = File::WRONLY | File::CREAT | File::EXCL
+		File.open filename, mode do |file|
+
+			# flock it
+
+			file.flock File::LOCK_EX | File::LOCK_NB \
+				or raise "Cannot obtain lock"
+
+			begin
+
+				# write our pid
+
+				file.puts $$
+				file.flush
+
+				# yield
+
+				yield
+
+			ensure
+
+				# delete lock file
+
+				File.unlink filename
+
+			end
+
+		end
+
+	end
+
+	def self.lock_remove filename
+
+		# open existing lock file
+
+		File.open filename, File::WRONLY do |file|
+
+			# lock it
+
+			file.flock File::LOCK_EX | File::LOCK_NB \
+				or raise "Cannot obtain lock"
+
+			# delete lock file
+
+			File.unlink filename
+
+		end
+
+	end
+
+	def self.lock filename, &proc
+
+		while true
+
+			begin
+
+				lock_simple filename, &proc
+				return
+
+			rescue Errno::EEXIST => e
+
+				lock_remove filename
+
+			end
+
+		end
+
+	end
+
+end
+end
+end

data/spec/hq/tools/lock-spec.rb

@@ -0,0 +1,33 @@
+require "tmpdir"
+
+require "hq/tools/lock"
+
+module HQ
+module Tools
+describe Lock do
+
+context "#lock" do
+
+	before do
+		@old_dir = Dir.pwd
+		@temp_dir = Dir.mktmpdir
+		Dir.chdir @temp_dir
+	end
+
+	it "calls the provided block" do
+		called = false
+		subject.lock "lockfile" do
+			called = true
+		end
+		called.should == true
+	end
+
+	after do
+		Dir.chdir @old_dir
+	end
+
+end
+
+end
+end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hq-tools
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
   prerelease: 
 platform: ruby
 authors:
@@ -100,8 +100,10 @@ extra_rdoc_files: []
 files:
 - lib/hq/tools/getopt.rb
 - lib/hq/tools/cron.rb
+- lib/hq/tools/lock.rb
 - lib/hq/tools/escape.rb
 - lib/hq/tools/check-script.rb
+- spec/hq/tools/lock-spec.rb
 - spec/hq/tools/getopt-spec.rb
 - spec/hq/tools/cron-spec.rb
 - spec/hq/tools/check-script-spec.rb
@@ -130,6 +132,7 @@ signing_key:
 specification_version: 3
 summary: HQ tools
 test_files:
+- spec/hq/tools/lock-spec.rb
 - spec/hq/tools/getopt-spec.rb
 - spec/hq/tools/cron-spec.rb
 - spec/hq/tools/check-script-spec.rb