karafka 2.3.1 → 2.3.3

data/lib/karafka/pro/routing/features/dead_letter_queue/contracts/topic.rb

@@ -28,6 +28,12 @@ module Karafka
                 ).fetch('en').fetch('validations').fetch('topic')
               end
 
+              nested(:dead_letter_queue) do
+                # We use strategy based DLQ for every case in Pro
+                # For default (when no strategy) a default `max_retries` based strategy is used
+                required(:strategy) { |val| val.respond_to?(:call) }
+              end
+
               # Make sure that when we use virtual partitions with DLQ, at least one retry is set
               # We cannot use VP with DLQ without retries as we in order to provide ordering
               # warranties on errors with VP, we need to collapse the VPs concurrency and retry

data/lib/karafka/pro/routing/features/dead_letter_queue/topic.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    module Routing
+      module Features
+        class DeadLetterQueue < Base
+          # Expansions to the topic API in DLQ
+          module Topic
+            # @param strategy [#call, nil] Strategy we want to use or nil if a default strategy
+            # (same as in OSS) should be applied
+            # @param args [Hash] OSS DLQ arguments
+            def dead_letter_queue(strategy: nil, **args)
+              return @dead_letter_queue if @dead_letter_queue
+
+              super(**args).tap do |config|
+                # If explicit strategy is not provided, use the default approach from OSS
+                config.strategy = strategy || lambda do |_errors_tracker, attempt|
+                  attempt > config.max_retries ? :dispatch : :retry
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/karafka/pro/routing/features/swarm/config.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    module Routing
+      module Features
+        class Swarm < Base
+          # Swarm feature configuration
+          Config = Struct.new(
+            :active,
+            :nodes,
+            keyword_init: true
+          ) do
+            alias_method :active?, :active
+          end
+        end
+      end
+    end
+  end
+end

data/lib/karafka/pro/routing/features/swarm/contracts/topic.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    module Routing
+      module Features
+        class Swarm < Base
+          # Namespace for swarm contracts
+          module Contracts
+            # Contract to validate configuration of the swarm feature
+            class Topic < Karafka::Contracts::Base
+              configure do |config|
+                config.error_messages = YAML.safe_load(
+                  File.read(
+                    File.join(Karafka.gem_root, 'config', 'locales', 'pro_errors.yml')
+                  )
+                ).fetch('en').fetch('validations').fetch('topic')
+              end
+
+              nested(:swarm) do
+                required(:active) { |val| val == true }
+
+                required(:nodes) do |val|
+                  val.is_a?(Range) || (
+                    val.is_a?(Array) &&
+                    val.all? { |id| id.is_a?(Integer) }
+                  )
+                end
+              end
+
+              # Make sure that if range is defined it fits number of nodes (except infinity)
+              # As it may be a common error to accidentally define a node that will never be
+              # reached
+              virtual do |data, errors|
+                next unless errors.empty?
+
+                nodes = data[:swarm][:nodes]
+
+                # Defaults
+                next if nodes.first.zero? && nodes.last == Float::INFINITY
+
+                # If our expectation towards which node should run things matches at least one
+                # node, then it's ok
+                next if Karafka::App.config.swarm.nodes.times.any? do |node_id|
+                  nodes.include?(node_id)
+                end
+
+                [[%i[swarm_nodes], :with_non_existent_nodes]]
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/karafka/pro/routing/features/swarm/topic.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    module Routing
+      module Features
+        class Swarm < Base
+          # Topic swarm API extensions
+          module Topic
+            # Allows defining swarm routing topic settings
+            # @param nodes [Range, Array] range of nodes ids or array with nodes ids for which we
+            #   should run given topic
+            def swarm(nodes: (0...Karafka::App.config.swarm.nodes))
+              @swarm ||= Config.new(active: true, nodes: nodes)
+            end
+
+            # @return [true] swarm setup is always true. May not be in use but is active
+            def swarm?
+              swarm.active?
+            end
+
+            # @return [Boolean] should this topic be active. In the context of swarm it is only
+            #   active when swarm routing setup does not limit nodes on which it should operate
+            def active?
+              node = Karafka::App.config.swarm.node
+
+              return super unless node
+
+              super && swarm.nodes.include?(node.id)
+            end
+
+            # @return [Hash] topic with all its native configuration options plus swarm
+            def to_h
+              super.merge(
+                swarm: swarm.to_h
+              ).freeze
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/karafka/pro/routing/features/swarm.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    module Routing
+      module Features
+        # Karafka Pro Swarm extensions to the routing
+        # They allow for more granular work assignment in the swarm
+        class Swarm < Base
+        end
+      end
+    end
+  end
+end

