puma-newrelic-codeur 0.1.11 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: deae6f29617e1568a8f6e22d40d6099453bd9d72a2884ea88383ac88834e6134
-  data.tar.gz: 4e2160ed9247856d16ac27c3031dee6d1478d6b9f37fde98493375d4d0e3b7ec
+  metadata.gz: deeb0397b1f0915290df6b38eaf216ea5d890d925686f84b0750fb2d08307a3a
+  data.tar.gz: 8f85e4fc25e3b8ff0928363ef12c333be2281f046465c9a0c7b0e58b11e6a787
 SHA512:
-  metadata.gz: a0a190611626d14f6128cac61e75715a20952623749a378c4738b4326766bbe019f90805a3d3c9501a093bee1bb3f4d15d0272caa894dbfd2e82f93e5fd5105a
-  data.tar.gz: da685073885aeed72e37375073aa9d6bb7b2503ca60eaced218a69e88627177ea0a4d4cba2fca168ab5a09bad5e81dc1325b01fbd7b111ad9f6487bc805fb97f
+  metadata.gz: 7eb3de70d9831ca1b60e85e7a122659f3eb55ce54aa4e66fecb834881c0e8afdcf2914f4c9a6cadba5b450f44dfcf3ca3c7f9c7b921468ae4f32f843939345a2
+  data.tar.gz: 4472ba6ce624f43f89c63f8e9e39a9c2c622f56477606a45f5ac118e9c5411ff76fbfaf56e194eb5c8932454e2cca0c4e99bfb3a9c9ced023fb8c1563a7dd498

data/lib/puma/newrelic/plugin.rb

@@ -11,7 +11,7 @@ module Puma
           end
         end
 
-        launcher.events.on_booted do
+        in_background do
           sampler.collect
         end
       end

data/lib/puma/newrelic/sampler.rb

@@ -1,28 +1,41 @@
-require "newrelic_rpm"
+require "yaml"
+require "faraday"
 
 module Puma
   module NewRelic
     class Sampler
       def initialize(launcher)
-        config = ::NewRelic::Agent.config[:puma] || {}
+        @newrelic_config = YAML.load_file("config/newrelic.yml", aliases: true)[ENV["RAILS_ENV"]]
+        config = @newrelic_config[:puma] || {}
         @launcher = launcher
         @sample_rate = config.fetch("sample_rate", 23)
         @keys = config.fetch("keys", %i[backlog running pool_capacity max_threads requests_count]).map(&:to_sym)
         @last_sample_at = Time.now
+        @conn = Faraday.new(url: "https://metric-api.eu.newrelic.com/metric/v1") do |faraday|
+          faraday.headers["Content-Type"] = "application/json"
+          faraday.headers["Api-Key"] = @newrelic_config["license_key"]
+          faraday.adapter Faraday.default_adapter
+        end
       end
 
       def collect
+        return unless agent_enabled?
+
         @running = true
         while @running
           sleep 1
 
           if should_sample?
-            record_metrics(@launcher.stats)
             @last_sample_at = Time.now
+            record_metrics(@launcher.stats)
           end
         end
       end
 
+      def agent_enabled?
+        @newrelic_config["agent_enabled"]
+      end
+
       def should_sample?
         Time.now - @last_sample_at > @sample_rate
       end
@@ -44,10 +57,38 @@ module Puma
           stats.each { |key, value| metrics[key] += value if @keys.include?(key) }
         end
 
+        payload = [
+          {
+            common: {
+              timestamp: Time.now.to_i,
+              "interval.ms": @sample_rate * 1000,
+              attributes: {
+                "app.name": @newrelic_config["app_name"]
+              }
+            },
+            metrics: []
+          }
+        ]
         metrics.each do |key, value|
-          ::NewRelic::Agent.logger.info("Recorded metric: Custom/Puma/#{key}=#{value}")
-          ::NewRelic::Agent.record_metric("Custom/Puma/#{key}", value)
+          payload[0]["metrics"] << {
+            name: "Custom/Puma/#{key}",
+            type: "count",
+            value: value
+          }
         end
+
+        response = @conn.post do |req|
+          req.body = payload.to_json
+        end
+
+        if response.status != 202
+          @launcher.log_writer.write("Failed to send metrics to New Relic: #{response.status} - #{response.body}")
+        end
+
+        # metrics.each do |key, value|
+        #   ::NewRelic::Agent.logger.info("Recorded metric: Custom/Puma/#{key}=#{value}")
+        #   ::NewRelic::Agent.record_metric("Custom/Puma/#{key}", value)
+        # end
       end
     end
   end

data/lib/puma/newrelic/version.rb

@@ -2,6 +2,6 @@
 
 module Puma
   module NewRelic
-    VERSION = "0.1.11"
+    VERSION = "0.1.12"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: puma-newrelic-codeur
 version: !ruby/object:Gem::Version
-  version: 0.1.11
+  version: 0.1.12
 platform: ruby
 authors:
 - Benoist Claassen