smplkit 1.0.5

data/lib/smplkit/logging/sources.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module Smplkit
+  module Logging
+    # Describes a logger to register via +Smplkit::ManagementClient#loggers#register+.
+    #
+    # Used both for buffered runtime discovery (called by +Smplkit::Client+ as
+    # adapters discover loggers) and for explicit registration from setup
+    # scripts that already know the +(service, environment)+ they belong to.
+    #
+    # Args:
+    #   - name: Logger name (e.g. +"sqlalchemy.engine"+). Normalized to
+    #     lowercase with slashes and colons replaced by dots before sending to
+    #     the API.
+    #   - resolved_level: Effective log level for this source.
+    #   - level: Explicit (configured) log level, if different from
+    #     +resolved_level+. Pass +nil+ when the level is inherited.
+    #   - service: Service name this source belongs to (optional).
+    #   - environment: Environment name this source belongs to (optional).
+    class LoggerSource
+      attr_reader :name, :resolved_level, :level, :service, :environment
+
+      def initialize(name:, resolved_level:, level: nil, service: nil, environment: nil)
+        @name = name
+        @resolved_level = resolved_level
+        @level = level
+        @service = service
+        @environment = environment
+        freeze
+      end
+
+      def ==(other)
+        other.is_a?(LoggerSource) && name == other.name && resolved_level == other.resolved_level &&
+          level == other.level && service == other.service && environment == other.environment
+      end
+      alias eql? ==
+
+      def hash = [name, resolved_level, level, service, environment].hash
+    end
+  end
+
+  # Top-level re-export.
+  LoggerSource = Logging::LoggerSource
+end

data/lib/smplkit/management/buffer.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+module Smplkit
+  module Management
+    CONTEXT_REGISTRATION_LRU_SIZE = 10_000
+    CONTEXT_BATCH_FLUSH_SIZE = 100
+    FLAG_BATCH_FLUSH_SIZE = 50
+    LOGGER_BATCH_FLUSH_SIZE = 50
+
+    # Thread-safe batch buffer for context registration.
+    class ContextRegistrationBuffer
+      def initialize
+        @seen = {}
+        @pending = []
+        @lock = Mutex.new
+      end
+
+      def observe(contexts)
+        @lock.synchronize do
+          contexts.each do |ctx|
+            cache_key = [ctx.type, ctx.key]
+            next if @seen.key?(cache_key)
+
+            @seen.shift if @seen.size >= CONTEXT_REGISTRATION_LRU_SIZE
+            @seen[cache_key] = ctx.attributes
+            @pending << { "type" => ctx.type, "key" => ctx.key, "attributes" => ctx.attributes.dup }
+          end
+        end
+      end
+
+      def drain
+        @lock.synchronize do
+          batch = @pending
+          @pending = []
+          batch
+        end
+      end
+
+      def pending_count
+        @lock.synchronize { @pending.length }
+      end
+    end
+
+    # Thread-safe batch buffer for flag declarations.
+    class FlagRegistrationBuffer
+      def initialize
+        @seen = {}
+        @pending = []
+        @lock = Mutex.new
+      end
+
+      def add(declaration)
+        @lock.synchronize do
+          next if @seen.key?(declaration.id)
+
+          @seen[declaration.id] = true
+          item = { "id" => declaration.id, "type" => declaration.type, "default" => declaration.default }
+          item["service"] = declaration.service if declaration.service
+          item["environment"] = declaration.environment if declaration.environment
+          @pending << item
+        end
+      end
+
+      def drain
+        @lock.synchronize do
+          batch = @pending
+          @pending = []
+          batch
+        end
+      end
+
+      def pending_count
+        @lock.synchronize { @pending.length }
+      end
+    end
+
+    # Thread-safe batch buffer for logger discovery.
+    class LoggerRegistrationBuffer
+      def initialize
+        @seen = {}
+        @pending = []
+        @lock = Mutex.new
+      end
+
+      def add(source)
+        @lock.synchronize do
+          next if @seen.key?(source.name)
+
+          @seen[source.name] = source.resolved_level
+          item = { "id" => source.name, "resolved_level" => source.resolved_level&.to_s }
+          item["level"] = source.level&.to_s if source.level
+          item["service"] = source.service if source.service
+          item["environment"] = source.environment if source.environment
+          @pending << item
+        end
+      end
+
+      def drain
+        @lock.synchronize do
+          batch = @pending
+          @pending = []
+          batch
+        end
+      end
+
+      def pending_count
+        @lock.synchronize { @pending.length }
+      end
+    end
+  end
+end