# frozen_string_literal: true

# Copyright The OpenTelemetry Authors
#
# SPDX-License-Identifier: Apache-2.0

module OpenTelemetry
  module Instrumentation
    module ConcurrentRuby
      module Patches
        # Concurrent::ThreadPoolExecutor patch for instrumentation
        module ThreadPoolExecutor
          # @see Concurrent::ExecutorService#post
          def post(*args, **kwargs, &task)
            context = OpenTelemetry::Context.current
            return super unless context

            super(*args, **kwargs) do
              OpenTelemetry::Context.with_current(context) do
                task.call(*args, **kwargs)
              end
            end
          end
        end
      end
    end
  end
end