resque-data 0.0.1

data/.gitignore

@@ -0,0 +1 @@
+/pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in resque-data.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,36 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    coderay (1.0.7)
+    interactive_editor (0.0.10)
+      spoon (>= 0.0.1)
+    method_source (0.8)
+    pry (0.9.10)
+      coderay (~> 1.0.5)
+      method_source (~> 0.8)
+      slop (~> 3.3.1)
+    rack (1.4.1)
+    rack-cors (0.2.7)
+      rack
+    rack-protection (1.2.0)
+      rack
+    redis (3.0.2)
+    redis-namespace (1.2.1)
+      redis (~> 3.0.0)
+    sinatra (1.3.3)
+      rack (~> 1.3, >= 1.3.6)
+      rack-protection (~> 1.2)
+      tilt (~> 1.3, >= 1.3.3)
+    slop (3.3.3)
+    spoon (0.0.1)
+    tilt (1.3.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  interactive_editor
+  pry
+  rack-cors
+  redis-namespace
+  sinatra

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 j-wilkins
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,65 @@
+# Resque::Data
+
+Resque Data is a small Sinatra based Rack app that will fetch basic information
+about a Resque instance.
+
+It has no security, and very little polish. Use at your own risk.
+
+That said, I'm using it to serve up Resque stats for internal apps, and it
+seems to work ok.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'resque-data'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install resque-data
+
+## Usage
+
+The easiest way to use `Resque::Data` is going to be mounting it as an engine.
+
+```
+   mount Resque::Data::Server, at: 'resque_data'
+```
+
+I'm hoping to get around to writing a `resque_data` executable, but until
+then there's a `config.ru` sitting the the `lib` folder that you can use.
+
+### Configuration
+
+You have a few configuration options off of `Resque::Data::Config`:
+
+ * `default_redis`  - set the redis to be used when none is specified.
+ * `multi_namespace` - specify whether to accept a namespace parameter. Default: true
+ * `multi_redis`    - specifiy whether to accept a redis parameter. Default: true
+
+Currently, you're going to get weird results if you allow `multi_redis` without
+allowing `multi_namespace`, but I'm not really sure why you'd want to do that.
+
+You set these options like:
+
+```
+   Resque::Data::Config.<option> = <whatever>
+```
+
+Also, the configuration is currently persisted when `Resque::Data::Server` is
+loaded, this is something to do with it being in a Sinatra `configuration`
+block. However, if all you do is configure it in an initializer then mount it
+you should never notice this.
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/lib/resque-data.rb

@@ -0,0 +1,13 @@
+require 'resque-data/version'
+require "resque-data/config"
+
+require "resque-data/fetcher"
+require "resque-data/manager"
+
+module Resque
+  module Data
+
+    autoload :Server, 'resque-data/server'
+
+  end
+end

data/lib/resque-data/config.rb

@@ -0,0 +1,38 @@
+
+module Resque
+  module Data
+
+    class Config
+      def self.default_redis=(arg)
+        Thread.current[:redis_data_default_redis] = arg
+      end
+
+      def self.default_redis
+        Thread.current[:redis_data_default_redis].tap do |ans|
+          return nil if ans.nil?
+        end
+      end
+
+      def self.multi_namespace=(true_or_false)
+        Thread.current[:redis_data_multi_namespace] = !!true_or_false
+      end
+
+      def self.multi_namespace
+        Thread.current[:redis_data_multi_namespace].tap do |ans|
+          return true if ans.nil?
+        end
+      end
+
+      def self.multi_redis=(true_or_false)
+        Thread.current[:redis_data_multi_redis] = !!true_or_false
+      end
+
+      def self.multi_redis
+        Thread.current[:redis_data_multi_redis].tap do |ans|
+          return true if ans.nil?
+        end
+      end
+
+    end
+  end
+end

data/lib/resque-data/config.ru

@@ -0,0 +1,17 @@
+require 'sinatra'
+require 'rack/cors'
+
+require 'resque-data'
+
+#Resque::Data::Config.default_redis = "localhost:6379/"
+#Resque::Data::Config.multi_namespace = false
+#Resque::Data::Config.multi_redis = false
+
+use Rack::Cors do |config|
+  config.allow do |allow|
+    allow.origins '*'
+    allow.resource '/', :headers => :any
+  end
+end
+
+run Resque::Data::Server

data/lib/resque-data/fetcher.rb

@@ -0,0 +1,83 @@
+
+class Resque::Data::Fetcher
+
+  def initialize(params)
+    @multi_namespace = params.delete(:multi_namespace)
+    @multi_namespace = true if @multi_namespace.nil?
+
+    (params[:redis] || 'redis://localhost:6379').tap do |r|
+      @connection, @default_namespace = parse_redis(r)
+    end
+
+    @default_namespace ||= params[:namespace] || :resque
+  end
+
+  def fetch
+    fetch_for(@default_namespace)
+  end
+
+  def fetch_for(ns)
+    ns = @default_namespace unless @multi_namespace
+
+    using_namespace(ns) do
+      queue_counts.merge(failed_count).merge(worker_counts)
+    end
+  end
+
+  def queue_counts
+    {queues: redis.smembers(:queues).map {|q|
+      {queue: q, count: redis.llen("queue:#{q}")}
+    }}
+  end
+
+  def failed_count
+    {failed: redis.llen(:failed)}
+  end
+
+  def worker_counts
+    workers = redis.smembers(:workers)
+    working = 0
+    workers.each {|w| working += 1 if redis.exists("worker:#{w}")}
+    {working: working, workers: workers.length}
+  end
+
+  def worker_stats
+    Hash[redis.smembers(:workers).map do |w|
+      [w, redis.exists("worker:#{w}") ? :working : :idle]
+    end]
+  end
+
+  def using_namespace(ns)
+    redis.namespace = ns
+    ret = yield
+  ensure
+    redis.namespace = @default_namespace
+    ret
+  end
+
+  def redis
+    @redis ||= Redis::Namespace.new(@default_namespace, redis: @connection)
+  end
+
+  private
+
+  def parse_redis(server)
+    case server
+    when String
+      if server['redis://']
+        redis = Redis.new(:url => server, :thread_safe => true)
+      else
+        server, namespace = server.split('/', 2)
+        host, port, db = server.split(':')
+        redis = Redis.new(:host => host, :port => port,
+          :thread_safe => true, :db => db)
+      end
+      [redis, namespace]
+    when Redis::Namespace
+      [server.redis, server.namespace]
+    else
+      [server, nil]
+    end
+  end
+
+end

data/lib/resque-data/manager.rb

@@ -0,0 +1,47 @@
+class Resque::Data::RedisManager
+
+  def initialize(default_redis = nil, multi_namespace = true, multi_redis = true, redi = [])
+    @default_redis = default_redis || 'redis://localhost:6379'
+
+    @multi_namespace, @multi_redis = multi_namespace, multi_redis
+
+    @redi = Hash[redi.map {|r| [r, new_redis(r)]}]
+    @redi[@default_redis] = new_redis(@default_redis) if @redi[@default_redis].nil?
+
+    @timestamps = Hash.new
+  end
+
+  def for(host = @default_redis)
+    return @redi[@default_redis] unless @multi_redis
+
+    (@redi[host] ||= new_redis(host)).tap do
+      @timestamps[host] = Time.now
+    end
+  end
+
+  def new_redis(host)
+    prune if needs_pruning?
+
+    Resque::Data::Fetcher.new(redis: host, multi_namespace: @multi_namespace)
+  end
+
+  def needs_pruning?
+    @redi.length >= 20
+  end
+
+  def prune
+    sorted = @timestamps.sort_by {|host, time| time}
+    sorted.pop if sorted.first.first == @default_redis
+
+    oldest = sorted.first.first
+    puts "deleting #{oldest}"
+
+    @timestamps.delete(oldest)
+    @redi.delete(oldest)
+  end
+
+  def to_json
+    @redi.to_json
+  end
+
+end

data/lib/resque-data/server.rb

@@ -0,0 +1,31 @@
+require 'sinatra'
+require 'json'
+require 'redis/namespace'
+
+module Resque
+  module Data
+    class Server < Sinatra::Application
+
+      configure do
+        $redis ||= RedisManager.new(Resque::Data::Config.default_redis,
+                                  Resque::Data::Config.multi_namespace,
+                                  Resque::Data::Config.multi_redis)
+      end
+
+      before do
+        content_type 'application/json'
+
+      end
+
+      get '/' do
+        begin
+          $redis.for(params['redis']).fetch_for(params['namespace']).to_json
+        rescue Redis::CannotConnectError => boom
+          status 404
+          {message: 'The passed redis url was invalid'}.to_json
+        end
+      end
+
+    end
+  end
+end

data/lib/resque-data/version.rb

@@ -0,0 +1,5 @@
+module Resque
+  module Data
+    VERSION = "0.0.1"
+  end
+end

data/resque-data.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'resque-data/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "resque-data"
+  gem.version       = Resque::Data::VERSION
+  gem.authors       = ["j-wilkins"]
+  gem.email         = ["pablo_honey@me.com"]
+  gem.description   = %q{Make Resque data accessible via HTTP.}
+  gem.summary       = %q{Make Resque data accessible via HTTP.}
+  gem.homepage      = "https://github.com/j-wilkins/resque-data"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "sinatra"
+  gem.add_dependency "rack-cors"
+  gem.add_dependency "redis-namespace"
+
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: resque-data
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- j-wilkins
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rack-cors
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: redis-namespace
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Make Resque data accessible via HTTP.
+email:
+- pablo_honey@me.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/resque-data.rb
+- lib/resque-data/config.rb
+- lib/resque-data/config.ru
+- lib/resque-data/fetcher.rb
+- lib/resque-data/manager.rb
+- lib/resque-data/server.rb
+- lib/resque-data/version.rb
+- resque-data.gemspec
+homepage: https://github.com/j-wilkins/resque-data
+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.24
+signing_key: 
+specification_version: 3
+summary: Make Resque data accessible via HTTP.
+test_files: []