data/lib/karafka/pro/swarm/liveness_listener.rb

@@ -0,0 +1,171 @@
+# frozen_string_literal: true
+
+# This Karafka component is a Pro component under a commercial license.
+# This Karafka component is NOT licensed under LGPL.
+#
+# All of the commercial components are present in the lib/karafka/pro directory of this
+# repository and their usage requires commercial license agreement.
+#
+# Karafka has also commercial-friendly license, commercial support and commercial components.
+#
+# By sending a pull request to the pro components, you are agreeing to transfer the copyright of
+# your code to Maciej Mensfeld.
+
+module Karafka
+  module Pro
+    # Pro Swarm components namespace
+    module Swarm
+      # Pro listener that monitors RSS usage and other heartbeat metrics (if configured) to ensure
+      # that everything operates.
+      #
+      # It can:
+      #   - monitor poll frequency to make sure things are not polled not often enough
+      #   - monitor consumption to make sure we do not process data for too long
+      #   - monitor RSS to make sure that we do not use too much memory
+      #
+      # By default it does **not** monitor memory and consuming and polling is configured in such
+      # a way to align with `max.poll.interval.ms` and other defaults.
+      #
+      # Failure statuses reported are as follows:
+      #   - 1 - polling ttl exceeded
+      #   - 2 - consuming ttl exceeded
+      #   - 3 - memory limit exceeded
+      #
+      # @note This listener should not break anything if subscribed in the supervisor prior to
+      #   forking as it relies on server events for operations.
+      class LivenessListener < Karafka::Swarm::LivenessListener
+        # @param memory_limit [Integer] max memory in MB for this process to be considered healthy
+        # @param consuming_ttl [Integer] time in ms after which we consider consumption hanging.
+        #   It allows us to define max consumption time after which supervisor should consider
+        #   given process as hanging
+        # @param polling_ttl [Integer] max time in ms for polling. If polling (any) does not
+        #   happen that often, process should be considered dead.
+        # @note The default TTL matches the default `max.poll.interval.ms`
+        def initialize(
+          memory_limit: Float::INFINITY,
+          consuming_ttl: 5 * 60 * 1_000,
+          polling_ttl: 5 * 60 * 1_000
+        )
+          @polling_ttl = polling_ttl
+          @consuming_ttl = consuming_ttl
+          # We cast it just in case someone would provide '10MB' or something similar
+          @memory_limit = memory_limit.is_a?(String) ? memory_limit.to_i : memory_limit
+          @pollings = {}
+          @consumptions = {}
+
+          super()
+        end
+
+        # Tick on each fetch
+        #
+        # @param _event [Karafka::Core::Monitoring::Event]
+        def on_connection_listener_fetch_loop(_event)
+          mark_polling_tick
+        end
+
+        {
+          consume: :consumed,
+          revoke: :revoked,
+          shutting_down: :shutdown,
+          tick: :ticked
+        }.each do |before, after|
+          class_eval <<~RUBY, __FILE__, __LINE__ + 1
+            # Tick on starting work
+            # @param _event [Karafka::Core::Monitoring::Event]
+            def on_consumer_#{before}(_event)
+              mark_consumption_tick
+            end
+
+            # Tick on finished work
+            # @param _event [Karafka::Core::Monitoring::Event]
+            def on_consumer_#{after}(_event)
+              clear_consumption_tick
+            end
+          RUBY
+        end
+
+        # @param _event [Karafka::Core::Monitoring::Event]
+        def on_error_occurred(_event)
+          clear_consumption_tick
+          clear_polling_tick
+        end
+
+        # Reports the current status once in a while
+        #
+        # @param _event [Karafka::Core::Monitoring::Event]
+        def on_statistics_emitted(_event)
+          periodically do
+            return unless node
+
+            current_status = status
+
+            current_status.positive? ? node.unhealthy(current_status) : node.healthy
+          end
+        end
+
+        private
+
+        # @return [Integer] object id of the current thread
+        def thread_id
+          Thread.current.object_id
+        end
+
+        # Update the polling tick time for current thread
+        def mark_polling_tick
+          synchronize do
+            @pollings[thread_id] = monotonic_now
+          end
+        end
+
+        # Clear current thread polling time tracker
+        def clear_polling_tick
+          synchronize do
+            @pollings.delete(thread_id)
+          end
+        end
+
+        # Update the processing tick time
+        def mark_consumption_tick
+          synchronize do
+            @consumptions[thread_id] = monotonic_now
+          end
+        end
+
+        # Clear current thread consumption time tracker
+        def clear_consumption_tick
+          synchronize do
+            @consumptions.delete(thread_id)
+          end
+        end
+
+        # Did we exceed any of the ttls
+        # @return [String] 204 string if ok, 500 otherwise
+        def status
+          time = monotonic_now
+
+          return 1 if @pollings.values.any? { |tick| (time - tick) > @polling_ttl }
+          return 2 if @consumptions.values.any? { |tick| (time - tick) > @consuming_ttl }
+          return 3 if rss_mb > @memory_limit
+
+          0
+        end
+
+        # @return [Integer] RSS in MB for the current process
+        # @note Since swarm is linux only, we do not have to worry about getting RSS on other OSes
+        def rss_mb
+          kb_rss = 0
+
+          IO.readlines("/proc/#{node.pid}/status").each do |line|
+            next unless line.start_with?('VmRSS:')
+
+            kb_rss = line.split[1].to_i
+
+            break
+          end
+
+          (kb_rss / 1_024.to_i).round
+        end
+      end
+    end
+  end
+end

