karafka 2.0.0.beta4 → 2.0.0.rc2

data/lib/karafka/cli/server.rb

@@ -5,6 +5,8 @@ module Karafka
   class Cli < Thor
     # Server Karafka Cli action
     class Server < Base
+      include Helpers::Colorize
+
       desc 'Start the Karafka server (short-cut alias: "s")'
       option aliases: 's'
       option :consumer_groups, type: :array, default: nil, aliases: :g
@@ -31,11 +33,11 @@ module Karafka
 
         if Karafka.pro?
           Karafka.logger.info(
-            "\033[0;32mThank you for investing in the Karafka Pro subscription!\033[0m\n"
+            green('Thank you for investing in the Karafka Pro subscription!')
           )
         else
           Karafka.logger.info(
-            "\033[0;31mYou like Karafka? Please consider getting a Pro subscription!\033[0m\n"
+            red('You like Karafka? Please consider getting a Pro version!')
           )
         end
       end

data/lib/karafka/connection/client.rb

@@ -36,6 +36,12 @@ module Karafka
         # Marks if we need to offset. If we did not store offsets, we should not commit the offset
         # position as it will crash rdkafka
         @offsetting = false
+        # We need to keep track of what we have paused for resuming
+        # In case we loose partition, we still need to resume it, otherwise it won't be fetched
+        # again if we get reassigned to it later on. We need to keep them as after revocation we
+        # no longer may be able to fetch them from Kafka. We could build them but it is easier
+        # to just keep them here and use if needed when cannot be obtained
+        @paused_tpls = Hash.new { |h, k| h[k] = {} }
       end
 
       # Fetches messages within boundaries defined by the settings (time, size, topics, etc).
@@ -45,12 +51,13 @@ module Karafka
       # @note This method should not be executed from many threads at the same time
       def batch_poll
         time_poll = TimeTrackers::Poll.new(@subscription_group.max_wait_time)
-        time_poll.start
 
         @buffer.clear
         @rebalance_manager.clear
 
         loop do
+          time_poll.start
+
           # Don't fetch more messages if we do not have any time left
           break if time_poll.exceeded?
           # Don't fetch more messages if we've fetched max as we've wanted
@@ -69,7 +76,11 @@ module Karafka
           # If partition revocation happens, we need to remove messages from revoked partitions
           # as well as ensure we do not have duplicated due to the offset reset for partitions
           # that we got assigned
-          remove_revoked_and_duplicated_messages if @rebalance_manager.revoked_partitions?
+          # We also do early break, so the information about rebalance is used as soon as possible
+          if @rebalance_manager.changed?
+            remove_revoked_and_duplicated_messages
+            break
+          end
 
           # Finally once we've (potentially) removed revoked, etc, if no messages were returned
           # we can break.
@@ -144,10 +155,14 @@ module Karafka
 
         internal_commit_offsets(async: false)
 
+        # Here we do not use our cached tpls because we should not try to pause something we do
+        # not own anymore.
         tpl = topic_partition_list(topic, partition)
 
         return unless tpl
 
+        @paused_tpls[topic][partition] = tpl
+
         @kafka.pause(tpl)
 
         @kafka.seek(pause_msg)
@@ -169,9 +184,13 @@ module Karafka
         # We can skip performance penalty since resuming should not happen too often
         internal_commit_offsets(async: false)
 
-        tpl = topic_partition_list(topic, partition)
+        # If we were not able, let's try to reuse the one we have (if we have)
+        tpl = topic_partition_list(topic, partition) || @paused_tpls[topic][partition]
 
         return unless tpl
+        # If we did not have it, it means we never paused this partition, thus no resume should
+        # happen in the first place
+        return unless @paused_tpls[topic].delete(partition)
 
         @kafka.resume(tpl)
       ensure
@@ -214,6 +233,7 @@ module Karafka
         @mutex.synchronize do
           @closed = false
           @offsetting = false
+          @paused_tpls.clear
           @kafka = build_consumer
         end
       end
@@ -296,37 +316,40 @@ module Karafka
 
         time_poll.start
 
-        @kafka.poll(time_poll.remaining)
+        @kafka.poll(timeout)
       rescue ::Rdkafka::RdkafkaError => e
