pinbo 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Andrew Rudenko
+
+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.rdoc

@@ -0,0 +1,18 @@
+= pinbo
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+   bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2009 Andrew Rudenko. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,52 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "pinbo"
+    gem.summary = %Q{ruby client for Pinba}
+    gem.description = %Q{ruby client for Pinba}
+    gem.email = "ceo@prepor.ru"
+    gem.homepage = "http://github.com/prepor/pinbo"
+    gem.authors = ["Andrew Rudenko"]
+    gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "pinbo #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/examples/sinatra_app.rb

@@ -0,0 +1,13 @@
+require 'rubygems'
+require 'sinatra'
+require 'pinbo'
+
+Object.send :include, Pinbo::Timer
+
+use Pinbo::Middleware
+get '/' do
+  timer :layout => :index, :action => :generate_all do
+    2 + 2 * 4
+  end
+  "Hello World!"
+end

data/lib/pinbo/middleware.rb

@@ -0,0 +1,14 @@
+module Pinbo
+  class Middleware
+    def initialize(app)
+      @app = app
+    end
+  
+    def call(env)
+      Pinbo.start :script_name => env['REQUEST_PATH']
+      res = @app.call(env)
+      Pinbo.stop :status => res[0], :server_name => env['SERVER_NAME'], :document_size => res[1]['Content-Length'].to_i
+      res
+    end
+  end
+end

data/lib/pinbo/request.rb

@@ -0,0 +1,47 @@
+module Pinbo
+  class Request
+    attr_accessor :options
+    def initialize(options)
+      self.options = options
+    end
+  
+    def cmd
+      req = Proto::Pinbo::Request.new 
+      req.hostname = Config[:host]
+      req.server_name = options[:data][:server_name]
+      req.script_name = options[:data][:script_name]
+      req.request_count = options[:data][:request_count]
+      req.document_size = options[:data][:document_size]
+      req.memory_peak = options[:data][:memory_peak]
+      req.request_time = options[:data][:request_time]
+      req.ru_utime = options[:data][:ru_utime]
+      req.ru_stime = options[:data][:ru_stime]
+      Pinbo.timers.each do |timer|
+         req.timer_hit_count << timer[:counter]
+         req.timer_value << timer[:period]
+         req.timer_tag_count << timer[:tags].size
+         timer[:normalized_tags].each do |tag|
+           req.timer_tag_name << tag[:name_id]
+           req.dictionary << tag[:name]
+           req.timer_tag_value << tag[:value_id]
+           req.dictionary << tag[:value]
+         end
+       end
+      req.status = options[:data][:status]
+      # require 'ruby-debug'
+      # debugger
+      req.serialize_to_string
+    end
+  
+    def perform
+      sock = nil
+      begin
+        sock = UDPSocket.open        
+        sock.send(cmd, 0, Config[:pinba_host], Config[:pinba_port])
+      rescue IOError, SystemCallError
+      ensure
+        sock.close if sock
+      end
+    end
+  end
+end

data/lib/pinbo/timer.rb

@@ -0,0 +1,5 @@
+module Pinbo::Timer
+  def timer(tags = {}, &block)
+    Pinbo.timer(tags, &block)
+  end
+end

data/lib/pinbo.rb

@@ -0,0 +1,100 @@
+require 'rubygems'
+require 'socket'
+require 'timeout'
+require 'ruby_protobuf'
+require "socket"
+require 'proto/pinbo.pb'
+
+module Pinbo
+
+  require 'pinbo/timer'
+  require 'pinbo/request'
+  require 'pinbo/middleware'
+  
+  Config = {
+    :host => Socket.gethostname,
+    :pinba_host => '74.86.33.154', #:pinba_host => 'localhost',
+    :pinba_port => 30002
+  }
+  
+  class << self
+  
+    def counter
+      Thread.current[:counter] ||= 1
+    end
+  
+    def counter=(val)
+      Thread.current[:counter] = val
+    end
+  
+    def data
+      Thread.current[:data]
+    end
+  
+    def data=(val)
+      Thread.current[:data] = val
+    end
+  
+    def timers
+      Thread.current[:timers]
+    end
+  
+    def timers=(val)
+      Thread.current[:timers] = val
+    end
+  
+    def increase_counter
+      self.counter += 1
+    end
+  
+    def start(options = {})
+      Thread.current[:request_start] = Time.now
+      Thread.current[:timer_counter] = 0
+      self.data = { 
+                    :server_name => Socket.gethostname,
+                    :script_name => '/test.rb',
+                    :request_count => counter,
+                    :document_size => 0,
+                    :memory_peak => 0,
+                    :request_time => 0,
+                    :ru_utime => 0,
+                    :ru_stime => 0,
+                    :status => 200 }.merge(options)
+      self.timers = []
+    end
+  
+    def timer_counter
+      Thread.current[:timer_counter] += 1
+      Thread.current[:timer_counter] - 1
+    end
+  
+    def stop(options = {})
+      data.merge!( :request_time => Time.now - Thread.current[:request_start] ).merge!(options)
+      Request.new( :data => data, :timers => timers).perform
+      increase_counter
+    end
+  
+    def timer(tags = {}, &block)
+      t = get_timer(tags)
+      t[:start] = Time.now
+      block.call
+      time = Time.now
+      t[:counter] += 1
+      t[:stop] = time
+      t[:period] += t[:stop] - t[:start]
+    end
+    
+    def normalize_tags(raw_tags)
+      raw_tags.map { |k, v| { :name_id => timer_counter, :value_id => timer_counter, :name => k.to_s, :value => v.to_s } }
+    end
+    
+    def get_timer(tags)
+      unless (t = self.timers.detect { |v| v[:tags] == tags } ) 
+        t = { :tags => tags, :normalized_tags => normalize_tags(tags), :counter => 0, :period => 0 }
+        self.timers << t
+      end
+      t
+    end
+  end
+  
+end

