bum 0.0.4 → 0.0.12

data/lib/role.rb

@@ -0,0 +1,99 @@
+require 'server'
+Thread.abort_on_exception = true
+
+class Role
+  INDEX = {}
+
+  class << self
+    def get_role(msg)
+      role = INDEX[msg['role']].new(msg)
+      role.start
+      role
+    end
+
+    def role(name)
+      INDEX[name] = self
+      self.my_role = name
+    end
+
+    attr_accessor :my_role
+  end
+
+  def state
+    { 'role' => self.class.my_role,
+      'active' => @active_links,
+      'disabled' => @disabled_links,
+      'broken' => @broken_links }
+  end
+
+  def initialize(manifest)
+    @manifest = manifest
+  end
+
+  def start
+    raise NotImplementedError
+  end
+
+  def stop
+    raise NotImplementedError
+  end
+
+  def disable
+    raise NotImplementedError
+  end
+
+  def enable
+    raise NotImplementedError
+  end
+
+  def disable_link(type, addr)
+    raise NotImplementedError
+  end
+
+  def enable_link(type, addr)
+    raise NotImplementedError
+  end
+
+  def init_links(type, links, &block)
+    @active_links ||= {}
+    @active_links[type] ||= (links || [])
+
+    @broken_links ||= {}
+    @broken_links[type] ||= []
+
+    @disabled_links ||= {}
+    @disabled_links[type] ||= []
+
+    @event_callbacks ||= {}
+    @event_callbacks[type] = block
+  end
+
+  def disable_link(type, addr)
+    types = @active_links.keys
+
+    types.each do |type|
+      if @active_links[type].include?(addr) || @broken_links[type].include?(addr)
+        @active_links[type].delete addr
+        @broken_links[type].delete addr
+    
+        @disabled_links[type] << addr unless @disabled_links[type].include?(addr)
+
+        @event_callbacks[type].call('disable', addr) if @event_callbacks[type]
+      end
+    end
+  end
+
+  def enable_link(type, addr)
+    types = @active_links.keys
+
+    types.each do |type|
+      if @disabled_links[type].include?(addr)
+        @disabled_links[type].delete addr
+
+        @active_links[type] << addr unless @active_links[type].include?(addr)
+
+        @event_callbacks[type].call('enable', addr) if @event_callbacks[type]
+      end
+    end
+  end
+end

data/lib/scenario.rb

