brainzlab 0.1.2 → 0.1.3

data/lib/brainzlab/instrumentation.rb

@@ -58,138 +58,138 @@ module BrainzLab
       end
 
       def install_net_http!
-        require_relative "instrumentation/net_http"
+        require_relative 'instrumentation/net_http'
         NetHttp.install!
       end
 
       def install_faraday!
         return unless defined?(::Faraday)
 
-        require_relative "instrumentation/faraday"
+        require_relative 'instrumentation/faraday'
         FaradayMiddleware.install!
       end
 
       def install_httparty!
         return unless defined?(::HTTParty)
 
-        require_relative "instrumentation/httparty"
+        require_relative 'instrumentation/httparty'
         HTTPartyInstrumentation.install!
       end
 
       def install_active_record!
-        require_relative "instrumentation/active_record"
+        require_relative 'instrumentation/active_record'
         ActiveRecord.install!
       end
 
       def install_redis!
         return unless defined?(::Redis)
 
-        require_relative "instrumentation/redis"
+        require_relative 'instrumentation/redis'
         RedisInstrumentation.install!
       end
 
       def install_sidekiq!
         return unless defined?(::Sidekiq)
 
-        require_relative "instrumentation/sidekiq"
+        require_relative 'instrumentation/sidekiq'
         SidekiqInstrumentation.install!
       end
 
       def install_graphql!
         return unless defined?(::GraphQL)
 
-        require_relative "instrumentation/graphql"
+        require_relative 'instrumentation/graphql'
         GraphQLInstrumentation.install!
       end
 
       def install_mongodb!
         return unless defined?(::Mongo) || defined?(::Mongoid)
 
-        require_relative "instrumentation/mongodb"
+        require_relative 'instrumentation/mongodb'
         MongoDBInstrumentation.install!
       end
 
       def install_elasticsearch!
         return unless defined?(::Elasticsearch) || defined?(::OpenSearch)
 
-        require_relative "instrumentation/elasticsearch"
+        require_relative 'instrumentation/elasticsearch'
         ElasticsearchInstrumentation.install!
       end
 
       def install_action_mailer!
         return unless defined?(::ActionMailer)
 
-        require_relative "instrumentation/action_mailer"
+        require_relative 'instrumentation/action_mailer'
         ActionMailerInstrumentation.install!
       end
 
       def install_delayed_job!
         return unless defined?(::Delayed::Job) || defined?(::Delayed::Backend)
 
-        require_relative "instrumentation/delayed_job"
+        require_relative 'instrumentation/delayed_job'
         DelayedJobInstrumentation.install!
       end
 
       def install_grape!
         return unless defined?(::Grape::API)
 
-        require_relative "instrumentation/grape"
+        require_relative 'instrumentation/grape'
         GrapeInstrumentation.install!
       end
 
       def install_solid_queue!
         return unless defined?(::SolidQueue)
 
-        require_relative "instrumentation/solid_queue"
+        require_relative 'instrumentation/solid_queue'
         SolidQueueInstrumentation.install!
       end
 
       def install_good_job!
         return unless defined?(::GoodJob)
 
-        require_relative "instrumentation/good_job"
+        require_relative 'instrumentation/good_job'
         GoodJobInstrumentation.install!
       end
 
       def install_resque!
         return unless defined?(::Resque)
 
-        require_relative "instrumentation/resque"
+        require_relative 'instrumentation/resque'
         ResqueInstrumentation.install!
       end
 
       def install_excon!
         return unless defined?(::Excon)
 
-        require_relative "instrumentation/excon"
+        require_relative 'instrumentation/excon'
         ExconInstrumentation.install!
       end
 
       def install_typhoeus!
         return unless defined?(::Typhoeus)
 
-        require_relative "instrumentation/typhoeus"
+        require_relative 'instrumentation/typhoeus'
         TyphoeusInstrumentation.install!
       end
 
       def install_dalli!
         return unless defined?(::Dalli::Client)
 
-        require_relative "instrumentation/dalli"
+        require_relative 'instrumentation/dalli'
         DalliInstrumentation.install!
       end
 
       def install_aws!
         return unless defined?(::Aws)
 
-        require_relative "instrumentation/aws"
+        require_relative 'instrumentation/aws'
         AWSInstrumentation.install!
       end
 
       def install_stripe!
         return unless defined?(::Stripe)
 
-        require_relative "instrumentation/stripe"
+        require_relative 'instrumentation/stripe'
         StripeInstrumentation.install!
       end
 

data/lib/brainzlab/nerve/client.rb

@@ -1,22 +1,22 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
-require "uri"
+require 'net/http'
+require 'json'
+require 'uri'
 
 module BrainzLab
   module Nerve
     class Client
       def initialize(config)
         @config = config
