brainzlab 0.1.1 → 0.1.3

data/lib/brainzlab/nerve.rb

@@ -0,0 +1,219 @@
+# frozen_string_literal: true
+
+require_relative 'nerve/client'
+require_relative 'nerve/provisioner'
+
+module BrainzLab
+  module Nerve
+    class << self
+      # Report a completed job
+      # @param job_class [String] Job class name
+      # @param job_id [String] Job ID
+      # @param queue [String] Queue name
+      # @param started_at [Time] When job started
+      # @param ended_at [Time] When job ended (defaults to now)
+      # @param attributes [Hash] Additional attributes
+      #
+      # @example
+      #   BrainzLab::Nerve.report_success(
+      #     job_class: "ProcessOrderJob",
+      #     job_id: "abc-123",
+      #     queue: "default",
+      #     started_at: 1.minute.ago
+      #   )
+      #
+      def report_success(job_class:, job_id:, queue:, started_at:, ended_at: Time.now, **attributes)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.report_job(
+          job_class: job_class,
+          job_id: job_id,
+          queue: queue,
+          status: 'completed',
+          started_at: started_at,
+          ended_at: ended_at,
+          **attributes
+        )
+      end
+
+      # Report a failed job
+      def report_failure(job_class:, job_id:, queue:, error:, started_at: nil, **attributes)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.report_failure(
+          job_class: job_class,
+          job_id: job_id,
+          queue: queue,
+          error_class: error.class.name,
+          error_message: error.message,
+          backtrace: error.backtrace,
+          started_at: started_at,
+          **attributes
+        )
+      end
+
+      # Report a job that's currently running
+      def report_started(job_class:, job_id:, queue:, **attributes)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.report_job(
+          job_class: job_class,
+          job_id: job_id,
+          queue: queue,
+          status: 'running',
+          started_at: Time.now,
+          ended_at: Time.now,
+          **attributes
+        )
+      end
+
+      # Get job statistics
+      # @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')
+        return nil unless enabled?
+
+        ensure_provisioned!
+        return nil unless BrainzLab.configuration.nerve_valid?
+
+        client.stats(queue: queue, job_class: job_class, period: period)
+      end
+
+      # List recent jobs
+      def jobs(queue: nil, status: nil, limit: 100)
+        return [] unless enabled?
+
+        ensure_provisioned!
+        return [] unless BrainzLab.configuration.nerve_valid?
+
+        client.list_jobs(queue: queue, status: status, limit: limit)
+      end
+
+      # List all queues
+      def queues
+        return [] unless enabled?
+
+        ensure_provisioned!
+        return [] unless BrainzLab.configuration.nerve_valid?
+
+        client.list_queues
+      end
+
+      # Get queue details
+      def queue(name)
+        return nil unless enabled?
+
+        ensure_provisioned!
+        return nil unless BrainzLab.configuration.nerve_valid?
+
+        client.get_queue(name)
+      end
+
+      # Retry a failed job
+      def retry(job_id)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.retry_job(job_id)
+      end
+
+      # Delete a job
+      def delete(job_id)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.delete_job(job_id)
+      end
+
+      # Report queue metrics (for custom job backends)
+      def report_metrics(queue:, size:, latency_ms: nil, workers: nil)
+        return false unless enabled?
+
+        ensure_provisioned!
+        return false unless BrainzLab.configuration.nerve_valid?
+
+        client.report_metrics(
+          queue: queue,
+          size: size,
+          latency_ms: latency_ms,
+          workers: workers
+        )
+      end
+
+      # Track a job execution (block helper)
+      # @example
+      #   BrainzLab::Nerve.track(job_class: "MyJob", job_id: "123", queue: "default") do
+      #     # job work
+      #   end
+      #
+      def track(job_class:, job_id:, queue: 'default', **attributes)
+        started_at = Time.now
+
+        begin
+          result = yield
+          report_success(
+            job_class: job_class,
+            job_id: job_id,
+            queue: queue,
+            started_at: started_at,
+            **attributes
+          )
+          result
+        rescue StandardError => e
+          report_failure(
+            job_class: job_class,
+            job_id: job_id,
+            queue: queue,
+            error: e,
+            started_at: started_at,
+            **attributes
+          )
+          raise
+        end
+      end
+
+      # === INTERNAL ===
+
+      def ensure_provisioned!
+        return if @provisioned
+
+        @provisioned = true
+        provisioner.ensure_project!
+      end
+
+      def provisioner
+        @provisioner ||= Provisioner.new(BrainzLab.configuration)
+      end
+
+      def client
+        @client ||= Client.new(BrainzLab.configuration)
+      end
+
+      def reset!
+        @client = nil
+        @provisioner = nil
+        @provisioned = false
+      end
+
+      private
+
+      def enabled?
+        BrainzLab.configuration.nerve_enabled
+      end
+    end
+  end
+end

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

data/lib/brainzlab/pulse/instrumentation.rb

@@ -22,20 +22,23 @@ module BrainzLab
         def install_active_record!
           return unless defined?(ActiveRecord)
 
