internethakai 0.2.0

data/lib/internethakai/scenario_handler.rb

@@ -0,0 +1,91 @@
+module InternetHakai
+    # = シナリオを扱うクラス
+    class ScenarioHandler < BaseHandler
+        #require 'resolv'
+        UNIQUE_BY_THREAD = false
+        attr_accessor :config
+        def init config
+            @config = config
+            @size = 0
+            @closed = false
+            @max_request = @config['max_scenario']
+            @max_request_counter = @max_request
+            @act_sum = 0
+            @scenarios = Hash::new
+            @logger = BaseHandler::get_handler(@config["logger"])
+        end
+        def create_scenario process_id=''
+            o = Scenario::new(@config)
+            @size += 1
+            o.scenario_id = (@size.to_s + process_id).to_i
+            o.create_scenario
+            #シナリオのサイズ
+            @scenarios[o.object_id] = o.size
+            @act_sum += o.size
+            o
+        end
+        #すべてのシナリオがつくられた
+        def on_scenarios_created
+            @loop_counter, @mod_counter = @act_sum.divmod(@max_request)
+            @queue = BaseHandler::get_handler(@config['queue'])
+        end
+        def set_on_complete &block
+            @on_complete_callback = block
+        end
+        def on_complete
+            @on_complete_callback.call() if @on_complete_callback
+        end
+        def set_on_turn_complete &block
+            @on_turn_complete_callback = block
+        end
+        def on_turn_complete
+            @queue.add([@on_turn_complete_callback,[]]) if @on_turn_complete_callback
+            @logger.run('o', 2)
+        end
+        def silence
+            def on_turn_complete
+                @queue.add([@on_turn_complete_callback,[]]) if @on_turn_complete_callback
+            end
+        end
+        # = ひとつのアクションが終了時に呼ばれる
+        def quit_action scenario_id
+            #ひとつのアクションが終了
+            @scenarios[scenario_id] -= 1
+            @max_request_counter -= 1
+            #puts "req_counter: #{@max_request_counter}"
+            if @max_request_counter == 0
+                #ひとつのアクション×リクエスト並列数が終了
+                on_turn_complete
+                @max_request_counter = @max_request
+                @loop_counter -= 1
+            end
+            #シナリオ並列数と、リクエスト並列数がちがうので、あまりの処理が必要
+            #puts "loop: #{@loop_counter}"
+            #if @loop_counter == 0
+            #    @mod_counter -= 1
+            #    if @mod_counter==0
+            #        on_turn_complete
+            #        on_complete
+            #    end
+            #end
+            if(@scenarios[scenario_id]==0)
+                #ひとつのシナリオが終了
+                @size -= 1
+                #@logger.run("size:#{@size}\n", 2)
+                if @size==0
+                    @logger.run("O", 2)
+                    on_complete
+                end
+            end
+        end
+        def closed?
+            @closed
+        end
+        def close
+            @closed = true
+        end
+        def complete?
+            @size <= 0
+        end
+    end
+end

data/lib/internethakai/time_register.rb

@@ -0,0 +1,53 @@
+module InternetHakai
+    # = 時間記録
+    class TimeRegister < BaseHandler
+        UNIQUE_BY_THREAD = false
+        def on_create
+            @report = ResponseRecord::new
+            @stack = []
+        end
+        def get_record
+            exec unless @stack.empty?
+            @report
+        end
+        def exec
+            for arg in @stack
+                do_regist(*arg)
+            end
+            @stack = []
+        end
+        def regist name, time, response, error
+            @stack << [name, time, response, error]
+        end
+        def do_regist name, time, response, error
+            time = time.to_f
+            v = @report[name]
+            if response.nil?
+                status = 0
+            else
+                status = response.status
+                v[:size]          += response.body.size
+            end
+
+            #if (response.nil? || (status != 200 && (status / 100 != 3)))
+            #    v[:errorcount] += 1 ## エラーカウント
+            #end
+            if v[:min].nil? and time != 0
+                v[:min] = time
+                v[:max] = time
+            elsif time != 0
+                v[:min] = v[:min] < time ? v[:min] : time
+            end
+            if time != 0
+                v[:max] = v[:max] > time ? v[:max] : time
+            end
+            v[:totaltime]     += time ## 総合時間
+            v[:accesscount]   += 1 if time > 0
+            v[:errorcount] += error
+            v["status:#{status}"] = v["status:#{status}"].to_i + 1 unless response.nil?
+            v["time:#{((time*1000).to_i/100)*100}"] = v["time:#{((time*1000).to_i/100)*100}"].to_i + 1
+            #detail は 一度だけ上書き
+            #v[:detail] += detail if(detail != "" && v[:detail]=="")
+        end
+    end
+end

data/lib/internethakai/util.rb

