internethakai 0.2.0 → 0.2.1

data/CHANGES

@@ -1,12 +1,16 @@
+0.2.1:
+* 実行時オプションのバグを修正
+* -g オプション、-n オプションを追加
+
 0.2.0:
 * accesslog2yml を削除
-* hakaigen を追加
-* <b>重要</b>: concurrency を max_scenario に、fork を max_process に、変更しました(しばらくは後方互換性を残します)。
+* 設定ファイル生成スクリプトhakaigen を追加
+* <b>重要</b>: concurrency を max_scenario に、fork を max_process に、変更しました(しばらくは後方互換性を残します)
 
 0.1.9:
 * 変数の記号を変更
-* <b>重要</b>: %%VAR%% という変数が使えなくなります。かわりに %(VAR)% を使用してください。
-* WAITする機能を追加。
+* <b>重要</b>: %%VAR%% という変数が使えなくなります。かわりに %(VAR)% を使用してください
+* WAITする機能を追加
 
 0.1.8:
 * オプション追加。cookie に対応

data/README

@@ -27,6 +27,8 @@ CUIで使える高機能な負荷試験スクリプトです。
 [-r --max-request] リクエスト並列数を上書きします。
 [-s --max-scenario] シナリオ並列数を上書きします。
 [-l --loop] ループ回数を上書きします。
+[-g --log-level] ログレベルを上書きします。
+[-n --nth] n 番目のアクションだけを実行(0から数えます)。
 [--test] テストモードで実行。テストモードで実行すると、強制的に並列数を1、ループ数1、1プロセス、ログレベル3で実行します。リクエストが正常に送られているか確認するために使用できます。
 
 

data/internethakai.gemspec

@@ -14,7 +14,7 @@ GEMSPEC = Gem::Specification::new do |s|
     s.executables = Dir::glob("bin/*").map{|f|File::basename(f)}
     s.default_executable = 'internethakai'
     s.add_dependency('rev', '= 0.3.2')
-    s.add_dependency('iobuffer', '= 0.1.3')
+    s.add_dependency('iobuffer', '>= 0.1.3')
     s.files = Dir::glob("{lib,bin}/**/*") + ['internethakai.gemspec']
     s.has_rdoc = true
     s.rdoc_options = %w(-U -c UTF-8 -t "インターネット破壊" --main=README)

data/lib/internethakai/concurrency_manager.rb

@@ -251,7 +251,7 @@ module InternetHakai
                 flag = true
                 puts "error: SIGINT"
                 @pids.each do |pid|