-          ActiveSupport::Notifications.subscribe("sql.active_record") do |*args|
+          ActiveSupport::Notifications.subscribe('sql.active_record') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
             next if skip_query?(event.payload)
 
+            sql = event.payload[:sql]
             record_span(
-              name: event.payload[:name] || "SQL",
-              kind: "db",
+              name: event.payload[:name] || 'SQL',
+              kind: 'db',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
               data: {
-                sql: truncate_sql(event.payload[:sql]),
+                sql: truncate_sql(sql),
                 name: event.payload[:name],
-                cached: event.payload[:cached] || false
+                cached: event.payload[:cached] || false,
+                table: extract_table(sql),
+                operation: extract_operation(sql)
               }
             )
           end
@@ -45,12 +48,12 @@ module BrainzLab
         def install_action_view!
           return unless defined?(ActionView)
 
-          ActiveSupport::Notifications.subscribe("render_template.action_view") do |*args|
+          ActiveSupport::Notifications.subscribe('render_template.action_view') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: short_path(event.payload[:identifier]),
-              kind: "render",
+              kind: 'render',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -61,12 +64,12 @@ module BrainzLab
             )
           end
 
-          ActiveSupport::Notifications.subscribe("render_partial.action_view") do |*args|
+          ActiveSupport::Notifications.subscribe('render_partial.action_view') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: short_path(event.payload[:identifier]),
-              kind: "render",
+              kind: 'render',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -77,12 +80,12 @@ module BrainzLab
             )
           end
 
-          ActiveSupport::Notifications.subscribe("render_collection.action_view") do |*args|
+          ActiveSupport::Notifications.subscribe('render_collection.action_view') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: short_path(event.payload[:identifier]),
-              kind: "render",
+              kind: 'render',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -100,11 +103,11 @@ module BrainzLab
           %w[cache_read.active_support cache_write.active_support cache_delete.active_support].each do |event_name|
             ActiveSupport::Notifications.subscribe(event_name) do |*args|
               event = ActiveSupport::Notifications::Event.new(*args)
