discourse/lib/autospec/spork_runner.rb

module Autospec
  class SporkRunner < BaseRunner

    def start
      if already_running?(pid_file)
        puts "autospec appears to be running, it is possible the pid file is old"
        puts "if you are sure it is not running, delete #{pid_file}"
        return
      end
      write_pid_file(pid_file, Process.pid)
      start_spork
      @spork_running = true
    end

    def running?
      @monitor_thread ||=
        Thread.new do
          Process.wait(@spork_pid)
          @spork_running = false
        end
      @spork_running
    end

    def stop
      stop_spork
    end

    def run(args,specs)
      spork_service.run(args,$stderr,$stdout)
    end

    def abort
      spork_service.abort
    end

    def reload
      stop_spork
      sleep 1
      start_spork
    end

    private
    def spork_pid_file
      Rails.root + "tmp/pids/spork.pid"
    end

    def pid_file
      Rails.root + "tmp/pids/autospec.pid"
    end

    def already_running?(pid_file)
      if File.exists? pid_file
        pid = File.read(pid_file).to_i
        Process.getpgid(pid) rescue nil
      end
    end

    def write_pid_file(file,pid)
      FileUtils.mkdir_p(Rails.root + "tmp/pids")
      File.open(file,'w') do |f|
        f.write(pid)
      end
    end

    def spork_running?
      spork_service.port rescue nil
    end

    def spork_service

      unless @drb_listener_running
        begin
          DRb.start_service("druby://127.0.0.1:0")
        rescue SocketError, Errno::EADDRNOTAVAIL
          DRb.start_service("druby://:0")
        end

        @drb_listener_running = true
      end

      @spork_service ||= DRbObject.new_with_uri("druby://127.0.0.1:8989")
    end

    def stop_spork
      pid = File.read(spork_pid_file).to_i
      Process.kill("SIGTERM",pid)
    end

    def start_spork
      if already_running?(spork_pid_file)
        puts "Killing old orphan spork instance"
        stop_spork
        sleep 1
      end

      @spork_pid = Process.spawn({'RAILS_ENV' => 'test'}, "bundle exec spork")
      write_pid_file(spork_pid_file, @spork_pid)

      running = false
      while !running
        running = spork_running?
        sleep 0.01
      end

    end
  end
end