-        @base_url = config.nerve_url || "https://nerve.brainzlab.ai"
+        @base_url = config.nerve_url || 'https://nerve.brainzlab.ai'
       end
 
       # Report a job execution
       def report_job(job_class:, job_id:, queue:, status:, started_at:, ended_at:, **attributes)
         response = request(
           :post,
-          "/api/v1/jobs",
+          '/api/v1/jobs',
           body: {
             job_class: job_class,
             job_id: job_id,
@@ -31,7 +31,7 @@ module BrainzLab
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("report_job", e)
+        log_error('report_job', e)
         false
       end
 
@@ -39,7 +39,7 @@ module BrainzLab
       def report_failure(job_class:, job_id:, queue:, error_class:, error_message:, backtrace: nil, **attributes)
         response = request(
           :post,
-          "/api/v1/jobs/failures",
+          '/api/v1/jobs/failures',
           body: {
             job_class: job_class,
             job_id: job_id,
@@ -54,23 +54,23 @@ module BrainzLab
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("report_failure", e)
+        log_error('report_failure', e)
         false
       end
 
       # Get job statistics
-      def stats(queue: nil, job_class: nil, period: "1h")
+      def stats(queue: nil, job_class: nil, period: '1h')
         params = { period: period }
         params[:queue] = queue if queue
         params[:job_class] = job_class if job_class
 
-        response = request(:get, "/api/v1/stats", params: params)
+        response = request(:get, '/api/v1/stats', params: params)
 
         return nil unless response.is_a?(Net::HTTPSuccess)
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("stats", e)
+        log_error('stats', e)
         nil
       end
 
@@ -80,27 +80,27 @@ module BrainzLab
         params[:queue] = queue if queue
         params[:status] = status if status
 
-        response = request(:get, "/api/v1/jobs", params: params)
+        response = request(:get, '/api/v1/jobs', params: params)
 
         return [] unless response.is_a?(Net::HTTPSuccess)
 
         data = JSON.parse(response.body, symbolize_names: true)
         data[:jobs] || []
       rescue StandardError => e
-        log_error("list_jobs", e)
+        log_error('list_jobs', e)
         []
       end
 
       # List queues
       def list_queues
-        response = request(:get, "/api/v1/queues")
+        response = request(:get, '/api/v1/queues')
 
         return [] unless response.is_a?(Net::HTTPSuccess)
 
         data = JSON.parse(response.body, symbolize_names: true)
         data[:queues] || []
       rescue StandardError => e
-        log_error("list_queues", e)
+        log_error('list_queues', e)
         []
       end
 
@@ -112,7 +112,7 @@ module BrainzLab
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("get_queue", e)
+        log_error('get_queue', e)
         nil
       end
 
@@ -121,7 +121,7 @@ module BrainzLab
         response = request(:post, "/api/v1/jobs/#{job_id}/retry")
         response.is_a?(Net::HTTPSuccess)
       rescue StandardError => e
-        log_error("retry_job", e)
+        log_error('retry_job', e)
         false
       end
 
@@ -130,7 +130,7 @@ module BrainzLab
         response = request(:delete, "/api/v1/jobs/#{job_id}")
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPNoContent)
       rescue StandardError => e
-        log_error("delete_job", e)
+        log_error('delete_job', e)
         false
       end
 
@@ -138,7 +138,7 @@ module BrainzLab
       def report_metrics(queue:, size:, latency_ms: nil, workers: nil)
         response = request(
           :post,
-          "/api/v1/metrics",
+          '/api/v1/metrics',
           body: {
             queue: queue,
             size: size,
@@ -150,21 +150,21 @@ module BrainzLab
 
         response.is_a?(Net::HTTPSuccess)
       rescue StandardError => e
-        log_error("report_metrics", e)
+        log_error('report_metrics', e)
         false
       end
 
       def provision(project_id:, app_name:)
         response = request(
           :post,
-          "/api/v1/projects/provision",
+          '/api/v1/projects/provision',
           body: { project_id: project_id, app_name: app_name },
           use_service_key: true
         )
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("provision", e)
+        log_error('provision', e)
         false
       end
 
@@ -173,34 +173,32 @@ module BrainzLab
       def request(method, path, headers: {}, body: nil, params: nil, use_service_key: false)
         uri = URI.parse("#{@base_url}#{path}")
 
-        if params
-          uri.query = URI.encode_www_form(params)
-        end
+        uri.query = URI.encode_www_form(params) if params
 
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 5
         http.read_timeout = 10
 
         request = case method
-        when :get
-          Net::HTTP::Get.new(uri)
-        when :post
-          Net::HTTP::Post.new(uri)
-        when :put
-          Net::HTTP::Put.new(uri)
-        when :delete
-          Net::HTTP::Delete.new(uri)
-        end
-
-        request["Content-Type"] = "application/json"
-        request["Accept"] = "application/json"
+                  when :get
+                    Net::HTTP::Get.new(uri)
+                  when :post
+                    Net::HTTP::Post.new(uri)
+                  when :put
+                    Net::HTTP::Put.new(uri)
+                  when :delete
+                    Net::HTTP::Delete.new(uri)
+                  end
+
+        request['Content-Type'] = 'application/json'
+        request['Accept'] = 'application/json'
 
         if use_service_key
-          request["X-Service-Key"] = @config.nerve_master_key || @config.secret_key
+          request['X-Service-Key'] = @config.nerve_master_key || @config.secret_key
         else
           auth_key = @config.nerve_api_key || @config.secret_key
-          request["Authorization"] = "Bearer #{auth_key}" if auth_key
+          request['Authorization'] = "Bearer #{auth_key}" if auth_key
         end
 
         headers.each { |k, v| request[k] = v }

data/lib/brainzlab/nerve/provisioner.rb

@@ -37,7 +37,7 @@ module BrainzLab
       end
 
       def detect_project_id
-        ENV["BRAINZLAB_PROJECT_ID"]
+        ENV.fetch('BRAINZLAB_PROJECT_ID', nil)
       end
     end
   end

data/lib/brainzlab/nerve.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "nerve/client"
-require_relative "nerve/provisioner"
+require_relative 'nerve/client'
+require_relative 'nerve/provisioner'
 
 module BrainzLab
   module Nerve
@@ -32,7 +32,7 @@ module BrainzLab
           job_class: job_class,
           job_id: job_id,
           queue: queue,
-          status: "completed",
+          status: 'completed',
           started_at: started_at,
           ended_at: ended_at,
           **attributes
@@ -69,7 +69,7 @@ module BrainzLab
           job_class: job_class,
           job_id: job_id,
           queue: queue,
-          status: "running",
+          status: 'running',
           started_at: Time.now,
           ended_at: Time.now,
           **attributes
@@ -80,7 +80,7 @@ module BrainzLab
       # @param queue [String] Filter by queue (optional)
       # @param job_class [String] Filter by job class (optional)
       # @param period [String] Time period: "1h", "24h", "7d", "30d"
-      def stats(queue: nil, job_class: nil, period: "1h")
+      def stats(queue: nil, job_class: nil, period: '1h')
         return nil unless enabled?
 
         ensure_provisioned!
@@ -160,7 +160,7 @@ module BrainzLab
       #     # job work
       #   end
       #
-      def track(job_class:, job_id:, queue: "default", **attributes)
+      def track(job_class:, job_id:, queue: 'default', **attributes)
         started_at = Time.now
 
         begin

data/lib/brainzlab/pulse/client.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "uri"
-require "json"
+require 'net/http'
+require 'uri'
+require 'json'
 
 module BrainzLab
   module Pulse
@@ -23,7 +23,7 @@ module BrainzLab
         if @config.pulse_buffer_size > 1
           buffer_trace(payload)
         else
-          post("/api/v1/traces", payload)
+          post('/api/v1/traces', payload)
         end
       end
 
@@ -31,13 +31,13 @@ module BrainzLab
         return unless @config.pulse_enabled && @config.pulse_valid?
         return if payloads.empty?
 
-        post("/api/v1/traces/batch", { traces: payloads })
+        post('/api/v1/traces/batch', { traces: payloads })
       end
 
       def send_metric(payload)
         return unless @config.pulse_enabled && @config.pulse_valid?
 
-        post("/api/v1/metrics", payload)
+        post('/api/v1/metrics', payload)
       end
 
       def flush
@@ -79,9 +79,9 @@ module BrainzLab
       def post(path, body)
         uri = URI.join(@config.pulse_url, path)
         request = Net::HTTP::Post.new(uri)
-        request["Content-Type"] = "application/json"
-        request["Authorization"] = "Bearer #{@config.pulse_auth_key}"
-        request["User-Agent"] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
+        request['Content-Type'] = 'application/json'
+        request['Authorization'] = "Bearer #{@config.pulse_auth_key}"
+        request['User-Agent'] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
         request.body = JSON.generate(body)
 
         execute_with_retry(uri, request)
@@ -94,7 +94,7 @@ module BrainzLab
         retries = 0
         begin
           http = Net::HTTP.new(uri.host, uri.port)
-          http.use_ssl = uri.scheme == "https"
+          http.use_ssl = uri.scheme == 'https'
           http.open_timeout = 5
           http.read_timeout = 10
 
@@ -102,7 +102,11 @@ module BrainzLab
 
           case response.code.to_i
           when 200..299
-            JSON.parse(response.body) rescue {}
+            begin
+              JSON.parse(response.body)
+            rescue StandardError
+              {}
+            end
           when 429, 500..599
             raise RetryableError, "Server error: #{response.code}"
           else