-                    Process::kill(:USR2, pid)
+                    Process::kill(:USR2, pid) if pid
                 end
             }
             Signal::trap(:USR2){

data/lib/internethakai/main.rb

@@ -1,11 +1,21 @@
 module InternetHakai
     class Main
         require 'optparse'
+        attr_reader :config
         def load_config configpath, basepath = nil, args = nil
-            args ||= {}
+            if args
+                setting = {}
+                setting['loop'] = args[:loop] if args.has_key? :loop
+                setting['max_request'] = args[:max_request] if args.has_key? :max_request
+                setting['max_scenario'] = args[:concurrency] if args.has_key? :concurrency
+                setting['log_level'] = args[:log_level] if args.has_key? :log_level
+            else
+                setting = nil
+            end
+
             @basepath = basepath || File::dirname(configpath)
             @scenario_builder = ScenarioBuilder::new(nil)
-            @scenario_builder.load(configpath)
+            @scenario_builder.load(configpath, setting)
             @scenario_handler = BaseHandler::get_handler("ScenarioHandler")
             @config = @scenario_builder.get_config
             BaseHandler::set_config(@config)
@@ -20,16 +30,16 @@ module InternetHakai
             start = starttime.strftime( '%Y%m%d_%H%M%S' )
             reporter.set_dir(@basepath)
             reporter.init_filename(start)
-            @config['loop'] = args[:loop] if args.has_key? :loop
-            @config['max_request'] = args[:max_request] if args.has_key? :max_request
-            @config['max_scenario'] = args[:concurrency] if args.has_key? :concurrency
+            if args && args.has_key?(:action_nth)
+                @config['actions'] = @config['actions'][args[:action_nth], 1]
+            end
             if @test
                 @config['loop'] = 1
                 @config['max_request'] = 1
                 @config['concurrency'] = 1
                 @config['max_scenario'] = 1
+                @config['max_process'] = 1
                 @config.delete('fork')
-                @config.delete('max_process')
                 @config['max_request_show'] = 1
                 @config['max_scenario_show'] = 1
                 @config['log_level'] = 3
@@ -47,13 +57,6 @@ module InternetHakai
         end
         def main config=nil
             #引数か、同じディレクトリの「scenario.yml」を読み込みます。
-            if config.nil?
-                config = ARGV[0]
-            end
-            unless config
-                puts "Usage: #{File::basename($0)} SCENARIO"
-                exit
-            end
             args = {}
             @test = false
             opts = OptionParser.new("Usage: #{File::basename($0)} SCENARIO")
@@ -65,14 +68,20 @@ module InternetHakai
             opts.on("--test", "exec in testmode") do
                 @test = true
             end
-            opts.on('-l', '--loop' 'loop') do |loop|
-                args[:loop] = loop
+            opts.on('-l NUM', '--loop' 'loop') do |loop|
+                args[:loop] = loop.to_i
             end
-            opts.on('-s', '--max-scenario', 'max-scenario') do |conc|
-                args[:concurrency] = conc
+            opts.on('-s NUM', '--max-scenario', 'max-scenario') do |conc|
+                args[:concurrency] = conc.to_o
             end
-            opts.on('-r', '--max-request', 'max-request') do |mr|
-                args[:max_request] = mr
+            opts.on('-r NUM', '--max-request', 'max-request') do |mr|
+                args[:max_request] = mr.to_i
+            end
+            opts.on('-g NUM', '--log-level', 'log-level') do |l|
+                args[:log_level] = l.to_i
+            end
+            opts.on('-n NUM', '--nth', 'run only nth action') do |l|
+                args[:action_nth] = l.to_i
             end
             opts.on_tail("-h", "--help", "show this message") do
                 puts opts
@@ -80,6 +89,13 @@ module InternetHakai
             end
             opts.version = InternetHakai::VERSION
             opts.parse!(ARGV)
+            if config.nil?
+                config = ARGV[0]
+            end
+            unless config
+                puts "Usage: #{File::basename($0)} SCENARIO"
+                exit
+            end
 
             load_config config, nil, args
 

data/lib/internethakai/scenario_builder.rb

@@ -148,16 +148,16 @@ module InternetHakai
             ch = BaseHandler::get_class(@config["client_handler"])
             @config = ch::on_config_load(@config)
         end
-        def load path
+        def load path, setting=nil
             begin
                 basedir = File::expand_path(File::dirname(path))
                 str = File::open(path){|io|io.read}
-                load_config(str, basedir)
+                load_config(str, basedir, setting)
             rescue
                 raise $!
             end
         end
-        def load_config str, basedir='.'
+        def load_config str, basedir='.', setting=nil
             default_config = DEFAULT_CONFIG
             if(RUBY_VERSION.to_f >= 1.9)
                 str.force_encoding('UTF-8')
@@ -167,6 +167,7 @@ module InternetHakai
                 str.gsub!('%%BASEDIR%%', basedir)
             end
             @config = ::YAML::load(str)
+            @config = @config.merge(setting) if setting
             raise "invalid file" unless @config
             @config = default_config.merge(@config)
             prepare_config

data/lib/internethakai.rb

@@ -18,7 +18,7 @@ require 'internethakai/generator'
 
 module InternetHakai
     # インターネット破壊
-    VERSION = '0.2.0'
+    VERSION = '0.2.1'
     GET="GET"
     POST="POST"
     PUT="PUT"

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 0
-  version: 0.2.0
+  - 1
+  version: 0.2.1
 platform: ruby
 authors: 
 - takada-at
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-26 00:00:00 +09:00
+date: 2011-03-04 00:00:00 +09:00
 default_executable: internethakai
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -36,7 +36,7 @@ dependencies:
   prerelease: false
   requirement: &id002 !ruby/object:Gem::Requirement 
     requirements: 
-    - - "="
+    - - ">="
       - !ruby/object:Gem::Version 
         segments: 
         - 0