internethakai 0.2.2 → 0.2.3

data/CHANGES

@@ -1,3 +1,6 @@
+0.2.3:
+* 一定時間経過するまで無限に試験をつづける機能(total_duration)を追加。
+
 0.2.2:
 * ファイルからランダムに選択した変数を使用できる機能(var_file, var_name)を追加。
 

data/README

@@ -29,6 +29,7 @@ CUIで使える高機能な負荷試験スクリプトです。
 [-l --loop] ループ回数を上書きします。
 [-g --log-level] ログレベルを上書きします。
 [-n --nth] n 番目のアクションだけを実行(0から数えます)。
+[-d --duration] 持続時間を上書きします。
 [--test] テストモードで実行。テストモードで実行すると、強制的に並列数を1、ループ数1、1プロセス、ログレベル3で実行します。リクエストが正常に送られているか確認するために使用できます。
 
 
@@ -37,6 +38,7 @@ YAMLの連想配列で設定項目を指定します。指定すべきキー名
 まずは hakaigen コマンドによる自動生成を試してみてください。
 
 [loop] 各シナリオのループ回数を指定します。
+[total_duration] 持続時間を指定します。total_durationを設定した場合、ループ回数は無視し、一定時間シナリオの実行をつづけます。単位は秒です。
 [max_request] HTTPクライアントの数を指定します。max_requestが10であればHTTPクライアントを10作成し、10のクライアントが同時にリクエストします。なおここで設定した値は全プロセスの合計値です。おすすめの値は "100" です(1プロセスあたり100コネクションくらいになるように調整しましょう)。
 [max_scenario] シナリオの数を指定します。デフォルトではmax_requestと同じ値になります。特に指定の必要はありませんが、アクセスするユーザーの数を増やしたい場合に多めに設定します。たとえば、max_requestが10、max_scenarioが20であれば、20*ループ数分のシナリオを10個のHTTPクラアントが順次実行していきます。<b>concurrencyというキー名から変更しました。</b>
 [log_level]
@@ -118,7 +120,7 @@ YAMLの連想配列で設定項目を指定します。指定すべきキー名
 ==== ファイルからランダムに取得
  user_id などをファイルからランダムに取得して使用できます。
  ファイルには改行区切りで、値のリストを書いておきます。シナリオごとに異なる値がセットされます。
- ファイルの名前、および変数の名前は var_file、var_name というキー名でセットしてください。
+ ファイルの名前、および変数の名前は var_file、var_name というキー名でセットします。
  var_file、var_name はactions の外に書きます。
 
  例.
@@ -126,7 +128,20 @@ YAMLの連想配列で設定項目を指定します。指定すべきキー名
  var_name: user_id
  actions:
    # user_id がファイルからランダムに選択される
-   - path: /somepath?user_id=%(user_id)%
+   - path: /somepath?id=%(user_id)%
+
+ 複数のファイルを使用したい場合は以下のように書いてください。
+
+  例.
+    vars:
+      - var_file: ./ids0
+        var_name: user_id
+      - var_file: ./ids1
+        var_name: item_id
+    actions:
+    # user_id と item_id がファイルからランダムに選択される
+      - path: /somepath?id=%(user_id)%
+      - path: /somepath?id=%(item_id)%
   
 
 ==== レスポンスで変数を設定

data/lib/internethakai.rb

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

data/lib/internethakai/client_handler.rb

@@ -2,7 +2,9 @@ module InternetHakai
     # = httpクライアントをいじるクラス
     class ClientHandler
         UNIQUE_BY_THREAD = true
+        @@var_stores = {}
         def self::on_config_load config
+            @@var_stores = {}
             config['actions'] = config['actions'].map do |actconfig|
                 actconfig['usr_var'] = false
                 if(/%\(.*\)%/ =~ actconfig['path'])
@@ -13,19 +15,38 @@ module InternetHakai
                 end
                 actconfig
             end
+            # ファイルでセットする変数
             if(config.has_key?('var_file') && config.has_key?('var_name'))
                 s = File::open(config['var_file']){|io|io.read}
+                var_ids = []
                 if s
                     lines = s.split("\n")
-                    @@var_ids = lines.sort_by{rand}
+                    var_ids = lines.sort_by{rand}
                 end
