Kouhei Sutou
null+****@clear*****
Mon Aug 12 11:59:59 JST 2013
Kouhei Sutou 2013-08-12 11:59:59 +0900 (Mon, 12 Aug 2013) New Revision: 9baf26d49275e6e9995130e75934f8649fe59cd3 https://github.com/groonga/grntest/commit/9baf26d49275e6e9995130e75934f8649fe59cd3 Message: Extract executors Added files: lib/grntest/executors.rb lib/grntest/executors/base-executor.rb lib/grntest/executors/http-executor.rb lib/grntest/executors/standard-io-executor.rb Modified files: lib/grntest/tester.rb Added: lib/grntest/executors.rb (+19 -0) 100644 =================================================================== --- /dev/null +++ lib/grntest/executors.rb 2013-08-12 11:59:59 +0900 (8cba3a5) @@ -0,0 +1,19 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2012-2013 Kouhei Sutou <kou �� clear-code.com> +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +require "grntest/exectors/standard-io-executor" +require "grntest/exectors/http-executor" Added: lib/grntest/executors/base-executor.rb (+328 -0) 100644 =================================================================== --- /dev/null +++ lib/grntest/executors/base-executor.rb 2013-08-12 11:59:59 +0900 (94aaeb4) @@ -0,0 +1,328 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2012-2013 Kouhei Sutou <kou �� clear-code.com> +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +require "pathname" +require "fileutils" +require "shellwords" + +require "groonga/command" + +require "grntest/error" +require "grntest/execution-context" + +module Grntest + module Executors + class BaseExecutor + module ReturnCode + SUCCESS = 0 + end + + attr_reader :context + def initialize(context=nil) + @loading = false + @pending_command = "" + @pending_load_command = nil + @current_command_name = nil + @output_type = nil + @long_timeout = default_long_timeout + @context = context || ExecutionContext.new + end + + def execute(script_path) + unless script_path.exist? + raise NotExist.new(script_path) + end + + @context.execute do + script_path.open("r:ascii-8bit") do |script_file| + parser = create_parser + script_file.each_line do |line| + begin + parser << line + rescue Error, Groonga::Command::ParseError + line_info = "#{script_path}:#{script_file.lineno}:#{line.chomp}" + log_error("#{line_info}: #{$!.message}") + if $!.is_a?(Groonga::Command::ParseError) + @context.abort + else + log_error("#{line_info}: #{$!.message}") + raise unles****@conte*****_level? + end + end + end + end + end + + @context.result + end + + private + def create_parser + parser = Groonga::Command::Parser.new + parser.on_command do |command| + execute_command(command) + end + parser.on_load_complete do |command| + execute_command(command) + end + parser.on_comment do |comment| + if /\A@/ =~ comment + directive_content = $POSTMATCH + execute_directive("\##{comment}", directive_content) + end + end + parser + end + + def resolve_path(path) + if path.relative? + @context.base_directory + path + else + path + end + end + + def execute_directive_suggest_create_dataset(line, content, options) + dataset_name = options.first + if dataset_name.nil? + log_input(line) + log_error("#|e| [suggest-create-dataset] dataset name is missing") + return + end + execute_suggest_create_dataset(dataset_name) + end + + def execute_directive_include(line, content, options) + path = options.first + if path.nil? + log_input(line) + log_error("#|e| [include] path is missing") + return + end + execute_script(Pathname(path)) + end + + def execute_directive_copy_path(line, content, options) + source, destination, = options + if source.nil? or destination.nil? + log_input(line) + if source.nil? + log_error("#|e| [copy-path] source is missing") + end + if destiantion.nil? + log_error("#|e| [copy-path] destination is missing") + end + return + end + source = resolve_path(Pathname(source)) + destination = resolve_path(Pathname(destination)) + FileUtils.cp_r(source.to_s, destination.to_s) + end + + def execute_directive_long_timeout(line, content, options) + long_timeout, = options + invalid_value_p = false + case long_timeout + when "default" + @long_timeout = default_long_timeout + when nil + invalid_value_p = true + else + begin + @long_timeout = Float(long_timeout) + rescue ArgumentError + invalid_value_p = true + end + end + + if invalid_value_p + log_input(line) + message = "long-timeout must be number or 'default': <#{long_timeout}>" + log_error("#|e| [long-timeout] #{message}") + end + end + + def execute_directive_on_error(line, content, options) + action, = options + invalid_value_p = false + valid_actions = ["default", "omit"] + if valid_actions.include?(action) + @context.on_error = action.to_sym + else + invalid_value_p = true + end + + if invalid_value_p + log_input(line) + valid_actions_label = "[#{valid_actions.join(', ')}]" + message = "on-error must be one of #{valid_actions_label}" + log_error("#|e| [on-error] #{message}: <#{action}>") + end + end + + def execute_directive_omit(line, content, options) + reason, = options + @output_type = "raw" + log_output("omit: #{reason}") + @context.omit + end + + def execute_directive(line, content) + command, *options = Shellwords.split(content) + case command + when "disable-logging" + @context.logging = false + when "enable-logging" + @context.logging = true + when "suggest-create-dataset" + execute_directive_suggest_create_dataset(line, content, options) + when "include" + execute_directive_include(line, content, options) + when "copy-path" + execute_directive_copy_path(line, content, options) + when "long-timeout" + execute_directive_long_timeout(line, content, options) + when "on-error" + execute_directive_on_error(line, content, options) + when "omit" + execute_directive_omit(line, content, options) + else + log_input(line) + log_error("#|e| unknown directive: <#{command}>") + end + end + + def execute_suggest_create_dataset(dataset_name) + command_line = [@context.groonga_suggest_create_dataset, + @context.db_path.to_s, + dataset_name] + packed_command_line = command_line.join(" ") + log_input("#{packed_command_line}\n") + begin + IO.popen(command_line, "r:ascii-8bit") do |io| + log_output(io.read) + end + rescue SystemCallError + raise Error.new("failed to run groonga-suggest-create-dataset: " + + "<#{packed_command_line}>: #{$!}") + end + end + + def execute_script(script_path) + executor = create_sub_executor(@context) + executor.execute(resolve_path(script_path)) + end + + def extract_command_info(command) + @current_command = command + if @current_command.name == "dump" + @output_type = "groonga-command" + else + @output_type = @current_command[:output_type] || @context.output_type + end + end + + def execute_command(command) + extract_command_info(command) + log_input("#{command.original_source}\n") + response = send_command(command) + type = @output_type + log_output(response) + log_error(read_error_log) + + @context.error if error_response?(response, type) + end + + def read_error_log + log = read_all_readable_content(context.log, :first_timeout => 0) + normalized_error_log = "" + log.each_line do |line| + timestamp, log_level, message = line.split(/\|\s*/, 3) + _ = timestamp # suppress warning + next unless error_log_level?(log_level) + next if backtrace_log_message?(message) + normalized_error_log << "\#|#{log_level}| #{message}" + end + normalized_error_log.chomp + end + + def read_all_readable_content(output, options={}) + content = "" + first_timeout = options[:first_timeout] || 1 + timeout = first_timeout + while IO.select([output], [], [], timeout) + break if output.eof? + request_bytes = 1024 + read_content = output.readpartial(request_bytes) + content << read_content + timeout = 0 if read_content.bytesize < request_bytes + end + content + end + + def error_log_level?(log_level) + ["E", "A", "C", "e"].include?(log_level) + end + + def backtrace_log_message?(message) + message.start_with?("/") + end + + def error_response?(response, type) + status = nil + begin + status, = ResponseParser.parse(response, type) + rescue ParseError + return false + end + + return_code, = status + return_code != ReturnCode::SUCCESS + end + + def log(tag, content, options={}) + return unles****@conte*****? + log_force(tag, content, options) + end + + def log_force(tag, content, options) + return if content.empty? + @context.result << [tag, content, options] + end + + def log_input(content) + log(:input, content) + end + + def log_output(content) + log(:output, content, + :command => @current_command, + :type => @output_type) + @current_command = nil + @output_type = nil + end + + def log_error(content) + log_force(:error, content, {}) + end + + def default_long_timeout + 180 + end + end + end +end Added: lib/grntest/executors/http-executor.rb (+60 -0) 100644 =================================================================== --- /dev/null +++ lib/grntest/executors/http-executor.rb 2013-08-12 11:59:59 +0900 (6cfc893) @@ -0,0 +1,60 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2012-2013 Kouhei Sutou <kou �� clear-code.com> +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +require "open-uri" + +require "grntest/executors/base-executor" + +module Grntest + module Executors + class HTTPExecutor < Executor + def initialize(host, port, context=nil) + super(context) + @host = host + @port = port + end + + def send_command(command) + url = "http://#{@host}:#{@port}#{command.to_uri_format}" + begin + open(url) do |response| + "#{response.read}\n" + end + rescue OpenURI::HTTPError + message = "Failed to get response from groonga: #{$!}: <#{url}>" + raise Error.new(message) + end + end + + def ensure_groonga_ready + n_retried = 0 + begin + send_command("status") + rescue SystemCallError + n_retried += 1 + sleep(0.1) + retry if n_retried < 10 + raise + end + end + + def create_sub_executor(context) + self.class.new(@host, @port, context) + end + end + end +end Added: lib/grntest/executors/standard-io-executor.rb (+71 -0) 100644 =================================================================== --- /dev/null +++ lib/grntest/executors/standard-io-executor.rb 2013-08-12 11:59:59 +0900 (3fdaaf7) @@ -0,0 +1,71 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2012-2013 Kouhei Sutou <kou �� clear-code.com> +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +require "grntest/executors/base-executor" + +module Grntest + module Executors + class StandardIOExecutor < BaseExecutor + def initialize(input, output, context=nil) + super(context) + @input = input + @output = output + end + + def send_command(command) + command_line = @current_command.original_source + unless @current_command.has_key?(:output_type) + command_line = command_line.sub(/$/, " --output_type #{@output_type}") + end + begin + @input.print(command_line) + @input.print("\n") + @input.flush + rescue SystemCallError + message = "failed to write to groonga: <#{command_line}>: #{$!}" + raise Error.new(message) + end + read_output + end + + def ensure_groonga_ready + @input.print("status\n") + @input.flush + @output.gets + end + + def create_sub_executor(context) + self.class.new(@input, @output, context) + end + + private + def read_output + options = {} + options[:first_timeout] = @long_timeout if may_slow_command? + read_all_readable_content(@output, options) + end + + MAY_SLOW_COMMANDS = [ + "column_create", + "register", + ] + def may_slow_command? + MAY_SLOW_COMMANDS.include?(@current_command) + end + end + end +end Modified: lib/grntest/tester.rb (+1 -385) =================================================================== --- lib/grntest/tester.rb 2013-08-12 11:57:47 +0900 (36173f4) +++ lib/grntest/tester.rb 2013-08-12 11:59:59 +0900 (2756739) @@ -31,7 +31,7 @@ require "groonga/command" require "grntest/version" require "grntest/error" require "grntest/reporters" -require "grntest/execution-context" +require "grntest/executors" module Grntest class Tester @@ -1171,389 +1171,5 @@ EOF end end end - - class Executor - module ReturnCode - SUCCESS = 0 - end - - attr_reader :context - def initialize(context=nil) - @loading = false - @pending_command = "" - @pending_load_command = nil - @current_command_name = nil - @output_type = nil - @long_timeout = default_long_timeout - @context = context || ExecutionContext.new - end - - def execute(script_path) - unless script_path.exist? - raise NotExist.new(script_path) - end - - @context.execute do - script_path.open("r:ascii-8bit") do |script_file| - parser = create_parser - script_file.each_line do |line| - begin - parser << line - rescue Error, Groonga::Command::ParseError - line_info = "#{script_path}:#{script_file.lineno}:#{line.chomp}" - log_error("#{line_info}: #{$!.message}") - if $!.is_a?(Groonga::Command::ParseError) - @context.abort - else - log_error("#{line_info}: #{$!.message}") - raise unles****@conte*****_level? - end - end - end - end - end - - @context.result - end - - private - def create_parser - parser = Groonga::Command::Parser.new - parser.on_command do |command| - execute_command(command) - end - parser.on_load_complete do |command| - execute_command(command) - end - parser.on_comment do |comment| - if /\A@/ =~ comment - directive_content = $POSTMATCH - execute_directive("\##{comment}", directive_content) - end - end - parser - end - - def resolve_path(path) - if path.relative? - @context.base_directory + path - else - path - end - end - - def execute_directive_suggest_create_dataset(line, content, options) - dataset_name = options.first - if dataset_name.nil? - log_input(line) - log_error("#|e| [suggest-create-dataset] dataset name is missing") - return - end - execute_suggest_create_dataset(dataset_name) - end - - def execute_directive_include(line, content, options) - path = options.first - if path.nil? - log_input(line) - log_error("#|e| [include] path is missing") - return - end - execute_script(Pathname(path)) - end - - def execute_directive_copy_path(line, content, options) - source, destination, = options - if source.nil? or destination.nil? - log_input(line) - if source.nil? - log_error("#|e| [copy-path] source is missing") - end - if destiantion.nil? - log_error("#|e| [copy-path] destination is missing") - end - return - end - source = resolve_path(Pathname(source)) - destination = resolve_path(Pathname(destination)) - FileUtils.cp_r(source.to_s, destination.to_s) - end - - def execute_directive_long_timeout(line, content, options) - long_timeout, = options - invalid_value_p = false - case long_timeout - when "default" - @long_timeout = default_long_timeout - when nil - invalid_value_p = true - else - begin - @long_timeout = Float(long_timeout) - rescue ArgumentError - invalid_value_p = true - end - end - - if invalid_value_p - log_input(line) - message = "long-timeout must be number or 'default': <#{long_timeout}>" - log_error("#|e| [long-timeout] #{message}") - end - end - - def execute_directive_on_error(line, content, options) - action, = options - invalid_value_p = false - valid_actions = ["default", "omit"] - if valid_actions.include?(action) - @context.on_error = action.to_sym - else - invalid_value_p = true - end - - if invalid_value_p - log_input(line) - valid_actions_label = "[#{valid_actions.join(', ')}]" - message = "on-error must be one of #{valid_actions_label}" - log_error("#|e| [on-error] #{message}: <#{action}>") - end - end - - def execute_directive_omit(line, content, options) - reason, = options - @output_type = "raw" - log_output("omit: #{reason}") - @context.omit - end - - def execute_directive(line, content) - command, *options = Shellwords.split(content) - case command - when "disable-logging" - @context.logging = false - when "enable-logging" - @context.logging = true - when "suggest-create-dataset" - execute_directive_suggest_create_dataset(line, content, options) - when "include" - execute_directive_include(line, content, options) - when "copy-path" - execute_directive_copy_path(line, content, options) - when "long-timeout" - execute_directive_long_timeout(line, content, options) - when "on-error" - execute_directive_on_error(line, content, options) - when "omit" - execute_directive_omit(line, content, options) - else - log_input(line) - log_error("#|e| unknown directive: <#{command}>") - end - end - - def execute_suggest_create_dataset(dataset_name) - command_line = [@context.groonga_suggest_create_dataset, - @context.db_path.to_s, - dataset_name] - packed_command_line = command_line.join(" ") - log_input("#{packed_command_line}\n") - begin - IO.popen(command_line, "r:ascii-8bit") do |io| - log_output(io.read) - end - rescue SystemCallError - raise Error.new("failed to run groonga-suggest-create-dataset: " + - "<#{packed_command_line}>: #{$!}") - end - end - - def execute_script(script_path) - executor = create_sub_executor(@context) - executor.execute(resolve_path(script_path)) - end - - def extract_command_info(command) - @current_command = command - if @current_command.name == "dump" - @output_type = "groonga-command" - else - @output_type = @current_command[:output_type] || @context.output_type - end - end - - def execute_command(command) - extract_command_info(command) - log_input("#{command.original_source}\n") - response = send_command(command) - type = @output_type - log_output(response) - log_error(read_error_log) - - @context.error if error_response?(response, type) - end - - def read_error_log - log = read_all_readable_content(context.log, :first_timeout => 0) - normalized_error_log = "" - log.each_line do |line| - timestamp, log_level, message = line.split(/\|\s*/, 3) - _ = timestamp # suppress warning - next unless error_log_level?(log_level) - next if backtrace_log_message?(message) - normalized_error_log << "\#|#{log_level}| #{message}" - end - normalized_error_log.chomp - end - - def read_all_readable_content(output, options={}) - content = "" - first_timeout = options[:first_timeout] || 1 - timeout = first_timeout - while IO.select([output], [], [], timeout) - break if output.eof? - request_bytes = 1024 - read_content = output.readpartial(request_bytes) - content << read_content - timeout = 0 if read_content.bytesize < request_bytes - end - content - end - - def error_log_level?(log_level) - ["E", "A", "C", "e"].include?(log_level) - end - - def backtrace_log_message?(message) - message.start_with?("/") - end - - def error_response?(response, type) - status = nil - begin - status, = ResponseParser.parse(response, type) - rescue ParseError - return false - end - - return_code, = status - return_code != ReturnCode::SUCCESS - end - - def log(tag, content, options={}) - return unles****@conte*****? - log_force(tag, content, options) - end - - def log_force(tag, content, options) - return if content.empty? - @context.result << [tag, content, options] - end - - def log_input(content) - log(:input, content) - end - - def log_output(content) - log(:output, content, - :command => @current_command, - :type => @output_type) - @current_command = nil - @output_type = nil - end - - def log_error(content) - log_force(:error, content, {}) - end - - def default_long_timeout - 180 - end - end - - class StandardIOExecutor < Executor - def initialize(input, output, context=nil) - super(context) - @input = input - @output = output - end - - def send_command(command) - command_line = @current_command.original_source - unless @current_command.has_key?(:output_type) - command_line = command_line.sub(/$/, " --output_type #{@output_type}") - end - begin - @input.print(command_line) - @input.print("\n") - @input.flush - rescue SystemCallError - message = "failed to write to groonga: <#{command_line}>: #{$!}" - raise Error.new(message) - end - read_output - end - - def ensure_groonga_ready - @input.print("status\n") - @input.flush - @output.gets - end - - def create_sub_executor(context) - self.class.new(@input, @output, context) - end - - private - def read_output - options = {} - options[:first_timeout] = @long_timeout if may_slow_command? - read_all_readable_content(@output, options) - end - - MAY_SLOW_COMMANDS = [ - "column_create", - "register", - ] - def may_slow_command? - MAY_SLOW_COMMANDS.include?(@current_command) - end - end - - class HTTPExecutor < Executor - def initialize(host, port, context=nil) - super(context) - @host = host - @port = port - end - - def send_command(command) - url = "http://#{@host}:#{@port}#{command.to_uri_format}" - begin - open(url) do |response| - "#{response.read}\n" - end - rescue OpenURI::HTTPError - message = "Failed to get response from groonga: #{$!}: <#{url}>" - raise Error.new(message) - end - end - - def ensure_groonga_ready - n_retried = 0 - begin - send_command("status") - rescue SystemCallError - n_retried += 1 - sleep(0.1) - retry if n_retried < 10 - raise - end - end - - def create_sub_executor(context) - self.class.new(@host, @port, context) - end - end end end -------------- next part -------------- HTML����������������������������...다운로드
