resque-web 0.0.1

data/app/controllers/resque_web/overview_controller.rb

@@ -0,0 +1,7 @@
+module ResqueWeb
+  class OverviewController < ApplicationController
+    def show
+      render :layout => !request.xhr?, :locals => { :polling => request.xhr? }
+    end
+  end
+end

data/app/controllers/resque_web/queues_controller.rb

@@ -0,0 +1,12 @@
+module ResqueWeb
+  class QueuesController < ApplicationController
+    def show
+      set_subtabs view_context.queue_names
+    end
+
+    def destroy
+      Resque.remove_queue(params[:id])
+      redirect_to queues_path
+    end
+  end
+end

data/app/controllers/resque_web/stats_controller.rb

@@ -0,0 +1,17 @@
+module ResqueWeb
+  class StatsController < ApplicationController
+    subtabs :resque, :redis, :keys
+
+    def index
+      redirect_to :action => 'resque'
+    end
+
+    def keys
+      if params[:id]
+        render 'key'
+      else
+        render 'keys'
+      end
+    end
+  end
+end

data/app/controllers/resque_web/workers_controller.rb

@@ -0,0 +1,18 @@
+module ResqueWeb
+  class WorkersController < ApplicationController
+    before_filter :display_subtabs
+
+    def show
+      if params[:id] && params[:id] != 'all'
+        @workers = view_context.worker_hosts[params[:id]].map { |id| Resque::Worker.find(id) }
+      else
+        @workers = Resque.workers
+      end
+    end
+
+    def display_subtabs
+      set_subtabs view_context.worker_hosts.map(&:first)
+    end
+    private :display_subtabs
+  end
+end

data/app/controllers/resque_web/working_controller.rb

@@ -0,0 +1,4 @@
+module ResqueWeb
+  class WorkingController < ApplicationController
+  end
+end

data/app/helpers/resque_web/application_helper.rb

