ddtrace 0.3.1

data/lib/ddtrace/writer.rb

@@ -0,0 +1,119 @@
+require 'ddtrace/buffer'
+require 'ddtrace/transport'
+require 'ddtrace/encoding'
+require 'ddtrace/workers'
+
+module Datadog
+  # Traces and services writer that periodically sends data to the trace-agent
+  class Writer
+    attr_reader :transport
+
+    HOSTNAME = 'localhost'.freeze
+    PORT = '7777'.freeze
+
+    def initialize(options = {})
+      # writer and transport parameters
+      @buff_size = options.fetch(:buffer_size, 100)
+      @span_interval = options.fetch(:spans_interval, 1)
+      @service_interval = options.fetch(:services_interval, 120)
+
+      # transport and buffers
+      @transport = options.fetch(:transport, Datadog::HTTPTransport.new(HOSTNAME, PORT))
+      @services = {}
+
+      # handles the thread creation after an eventual fork
+      @mutex_after_fork = Mutex.new
+      @pid = nil
+
+      @traces_flushed = 0
+      @services_flushed = 0
+
+      # one worker for both services and traces, each have their own queues
+      @worker = nil
+    end
+
+    # spawns two different workers for spans and services;
+    # they share the same transport which is thread-safe
+    def start
+      @trace_handler = ->(items, transport) { send_spans(items, transport) }
+      @service_handler = ->(items, transport) { send_services(items, transport) }
+      @worker = Datadog::Workers::AsyncTransport.new(@span_interval,
+                                                     @service_interval,
+                                                     @transport,
+                                                     @buff_size,
+                                                     @trace_handler,
+                                                     @service_handler)
+
+      @worker.start()
+    end
+
+    # stops both workers for spans and services.
+    def stop
+      @worker.stop()
+      @worker = nil
+    end
+
+    # flush spans to the trace-agent, handles spans only
+    def send_spans(traces, transport)
+      return true if traces.empty?
+
+      code = transport.send(:traces, traces)
+
+      if transport.server_error? code # requeue on server error, skip on success or client error
+        traces[0..@buff_size].each do |trace|
+          @worker.enqueue_trace trace
+        end
+        return false
+      end
+
+      @traces_flushed += traces.length()
+      true
+    end
+
+    # flush services to the trace-agent, handles services only
+    def send_services(services, transport)
+      return true if services.empty?
+
+      code = transport.send(:services, services)
+      if transport.server_error? code # requeue on server error, skip on success or client error
+        @worker.enqueue_service services
+        return false
+      end
+
+      @services_flushed += 1
+      true
+    end
+
+    # enqueue the trace for submission to the API
+    def write(trace, services)
+      # In multiprocess environments, the main process initializes the +Writer+ instance and if
+      # the process forks (i.e. a web server like Unicorn or Puma with multiple workers) the new
+      # processes will share the same +Writer+ until the first write (COW). Because of that,
+      # each process owns a different copy of the +@buffer+ after each write and so the
+      # +AsyncTransport+ will not send data to the trace agent.
+      #
+      # This check ensures that if a process doesn't own the current +Writer+, async workers
+      # will be initialized again (but only once for each process).
+      pid = Process.pid
+      @mutex_after_fork.synchronize do
+        if pid != @pid
+          @pid = pid
+          # we should start threads because the worker doesn't own this
+          start()
+        end
+      end
+
+      @worker.enqueue_trace(trace)
+      @worker.enqueue_service(services)
+    end
+
+    # stats returns a dictionary of stats about the writer.
+    def stats
+      {
+        traces_flushed: @traces_flushed,
+        services_flushed: @services_flushed,
+        transport: @transport.stats
+      }
+    end
+  end
+end

metadata

@@ -0,0 +1,187 @@
+--- !ruby/object:Gem::Specification
+name: ddtrace
+version: !ruby/object:Gem::Version
+  version: 0.3.1
+platform: ruby
+authors:
+- Datadog, Inc.
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-01-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: msgpack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.43'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.43'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: appraisal
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+description: |
+  ddtrace is Datadog’s tracing client for Ruby. It is used to trace requests
+  as they flow across web servers, databases and microservices so that developers
+  have great visiblity into bottlenecks and troublesome requests.
+email:
+- dev@datadoghq.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".env"
+- ".gitignore"
+- ".rubocop.yml"
+- Appraisals
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- circle.yml
+- ddtrace.gemspec
+- docker-compose.yml
+- docs/GettingStarted
+- gemfiles/contrib.gemfile
+- gemfiles/rails3_mysql2.gemfile
+- gemfiles/rails3_postgres.gemfile
+- gemfiles/rails3_postgres_redis.gemfile
+- gemfiles/rails4_mysql2.gemfile
+- gemfiles/rails4_postgres.gemfile
+- gemfiles/rails4_postgres_redis.gemfile
+- gemfiles/rails5_mysql2.gemfile
+- gemfiles/rails5_postgres.gemfile
+- gemfiles/rails5_postgres_redis.gemfile
+- lib/ddtrace.rb
+- lib/ddtrace/buffer.rb
+- lib/ddtrace/contrib/elasticsearch/core.rb
+- lib/ddtrace/contrib/elasticsearch/patcher.rb
+- lib/ddtrace/contrib/elasticsearch/quantize.rb
+- lib/ddtrace/contrib/rails/action_controller.rb
+- lib/ddtrace/contrib/rails/action_view.rb
+- lib/ddtrace/contrib/rails/active_record.rb
+- lib/ddtrace/contrib/rails/active_support.rb
+- lib/ddtrace/contrib/rails/core_extensions.rb
+- lib/ddtrace/contrib/rails/framework.rb
+- lib/ddtrace/contrib/rails/utils.rb
+- lib/ddtrace/contrib/redis/core.rb
+- lib/ddtrace/contrib/redis/patcher.rb
+- lib/ddtrace/contrib/redis/quantize.rb
+- lib/ddtrace/contrib/redis/tags.rb
+- lib/ddtrace/encoding.rb
+- lib/ddtrace/ext/app_types.rb
+- lib/ddtrace/ext/cache.rb
+- lib/ddtrace/ext/errors.rb
+- lib/ddtrace/ext/http.rb
+- lib/ddtrace/ext/net.rb
+- lib/ddtrace/ext/redis.rb
+- lib/ddtrace/ext/sql.rb
+- lib/ddtrace/monkey.rb
+- lib/ddtrace/pin.rb
+- lib/ddtrace/span.rb
+- lib/ddtrace/tracer.rb
+- lib/ddtrace/transport.rb
+- lib/ddtrace/utils.rb
+- lib/ddtrace/version.rb
+- lib/ddtrace/workers.rb
+- lib/ddtrace/writer.rb
+homepage: https://github.com/DataDog/dd-trace-rb
+licenses:
+- BSD-3-Clause
+metadata:
+  allowed_push_host: https://rubygems.org
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2
+signing_key: 
+specification_version: 4
+summary: Datadog tracing code for your Ruby applications
+test_files: []