data/lib/karafka/process.rb

@@ -14,6 +14,8 @@ module Karafka
       SIGTERM
       SIGTTIN
       SIGTSTP
+      SIGCHLD
+      SIGUSER1
     ].freeze
 
     HANDLED_SIGNALS.each do |signal|
@@ -32,16 +34,40 @@ module Karafka
       RUBY
     end
 
+    # Assigns a callback that will run on any supported signal that has at least one callback
+    # registered already.
+    # @param block [Proc] code we want to run
+    # @note This will only bind to signals that already have at least one callback defined
+    def on_any_active(&block)
+      HANDLED_SIGNALS.each do |signal|
+        next unless @callbacks.key?(signal)
+
+        public_send(:"on_#{signal.to_s.downcase}", &block)
+      end
+    end
+
     # Creates an instance of process and creates empty hash for callbacks
     def initialize
       @callbacks = Hash.new { |hsh, key| hsh[key] = [] }
       @supervised = false
     end
 
+    # Clears all the defined callbacks. Useful for post-fork cleanup when parent already defined
+    # some signals
+    def clear
+      @callbacks.clear
+    end
+
     # Method catches all HANDLED_SIGNALS and performs appropriate callbacks (if defined)
     # @note If there are no callbacks, this method will just ignore a given signal that was sent
     def supervise
-      HANDLED_SIGNALS.each { |signal| trap_signal(signal) }
+      HANDLED_SIGNALS.each do |signal|
+        # Supervise only signals for which we have defined callbacks
+        next unless @callbacks.key?(signal)
+
+        trap_signal(signal)
+      end
+
       @supervised = true
     end
 

data/lib/karafka/routing/features/dead_letter_queue/config.rb

@@ -15,6 +15,8 @@ module Karafka
           :independent,
           # Move to DLQ and mark as consumed in transactional mode (if applicable)
           :transactional,
+          # Strategy to apply (if strategies supported)
+          :strategy,
           keyword_init: true
         ) do
           alias_method :active?, :active

data/lib/karafka/routing/subscription_group.rb

@@ -8,6 +8,12 @@ module Karafka
     # @note One subscription group will always belong to one consumer group, but one consumer
     #   group can have multiple subscription groups.
     class SubscriptionGroup
+      include Helpers::ConfigImporter.new(
+        activity_manager: %i[internal routing activity_manager],
+        client_id: %i[client_id],
+        node: %i[swarm node]
+      )
+
       attr_reader :id, :name, :topics, :kafka, :consumer_group
 
       # Lock for generating new ids safely
@@ -67,7 +73,7 @@ module Karafka
 
       # @return [Boolean] is this subscription group one of active once
       def active?
-        Karafka::App.config.internal.routing.activity_manager.active?(:subscription_groups, name)
+        activity_manager.active?(:subscription_groups, name)
       end
 
       # @return [Array<String>] names of topics to which we should subscribe.
