velociraptor 0.1.0

data/README.md

@@ -0,0 +1,14 @@
+= Velociraptor
+
+    $ velociraptor start &
+    $ curl localhost:3001
+    {"backends": "/backends"}
+    $ curl localhost:3001/backends
+    []
+    $ echo 'run(lambda { |env| [200, {"Content-Type" => "text/plain"}, ["Hello, world!\n"]] })' > config.ru
+    $ thin -p 5000 start &
+    $ curl -X PUT -d '{"host": "localhost", "port": "5000"}' localhost:3001/backends/web.0
+    {"host": "localhost", "port": "5000", "url": "/backends/web.0"}
+    $ curl localhost:3000
+    Hello, world!
+

data/bin/velociraptor

@@ -0,0 +1,205 @@
+#!/usr/bin/env ruby
+
+require 'enumerator'
+require 'eventmachine'
+require 'json'
+require 'sinatra/base'
+require 'thor'
+
+module ProxyConnection
+  def initialize client, header
+    @client, @header = client, header
+  end
+  def post_init
+    EM::enable_proxy self, @client
+  end
+  def connection_completed
+    send_data @header
+  end
+  def proxy_target_unbound
+    close_connection
+  end
+	def on_unbind &b
+		@on_unbind = b
+	end
+  def unbind
+    @client.close_connection_after_writing
+		@on_unbind.call if @on_unbind
+  end
+end
+
+class Backend
+	attr_accessor :id, :host, :port
+	attr_writer :busy
+	def busy?
+		@busy
+	end
+	def initialize(id, host, port)
+		self.id = id
+		self.host = host
+		self.port = port
+		self.busy = false
+	end
+	def proxy client, header
+		self.busy = true
+		EM.connect(self.host, self.port, ProxyConnection, client, header) do |c|
+			c.on_unbind &method(:ready!)
+		end
+	end
+	def ready!
+		self.busy = false
+		@on_ready.call(self) if @on_ready
+	end
+	def on_ready &b
+		@on_ready = b
+	end
+end
+
+class RoundRobinStrategy
+	def initialize(list)
+		@list = list
+		@i = 0
+	end
+	def next
+		enum = Enumerator.new do |y|
+			start = @i
+			while @i < @list.length
+				y.yield @list[@i]
+				@i += 1
+			end
+			@i = 0
+			while @i < start and @i < @list.length
+				y.yield @list[@i]
+				@i += 1
+			end
+		end
+		yield(enum).tap { @i += 1}
+	end
+end
+
+class BackendManager
+	class <<self
+		attr_accessor :backends, :strategy, :waiting
+		def init
+			self.backends = []
+			self.strategy = RoundRobinStrategy.new(self.backends)
+			self.waiting = []
+		end
+		def add backend
+			self.backends.push backend
+			backend.on_ready &method(:ready)
+			ready backend
+		end
+		def ready backend
+			if (b = self.waiting.shift)
+				b.call backend
+			end
+		end
+		def find id
+			self.backends.find { |b| b.id == id }
+		end
+		def remove id
+			self.backends.delete_if { |b| b.id == id }
+		end
+		def next(&b)
+			backend = self.strategy.next do |enum|
+				enum.find { |backend| not backend.busy? }
+			end
+			backend ? b.call(backend) : self.waiting.push(b)
+		end
+	end
+end
+
+module ProxyServer
+  def receive_data(data)
+    (@header ||= "") << data
+    if @header =~ /\r\n\r\n/ # all http headers received
+			BackendManager.next do |backend|
+				backend.proxy self, @header
+			end
+    end
+  end
+end
+
+class Admin < Sinatra::Base
+	set :run, false
+	set :logging, true
+	before { content_type :json }
+	
+	module BackendToHash
+		def to_hash
+			{
+				"host" => self.host,
+				"url" => "/backends/" + self.id,
+				"port" => self.port
+			}
+		end
+	end
+	
+	get '/' do
+		{ "backends" => "/backends" }.to_json
+	end
+	get '/backends' do
+		BackendManager.backends.map do |b|
+			b.extend(BackendToHash).to_hash
+		end.to_json
+	end
+	put '/backends/:id' do
+		request.body.rewind
+		data = JSON.parse request.body.read
+		id = params[:id]
+		host = data['host']
+		port = data['port'] || 80
+		
+		if (b = BackendManager.find id)
+			b.host = host
+			b.port = port
+		else
+			b = Backend.new id, host, port
+			BackendManager.add b
+		end
+		b.extend(BackendToHash).to_hash.to_json
+	end
+	delete '/backends/:id' do
+		BackendManager.remove params[:id]
+		202
+	end
+end
+
+class Printer
+	def self.out id, msg
+		puts "#{Time.now.strftime '%H:%M:%S'} #{id}#{' ' * (9 - id.size)} | #{msg}"
+	end
+end
+
+VERSION = "0.1.0"
+
+class CLI < Thor
+	
+	desc "start [PORT] [ADMIN_PORT]", "Start the proxy server on PORT(default 3000), with the admin server on ADMIN_PORT(default PORT+1)"
+
+	def start(*args)
+		port = args.shift || 3000
+		admin_port = args.shift || port + 1
+		
+		BackendManager.init
+		EM.run {
+			trap("INT") { EM.stop }
+
+			Printer.out "system", "proxy running on 127.0.0.1:#{port}"
+		  EM.start_server("127.0.0.1", port, ProxyServer)
+
+			Printer.out "system", "admin running on 0.0.0.0:#{admin_port}"
+			Admin.run!({:host => "0.0.0.0", :port => admin_port})
+		}
+	end
+	
+	def help(*args)
+		puts "Velociraptor #{VERSION}"
+		puts
+		super
+	end
+	
+end
+
+CLI.start

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: velociraptor
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Michael Maltese
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: &70300733791240 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70300733791240
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: &70300733790580 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70300733790580
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &70300733789680 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70300733789680
+description: ''
+email: michael.maltese@pomona.edu
+executables:
+- velociraptor
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/velociraptor
+- README.md
+homepage: http://github.com/mikemaltese/velociraptor
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Managed proxy server
+test_files: []