upfluence-utils 0.5.2 → 0.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b23e691f998914c5bc1d98d97af634f4361f3587
-  data.tar.gz: 0244e8237ceda7c2dcc463fc704ecb2aab689f47
+  metadata.gz: 96b4ca9a69d062c47d1338e7abed5ecc8077cb39
+  data.tar.gz: 2fb9fb60bfb19b53fb9204e36dc2cf46adf64aa9
 SHA512:
-  metadata.gz: e3fbb887339d789a60cdec17461fb64e9c00142585f752e9ea472cea509871d6173376d70d895e1bb4bfeb6fa9efd9670a16200bd44f5aea7d988bfa2bc672cf
-  data.tar.gz: 488d26965cfd791f160376afcaa493e956cc26cffe74bce0c2e511d429da3a9089cba42b68e0102ef37b0dd1f2717cf9f639f28a76b07638f793cab41ce584a1
+  metadata.gz: 777f3f18e6305964baebe31806b43e397b381eceea0ec98fb4397466dd5b1a928a70b7a4424a7670bbdab50cc8180557105a278ac942bf0bd5b8010e2251df93
+  data.tar.gz: c46b9b7d281e884b719bad94d214247393512b8698c4366e1147739d049ff7b30136328423ed42460cd0182a213908da768f262d19eda72a315921e937ad84a4

data/lib/upfluence/http/endpoint/profiler.rb

@@ -0,0 +1,51 @@
+require 'stackprof'
+
+module Upfluence
+  module HTTP
+    module Endpoint
+      class Profiler
+        def initialize(opts = {})
+          @interval = opts[:interval] || 50
+          @mode = opts[:mode] || :cpu
+          @mapping = Rack::URLMap.new(
+            '/start' => lambda { |env|  start(env) },
+            '/stop' => lambda { |env| stop(env) },
+            '/profile' => lambda { |env| profile(env) }
+          )
+        end
+
+        def call(env)
+          @mapping.call(env)
+        end
+
+        private
+
+        def stop(env)
+          StackProf.stop
+          return ok
+        end
+
+        def start(env)
+          StackProf.start(mode: @mode, interval: @interval, raw: false)
+          return ok
+        end
+
+        def profile(env)
+          results = StackProf.results
+          @last_results = Marshal.dump(results) if results
+
+          return [200, {}, [@last_results]] if @last_results
+          return [404, {}, ['No profile available']]
+        end
+
+        def ok
+          [200, {}, ['ok']]
+        end
+
+        def path(env)
+          Rack::Request.new(env).path
+        end
+      end
+    end
+  end
+end

data/lib/upfluence/http/server.rb

@@ -8,6 +8,7 @@ require 'upfluence/error_logger'
 
 require 'upfluence/http/builder'
 require 'upfluence/http/endpoint/healthcheck'
+require 'upfluence/http/endpoint/profiler'
 require 'upfluence/http/middleware/logger'
 require 'upfluence/http/middleware/application_headers'
 require 'upfluence/http/middleware/handle_exception'
@@ -26,7 +27,8 @@ module Upfluence
         push_gateway_url: ENV['PUSH_GATEWAY_URL'],
         push_gateway_interval: 15, # sec
         app_name: ENV['APP_NAME'] || 'uhttp-rb-server',
-        unit_name: ENV['UNIT_NAME'] || 'uhttp-rb-server-anonymous'
+        unit_name: ENV['UNIT_NAME'] || 'uhttp-rb-server-anonymous',
+        debug: ENV['DEBUG']
       }.freeze
 
       def initialize(options = {}, &block)
@@ -54,6 +56,10 @@ module Upfluence
             run_thrift Base::Base_service::BaseService::Processor, base_handler
           end
 
+          map '/debug' do
+            run Endpoint::Profiler.new
+          end if opts[:debug]
+
           instance_eval(&block)
         end
 

data/lib/upfluence/utils/version.rb

@@ -1,5 +1,5 @@
 module Upfluence
   module Utils
-    VERSION = '0.5.2'.freeze
+    VERSION = '0.5.3'.freeze
   end
 end

data/rbutils.gemspec

@@ -28,6 +28,7 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency 'activesupport'
   spec.add_runtime_dependency 'puma'
   spec.add_runtime_dependency 'rack'
+  spec.add_runtime_dependency 'stackprof'
   spec.add_runtime_dependency 'prometheus-client'
   spec.add_runtime_dependency 'active_model_serializers', '~> 0.9.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: upfluence-utils
 version: !ruby/object:Gem::Version
-  version: 0.5.2
+  version: 0.5.3
 platform: ruby
 authors:
 - Upfluence
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-12-21 00:00:00.000000000 Z
+date: 2018-05-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -184,6 +184,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: stackprof
+  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: prometheus-client
   requirement: !ruby/object:Gem::Requirement
@@ -238,6 +252,7 @@ files:
 - lib/upfluence/http/builder.rb
 - lib/upfluence/http/endpoint/api_endpoint.rb
 - lib/upfluence/http/endpoint/healthcheck.rb
+- lib/upfluence/http/endpoint/profiler.rb
 - lib/upfluence/http/middleware/application_headers.rb
 - lib/upfluence/http/middleware/handle_exception.rb
 - lib/upfluence/http/middleware/logger.rb