-        raise if time_poll.attempts > MAX_POLL_RETRIES
-        raise unless time_poll.retryable?
-
+        # We return nil, so we do not restart until running the whole loop
+        # This allows us to run revocation jobs and other things and we will pick up new work
+        # next time after dispatching all the things that are needed
+        #
+        # If we would retry here, the client reset would become transparent and we would not have
+        # a chance to take any actions
         case e.code
         when :max_poll_exceeded # -147
           reset
+          return nil
         when :transport # -195
           reset
+          return nil
         when :rebalance_in_progress # -27
           reset
+          return nil
         when :not_coordinator # 16
           reset
+          return nil
         when :network_exception # 13
           reset
+          return nil
         end
 
-        time_poll.checkpoint
-
+        raise if time_poll.attempts > MAX_POLL_RETRIES
         raise unless time_poll.retryable?
 
+        time_poll.checkpoint
         time_poll.backoff
 
-        # We return nil, so we do not restart until running the whole loop
-        # This allows us to run revocation jobs and other things and we will pick up new work
-        # next time after dispatching all the things that are needed
-        #
-        # If we would retry here, the client reset would become transparent and we would not have
-        # a chance to take any actions
-        nil
+        # On unknown errors we do our best to retry and handle them before raising
+        retry
       end
 
       # Builds a new rdkafka consumer instance based on the subscription group configuration
@@ -369,7 +392,7 @@ module Karafka
       # we are no longer responsible in a given process for processing those messages and they
       # should have been picked up by a different process.
       def remove_revoked_and_duplicated_messages
-        @rebalance_manager.revoked_partitions.each do |topic, partitions|
+        @rebalance_manager.lost_partitions.each do |topic, partitions|
           partitions.each do |partition|
             @buffer.delete(topic, partition)
           end

data/lib/karafka/connection/listener.rb

@@ -18,15 +18,18 @@ module Karafka
       # @param jobs_queue [Karafka::Processing::JobsQueue] queue where we should push work
       # @return [Karafka::Connection::Listener] listener instance
       def initialize(subscription_group, jobs_queue)
+        proc_config = ::Karafka::App.config.internal.processing
+
         @id = SecureRandom.uuid
         @subscription_group = subscription_group
         @jobs_queue = jobs_queue
-        @jobs_builder = ::Karafka::App.config.internal.jobs_builder
-        @pauses_manager = PausesManager.new
+        @coordinators = Processing::CoordinatorsBuffer.new
         @client = Client.new(@subscription_group)
         @executors = Processing::ExecutorsBuffer.new(@client, subscription_group)
+        @jobs_builder = proc_config.jobs_builder
+        @partitioner = proc_config.partitioner_class.new(subscription_group)
         # We reference scheduler here as it is much faster than fetching this each time
-        @scheduler = ::Karafka::App.config.internal.scheduler
+        @scheduler = proc_config.scheduler
         # We keep one buffer for messages to preserve memory and not allocate extra objects
         # We can do this that way because we always first schedule jobs using messages before we
         # fetch another batch.
@@ -86,6 +89,9 @@ module Karafka
           build_and_schedule_revoke_lost_partitions_jobs
 
           # We wait only on jobs from our subscription group. Other groups are independent.
+          # This will block on revoked jobs until they are finished. Those are not meant to last
+          # long and should not have any bigger impact on the system. Doing this in a blocking way
+          # simplifies the overall design and prevents from race conditions
           wait
 
           build_and_schedule_consumption_jobs
@@ -136,7 +142,7 @@ module Karafka
 
       # Resumes processing of partitions that were paused due to an error.
       def resume_paused_partitions
-        @pauses_manager.resume do |topic, partition|
+        @coordinators.resume do |topic, partition|
           @client.resume(topic, partition)
         end
       end
@@ -152,9 +158,21 @@ module Karafka
 
         revoked_partitions.each do |topic, partitions|
           partitions.each do |partition|
-            pause_tracker = @pauses_manager.fetch(topic, partition)
-            executor = @executors.fetch(topic, partition, pause_tracker)
-            jobs << @jobs_builder.revoked(executor)
+            @coordinators.revoke(topic, partition)
+
+            # There may be a case where we have lost partition of which data we have never
+            # processed (if it was assigned and revoked really fast), thus we may not have it
+            # here. In cases like this, we do not run a revocation job
+            @executors.find_all(topic, partition).each do |executor|
+              jobs << @jobs_builder.revoked(executor)
+            end
+
+            # We need to remove all the executors of a given topic partition that we have lost, so
+            # next time we pick up it's work, new executors kick in. This may be needed especially
+            # for LRJ where we could end up with a race condition
+            # This revocation needs to happen after the jobs are scheduled, otherwise they would
+            # be scheduled with new executors instead of old
+            @executors.revoke(topic, partition)
           end
         end
 