@@ -0,0 +1,186 @@
+class Scenario
+  class << self
+    attr_accessor :index
+    Scenario.index = {}
+
+    def register(name)
+      Scenario.index[name] = self
+    end
+  end
+end
+
+class ScenarioA < Scenario
+  register 'a'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxy = rand_clients.pop
+    rails = rand_clients.pop
+    db = rand_clients.pop
+
+    db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => []
+    rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_write' => [db.addr], 'db_read' => [db.addr]
+    proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => [rails.addr]
+  end
+end
+
+class ScenarioB < Scenario
+  register 'b'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxy = rand_clients.pop
+    db = rand_clients.pop
+    rails_nodes = rand_clients.slice!(0,5) # up to 5 of the rest get to be rails nodes
+
+    db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => []
+
+    rails_nodes.each do |rails|
+      rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_write' => [db.addr], 'db_read' => [db.addr]
+    end
+
+    proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => rails_nodes.map { |n| n.addr }
+  end
+end
+
+class ScenarioC < Scenario
+  register 'c'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxy = rand_clients.pop
+    master_db = rand_clients.pop
+    slave_dbs = rand_clients.slice!(0,2)
+    rails_nodes = rand_clients.slice!(0,5) # up to 5 of the rest get to be rails nodes
+
+    slave_dbs.each do |slave_db|
+      slave_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => [], 'slave' => true
+    end
+
+    master_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => slave_dbs.map { |n| n.addr }
+
+    rails_nodes.each do |rails|
+      rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_write' => [master_db.addr], 'db_read' => slave_dbs.map { |n| n.addr }
+    end
+
+    proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => rails_nodes.map { |n| n.addr }
+  end
+end
+
+
+class ScenarioD < Scenario
+  register 'd'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxy = rand_clients.pop
+    master_db = rand_clients.pop
+    slave_dbs = rand_clients.slice!(0,2)
+    stats = rand_clients.pop
+    rails_nodes = rand_clients.slice!(0,5) # up to 5 of the rest get to be rails nodes
+
+    slave_dbs.each do |slave_db|
+      slave_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => [], 'slave' => true
+    end
+
+    master_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => slave_dbs.map { |n| n.addr }
+
+    stats.msg 't' => 'ROLE', 'role' => 'stats', 'db_read' => slave_dbs.map { |n| n.addr }
+
+    rails_nodes.each do |rails|
+      rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_write' => [master_db.addr], 'db_read' => slave_dbs.map { |n| n.addr }
+    end
+
+    proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => rails_nodes.map { |n| n.addr }, 'stats_backends' => [stats.addr]
+  end
+end
+
+class ScenarioE < Scenario
+  register 'e'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxies = rand_clients.slice!(0,2)
+    master_db = rand_clients.pop
+    slave_dbs = rand_clients.slice!(0,2)
+    stats_nodes = rand_clients.slice!(0,2)
+    rails_nodes = rand_clients.slice!(0,5) # up to 5 of the rest get to be rails nodes
+
+    slave_dbs.each do |slave_db|
+      slave_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => [], 'slave' => true
+    end
+
+    master_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => slave_dbs.map { |n| n.addr }
+
+    stats_nodes.each do |stats|
+      stats.msg 't' => 'ROLE', 'role' => 'stats', 'db_read' => slave_dbs.map { |n| n.addr }
+    end
+
+    rails_nodes.each do |rails|
+      rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_write' => [master_db.addr], 'db_read' => slave_dbs.map { |n| n.addr }
+    end
+
+    proxies.each do |proxy|
+      proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => rails_nodes.map { |n| n.addr }, 'stats_backends' => stats_nodes.map { |n| n.addr }
+    end
+  end
+end
+
+class ScenarioF < Scenario
+  register 'f'
+
+  def run(clients)
+    clients.each do |client|
+      client.msg 't' => 'STOP'
+    end
+
+    rand_clients = clients.sort_by { rand }
+
+    proxies = rand_clients.slice!(0,2)
+    peer_dbs = rand_clients.slice!(0,3)
+    stats_nodes = rand_clients.slice!(0,2)
+    rails_nodes = rand_clients.slice!(0,5) # up to 5 of the rest get to be rails nodes
+
+    peer_dbs.each do |peer_db|
+      peer_db.msg 't' => 'ROLE', 'role' => 'db', 'slaves' => [], 'slave' => false
+    end
+
+    stats_nodes.each do |stats|
+      stats.msg 't' => 'ROLE', 'role' => 'stats', 'db_peer' => peer_dbs.map { |n| n.addr }
+    end
+
+    rails_nodes.each do |rails|
+      rails.msg 't' => 'ROLE', 'role' => 'rails', 'db_peer' => peer_dbs.map { |n| n.addr }
+    end
+
+    proxies.each do |proxy|
+      proxy.msg 't' => 'ROLE', 'role' => 'proxy', 'backends' => rails_nodes.map { |n| n.addr }, 'stats_backends' => stats_nodes.map { |n| n.addr }
+    end
+  end
+end
+

data/lib/server.rb

@@ -0,0 +1,30 @@
+require 'demo_socket'
+require 'socket'
+
+module Server
+  def run_server(addr, port, &block)
+    @_server_threads ||= {}
+    @_server_instances ||= {}
+
+    key = "#{addr}:#{port}"
+
+    @_server_threads[key] = Thread.new {
+      server = TCPServer.new(addr, port)
+      @_server_instances[key] = server
+      loop do
+        client_socket = DemoSocket.new(server.accept)
+        yield client_socket
+      end
+    }
+  end
+
+  def shutdown_server(addr, port)
+    key = "#{addr}:#{port}"
+
+    inst = @_server_instances[key]
+    inst.shutdown if inst
+
+    thread = @_server_threads[key]
+    thread.kill if thread
+  end
+end

data/lib/stats_role.rb

