zipkin-tracer 0.0.1

data/lib/zipkin-tracer/careless_scribe.rb

@@ -0,0 +1,35 @@
+# Copyright 2012 Twitter Inc.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#      http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+class CarelessScribe
+  def initialize(scribe)
+    @scribe = scribe
+  end
+
+  def log(*args)
+    @scribe.log(*args)
+  rescue ThriftClient::NoServersAvailable, Thrift::Exception
+    # I couldn't care less
+  end
+
+  def batch(&block)
+    @scribe.batch(&block)
+  rescue ThriftClient::NoServersAvailable, Thrift::Exception
+    # I couldn't care less
+  end
+
+  def method_missing(name, *args)
+    @scribe.send(name, *args)
+  end
+end

data/lib/zipkin-tracer/version.rb

@@ -0,0 +1,17 @@
+# Copyright 2012 Twitter Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+module ZipkinTracer
+  VERSION = "0.0.1"
+end
+

data/lib/zipkin-tracer.rb

@@ -0,0 +1,79 @@
+# Copyright 2012 Twitter Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+require 'finagle-thrift'
+require 'finagle-thrift/trace'
+require 'scribe'
+
+require 'zipkin-tracer/careless_scribe'
+
+module ZipkinTracer extend self
+
+  class RackHandler
+    def initialize(app)
+      @app = app
+      @lock = Mutex.new
+
+      config = app.config.zipkin_tracer
+      @service_name = config[:service_name]
+      @service_port = config[:service_port]
+
+      scribe =
+        if config[:scribe_server] then
+          Scribe.new(config[:scribe_server])
+        else
+          Scribe.new()
+        end
+
+      scribe_max_buffer =
+        if config[:scribe_max_buffer] then
+          config[:scribe_max_buffer]
+        else
+          10
+        end
+
+      @sample_rate =
+        if config[:sample_rate] then
+          config[:sample_rate]
+        else
+          0.1
+        end
+
+      ::Trace.tracer = ::Trace::ZipkinTracer.new(CarelessScribe.new(scribe), scribe_max_buffer)
+    end
+
+    def call(env)
+      id = ::Trace::TraceId.new(::Trace.generate_id, nil, ::Trace.generate_id, true)
+      ::Trace.default_endpoint = ::Trace.default_endpoint.with_service_name(@service_name).with_port(@service_port)
+      ::Trace.sample_rate=(@sample_rate)
+      tracing_filter(id, env) { @app.call(env) }
+    end
+
+    private
+    def tracing_filter(trace_id, env)
+      @lock.synchronize do
+        ::Trace.push(trace_id)
+        ::Trace.set_rpc_name(env["REQUEST_METHOD"]) # get/post and all that jazz
+        ::Trace.record(::Trace::BinaryAnnotation.new("http.uri", env["PATH_INFO"], "STRING", ::Trace.default_endpoint))
+        ::Trace.record(::Trace::Annotation.new(::Trace::Annotation::SERVER_RECV, ::Trace.default_endpoint))
+      end
+      yield if block_given?
+    ensure
+      @lock.synchronize do
+        ::Trace.record(::Trace::Annotation.new(::Trace::Annotation::SERVER_SEND, ::Trace.default_endpoint))
+        ::Trace.pop
+      end
+    end
+  end
+
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification 
+name: zipkin-tracer
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Franklin Hu
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-06-12 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: finagle-thrift
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 1
+        - 2
+        - 0
+        version: 1.2.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: scribe
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 0
+        - 2
+        - 4
+        version: 0.2.4
+  type: :runtime
+  version_requirements: *id002
+description: Adds tracing instrumentation for ruby applications
+email: 
+- franklin@twitter.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/zipkin-tracer/careless_scribe.rb
+- lib/zipkin-tracer/version.rb
+- lib/zipkin-tracer.rb
+homepage: https://github.com/twitter/zipkin
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 17
+      segments: 
+      - 1
+      - 3
+      - 5
+      version: 1.3.5
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Ruby tracing via Zipkin
+test_files: []
+