@@ -0,0 +1,65 @@
+module ResqueWeb
+  module ApplicationHelper
+
+    PER_PAGE = 20
+
+    def tabs
+      {'overview' => overview_path,
+       'working'  => working_index_path,
+       'failures' => failures_path,
+       'queues' => queues_path,
+       'workers' => workers_path,
+       'stats' => stats_path
+      }
+    end
+
+    def tab(name,path)
+      content_tag :li, link_to(name.capitalize, path), :class => current_tab?(name) ? "active" : nil
+    end
+
+    def current_tab
+      params[:controller].gsub(/resque_web\//, "#{root_path}")
+    end
+
+    def current_tab?(name)
+      params[:controller] == name.to_s
+    end
+
+    attr_reader :subtabs
+
+    def subtab(name)
+      content_tag :li, link_to(name, "#{current_tab}/#{name}"), :class => current_subtab?(name) ? "current" : nil
+    end
+
+    def current_subtab?(name)
+      params[:id] == name.to_s
+    end
+
+    def pagination(options = {})
+      start    = options[:start] || 1
+      per_page = options[:per_page] || PER_PAGE
+      total    = options[:total] || 0
+      return if total < per_page
+
+      markup = ""
+      if start - per_page >= 0
+        markup << link_to(raw("&laquo; less"), params.merge(:start => start - per_page), :class => 'btn less')
+      end
+
+      if start + per_page <= total
+        markup << link_to(raw("more &raquo;"), params.merge(:start => start + per_page), :class => 'btn more')
+      end
+
+      content_tag :p, raw(markup), :class => 'pagination'
+    end
+
+    def poll(polling=false)
+      if polling
+        text = "Last Updated: #{Time.now.strftime("%H:%M:%S")}".html_safe
+      else
+        text = "<a href='#{h(request.path)}' rel='poll'>Live Poll</a>".html_safe
+      end
+      content_tag :p, text, :class => 'poll'
+    end
+  end
+end

data/app/helpers/resque_web/failures_helper.rb

@@ -0,0 +1,54 @@
+module ResqueWeb
+  module FailuresHelper
+    def each_failure(&block)
+      Resque::Failure.each(failure_start_at, failure_per_page, params[:queue], params[:class], &block)
+    end
+
+    def failure_date_format
+      "%Y/%m/%d %T %z"
+    end
+
+    def multiple_failure_queues?
+      @multiple_failure_queues ||= Resque::Failure.queues.size > 1
+    end
+
+    def failure_queue
+      multiple_failure_queues? ? params[:id] : 'failed'
+    end
+
+    def failure_size
+      @failure_size ||= Resque::Failure.count(params[:id], params[:class])
+    end
+
+    def failure_per_page
+      @failures_per_page ||= params[:class] ? failure_size : 20
+    end
+
+    def failure_start_at
+      params[:start].to_i
+    end
+
+    def failure_end_at
+      if failure_start_at + failure_per_page > failure_size
+        failure_size
+      else
+        failure_start_at + failure_per_page
+      end
+    end
+
+    def failure_class_counts(queue = params[:id])
+      classes = Hash.new(0)
+      Resque::Failure.each(0, Resque::Failure.count(queue), queue) do |_, item|
+        class_name = item['payload']['class'] if item['payload']
+        class_name ||= "nil"
+        classes[class_name] += 1
+      end
+      classes.sort_by { |name,_| name }
+    end
+
+    def job_arguments(job)
+      return 'nil' unless job['payload']
+      Array(job['payload']['args']).map { |arg| arg.inspect }.join("\n")
+    end
+  end
+end

data/app/helpers/resque_web/overview_helper.rb

@@ -0,0 +1,6 @@
+module ResqueWeb
+  module OverviewHelper
+    include QueuesHelper
+    include WorkingHelper
+  end
+end

data/app/helpers/resque_web/queues_helper.rb

@@ -0,0 +1,72 @@
+require 'resque/failure/redis_multi_queue'
+
+module ResqueWeb
+  module QueuesHelper
+    def queues_partial_name
+      if Resque::Failure.backend == Resque::Failure::RedisMultiQueue
+        'resque_web/queues/queues_advanced'
+      else
+        'resque_web/queues/queues_basic'
+      end
+    end
+
+    def queue_names
+      Resque.queues.sort_by(&:to_s)
+    end
+
+    def queue_start_at
+      params[:start].to_i
+    end
+
+    def queue_end_at
+      if queue_start_at + queue_per_page > queue_size
+        queue_size
+      else
+        queue_start_at + queue_per_page
+      end
+    end
+
+    def queue_per_page
+      20
+    end
+
+    def queue_size(queue_name = params[:id])
+      Resque.size queue_name
+    end
+
+    def queue_jobs
+      @queue_jobs ||= Resque.peek(params[:id], queue_start_at, queue_per_page)
+    end
+
+    def failed_queue_names
+      Resque::Failure.queues.sort_by(&:to_s)
+    end
+
+    def failed_queue_name(original_queue_name)
+      "#{original_queue_name}_failed"
+    end
+
+    def failed_queue_class(queue_name)
+      Resque::Failure.count(queue_name).zero? ? "failed" : "failure"
+    end
+
+    def failed_queue_size(queue_name)
+      Resque::Failure.count(queue_name)
+    end
+
+    def failed_queue_info(queue_name)
+      failed_queue = failed_queue_name(queue_name)
+      size = failed_queue_size(failed_queue)
+
+      if size > 0
+        css_class = "badge badge-important"
+        badge = link_to(size, failure_path(failed_queue))
+      else
+        css_class = "badge"
+        badge = size.to_s
+      end
+
+      raw "<span class=\"#{css_class}\">#{badge}</span>"
+    end
+  end
+end

data/app/helpers/resque_web/stats_helper.rb

@@ -0,0 +1,54 @@
+module ResqueWeb
+  module StatsHelper
+    def resque_info
+      Resque.info.sort_by { |i| i[0].to_s }
+    end
+
+    def redis_info
+      Resque.redis.info.to_a.sort_by { |i| i[0].to_s }
+    end
+
+    def redis_key_type(key)
+      Resque.redis.type(key)
+    end
+
+    def redis_key_size(key)
+      # FIXME: there's a potential race in this method if a key is modified
+      # "in flight". Not sure how to fix it, unfortunately :(
+      case redis_key_type(key)
+      when 'none'
+        0
+      when 'list'
+        Resque.redis.llen(key)
+      when 'set'
+        Resque.redis.scard(key)
+      when 'string'
+        string = Resque.redis.get(key)
+        string ? string.length : 0
+      when 'zset'
+        Resque.redis.zcard(key)
+      end
+    end
+
+    def redis_get_array(key, start=0)
+      case redis_key_type(key)
+      when 'none'
+        []
+      when 'list'
+        Resque.redis.lrange(key, start, start + 20)
+      when 'set'
+        Resque.redis.smembers(key)[start..(start + 20)]
+      when 'string'
+        [Resque.redis.get(key)]
+      when 'zset'
+        Resque.redis.zrange(key, start, start + 20)
+      when 'hash'
+        Resque.redis.hgetall(key)
+      end
+    end
+
+    def current_subtab?(name)
+      params[:action] == name.to_s
+    end
+  end
+end

data/app/helpers/resque_web/workers_helper.rb

@@ -0,0 +1,16 @@
+module ResqueWeb
+  module WorkersHelper
+    def worker_hosts
+      @worker_hosts ||= begin
+        hosts = Hash.new { [] }
+
+        Resque.workers.each do |worker|
+          host, _ = worker.to_s.split(':')
+          hosts[host] += [worker.to_s]
+        end
+
+        hosts
+      end
+    end
+  end
+end

data/app/helpers/resque_web/working_helper.rb

@@ -0,0 +1,19 @@
+module ResqueWeb
+  module WorkingHelper
+    def workers
+      @workers ||= Resque.workers
+    end
+
+    def jobs
+      @jobs ||= workers.map(&:job)
+    end
+
+    def worker_jobs
+      @worker_jobs ||= workers.zip(jobs).reject { |w, j| w.idle? || j['queue'].nil? }
+    end
+
+    def sorted_worker_jobs
+      @sorted_worker_jobs ||= worker_jobs.sort_by { |w, j| j['run_at'] || '' }
+    end
+  end
+end

data/app/views/layouts/resque_web/application.html.erb

@@ -0,0 +1,58 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="utf-8">
+  <meta http-equiv="X-UA-Compatible" content="IE=edge,chrome=1">
+  <meta name="description" content="">
+  <meta name="viewport" content="width=device-width">
+  <title>Resque.</title>
+  <%= stylesheet_link_tag    "resque_web/application", :media => "all" %>
+  <%= javascript_include_tag "resque_web/application" %>
+  <%= csrf_meta_tags %>
+</head>
+<body>
+
+<div class="navbar navbar-inverse navbar-fixed-top">
+  <div class="navbar-inner">
+    <div class="container">
+      <a class="btn btn-navbar" data-toggle="collapse" data-target=".nav-collapse">
+        <span class="icon-bar"></span>
+        <span class="icon-bar"></span>
+        <span class="icon-bar"></span>
+      </a>
+      <img src="/assets/resque_web/lifebuoy.png" alt="" class="logo">
+      <%= link_to "Resque.", root_path, :class => "brand" %>
+      <div class="nav-collapse collapse">
+        <ul class="nav">
+          <% tabs.each do |tab_name,path| %>
+            <%= tab tab_name,path %>
+          <% end %>
+        </ul>
+      </div>
+    </div>
+  </div>
+</div>
+
+<% unless subtabs.empty? %>
+<ul class="nav subnav">
+  <div class="container">
+    <% subtabs.each do |tab_name| %>
+      <%= subtab tab_name %>
+    <% end %>
+  </div>
+</ul>
+<% end %>
+
+<div class="container" id="main">
+  <%= yield %>
+</div>
+
+<footer id="footer">
+  <div class="container">
+    <p>Powered by <a href="http://github.com/resque/resque">Resque</a> v<%=Resque::Version%></p>
+    <p>Connected to Redis namespace <%= Resque.redis.namespace %> on <%=Resque.redis_id%></p>
+  </div>
+</footer>
+
+</body>
+</html>

data/app/views/resque_web/failures/_failed_job.html.erb

@@ -0,0 +1,53 @@
+<li>
+  <dl>
+    <% if job.nil? %>
+    <dt>Error</dt>
+    <dd>Job <%= id %> could not be parsed; perhaps it contains invalid JSON?</dd>
+    <% else %>
+    <dt>Worker</dt>
+    <dd>
+      <%= job['worker'].split(':')[0...2].join(':') %>
+      on <b class="label label-info"><%= job['queue'] %></b >
+      at <b><span class="time"><%= Time.parse(job['failed_at']).strftime(failure_date_format) %></span></b>
+
+      <% if job['retried_at'] %>
+        <div class="retried">
+          Retried <b><span class="time"><%= Time.parse(job['retried_at']).strftime(failure_date_format) %></span></b>
+          <%= link_to "Remove", failure_path(:queue=>failure_queue,:id=>id), :method => :delete, :class => 'remove', :rel => 'remove' %>
+        </div>
+      <% else %>
+        <div class="controls">
+          <%= link_to "Retry", retry_failure_path(:queue=>failure_queue,:id=>id), :method => :put %>
+          or
+          <%= link_to "Remove", failure_path(:queue=>failure_queue,:id=>id), :method => :delete %>
+        </div>
+      <% end %>
+    </dd>
+    <dt>Class</dt>
+    <dd>
+      <% if job['payload'] && job['payload']['class'] %>
+        <%= link_to failures_path(:class=>job['payload']['class'],:queue=>params[:queue]) do %>
+          <code><%= job['payload']['class'] %></code>
+        <% end %>
+      <% else %>
+        <code>nil</code>
+      <% end %>
+    </dd>
+    <dt>Arguments</dt>
+    <dd><pre><%= job_arguments(job) %></pre></dd>
+    <dt>Exception</dt>
+    <dd><code><%= job['exception'] %></code></dd>
+    <dt>Error</dt>
+    <dd class="error">
+      <% if job['backtrace'] %>
+        <a href="#" class="backtrace"><%= job['error'] %></a>
+        <pre style='display:none'><%=h job['backtrace'].join("\n") %></pre>
+      <% else %>
+        <%=h job['error'] %>
+      <% end %>
+    </dd>
+    <% end %>
+  </dl>
+  <div class="r">
+  </div>
+</li>