@@ -191,11 +209,19 @@ module Karafka
         jobs = []
 
         @messages_buffer.each do |topic, partition, messages|
-          pause_tracker = @pauses_manager.fetch(topic, partition)
+          coordinator = @coordinators.find_or_create(topic, partition)
+
+          # Start work coordination for this topic partition
+          coordinator.start(messages)
 
-          executor = @executors.fetch(topic, partition, pause_tracker)
+          @partitioner.call(topic, messages) do |group_id, partition_messages|
+            # Count the job we're going to create here
+            coordinator.increment
 
-          jobs << @jobs_builder.consume(executor, messages)
+            executor = @executors.find_or_create(topic, partition, group_id)
+
+            jobs << @jobs_builder.consume(executor, partition_messages, coordinator)
+          end
         end
 
         @scheduler.schedule_consumption(@jobs_queue, jobs)
@@ -231,7 +257,7 @@ module Karafka
         @jobs_queue.wait(@subscription_group.id)
         @jobs_queue.clear(@subscription_group.id)
         @client.reset
-        @pauses_manager = PausesManager.new
+        @coordinators.reset
         @executors = Processing::ExecutorsBuffer.new(@client, @subscription_group)
       end
     end

data/lib/karafka/connection/rebalance_manager.rb

@@ -18,13 +18,15 @@ module Karafka
       # Empty array for internal usage not to create new objects
       EMPTY_ARRAY = [].freeze
 
+      attr_reader :assigned_partitions, :revoked_partitions
+
       private_constant :EMPTY_ARRAY
 
       # @return [RebalanceManager]
       def initialize
         @assigned_partitions = {}
         @revoked_partitions = {}
-        @lost_partitions = {}
+        @changed = false
       end
 
       # Resets the rebalance manager state
@@ -33,26 +35,12 @@ module Karafka
       def clear
         @assigned_partitions.clear
         @revoked_partitions.clear
-        @lost_partitions.clear
-      end
-
-      # @return [Hash<String, Array<Integer>>] hash where the keys are the names of topics for
-      #   which we've lost partitions and array with ids of the partitions as the value
-      # @note We do not consider as lost topics and partitions that got revoked and assigned
-      def revoked_partitions
-        return @revoked_partitions if @revoked_partitions.empty?
-        return @lost_partitions unless @lost_partitions.empty?
-
-        @revoked_partitions.each do |topic, partitions|
-          @lost_partitions[topic] = partitions - @assigned_partitions.fetch(topic, EMPTY_ARRAY)
-        end
-
-        @lost_partitions
+        @changed = false
       end
 
-      # @return [Boolean] true if any partitions were revoked
-      def revoked_partitions?
-        !revoked_partitions.empty?
+      # @return [Boolean] indicates a state change in the partitions assignment
+      def changed?
+        @changed
       end
 
       # Callback that kicks in inside of rdkafka, when new partitions are assigned.
@@ -62,6 +50,7 @@ module Karafka
       # @param partitions [Rdkafka::Consumer::TopicPartitionList]
       def on_partitions_assigned(_, partitions)
         @assigned_partitions = partitions.to_h.transform_values { |part| part.map(&:partition) }
+        @changed = true
       end
 
       # Callback that kicks in inside of rdkafka, when partitions are revoked.
@@ -71,6 +60,18 @@ module Karafka
       # @param partitions [Rdkafka::Consumer::TopicPartitionList]
       def on_partitions_revoked(_, partitions)
         @revoked_partitions = partitions.to_h.transform_values { |part| part.map(&:partition) }
+        @changed = true
+      end
+
+      # We consider as lost only partitions that were taken away and not re-assigned back to us
+      def lost_partitions
+        lost_partitions = {}
+
+        revoked_partitions.each do |topic, partitions|
+          lost_partitions[topic] = partitions - assigned_partitions.fetch(topic, EMPTY_ARRAY)
+        end
+
+        lost_partitions
       end
     end
   end

data/lib/karafka/contracts/base.rb

@@ -3,20 +3,14 @@
 module Karafka
   module Contracts
     # Base contract for all Karafka contracts
