Class: GroongaQueryLog::Command::RunRegressionTest::GroongaServer

Inherits:

Object

• Object
• GroongaQueryLog::Command::RunRegressionTest::GroongaServer

Includes:

Loggable

Defined in:

lib/groonga-query-log/command/run-regression-test.rb

Instance Attribute Summary

• #host ⇒ Object readonly

  Returns the value of attribute host.

• #port ⇒ Object readonly

  Returns the value of attribute port.

Instance Method Summary

• #ensure_database ⇒ Object
• #initialize(groonga, groonga_options, groonga_env, database_path, options) ⇒ GroongaServer constructor

  A new instance of GroongaServer.

• #n_leaked_objects ⇒ Object
• #run ⇒ Object
• #shutdown ⇒ Object
• #use_persistent_cache? ⇒ Boolean

Methods included from Loggable

#puts

Constructor Details

#initialize(groonga, groonga_options, groonga_env, database_path, options) ⇒ GroongaServer

Returns a new instance of GroongaServer.

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 548

def initialize(groonga,
               groonga_options,
               groonga_env,
               database_path,
               options)
  @input_directory = options[:input_directory] || Pathname.new(".")
  @working_directory = options[:working_directory] || Pathname.new(".")
  @groonga = groonga
  @groonga_options = groonga_options
  @groonga_env = groonga_env
  @database_path = @working_directory + database_path
  @host = "127.0.0.1"
  @port = find_unused_port
  @options = options
  @pid = nil
end

Instance Attribute Details

#host ⇒ Object (readonly)

Returns the value of attribute host.

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 547

def host
  @host
end

#port ⇒ Object (readonly)

Returns the value of attribute port.

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 547

def port
  @port
end

Instance Method Details

#ensure_database ⇒ Object

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 607

def ensure_database
  if @options[:recreate_database]
    FileUtils.rm_rf(@database_path.dirname.to_s)
  end
  return if @database_path.exist?

  FileUtils.mkdir_p(@database_path.dirname.to_s)
  create_db_command = [@groonga, "-n", @database_path.to_s, "quit"]
  unless system(*create_db_command)
    create_db_command_line = create_db_command.join(" ")
    raise "Failed to run: #{create_db_command_line}"
  end

  load_files.each do |load_file|
    filter_command = nil
    case load_file.extname
    when ".rb"
      env = {
        "GROONGA_LOG_PATH" => log_path.to_s,
      }
      command = [
        RbConfig.ruby,
        load_file.to_s,
        @database_path.to_s,
      ]
    when ".zst"
      env = {}
      command = [
        @groonga,
        "--log-path", log_path.to_s,
        @database_path.to_s,
      ]
      filter_command = [
        "zstdcat",
        load_file.to_s,
      ]
    else
      env = {}
      command = [
        @groonga,
        "--log-path", log_path.to_s,
        "--file", load_file.to_s,
        @database_path.to_s,
      ]
    end
    command_line = command.join(" ")
    if filter_command
      filter_command_line = filter_command.join(" ")
      command_line = "#{filter_command_line} | #{command_line}"
    end
    puts("Running...: #{command_line}")
    status = nil
    if filter_command
      IO.pipe do |input, output|
        filter_pid = spawn(*filter_command, out: output)
        output.close
        pid = spawn(env, *command, in: input)
        input.close
        begin
          pid, status = Process.waitpid2(pid)
          filter_pid, _filter_status = Process.waitpid2(filter_pid)
        rescue Interrupt
          Process.kill(:TERM, pid)
          Process.kill(:TERM, filter_pid)
          pid, status = Process.waitpid2(pid)
          filter_pid, _filter_status = Process.waitpid2(filter_pid)
        end
      end
    else
      pid = spawn(env, *command)
      begin
        pid, status = Process.waitpid2(pid)
      rescue Interrupt
        Process.kill(:TERM, pid)
        pid, status = Process.waitpid2(pid)
      end
    end
    unless status.success?
      raise "Failed to run: #{command_line}"
    end
  end
end

#n_leaked_objects ⇒ Object

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 705

def n_leaked_objects
  n = 0
  File.open(log_path, encoding: "UTF-8") do |log|
    log.each_line do |line|
      next unless line.valid_encoding?
      case line
      when /grn_fin \((\d+)\)/
        n += Integer($1, 10)
      end
    end
  end
  n
end

#run ⇒ Object

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 565

def run
  return unless @options[:run_queries]

  spawn_args = []
  spawn_args << @groonga_env if @groonga_env
  spawn_args << @groonga
  spawn_args.concat(@groonga_options)
  spawn_args.concat(["--bind-address", @host])
  spawn_args.concat(["--port", @port.to_s])
  spawn_args.concat(["--protocol", "http"])
  spawn_args.concat(["--log-path", log_path.to_s])
  if @options[:output_query_log]
    spawn_args.concat(["--query-log-path", query_log_path.to_s])
  end
  spawn_args << "-s"
  spawn_args << @database_path.to_s
  @pid = spawn(*spawn_args)

  begin
    n_retries = 60
    begin
      send_command("status")
    rescue SystemCallError
      sleep(1)
      n_retries -= 1
      raise if n_retries.zero?
      retry
    end

    if @options[:warm_up]
      send_command("dump?dump_records=no")
      warm_up_commands = @options[:warm_up_commands] || []
      warm_up_commands.each do |command|
        send_command(command)
      end
    end
  rescue
    shutdown
    raise
  end
end

#shutdown ⇒ Object

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 694

def shutdown
  return if @pid.nil?
  begin
    send_command("shutdown")
  rescue SystemCallError
    Process.kill(:KILL, @pid)
  end
  Process.waitpid(@pid)
  @pid = nil
end

#use_persistent_cache? ⇒ Boolean

Returns:

• (Boolean)

# File 'lib/groonga-query-log/command/run-regression-test.rb', line 690

def use_persistent_cache?
  @groonga_options.include?("--cache-base-path")
end