Skip to content
This repository has been archived by the owner on Oct 4, 2018. It is now read-only.

Commit

Permalink
make all utilities and examples work
Browse files Browse the repository at this point in the history
  • Loading branch information
fredemmott committed May 16, 2011
1 parent 2554556 commit db9820b
Show file tree
Hide file tree
Showing 12 changed files with 23 additions and 32 deletions.
4 changes: 2 additions & 2 deletions examples/direct/TextConsumer.rb
@@ -1,7 +1,7 @@
#!/usr/bin/env ruby

$LOAD_PATH.push File.dirname(__FILE__) + '/../../gen-rb/'
require 'job_pool'
$LOAD_PATH.push File.dirname(__FILE__) + '/../../lib/rb/'
require 'jp/thrift'
include Jp

socket = Thrift::Socket.new 'localhost', 9090
Expand Down
4 changes: 2 additions & 2 deletions examples/direct/TextProducer.rb
@@ -1,7 +1,7 @@
#!/usr/bin/env ruby

$LOAD_PATH.push File.dirname(__FILE__) + '/../../gen-rb/'
require 'job_pool'
$LOAD_PATH.push File.dirname(__FILE__) + '/../../lib/rb/'
require 'jp/thrift'
include Jp

socket = Thrift::Socket.new 'localhost', 9090
Expand Down
2 changes: 1 addition & 1 deletion examples/simple/JsonConsumer.rb
@@ -1,6 +1,6 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb'
require 'consumer'
require 'jp/consumer'

c = Jp::JsonConsumer.new 'json' do |message|
print "I consumed:\n%s\n" % message.inspect
Expand Down
2 changes: 1 addition & 1 deletion examples/simple/JsonProducer.rb
@@ -1,6 +1,6 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb'
require 'producer'
require 'jp/producer'

p = Jp::JsonProducer.new 'json'
doc = {
Expand Down
2 changes: 1 addition & 1 deletion examples/simple/TextConsumer.rb
@@ -1,6 +1,6 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb'
require 'consumer'
require 'jp/consumer'

c = Jp::TextConsumer.new 'text' do |message|
print "I consumed a %s.\n" % message
Expand Down
2 changes: 1 addition & 1 deletion examples/simple/TextProducer.rb
@@ -1,6 +1,6 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb'
require 'producer'
require 'jp/producer'

p = Jp::TextProducer.new 'text'
p.add 'simple pie'
2 changes: 1 addition & 1 deletion examples/simple/ThriftConsumer.rb
@@ -1,7 +1,7 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb/'
$LOAD_PATH.push '../gen-rb/'
require 'consumer'
require 'jp/consumer'
require 'example_types'

c = Jp::ThriftConsumer.new 'thrift', ExampleData do |message|
Expand Down
2 changes: 1 addition & 1 deletion examples/simple/ThriftProducer.rb
@@ -1,7 +1,7 @@
#!/usr/bin/env ruby
$LOAD_PATH.push '../../lib/rb'
$LOAD_PATH.push '../gen-rb/'
require 'producer'
require 'jp/producer'
require 'example_types'

p = Jp::ThriftProducer.new 'thrift'
Expand Down
3 changes: 2 additions & 1 deletion lib/rb/jp/consumer.rb
Expand Up @@ -2,9 +2,10 @@

require 'jp/client'

autoload :Rev, 'rev'
require 'rev'
autoload :JSON, 'json'


module Jp
class AbstractConsumer < AbstractClient
def initialize queue, options = {}, &block
Expand Down
1 change: 1 addition & 0 deletions lib/rb/jp/thrift.rb
@@ -1,3 +1,4 @@
$LOAD_PATH.unshift File.dirname(__FILE__) + '/gen-rb'
require 'job_pool'
require 'job_pool_instrumented'
$LOAD_PATH.shift
24 changes: 7 additions & 17 deletions utils/jp-pool-size.rb
@@ -1,37 +1,27 @@
#!/usr/bin/env ruby

$LOAD_PATH.push File.dirname(__FILE__) + '/../classes/'
$LOAD_PATH.push File.dirname(__FILE__) + '/../lib/rb/'

require 'ruby-1.9.0-compat'
require 'jp_config_loader'
require 'jp/server/configuration_loader'
require 'jp/server/handler'
require 'mongo'


module Jp
def self.pool_sizes
# Load options & defaults
options = Jp::load_server_config

options[:mongo_uri] ||= 'mongodb://localhost'
raise ArgumentError.new "mongo_db option must be specified" unless options[:mongo_db]
raise ArgumentError.new "pools option must be specified" unless options[:pools]
raise ArgumentError.new "pools option must not be empty" unless ! options[:pools].empty?

# Connect to mongodb
connection = Mongo::Connection.from_uri options[:mongo_uri]
database = connection.db options[:mongo_db]
options = Jp::Server::ConfigurationLoader.server_config
server = Jp::Server::Handler.new(options)

out = Hash.new
options[:pools].each do |name, data|
out[name] = database[name].count
server.pools.each do |name, data|
out[name] = server.database[name].count
end
out
end
end

if File.expand_path($0) == File.expand_path(__FILE__)
Jp.pool_sizes.each do |name, size|
next if ARGV[0] && ARGV[0] != name
print "%s: %d\n" % [name, size]
end
end
7 changes: 3 additions & 4 deletions utils/jp-stats.rb
@@ -1,14 +1,13 @@
#!/usr/bin/env ruby1.9
#!/usr/bin/env ruby
# Requires:
# - Thrift
$LOAD_PATH.push File.dirname(__FILE__) + '/../lib/rb'
require 'jp/thrift'

module Jp
def self.stats host, port
##### Thrift setup #####

$LOAD_PATH.push File.dirname(__FILE__) + '/../gen-rb/'
require 'job_pool_instrumented'

transport = Thrift::BufferedTransport.new(Thrift::Socket.new(host, port))
client = JobPoolInstrumented::Client.new(Thrift::BinaryProtocol.new(transport))
transport.open
Expand Down

0 comments on commit db9820b

Please sign in to comment.