-    class Base < Dry::Validation::Contract
-      config.messages.load_paths << File.join(Karafka.gem_root, 'config', 'errors.yml')
-
+    class Base < ::WaterDrop::Contractable::Contract
       # @param data [Hash] data for validation
       # @return [Boolean] true if all good
       # @raise [Errors::InvalidConfigurationError] invalid configuration error
       # @note We use contracts only in the config validation context, so no need to add support
       #   for multiple error classes. It will be added when it will be needed.
       def validate!(data)
-        result = call(data)
-
-        return true if result.success?
-
-        raise Errors::InvalidConfigurationError, result.errors.to_h
+        super(data, Errors::InvalidConfigurationError)
       end
     end
   end

data/lib/karafka/contracts/config.rb

@@ -9,57 +9,90 @@ module Karafka
     #   validated per each route (topic + consumer_group) because they can be overwritten,
     #   so we validate all of that once all the routes are defined and ready.
     class Config < Base
-      params do
-        # License validity happens in the licenser. Here we do only the simple consistency checks
-        required(:license).schema do
-          required(:token) { bool? | str? }
-          required(:entity) { str? }
-          required(:expires_on) { date? }
+      configure do |config|
+        config.error_messages = YAML.safe_load(
+          File.read(
+            File.join(Karafka.gem_root, 'config', 'errors.yml')
+          )
+        ).fetch('en').fetch('validations').fetch('config')
+      end
+
+      # License validity happens in the licenser. Here we do only the simple consistency checks
+      nested(:license) do
+        required(:token) { |val| [true, false].include?(val) || val.is_a?(String) }
+        required(:entity) { |val| val.is_a?(String) }
+        required(:expires_on) { |val| val.is_a?(Date) }
+      end
+
+      required(:client_id) { |val| val.is_a?(String) && Contracts::TOPIC_REGEXP.match?(val) }
+      required(:concurrency) { |val| val.is_a?(Integer) && val.positive? }
+      required(:consumer_mapper) { |val| !val.nil? }
+      required(:consumer_persistence) { |val| [true, false].include?(val) }
+      required(:pause_timeout) { |val| val.is_a?(Integer) && val.positive? }
+      required(:pause_max_timeout) { |val| val.is_a?(Integer) && val.positive? }
+      required(:pause_with_exponential_backoff) { |val| [true, false].include?(val) }
+      required(:shutdown_timeout) { |val| val.is_a?(Integer) && val.positive? }
+      required(:max_wait_time) { |val| val.is_a?(Integer) && val.positive? }
+      required(:kafka) { |val| val.is_a?(Hash) && !val.empty? }
+
+      # We validate internals just to be sure, that they are present and working
+      nested(:internal) do
+        required(:status) { |val| !val.nil? }
+        required(:process) { |val| !val.nil? }
+
+        nested(:routing) do
+          required(:builder) { |val| !val.nil? }
+          required(:subscription_groups_builder) { |val| !val.nil? }
+        end
+
+        nested(:processing) do
+          required(:jobs_builder) { |val| !val.nil? }
+          required(:scheduler) { |val| !val.nil? }
+          required(:coordinator_class) { |val| !val.nil? }
+          required(:partitioner_class) { |val| !val.nil? }
         end
 
-        required(:client_id).filled(:str?, format?: Karafka::Contracts::TOPIC_REGEXP)
-        required(:concurrency) { int? & gt?(0) }
-        required(:consumer_mapper).filled
-        required(:consumer_persistence).filled(:bool?)
-        required(:pause_timeout) { int? & gt?(0) }
-        required(:pause_max_timeout) { int? & gt?(0) }
-        required(:pause_with_exponential_backoff).filled(:bool?)
-        required(:shutdown_timeout) { int? & gt?(0) }
-        required(:max_wait_time) { int? & gt?(0) }
-        required(:kafka).filled(:hash)
-
-        # We validate internals just to be sure, that they are present and working
-        required(:internal).schema do
-          required(:routing_builder)
-          required(:subscription_groups_builder)
-          required(:jobs_builder)
-          required(:status)
-          required(:process)
-          required(:scheduler)
+        nested(:active_job) do
+          required(:dispatcher) { |val| !val.nil? }
+          required(:job_options_contract) { |val| !val.nil? }
+          required(:consumer_class) { |val| !val.nil? }
         end
       end
 