@@ -0,0 +1,89 @@
+require 'role'
+require 'sinatra/base'
+require 'db_crap'
+
+class StatsRole < Role
+  PORT = 9500
+  role 'stats'
+
+  def initialize(manifest)
+    super
+    init_links 'db_read', @manifest['db_read']
+    init_links 'db_peer', @manifest['db_peer']
+  end
+
+  def start
+    start_server
+  end
+
+  def stop
+    @http_server.shutdown
+  end
+
+  alias :enable :start
+  alias :disable :stop
+
+  private
+
+  def start_server
+    my_app = StatsApp.new(@active_links, @manifest)
+
+    trap('INT') do
+      @http_server.shutdown if @http_server
+      exit
+    end
+
+    Thread.new {
+      Rack::Handler::WEBrick.run(my_app, :Port => PORT.to_s) do |server|
+        @http_server = server
+      end
+    }
+  end
+
+  class StatsApp < Sinatra::Base
+    def initialize(active_links, manifest)
+      @active_links = active_links
+      @manifest = manifest
+      super()
+    end
+
+    get '/stats' do
+      stats_html
+    end
+
+    get '/ping' do
+      'pong'
+    end
+
+    private
+
+    def stats_html
+      resp = list
+
+      key_count = resp.size
+      last_updated = key_count > 0 ? resp.sort_by { |rec| rec['updated'] }[-1]['key'] : nil
+      largest_value = key_count > 0 ? resp.sort_by { |rec| rec['value'].to_s.size }[-1]['key'] : nil
+
+      <<-EOF
+        <div class="stats">
+          <h3>DB Stats</h3>
+          <dl>
+            <dt>DB Size</dt>
+            <dd>#{key_count}</dd>
+
+            <dt>Last updated key</dt>
+            <dd>#{last_updated}</dd>
+
+            <dt>Largest value</dt>
+            <dd>#{largest_value}</dd>
+          </dl>
+        </div>
+      EOF
+    end
+
+    private
+
+    include DbCrap
+  end
+end
+

data/public/app.css

@@ -0,0 +1,91 @@
+body {
+    width: 700px;
+    margin: 20px auto;
+    background: #ddd;
+    font-family: helvetica;
+}
+
+table {
+    float: right;
+    width: 458px;
+
+    border-radius: 10px;
+    padding: 10px 15px 15px 15px;
+    background: #f5f5f5;
+    border: 1px solid rgba(0, 0, 0, 0.15);
+    box-shadow: 0px 4px 20px rgba(0, 0, 0, 0.25);
+    z-index: 10;
+    position:relative;
+}
+
+table th {
+    font-size: 1.25em;
+    text-align: left;
+    padding: 5px 10px;
+    
+}
+table td {
+    font-family: Helvetica;
+    font-size: 1em;
+    padding: 5px 10px;
+}
+
+table tr {
+    background: #f3f3f3;
+}
+table tr:nth-child(2n) {
+    background: #fefefe;
+}
+
+
+form {
+    float: right;
+    margin-top: 48px;
+    border-bottom-left-radius: 10px;
+    border-top-left-radius: 10px;
+    padding: 20px;
+    background: #f5f5f5;
+    border: 1px solid rgba(0, 0, 0, 0.15);
+    box-shadow: 0px 4px 20px rgba(0, 0, 0, 0.25);
+    z-index: 1;
+}
+
+form label {
+    display:block;
+    font-size: 0.8em;
+    font-weight: bold;
+}
+form input[type=text] {
+    font-size: 1.2em;
+    width: 200px;
+    margin-bottom: 15px;
+}
+
+.stats {
+    width: 190px;
+    float: left;
+    border-radius: 10px;
+    padding: 10px 15px 10px 15px;
+    background: #f5f5f5;
+    border: 1px solid rgba(0, 0, 0, 0.15);
+    box-shadow: 0px 4px 20px rgba(0, 0, 0, 0.25);
+}
+.stats h3 {
+    margin: 5px 0 15px 0;
+}
+.stats dl {
+    overflow: hidden;
+    line-height: 1.5em;
+}
+.stats dl dt {
+    font-size: 0.8em;
+    margin-right: 10px;
+    width: 110px;
+    display: inline-block;
+}
+.stats dl dd {
+    font-size: 0.8em;
+    font-weight: bold;
+    margin: 0;
+    display: inline;
+}