Class: BinScript
- Inherits:
-
Object
- Object
- BinScript
- Includes:
- ClassLevelInheritableAttributes
- Defined in:
- lib/bin_script/bin_script.rb,
lib/bin_script/version.rb
Overview
require ‘active_support/core_ext/class/attribute.rb’
Constant Summary collapse
- VERSION =
"0.1.4"
- PARAMETER_TYPES =
Allowed parameter types. Equivalence aliases with GetoptLong constants.
{ :noarg => GetoptLong::NO_ARGUMENT, :optional => GetoptLong::OPTIONAL_ARGUMENT, :required => GetoptLong::REQUIRED_ARGUMENT }
Instance Attribute Summary collapse
-
#exit_status ⇒ Object
Place for store exit status.
-
#logger ⇒ Object
Place for logger object.
Class Method Summary collapse
-
.calculate_script_class_filename(parts) ⇒ Object
Prepare class name from filename parts.
-
.calculate_script_class_name(parts) ⇒ Object
Prepare class name from filename parts.
-
.get_argv_values ⇒ Object
Prepare ARGV parameters as hash.
-
.get_parameter(key) ⇒ Object
Get parameter by key.
-
.parse_script_file_name(filename) ⇒ Object
Parse script filename.
-
.remove_parameter(key) ⇒ Object
Remove parameter.
-
.run_script(filename = $0) ⇒ Object
Run script detected by the filename of source script file.
-
.script_name ⇒ Object
Prepare readable script name.
-
.usage(message = nil) ⇒ Object
Prepare usage message.
Instance Method Summary collapse
-
#do! ⇒ Object
Dummy for do! method.
-
#initialize ⇒ BinScript
constructor
Initialize script.
-
#lock_filename ⇒ Object
Prepare filename of log file.
-
#log_filename ⇒ Object
Prepare filename of log file.
-
#override_parameters(args) ⇒ Object
Override one or more parameters for testing purposes.
-
#params(key) ⇒ Object
Return parameter value by key.
- #puts(*arg) ⇒ Object
-
#run! ⇒ Object
Create lock file, call script code and unlock file even if error happend.
-
#usage_exit(message = nil) ⇒ Object
Print usage message and exit.
Methods included from ClassLevelInheritableAttributes
Constructor Details
#initialize ⇒ BinScript
Initialize script
217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 |
# File 'lib/bin_script/bin_script.rb', line 217 def initialize begin @source_argv = ARGV.dup @overrided_parameters = {} @params_values = (RailsStub.env == 'test' ? {} : self.class.get_argv_values) # Create logger if logging enabled if self.class.enable_logging @logger = XLogger.new(:file => log_filename, :dont_touch_rails_logger => (RailsStub.env == 'test')) @logger.level = self.class.log_level end rescue GetoptLong::InvalidOption, GetoptLong::MissingArgument, GetoptLong::NeedlessArgument => e usage_exit e. end end |
Instance Attribute Details
#exit_status ⇒ Object
Place for store exit status.
54 55 56 |
# File 'lib/bin_script/bin_script.rb', line 54 def exit_status @exit_status end |
#logger ⇒ Object
Place for logger object
51 52 53 |
# File 'lib/bin_script/bin_script.rb', line 51 def logger @logger end |
Class Method Details
.calculate_script_class_filename(parts) ⇒ Object
Prepare class name from filename parts
117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
# File 'lib/bin_script/bin_script.rb', line 117 def calculate_script_class_filename(parts) files = [] # Calculate and add to list file with script class itself class_file = File.join(%w{app models bin}, parts) class_file += '_nagios' if(parts.length > 1) class_file += '_script.rb' files << class_file # Add intermediate classes parts[0..-2].each { |p| files << "app/models/#{p}_script.rb" } files.reverse end |
.calculate_script_class_name(parts) ⇒ Object
Prepare class name from filename parts
106 107 108 109 110 111 112 113 114 |
# File 'lib/bin_script/bin_script.rb', line 106 def calculate_script_class_name(parts) # Calculate class name and paths from source script filename parts if(parts.length > 1) class_name = parts.map{|p| p.camelize}.join('::') + parts.first.camelize else class_name = parts.first.camelize end class_name += "Script" end |
.get_argv_values ⇒ Object
Prepare ARGV parameters as hash
171 172 173 174 175 176 177 |
# File 'lib/bin_script/bin_script.rb', line 171 def get_argv_values values = {} o = GetoptLong.new(*get_getoptlong_params) o.quiet = true # Don't write arg error to STDERR o.each { |opt, arg| values[opt[1..-1].to_sym] = arg } values end |
.get_parameter(key) ⇒ Object
Get parameter by key
78 79 80 81 82 |
# File 'lib/bin_script/bin_script.rb', line 78 def get_parameter(key) param = @parameters.find{|p| p[:key] == key || (p[:alias].present? && p[:alias].include?(key))} raise "Can't find parameter with key #{key.inspect} for class #{self.inspect}!" if param.nil? param end |
.parse_script_file_name(filename) ⇒ Object
Parse script filename. Extract important path parts
90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
# File 'lib/bin_script/bin_script.rb', line 90 def parse_script_file_name(filename) result = {} # Prepare important parts of source script filename parts = filename.split(File::SEPARATOR) parts = parts[parts.index('bin')+1..-1] parts.map!{|p| File.basename(p).split('.').first} result[:parts] = parts result[:class] = calculate_script_class_name(parts) result[:files] = calculate_script_class_filename(parts) result end |
.remove_parameter(key) ⇒ Object
Remove parameter
164 165 166 167 168 |
# File 'lib/bin_script/bin_script.rb', line 164 def remove_parameter(key) new = [] @parameters.each { |p| new << p if p[:key] != key } @parameters = new end |
.run_script(filename = $0) ⇒ Object
Run script detected by the filename of source script file
133 134 135 136 137 138 139 140 141 142 143 144 |
# File 'lib/bin_script/bin_script.rb', line 133 def run_script(filename = $0) cfg = parse_script_file_name(Pathname.new(filename).realpath.to_s) cfg[:files].each { |f| require File.join(RailsStub.root, f) } # Create instance and call run! for script class klass = cfg[:class].constantize script = klass.new script.run! # Exit with specified exit status exit script.exit_status || 0 end |
.script_name ⇒ Object
Prepare readable script name
85 86 87 |
# File 'lib/bin_script/bin_script.rb', line 85 def script_name self.to_s.underscore.gsub('/','__') end |
.usage(message = nil) ⇒ Object
Prepare usage message
180 181 182 183 184 185 186 187 188 189 |
# File 'lib/bin_script/bin_script.rb', line 180 def usage( = nil) usage_msg = '' usage_msg += "Error: #{}\n\n" unless .nil? usage_msg += "Use: ./bin/#{script_name}.rb [OPTIONS]\n\nAvailible options:\n\n" @parameters.each do |param| usage_msg += " #{prefix_key param[:key]} #{param[:description]}\n" end usage_msg += "\n" usage_msg end |
Instance Method Details
#do! ⇒ Object
Dummy for do! method
297 |
# File 'lib/bin_script/bin_script.rb', line 297 def do!; end |
#lock_filename ⇒ Object
Prepare filename of log file
330 331 332 |
# File 'lib/bin_script/bin_script.rb', line 330 def lock_filename params(:L).blank? ? File.join(RailsStub.root, 'locks', "#{self.class.script_name}.lock") : params(:L) end |
#log_filename ⇒ Object
Prepare filename of log file
335 336 337 |
# File 'lib/bin_script/bin_script.rb', line 335 def log_filename params(:l).blank? ? File.join(RailsStub.root, 'log', "#{self.class.script_name}#{log_filename_time_part}.log") : params(:l) end |
#override_parameters(args) ⇒ Object
Override one or more parameters for testing purposes
300 301 302 303 304 305 306 307 308 |
# File 'lib/bin_script/bin_script.rb', line 300 def override_parameters(args) if args.is_a?(Symbol) override_parameter(self.class.get_parameter(args)) elsif args.is_a?(Hash) args.each{|key, value| override_parameter(self.class.get_parameter(key), value)} else raise "Parameter should be Symbol or Hash" end end |
#params(key) ⇒ Object
Return parameter value by key
311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 |
# File 'lib/bin_script/bin_script.rb', line 311 def params(key) param = self.class.get_parameter(key) # Use dafault key (if call by alias) key = param[:key] case param[:type] when :noarg return (@overrided_parameters.has_key?(key) && @overrided_parameters[key]) || !@params_values[key].nil? when :optional #return nil unless @overrided_parameters.has_key?(key) || @params_values.has_key?(key) return @overrided_parameters[key] || @params_values[key] || param[:default] when :required value = @overrided_parameters[key] || @params_values[key] || param[:default] return value end end |
#puts(*arg) ⇒ Object
211 212 213 214 |
# File 'lib/bin_script/bin_script.rb', line 211 def puts(*arg) return if self.class.disable_puts_for_tests && RailsStub.env == 'test' Kernel.puts(*arg) end |
#run! ⇒ Object
Create lock file, call script code and unlock file even if error happend.
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 |
# File 'lib/bin_script/bin_script.rb', line 235 def run! # Print usage and exit if asked usage_exit if params(:h) # Create and check lock file if enabled if self.class.enable_locking @lock = LockFile.new(lock_filename) @lock.quiet = true # Don't write errors to STDERR info "Use lock file: #{@lock.path}" if(@lock.lock) warn "Lock file '#{@lock.path}' already open in exclusive mode. Exit!" exit end end begin # Log important info and call script job info "Log level = #{@logger.level}" if self.class.enable_logging info "Parameters: #{@params_values.inspect}" info "Starting script #{self.class.script_name}..." start = Time.now # Инкрементируем счетчик запусков этого скрипта inc_counter("#{self.class.script_name}_times") do! duration = Time.now - start info "Script #{self.class.script_name} finished!" info "Script job duration: #{duration}" info "Exit status: #{@exit_status}" if @exit_status # Инкрементируем время работы э inc_counter("#{self.class.script_name}_long", duration) # Log benchmarker info if it's not empty log_benchmarker_data rescue Exception => e # Print error info if it's not test env or exit if RailsStub.env != 'test' && e.class != SystemExit && e.class != Interrupt msg = self.class.(e) puts "\n" + msg fatal msg notify_about_error(e) end # Инкрементируем счетчик ошибок этого скрипта inc_counter("#{self.class.script_name}_raised") ensure # Unlock lock file @lock.unlock if self.class.enable_locking && @lock end end |
#usage_exit(message = nil) ⇒ Object
Print usage message and exit
290 291 292 293 294 |
# File 'lib/bin_script/bin_script.rb', line 290 def usage_exit( = nil) error "Exit with error message: #{}" Kernel.puts(self.class.usage()) exit end |