-      # rdkafka requires all the keys to be strings, so we ensure that
-      rule(:kafka) do
-        next unless value.is_a?(Hash)
+      virtual do |data, errors|
+        next unless errors.empty?
+
+        detected_errors = []
 
-        value.each_key do |key|
+        data.fetch(:kafka).each_key do |key|
           next if key.is_a?(Symbol)
 
-          key(:"kafka.#{key}").failure(:kafka_key_must_be_a_symbol)
+          detected_errors << [[:kafka, key], :key_must_be_a_symbol]
         end
+
+        detected_errors
       end
 
-      rule(:pause_timeout, :pause_max_timeout) do
-        if values[:pause_timeout].to_i > values[:pause_max_timeout].to_i
-          key(:pause_timeout).failure(:max_timeout_vs_pause_max_timeout)
-        end
+      virtual do |data, errors|
+        next unless errors.empty?
+
+        pause_timeout = data.fetch(:pause_timeout)
+        pause_max_timeout = data.fetch(:pause_max_timeout)
+
+        next if pause_timeout <= pause_max_timeout
+
+        [[%i[pause_timeout], :max_timeout_vs_pause_max_timeout]]
       end
 
-      rule(:shutdown_timeout, :max_wait_time) do
-        if values[:max_wait_time].to_i >= values[:shutdown_timeout].to_i
-          key(:shutdown_timeout).failure(:shutdown_timeout_vs_max_wait_time)
-        end
+      virtual do |data, errors|
+        next unless errors.empty?
+
+        shutdown_timeout = data.fetch(:shutdown_timeout)
+        max_wait_time = data.fetch(:max_wait_time)
+
+        next if max_wait_time < shutdown_timeout
+
+        [[%i[shutdown_timeout], :shutdown_timeout_vs_max_wait_time]]
       end
     end
   end

data/lib/karafka/contracts/consumer_group.rb

@@ -4,32 +4,39 @@ module Karafka
   module Contracts
     # Contract for single full route (consumer group + topics) validation.
     class ConsumerGroup < Base
-      # Internal contract for sub-validating topics schema
-      TOPIC_CONTRACT = ConsumerGroupTopic.new.freeze
+      configure do |config|
+        config.error_messages = YAML.safe_load(
+          File.read(
+            File.join(Karafka.gem_root, 'config', 'errors.yml')
+          )
+        ).fetch('en').fetch('validations').fetch('consumer_group')
+      end
 
-      private_constant :TOPIC_CONTRACT
+      required(:id) { |id| id.is_a?(String) && Contracts::TOPIC_REGEXP.match?(id) }
+      required(:topics) { |topics| topics.is_a?(Array) && !topics.empty? }
 
-      params do
-        required(:id).filled(:str?, format?: Karafka::Contracts::TOPIC_REGEXP)
-        required(:topics).value(:array, :filled?)
-      end
+      virtual do |data, errors|
+        next unless errors.empty?
 
-      rule(:topics) do
-        if value.is_a?(Array)
-          names = value.map { |topic| topic[:name] }
+        names = data.fetch(:topics).map { |topic| topic[:name] }
 
-          key.failure(:topics_names_not_unique) if names.size != names.uniq.size
-        end
+        next if names.size == names.uniq.size
+
+        [[%i[topics], :names_not_unique]]
       end
 
-      rule(:topics) do
-        if value.is_a?(Array)
-          value.each_with_index do |topic, index|
-            TOPIC_CONTRACT.call(topic).errors.each do |error|
-              key([:topics, index, error.path[0]]).failure(error.text)
-            end
+      virtual do |data, errors|
+        next unless errors.empty?
+
+        fetched_errors = []
+
+        data.fetch(:topics).each do |topic|
+          ConsumerGroupTopic.new.call(topic).errors.each do |key, value|
+            fetched_errors << [[topic, key].flatten, value]
           end
         end
+
+        fetched_errors
       end
     end
   end

data/lib/karafka/contracts/consumer_group_topic.rb

@@ -4,24 +4,38 @@ module Karafka
   module Contracts
     # Consumer group topic validation rules.
     class ConsumerGroupTopic < Base
