Show More
Commit Description:
add option -A <opt> to box. This options allow more argument to be explicitly passed to the program...
Commit Description:
add option -A <opt> to box. This options allow more argument to be explicitly passed to the program
We have to use this because if the argument we wish to pass to the program is option (in -? format),
box will intepret it as its option and failed accordingly.
be noted that, by the definition of getopt, these options will be put after original argument
(check the code for more info)
References:
File last commit:
Show/Diff file:
Action:
Last Author
Show All-
Jittat Fakcharoenphol-
lib/dir_init.rb
| 113 lines
| 2.7 KiB
| text/x-ruby
| RubyLexer
|
Jittat Fakcharoenphol
|
r137 | require 'fileutils' | ||
jittat
|
r71 | |||
| # DirInit::Manager handles directory initialization and clean-up when | ||||
| # there are many concurrent processes that wants to modify the | ||||
| # directory in the same way. | ||||
| # | ||||
| # An example usage is when each process wants to copy some temporary | ||||
| # files to the directory and delete these files after finishing its | ||||
| # job. Problems may occur when the first process delete the files | ||||
| # while the second process is still using the files. | ||||
| # | ||||
| # This library maintain a reference counter on the processes using the | ||||
| # directory. It locks the dir to manage critical section when | ||||
| # updating the reference counter. | ||||
| module DirInit | ||||
| class Manager | ||||
| def initialize(dir_name, usage_filename='.usage_counter') | ||||
| @dir_name = dir_name | ||||
| @usage_filename = usage_filename | ||||
| end | ||||
|
Jittat Fakcharoenphol
|
r109 | def lock_filename | ||
| return @dir_name + '/lockfile' | ||||
| end | ||||
|
jittat
|
r71 | # Check if someone has initialized the dir. If not, call block. | ||
| def setup # :yields: block | ||||
|
Jittat Fakcharoenphol
|
r109 | dir = File.new(lock_filename,"w+") | ||
|
jittat
|
r71 | dir.flock(File::LOCK_EX) | ||
| begin | ||||
| counter_filename = get_counter_filename | ||||
| if File.exist? counter_filename | ||||
| # someone is here | ||||
| f = File.new(counter_filename,"r+") | ||||
| counter = f.read.to_i | ||||
| f.seek(0) | ||||
| f.write("#{counter+1}\n") | ||||
| f.close | ||||
| else | ||||
| # i'm the first, create the counter file | ||||
| counter = 0 | ||||
| f = File.new(counter_filename,"w") | ||||
| f.write("1\n") | ||||
| f.close | ||||
| end | ||||
| # if no one is here | ||||
| if counter == 0 | ||||
| if block_given? | ||||
| yield | ||||
| end | ||||
| end | ||||
| rescue | ||||
| raise | ||||
| ensure | ||||
| # make sure it unlock the directory | ||||
| dir.flock(File::LOCK_UN) | ||||
|
Jittat Fakcharoenphol
|
r96 | dir.close | ||
|
jittat
|
r71 | end | ||
| end | ||||
| # Check if I am the last one using the dir. If true, call block. | ||||
| def teardown | ||||
|
Jittat Fakcharoenphol
|
r109 | dir = File.new(lock_filename) | ||
|
jittat
|
r71 | dir.flock(File::LOCK_EX) | ||
| begin | ||||
| counter_filename = get_counter_filename | ||||
| if File.exist? counter_filename | ||||
| # someone is here | ||||
| f = File.new(counter_filename,"r+") | ||||
| counter = f.read.to_i | ||||
| f.seek(0) | ||||
| f.write("#{counter-1}\n") | ||||
| f.close | ||||
| if counter == 1 | ||||
| # i'm the last one | ||||
| File.delete(counter_filename) | ||||
| if block_given? | ||||
| yield | ||||
| end | ||||
| end | ||||
| else | ||||
| # This is BAD | ||||
| raise "Error: reference count missing" | ||||
| end | ||||
| rescue | ||||
| raise | ||||
| ensure | ||||
| # make sure it unlock the directory | ||||
| dir.flock(File::LOCK_UN) | ||||
|
Jittat Fakcharoenphol
|
r96 | dir.close | ||
|
jittat
|
r71 | end | ||
| end | ||||
| protected | ||||
| def get_counter_filename | ||||
| return File.join(@dir_name,@usage_filename) | ||||
| end | ||||
| end | ||||
| end | ||||