@@ -85,6 +91,19 @@ module Karafka
         id
       end
 
+      # Refreshes the configuration of this subscription group if needed based on the execution
+      # context.
+      #
+      # Since the initial routing setup happens in the supervisor, it is inherited by the children.
+      # This causes incomplete assignment of `group.instance.id` which is not expanded with proper
+      # node identifier. This refreshes this if needed when in swarm.
+      def refresh
+        return unless node
+        return unless kafka.key?(:'group.instance.id')
+
+        @kafka = build_kafka
+      end
+
       private
 
       # @return [Hash] kafka settings are a bit special. They are exactly the same for all of the
@@ -93,15 +112,9 @@ module Karafka
       def build_kafka
         kafka = Setup::AttributesMap.consumer(@topics.first.kafka.dup)
 
-        # If we use static group memberships, there can be a case, where same instance id would
-        # be set on many subscription groups as the group instance id from Karafka perspective is
-        # set per config. Each instance even if they are subscribed to different topics needs to
-        # have it fully unique. To make sure of that, we just add extra postfix at the end that
-        # increments.
-        group_instance_id = kafka.fetch(:'group.instance.id', false)
+        inject_group_instance_id(kafka)
 
-        kafka[:'group.instance.id'] = "#{group_instance_id}_#{@position}" if group_instance_id
-        kafka[:'client.id'] ||= Karafka::App.config.client_id
+        kafka[:'client.id'] ||= client_id
         kafka[:'group.id'] ||= @consumer_group.id
         kafka[:'auto.offset.reset'] ||= @topics.first.initial_offset
         # Karafka manages the offsets based on the processing state, thus we do not rely on the
@@ -110,6 +123,28 @@ module Karafka
         kafka.freeze
         kafka
       end
+
+      # If we use static group memberships, there can be a case, where same instance id would
+      # be set on many subscription groups as the group instance id from Karafka perspective is
+      # set per config. Each instance even if they are subscribed to different topics needs to
+      # have it fully unique. To make sure of that, we just add extra postfix at the end that
+      # increments.
+      #
+      # We also handle a swarm case, where the same setup would run from many forked nodes, hence
+      # affecting the instance id and causing conflicts
+      # @param kafka [Hash] kafka level config
+      def inject_group_instance_id(kafka)
+        group_instance_prefix = kafka.fetch(:'group.instance.id', false)
+
+        # If group instance id was not even configured, do nothing
+        return unless group_instance_prefix
+
+        # If there is a node, we need to take its id and inject it as well so multiple forks can
+        # have different instances ids but they are reproducible
+        components = [group_instance_prefix, node ? node.id : nil, @position]
+
+        kafka[:'group.instance.id'] = components.compact.join('_')
+      end
     end
   end
 end

data/lib/karafka/server.rb

@@ -3,16 +3,6 @@
 module Karafka
   # Karafka consuming server class
   class Server
-    # How long should we sleep between checks on shutting down consumers
-    SUPERVISION_SLEEP = 0.1
-    # What system exit code should we use when we terminated forcefully
-    FORCEFUL_EXIT_CODE = 2
-    # This factor allows us to calculate how many times we have to sleep before
-    # a forceful shutdown
-    SUPERVISION_CHECK_FACTOR = (1 / SUPERVISION_SLEEP)
-
-    private_constant :SUPERVISION_SLEEP, :FORCEFUL_EXIT_CODE, :SUPERVISION_CHECK_FACTOR
-
     class << self
       # Set of consuming threads. Each consumer thread contains a single consumer
       attr_accessor :listeners
@@ -36,12 +26,20 @@ module Karafka
           config.internal.routing.activity_manager.to_h
         )
 
+        # We clear as we do not want parent handlers in case of working from fork
+        process.clear
         process.on_sigint { stop }
         process.on_sigquit { stop }
         process.on_sigterm { stop }
         process.on_sigtstp { quiet }
+        # Needed for instrumentation
+        process.on_sigttin {}
         process.supervise
 
+        # This will only run when not in a swarm mode. In swarm mode the server runs post-fork, so
+        # warmup will do nothing
+        Karafka::App.warmup
+
         # Start is blocking until stop is called and when we stop, it will wait until
         # all of the things are ready to stop
         start
@@ -86,13 +84,13 @@ module Karafka
         # We check from time to time (for the timeout period) if all the threads finished
         # their work and if so, we can just return and normal shutdown process will take place
         # We divide it by 1000 because we use time in ms.