-      params do
-        required(:consumer).filled
-        required(:deserializer).filled
-        required(:id).filled(:str?, format?: Karafka::Contracts::TOPIC_REGEXP)
-        required(:kafka).filled
-        required(:max_messages) { int? & gteq?(1) }
-        required(:initial_offset).filled(included_in?: %w[earliest latest])
-        required(:max_wait_time).filled { int? & gteq?(10) }
-        required(:manual_offset_management).filled(:bool?)
-        required(:name).filled(:str?, format?: Karafka::Contracts::TOPIC_REGEXP)
+      configure do |config|
+        config.error_messages = YAML.safe_load(
+          File.read(
+            File.join(Karafka.gem_root, 'config', 'errors.yml')
+          )
+        ).fetch('en').fetch('validations').fetch('consumer_group_topic')
       end
 
-      rule(:kafka) do
-        # This will trigger rdkafka validations that we catch and re-map the info and use dry
-        # compatible format
-        Rdkafka::Config.new(value).send(:native_config)
-      rescue Rdkafka::Config::ConfigError => e
-        key(:kafka).failure(e.message)
+      required(:consumer) { |consumer_group| !consumer_group.nil? }
+      required(:deserializer) { |deserializer| !deserializer.nil? }
+      required(:id) { |id| id.is_a?(String) && Contracts::TOPIC_REGEXP.match?(id) }
+      required(:kafka) { |kafka| kafka.is_a?(Hash) && !kafka.empty? }
+      required(:max_messages) { |mm| mm.is_a?(Integer) && mm >= 1 }
+      required(:initial_offset) { |io| %w[earliest latest].include?(io) }
+      required(:max_wait_time) { |mwt| mwt.is_a?(Integer) && mwt >= 10 }
+      required(:manual_offset_management) { |mmm| [true, false].include?(mmm) }
+      required(:name) { |name| name.is_a?(String) && Contracts::TOPIC_REGEXP.match?(name) }
+
+      virtual do |data, errors|
+        next unless errors.empty?
+
+        value = data.fetch(:kafka)
+
+        begin
+          # This will trigger rdkafka validations that we catch and re-map the info and use dry
+          # compatible format
+          Rdkafka::Config.new(value).send(:native_config)
+
+          nil
+        rescue Rdkafka::Config::ConfigError => e
+          [[%w[kafka], e.message]]
+        end
       end
     end
   end

data/lib/karafka/contracts/server_cli_options.rb

@@ -4,17 +4,28 @@ module Karafka
   module Contracts
     # Contract for validating correctness of the server cli command options.
     class ServerCliOptions < Base
-      params do
-        optional(:consumer_groups).value(:array, :filled?)
+      configure do |config|
+        config.error_messages = YAML.safe_load(
+          File.read(
+            File.join(Karafka.gem_root, 'config', 'errors.yml')
+          )
+        ).fetch('en').fetch('validations').fetch('server_cli_options')
       end
 
-      rule(:consumer_groups) do
+      optional(:consumer_groups) { |cg| cg.is_a?(Array) && !cg.empty? }
+
+      virtual do |data, errors|
+        next unless errors.empty?
+        next unless data.key?(:consumer_groups)
+
+        value = data.fetch(:consumer_groups)
+
         # If there were no consumer_groups declared in the server cli, it means that we will
         # run all of them and no need to validate them here at all
-        if !value.nil? &&
-           !(value - Karafka::App.config.internal.routing_builder.map(&:name)).empty?
-          key(:consumer_groups).failure(:consumer_groups_inclusion)
-        end
+        next if value.nil?
+        next if (value - Karafka::App.config.internal.routing.builder.map(&:name)).empty?
+
+        [[%i[consumer_groups], :consumer_groups_inclusion]]
       end
     end
   end

data/lib/karafka/errors.rb

@@ -47,5 +47,8 @@ module Karafka
     # Used to instrument this error into the error notifications
     # We do not raise it so we won't crash deployed systems
     ExpiredLicenseTokenError = Class.new(BaseError)
+
+    # This should never happen. Please open an issue if it does.
+    InvalidCoordinatorState = Class.new(BaseError)
   end
 end

data/lib/karafka/helpers/colorize.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Karafka
+  module Helpers
+    # Simple wrapper for adding colors to strings
+    module Colorize
+      # @param string [String] string we want to have in green
+      # @return [String] green string
+      def green(string)
+        "\033[0;32m#{string}\033[0m"
+      end
+
+      # @param string [String] string we want to have in red
+      # @return [String] red string
+      def red(string)
+        "\033[0;31m#{string}\033[0m"
+      end
+    end
+  end
+end