-                @@var_name = config['var_name']
+                unless var_ids.empty?
+                    var_name = config['var_name']
+                    @@var_stores[var_name] = var_ids
+                end
+            end
+            # 複数ファイルの場合
+            if(config.has_key?('vars') && config['vars'].is_a?(Array))
+                config['vars'].each do |c|
+                    next unless c.has_key?('var_file') && c.has_key?('var_name')
+                    s = File::open(c['var_file']){|io|io.read}
+                    next if s.nil? || s==''
+                    ids = s.split("\n").sort_by{rand}
+                    @@var_stores[c['var_name']] = ids unless ids.empty?
+                end
+            end
+            unless @@var_stores.empty?
                 def set_var_id
-                    ids = @@var_ids
-                    id = ids[@performance_id % ids.size]
-                    set_var(@@var_name, id)
+                    @@var_stores.each do |var_name, ids|
+                        id = ids[@performance_id % ids.size]
+                        set_var(var_name, id)
+                    end
                 end
             else
+                #変数がないときは何もしない
                 def set_var_id
                 end
             end
@@ -78,6 +99,8 @@ module InternetHakai
         def handle_client client
             @client = client
         end
+        def set_var_id
+        end
     end
 
     # = ケータイサイト用

data/lib/internethakai/concurrency_manager.rb

@@ -76,10 +76,7 @@ module InternetHakai
                 end
             rescue SystemExit
                 return
-            rescue Exception => e
-                raise e
-                puts "error: #{e.message}"
-                puts e.backtrace[0]
+            rescue Interrupt => e
                 if @fork_mode
                     #死ぬ前に親に連絡
                     Process::kill(:USR1, Process::ppid)

data/lib/internethakai/hakairev.rb

@@ -66,6 +66,13 @@ module InternetHakai
                 #end
                 sc.init
             end
+            if @config.has_key?('total_duration')
+                @config['infinity'] = true
+                timer = Rev::TimerWatcher::new(@config['total_duration'], false)
+                timer.on_timer(&method(:rev_on_complete))
+                timer.attach(Rev::Loop::default)
+                timer.enable unless timer.enabled?
+            end
 
             @threads[:main] = Thread::current
             @starttime = Time::now

data/lib/internethakai/main.rb

@@ -9,6 +9,7 @@ module InternetHakai
                 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
+                setting['duration'] = args[:duration] if args.has_key? :duration
             else
                 setting = nil
             end
@@ -51,7 +52,11 @@ module InternetHakai
             @logger.run(start + "\n", 2)
             @logger.run("Rev Mode\n", 2) if @config['rev']
             @logger.run("Target domain = #{@config['domain']}\n", 2)
-            @logger.run("Loop: #{@config['loop']}\n", 2)
+            if @config.has_key?('total_duration')
+                @logger.run("Duration: #{@config['total_duration']}\n", 2)
+            else
+                @logger.run("Loop: #{@config['loop']}\n", 2)
+            end
             @logger.run("Request Concurrency: #{@config['max_request_show']}\n", 2)
             @logger.run("Scenario Concurrency: #{@config['max_scenario_show']}\n", 2)
         end
@@ -83,6 +88,9 @@ module InternetHakai
             opts.on('-n NUM', '--nth', 'run only nth action') do |l|
                 args[:action_nth] = l.to_i
             end
+            opts.on('-d NUM', '--duration', 'duration-time') do |l|
+                args[:duration] = l.to_i
+            end
             opts.on_tail("-h", "--help", "show this message") do
                 puts opts
                 exit

data/lib/internethakai/scenario.rb

@@ -38,7 +38,13 @@ module InternetHakai
             if @action_counter == 0
                 @loop -= 1
                 #これで最後
-                return if @loop == 0
+                if @loop == 0
+                    if @config['infinity']
+                        @loop += 100 # 無限ループ!
+                    else
+                        return 
+                    end
+                end
                 @action_counter = @actions_size
                 @performance_id = (@scenario_id.to_s + sprintf(@tmpl, @loop)).to_i
             end

data/lib/internethakai/scenario_handler.rb

@@ -72,6 +72,10 @@ module InternetHakai
                 #ひとつのシナリオが終了
                 @size -= 1
                 #@logger.run("size:#{@size}\n", 2)
+                if @config['infinity']
+                    @scenarios[scenario_id] += 100
+                    return
+                end
                 if @size==0
                     @logger.run("O", 2)
                     on_complete

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 2
-  version: 0.2.2
+  - 3
+  version: 0.2.3
 platform: ruby
 authors: 
 - takada-at