-        ((timeout / 1_000) * SUPERVISION_CHECK_FACTOR).to_i.times do
+        ((timeout / 1_000) * (1 / config.internal.supervision_sleep)).to_i.times do
           all_listeners_stopped = listeners.all?(&:stopped?)
           all_workers_stopped = workers.none?(&:alive?)
 
           return if all_listeners_stopped && all_workers_stopped
 
-          sleep SUPERVISION_SLEEP
+          sleep(config.internal.supervision_sleep)
         end
 
         raise Errors::ForcefulShutdownError
@@ -116,7 +114,7 @@ module Karafka
         return unless process.supervised?
 
         # exit! is not within the instrumentation as it would not trigger due to exit
-        Kernel.exit!(FORCEFUL_EXIT_CODE)
+        Kernel.exit!(config.internal.forceful_exit_code)
       ensure
         # We need to check if it wasn't an early exit to make sure that only on stop invocation
         # can change the status after everything is closed

data/lib/karafka/setup/config.rb

@@ -105,6 +105,17 @@ module Karafka
       # @see https://github.com/edenhill/librdkafka/blob/master/CONFIGURATION.md
       setting :kafka, default: {}
 
+      # Public configuration for swarm operations
+      setting :swarm do
+        # option [Integer] how many processes do we want to run in a swarm mode
+        # Keep in mind this is only applicable when running in a swarm mode
+        setting :nodes, default: 3
+        # This is set automatically when we fork. Used to hold reference that may be needed
+        # for static group membership, supervision and more. If set to `false`, it means this
+        # process is not a fork
+        setting :node, default: false
+      end
+
       # Admin specific settings.
       #
       # Since admin operations are often specific, they may require specific librdkafka settings
@@ -151,7 +162,6 @@ module Karafka
         # @note In the future, we need to have a single process representation for all the karafka
         #   instances
         setting :process, default: Process.new
-
         # Interval of "ticking". This is used to define the maximum time between consecutive
         # polling of the main rdkafka queue. It should match also the `statistics.interval.ms`
         # smallest value defined in any of the per-kafka settings, so metrics are published with
@@ -162,6 +172,36 @@ module Karafka
         # not to have enough time to run. This (not directly) defines also a single poll
         # max timeout as to allow for frequent enough events polling
         setting :tick_interval, default: 5_000
+        # How long should we sleep between checks on shutting down consumers
+        setting :supervision_sleep, default: 0.1
+        # What system exit code should we use when we terminated forcefully
+        setting :forceful_exit_code, default: 2
+
+        setting :swarm do
+          # Manager for swarm nodes control
+          setting :manager, default: Swarm::Manager.new
+          # Exit code we exit an orphaned child with to indicate something went wrong
+          setting :orphaned_exit_code, default: 3
+          # syscall number for https://man7.org/linux/man-pages/man2/pidfd_open.2.html
+          setting :pidfd_open_syscall, default: 434
+          # syscall number for https://man7.org/linux/man-pages/man2/pidfd_send_signal.2.html
+          setting :pidfd_signal_syscall, default: 424
+          # How often (in ms) should we control our nodes
+          # This is maximum time after which we will check. This can happen more often in case of
+          # system events.
+          setting :supervision_interval, default: 30_000
+          # How often should each node report its status
+          setting :liveness_interval, default: 10_000
+          # Listener used to report nodes state to the supervisor
+          setting :liveness_listener, default: Swarm::LivenessListener.new
+          # How long should we wait for any info from the node before we consider it hanging at
+          # stop it
+          setting :node_report_timeout, default: 60_000
+          # How long should we wait before restarting a node. This can prevent us from having a
+          # case where for some external reason our spawned process would die immediately and we
+          # would immediately try to start it back in an endless loop
+          setting :node_restart_timeout, default: 5_000
+        end
 
         # Namespace for CLI related settings
         setting :cli do
@@ -176,7 +216,6 @@ module Karafka
           # option subscription_groups_builder [Routing::SubscriptionGroupsBuilder] subscription
           #   group builder
           setting :subscription_groups_builder, default: Routing::SubscriptionGroupsBuilder.new
-
           # Internally assigned list of limits on routings active for the current process
           # This can be altered by the CLI command
           setting :activity_manager, default: Routing::ActivityManager.new