-              operation = event_name.split(".").first.sub("cache_", "")
+              operation = event_name.split('.').first.sub('cache_', '')
 
               record_span(
                 name: "Cache #{operation}",
-                kind: "cache",
+                kind: 'cache',
                 started_at: event.time,
                 ended_at: event.end,
                 duration_ms: event.duration,
@@ -122,7 +125,7 @@ module BrainzLab
         def install_action_controller!
           return unless defined?(ActionController)
 
-          ActiveSupport::Notifications.subscribe("process_action.action_controller") do |*args|
+          ActiveSupport::Notifications.subscribe('process_action.action_controller') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
             payload = event.payload
 
@@ -138,12 +141,12 @@ module BrainzLab
         def install_http_clients!
           # Net::HTTP instrumentation
           if defined?(Net::HTTP)
-            ActiveSupport::Notifications.subscribe("request.net_http") do |*args|
+            ActiveSupport::Notifications.subscribe('request.net_http') do |*args|
               event = ActiveSupport::Notifications::Event.new(*args)
 
               record_span(
                 name: "HTTP #{event.payload[:method]} #{event.payload[:host]}",
-                kind: "http",
+                kind: 'http',
                 started_at: event.time,
                 ended_at: event.end,
                 duration_ms: event.duration,
@@ -158,26 +161,26 @@ module BrainzLab
           end
 
           # Faraday instrumentation
-          if defined?(Faraday)
-            ActiveSupport::Notifications.subscribe("request.faraday") do |*args|
-              event = ActiveSupport::Notifications::Event.new(*args)
-              env = event.payload[:env]
-              next unless env
+          return unless defined?(Faraday)
 
-              record_span(
-                name: "HTTP #{env.method.to_s.upcase} #{env.url.host}",
-                kind: "http",
-                started_at: event.time,
-                ended_at: event.end,
-                duration_ms: event.duration,
-                data: {
-                  method: env.method.to_s.upcase,
-                  host: env.url.host,
-                  path: env.url.path,
-                  status: env.status
-                }
-              )
-            end
+          ActiveSupport::Notifications.subscribe('request.faraday') do |*args|
+            event = ActiveSupport::Notifications::Event.new(*args)
+            env = event.payload[:env]
+            next unless env
+
+            record_span(
+              name: "HTTP #{env.method.to_s.upcase} #{env.url.host}",
+              kind: 'http',
+              started_at: event.time,
+              ended_at: event.end,
+              duration_ms: event.duration,
+              data: {
+                method: env.method.to_s.upcase,
+                host: env.url.host,
+                path: env.url.path,
+                status: env.status
+              }
+            )
           end
         end
 
@@ -186,13 +189,13 @@ module BrainzLab
           return unless defined?(ActiveJob)
 
           # Track job enqueuing
-          ActiveSupport::Notifications.subscribe("enqueue.active_job") do |*args|
+          ActiveSupport::Notifications.subscribe('enqueue.active_job') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
             job = event.payload[:job]
 
             record_span(
               name: "Enqueue #{job.class.name}",
-              kind: "job",
+              kind: 'job',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -205,14 +208,14 @@ module BrainzLab
           end
 
           # Track job retry
-          ActiveSupport::Notifications.subscribe("retry_stopped.active_job") do |*args|
+          ActiveSupport::Notifications.subscribe('retry_stopped.active_job') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
             job = event.payload[:job]
             error = event.payload[:error]
 
             record_span(
               name: "Retry stopped #{job.class.name}",
-              kind: "job",
+              kind: 'job',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -229,14 +232,14 @@ module BrainzLab
           end
 
           # Track job discard
-          ActiveSupport::Notifications.subscribe("discard.active_job") do |*args|
+          ActiveSupport::Notifications.subscribe('discard.active_job') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
             job = event.payload[:job]
             error = event.payload[:error]
 
             record_span(
               name: "Discarded #{job.class.name}",
-              kind: "job",
+              kind: 'job',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -257,12 +260,12 @@ module BrainzLab
         def install_action_cable!
           return unless defined?(ActionCable)
 
-          ActiveSupport::Notifications.subscribe("perform_action.action_cable") do |*args|
+          ActiveSupport::Notifications.subscribe('perform_action.action_cable') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: "Cable #{event.payload[:channel_class]}##{event.payload[:action]}",
-              kind: "cable",
+              kind: 'cable',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -273,12 +276,12 @@ module BrainzLab
             )
           end
 
-          ActiveSupport::Notifications.subscribe("transmit.action_cable") do |*args|
+          ActiveSupport::Notifications.subscribe('transmit.action_cable') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: "Cable transmit #{event.payload[:channel_class]}",
-              kind: "cable",
+              kind: 'cable',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -289,12 +292,12 @@ module BrainzLab
             )
           end
 
-          ActiveSupport::Notifications.subscribe("broadcast.action_cable") do |*args|
+          ActiveSupport::Notifications.subscribe('broadcast.action_cable') do |*args|
             event = ActiveSupport::Notifications::Event.new(*args)
 
             record_span(
               name: "Cable broadcast #{event.payload[:broadcasting]}",
-              kind: "cable",
+              kind: 'cable',
               started_at: event.time,
               ended_at: event.end,
               duration_ms: event.duration,
@@ -306,7 +309,8 @@ module BrainzLab
           end
         end
 
-        def record_span(name:, kind:, started_at:, ended_at:, duration_ms:, error: false, error_class: nil, error_message: nil, data: {})
+        def record_span(name:, kind:, started_at:, ended_at:, duration_ms:, error: false, error_class: nil,
+                        error_message: nil, data: {})
           spans = Thread.current[:brainzlab_pulse_spans]
           return unless spans
 
@@ -331,10 +335,10 @@ module BrainzLab
 
         def skip_query?(payload)
           # Skip SCHEMA queries and internal Rails queries
-          return true if payload[:name] == "SCHEMA"
-          return true if payload[:name]&.start_with?("EXPLAIN")
-          return true if payload[:sql]&.include?("pg_")
-          return true if payload[:sql]&.include?("information_schema")
+          return true if payload[:name] == 'SCHEMA'
+          return true if payload[:name]&.start_with?('EXPLAIN')
+          return true if payload[:sql]&.include?('pg_')
+          return true if payload[:sql]&.include?('information_schema')
           return true if payload[:cached] && !include_cached_queries?
 
           false
@@ -346,17 +350,50 @@ module BrainzLab
 
         def truncate_sql(sql)
           return nil unless sql
+
           sql.to_s[0, 1000]
         end
 
         def truncate_key(key)
           return nil unless key
+
           key.to_s[0, 200]
         end
 
         def short_path(path)
           return nil unless path
-          path.to_s.split("/").last(2).join("/")
+
+          path.to_s.split('/').last(2).join('/')
+        end
+
+        def extract_table(sql)
+          return nil unless sql
+
+          # Match FROM "table" or FROM table patterns
+          # Also handles INSERT INTO, UPDATE, DELETE FROM
+          case sql.to_s
+          when /\bFROM\s+["'`]?(\w+)["'`]?/i
+            Regexp.last_match(1)
+          when /\bINTO\s+["'`]?(\w+)["'`]?/i
+            Regexp.last_match(1)
+          when /\bUPDATE\s+["'`]?(\w+)["'`]?/i
+            Regexp.last_match(1)
+          when /\bJOIN\s+["'`]?(\w+)["'`]?/i
+            Regexp.last_match(1)
+          end
+        end
+
+        def extract_operation(sql)
+          return nil unless sql
+
+          case sql.to_s.strip.upcase
+          when /\ASELECT/i then 'SELECT'
+          when /\AINSERT/i then 'INSERT'
+          when /\AUPDATE/i then 'UPDATE'
+          when /\ADELETE/i then 'DELETE'
+          when /\ABEGIN/i, /\ACOMMIT/i, /\AROLLBACK/i then 'TRANSACTION'
+          else 'QUERY'
+          end
         end
       end
     end