@@ -0,0 +1,130 @@
+module InternetHakai
+   class Util
+        require 'uri'
+        require 'kconv'
+        require 'nkf'
+       def self::encode_method(meth)
+           case meth
+               when 'GET'
+               GET
+               when 'POST'
+               POST
+               when 'PUT'
+               PUT
+               when 'DELETE'
+               DELETE
+           end
+       end
+        def self::time(msg, &b)
+            t = Time::now
+            b.call
+            d = (Time::now - t)*1000
+            puts "#{msg}: #{d}"
+        end
+        def self::hash2poststring(body)
+            body.map do |key,value|
+                "#{URI.encode(key)}=#{URI.encode(value.to_s)}"
+            end.join("&")
+        end
+        def self::set_params(url, params)
+            unless url.include? "?"
+                url += "?"
+            else
+                url[url.size-1] != "&"
+                url += "&"
+            end
+            q = []
+            params.each do |k,v|
+                one = URI::encode(k.to_s)
+                one += "="
+                one += URI::encode(v.to_s)
+                q << one
+            end
+            url + q.join("&")
+        end
+        def self::parse_url url
+            hostpath = url.slice(url.index('://')+3, url.size)
+            if pathstart = hostpath.index('/')
+                host = hostpath.slice(0, pathstart)
+                path = hostpath.slice(pathstart, hostpath.size)
+            else
+                host = hostpath
+                path = '/'
+            end
+            if portstart = host.index(':')
+                host, port = host.split(':')
+                port = port.to_i
+            else
+                port = 80
+            end
+            [host, port, path]
+        end
+        def self::parse_post_string ar
+            ar.map do |key,value|
+                "#{URI.encode(key)}=#{URI.encode(value.to_s)}"
+            end.join("&")
+        end
+        def self::get_query path
+            #ワンライナーで書けた!
+            Hash[ *path[/\?(.*)/, 1].to_s.split("&").map{|kv| ((v = kv.split("=")).size==2 ? v : []).map(&URI::method(:decode))}.flatten ]
+        end
+        def self::convert_encoding(str, params)
+            return if str.nil?
+            encoding = params["encoding"].upcase
+            opt = "-x#{self::get_encode_nkf(encoding)}W"
+            NKF::nkf(opt, str)
+            #Kconv::kconv(str, self::get_encode(encoding), Kconv::UTF8)
+        end
+        def self::recover_encoding(str, params)
+            return if str.nil?
+            encoding = params["encoding"].upcase
+            opt = "-xw#{self::get_encode_nkf_input(encoding)}"
+            NKF::nkf(opt, str)
+            #Kconv::kconv(str, Kconv::UTF8, self::get_encode(encoding))
+        end
+        def self::get_encode encode
+            case encode
+                when "SHIFT_JIS", "SJIS-WIN", "SHIFTJIS"
+                Kconv::SJIS
+                when "UTF-8"
+                Kconv::UTF8
+                when "JIS"
+                Kconv::JIS
+                when "EUC-JP"
+                Kconv::EUC
+                else
+                Kconv::UNKNOWN
+            end
+        end
+        def self::get_encode_nkf encode
+            case encode
+                when "SHIFT_JIS", "SJIS-WIN", "SHIFTJIS"
+                's'
+                when "UTF-8"
+                'w'
+                when "JIS"
+                'j'
+                when "EUC-JP"
+                'e'
+                else
+                nil
+            end
+        end
+        def self::get_encode_nkf_input encode
+            case encode
+                when "SHIFT_JIS", "SJIS-WIN", "SHIFTJIS"
+                'S'
+                when "UTF-8"
+                'W'
+                when "JIS"
+                'J'
+                when "EUC-JP"
+                'E'
+                else
+                nil
+            end
+        end
+    end
+
+
+end

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification 
+name: internethakai
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 2
+  - 0
+  version: 0.2.0
+platform: ruby
+authors: 
+- takada-at
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-02-26 00:00:00 +09:00
+default_executable: internethakai
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rev
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 3
+        - 2
+        version: 0.3.2
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: iobuffer
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 1
+        - 3
+        version: 0.1.3
+  type: :runtime
+  version_requirements: *id002
+description: "    scenario test tool for web application.\n"
+email: takada-at@klab.jp
+executables: 
+- hakaigen
+- internethakai
+extensions: []
+
+extra_rdoc_files: 
+- README
+- CHANGES
+files: 
+- lib/internethakai/logger.rb
+- lib/internethakai/executor.rb
+- lib/internethakai/hakairev.rb
+- lib/internethakai/scenario_handler.rb
+- lib/internethakai/main.rb
+- lib/internethakai/scenario_builder.rb
+- lib/internethakai/reporter.rb
+- lib/internethakai/scenario.tmpl
+- lib/internethakai/hakairev/http_client.rb.~1~
+- lib/internethakai/hakairev/executor.rb
+- lib/internethakai/hakairev/revpipe.rb
+- lib/internethakai/hakairev/monkey.rb
+- lib/internethakai/hakairev/http_client.rb
+- lib/internethakai/hakairev/time_register.rb
+- lib/internethakai/hakairev/task.rb
+- lib/internethakai/hakairev/timer_factory.rb
+- lib/internethakai/http_client.rb
+- lib/internethakai/client_handler.rb
+- lib/internethakai/action.rb
+- lib/internethakai/core.rb
+- lib/internethakai/time_register.rb
+- lib/internethakai/generator.rb
+- lib/internethakai/util.rb
+- lib/internethakai/concurrency_manager.rb
+- lib/internethakai/concurrency_manager.rb.~1~
+- lib/internethakai/response.rb
+- lib/internethakai/client_queue.rb
+- lib/internethakai/scenario.rb
+- lib/internethakai.rb.~1~
+- lib/internethakai.rb
+- bin/hakaigen
+- bin/internethakai
+- internethakai.gemspec
+- README
+- CHANGES
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- -U
+- -c
+- UTF-8
+- -t
+- !binary |
+  IuOCpOODs+OCv+ODvOODjeODg+ODiOegtOWjiiI=
+
+- --main=README
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 8
+      - 6
+      version: 1.8.6
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: internethakai
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: scenario test tool for web application
+test_files: []
+