data/lib/proto/pinbo.pb.rb

@@ -0,0 +1,54 @@
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: pinbo.proto>
+# package Proto.Pinbo;
+# option optimize_for = SPEED;
+# 
+# message Request {
+# 	required string hostname		= 1;
+# 	required string server_name		= 2;
+# 	required string script_name		= 3;
+# 	required uint32 request_count	= 4;
+# 	required uint32 document_size	= 5;
+# 	required uint32 memory_peak		= 6;
+# 	required float request_time		= 7;
+# 	required float ru_utime			= 8;
+# 	required float ru_stime			= 9;
+# 
+# 	repeated uint32 timer_hit_count = 10;
+# 	repeated float timer_value      = 11;
+# 	repeated uint32 timer_tag_count = 12;
+# 	repeated uint32 timer_tag_name  = 13;
+# 	repeated uint32 timer_tag_value = 14;
+# 	repeated string dictionary      = 15;
+# 	optional uint32 status          = 16;
+# }
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+module Proto
+  module Pinbo
+    ::Protobuf::OPTIONS[:"optimize_for"] = :SPEED
+    class Request < ::Protobuf::Message
+      defined_in __FILE__
+      required :string, :hostname, 1
+      required :string, :server_name, 2
+      required :string, :script_name, 3
+      required :uint32, :request_count, 4
+      required :uint32, :document_size, 5
+      required :uint32, :memory_peak, 6
+      required :float, :request_time, 7
+      required :float, :ru_utime, 8
+      required :float, :ru_stime, 9
+      repeated :uint32, :timer_hit_count, 10
+      repeated :float, :timer_value, 11
+      repeated :uint32, :timer_tag_count, 12
+      repeated :uint32, :timer_tag_name, 13
+      repeated :uint32, :timer_tag_value, 14
+      repeated :string, :dictionary, 15
+      optional :uint32, :status, 16
+    end
+  end
+end

data/lib/proto/pinbo.proto

@@ -0,0 +1,22 @@
+package Proto.Pinbo;
+option optimize_for = SPEED;
+
+message Request {
+	required string hostname		= 1;
+	required string server_name		= 2;
+	required string script_name		= 3;
+	required uint32 request_count	= 4;
+	required uint32 document_size	= 5;
+	required uint32 memory_peak		= 6;
+	required float request_time		= 7;
+	required float ru_utime			= 8;
+	required float ru_stime			= 9;
+
+	repeated uint32 timer_hit_count = 10;
+	repeated float timer_value      = 11;
+	repeated uint32 timer_tag_count = 12;
+	repeated uint32 timer_tag_name  = 13;
+	repeated uint32 timer_tag_value = 14;
+	repeated string dictionary      = 15;
+	optional uint32 status          = 16;
+}

data/pinbo.gemspec

@@ -0,0 +1,60 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{pinbo}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Andrew Rudenko"]
+  s.date = %q{2009-11-08}
+  s.description = %q{ruby client for Pinba}
+  s.email = %q{ceo@prepor.ru}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "examples/sinatra_app.rb",
+     "lib/pinbo.rb",
+     "lib/pinbo/middleware.rb",
+     "lib/pinbo/request.rb",
+     "lib/pinbo/timer.rb",
+     "lib/proto/pinbo.pb.rb",
+     "lib/proto/pinbo.proto",
+     "spec/pinbo_spec.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/prepor/pinbo}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{ruby client for Pinba}
+  s.test_files = [
+    "spec/pinbo_spec.rb",
+     "spec/spec_helper.rb",
+     "examples/sinatra_app.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+  end
+end

data/spec/pinbo_spec.rb

@@ -0,0 +1,11 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Pinbo" do
+  
+  describe "common statistics" do
+    it "should description" do
+      
+    end
+  end
+
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'pinbo'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification 
+name: pinbo
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Andrew Rudenko
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-18 00:00:00 +03:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: ruby client for Pinba
+email: ceo@prepor.ru
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- examples/sinatra_app.rb
+- lib/pinbo.rb
+- lib/pinbo/middleware.rb
+- lib/pinbo/request.rb
+- lib/pinbo/timer.rb
+- lib/proto/pinbo.pb.rb
+- lib/proto/pinbo.proto
+- pinbo.gemspec
+- spec/pinbo_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/prepor/pinbo
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: ruby client for Pinba
+test_files: 
+- spec/pinbo_spec.rb
+- spec/spec_helper.rb
+- examples/sinatra_app.rb