theine 0.0.1

data/bin/theine

@@ -0,0 +1,2 @@
+#!/usr/bin/env ruby
+require 'theine/client'

data/bin/theine_server

@@ -0,0 +1,2 @@
+#!/usr/bin/env ruby
+require 'theine/server'

data/lib/theine/client.rb

@@ -0,0 +1,90 @@
+require 'drb/drb'
+require 'readline'
+
+PRAILS_BASE_PORT = 11000
+
+class IOUndumpedProxy
+  include DRb::DRbUndumped
+
+  def initialize(obj)
+    @obj = obj
+  end
+
+  def completion_proc=(val)
+    if @obj.respond_to? :completion_proc=
+      @obj.completion_proc = val
+    end
+  end
+
+  def completion_proc
+    @obj.completion_proc if @obj.respond_to? :completion_proc
+  end
+
+  def readline(prompt)
+    if ::Readline == @obj
+      @obj.readline(prompt, true)
+    elsif @obj.method(:readline).arity == 1
+      @obj.readline(prompt)
+    else
+      $stdout.print prompt
+      @obj.readline
+    end
+  end
+
+  def puts(*lines)
+    @obj.puts(*lines)
+  end
+
+  def print(*objs)
+    @obj.print(*objs)
+  end
+
+  def write(data)
+    @obj.write data
+  end
+
+  def <<(data)
+    @obj << data
+    self
+  end
+
+  def flush
+    @obj.flush
+  end
+
+  # Some versions of Pry expect $stdout or its output objects to respond to
+  # this message.
+  def tty?
+    false
+  end
+end
+
+argv = ARGV.dup
+ARGV.clear
+
+DRb.start_service
+
+i = 1
+begin
+  balancer = DRbObject.new_with_uri("druby://localhost:#{PRAILS_BASE_PORT}")
+  sleep 0.1 until port = balancer.get_port
+  prails = DRbObject.new_with_uri("druby://localhost:#{port}")
+rescue DRb::DRbConnError
+  sleep 0.5
+  putc "."
+  i += 1
+  retry
+end
+putc "\n"
+
+trap('INT') {
+  %x[kill -2 #{prails.pid}]
+}
+
+prails.stdin = IOUndumpedProxy.new($stdin)
+prails.stdout = IOUndumpedProxy.new($stdout)
+prails.stderr = IOUndumpedProxy.new($stderr)
+begin
+  prails.command_rails(argv)
+rescue DRb::DRbConnError
+end

data/lib/theine/instance.rb

@@ -0,0 +1,114 @@
+root_path = ARGV[2]
+APP_PATH = "#{root_path}/config/application"
+require "#{root_path}/config/boot"
+require "#{root_path}/config/environment"
+require 'drb/drb'
+
+$real_stdout = $stdout
+$real_stderr = $stderr
+
+class PrailsServer
+  attr_reader :stdin, :stdout, :stderr
+  InputProxy = Struct.new :input do
+    # Reads a line from the input
+    def readline(prompt)
+      case readline_arity
+      when 1 then input.readline(prompt)
+      else        input.readline
+      end
+    end
+
+    def completion_proc=(val)
+      input.completion_proc = val
+    end
+
+    def readline_arity
+      input.method_missing(:method, :readline).arity
+    rescue NameError
+      0
+    end
+  end
+
+  def initialize
+    @pumps = []
+  end
+
+  def command_rails(argv)
+    rails_reload!
+
+    ARGV.clear
+    ARGV.concat(argv)
+
+    require 'pry'
+    ::Rails.application.config.console = Pry
+
+    Pry.config.input = stdin
+    Pry.config.output = stdout
+
+    require 'rails/commands'
+    sleep 0.1 # allow Pumps to finish
+    DRb.stop_service
+  end
+
+  def stdin=(value)
+    @stdin = InputProxy.new(value)
+    $stdin = @stdin
+  end
+
+  def stdout=(value)
+    $orig_stdout = $stdout
+    @stdout = value
+    r, w = IO.pipe
+    $stdout = w
+    @pumps << Pump.new(r, @stdout)
+  end
+
+  def stderr=(value)
+    $orig_stderr = $stderr
+    @stderr = value
+    r, w = IO.pipe
+    $stderr = w
+    @pumps << Pump.new(r, @stderr)
+  end
+
+  def pid
+    ::Process.pid
+  end
+
+private
+  def rails_reload!
+    ActionDispatch::Reloader.cleanup!
+    ActionDispatch::Reloader.prepare!
+  end
+
+  class Pump < Thread
+    def initialize(input, output)
+      if output
+        @input = input
+        @output = output
+        super(&method(:main))
+      else
+        close_stream(input)
+      end
+    end
+
+  private
+    def main
+      while buf = @input.sysread(1024)
+        @output.print(buf)
+        @output.flush
+      end
+    ensure
+      @output.close
+    end
+  end
+end
+
+base_port = ARGV[0]
+instance_port = ARGV[1]
+DRb.start_service("druby://localhost:#{instance_port}", PrailsServer.new)
+
+balancer = DRbObject.new_with_uri("druby://localhost:#{base_port}")
+balancer.instance_boot(instance_port)
+
+DRb.thread.join

data/lib/theine/server.rb

@@ -0,0 +1,75 @@
+require 'drb/drb'
+require 'thread'
+
+PRAILS_BASE_PORT = 11000
+PRAILS_MAX_PORT = 11100
+MIN_FREE_INSTANCES = 2
+RAILS_APP_ROOT = Dir.pwd
+
+class PrailsServer
+  include DRb::DRbUndumped
+
+  def initialize
+    @instances = []
+    @spawning = []
+
+    @available_ports = ((PRAILS_BASE_PORT + 1)..PRAILS_MAX_PORT).to_a
+    @check_mutex = Mutex.new
+    @instances_mutex = Mutex.new
+  end
+
+  def add_instance
+    path = File.expand_path('../instance.rb', __FILE__)
+    port = @available_ports.shift
+    puts "(spawn #{port})"
+    spawn("ruby", path, PRAILS_BASE_PORT.to_s, port.to_s, RAILS_APP_ROOT)
+    @instances_mutex.synchronize { @spawning << 1 }
+  end
+
+  def instance_boot(port)
+    puts "+ instance #{port}"
+
+    @instances_mutex.synchronize do
+      @spawning.pop
+      @instances << port
+    end
+  end
+
+  def get_port
+    add_instance if all_size == 0
+
+    port = nil
+    while port.nil? && all_size > 0
+      @instances_mutex.synchronize do
+        port = @instances.shift
+      end
+      sleep 0.1 unless port
+    end
+
+    Thread.new { check_min_free_instances }
+
+    port
+  end
+
+  def check_min_free_instances
+    if @check_mutex.try_lock
+      # TODO: mutex, and dont do it if already in progress
+      # do this in thread
+      while all_size < MIN_FREE_INSTANCES
+        sleep 0.1 until @instances_mutex.synchronize { @spawning.empty? }
+        add_instance
+      end
+      @check_mutex.unlock
+    end
+  end
+
+  def all_size
+    @instances_mutex.synchronize { @instances.size + @spawning.size }
+  end
+end
+
+server = PrailsServer.new
+DRb.start_service("druby://localhost:#{PRAILS_BASE_PORT}", server)
+
+server.check_min_free_instances
+DRb.thread.join

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification
+name: theine
+version: !ruby/object:Gem::Version
+  prerelease:
+  version: 0.0.1
+platform: ruby
+authors:
+- Jan Berdajs
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2013-10-01 00:00:00.000000000 Z
+dependencies: []
+description: A Rails preloader for JRuby
+email: mrbrdo@mrbrdo.net
+executables:
+- theine
+- theine_server
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/theine.rb
+- lib/theine/client.rb
+- lib/theine/server.rb
+- lib/theine/instance.rb
+- bin/theine
+- bin/theine_server
+homepage: https://github.com/mrbrdo/theine
+licenses:
+- MIT
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+  none: false
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+  none: false
+requirements: []
+rubyforge_project:
+rubygems_version: 1.8.24
+signing_key:
+specification_version: 3
+summary: Theine
+test_files: []