add exclude target for the sampler

lib/app_profiler/sampler.rb

@@ -3,6 +3,7 @@
 require "app_profiler/sampler/config"
 module AppProfiler
   module Sampler
+    @exclude_targets_result = {}
     class << self
       def profile_params(request, config)
         profile_params_for(request.path, config)
@@ -18,6 +19,12 @@ def profile_params_for(target, config)
 
       def sample?(config, target)
         return false if Kernel.rand > config.sample_rate
+        return false if @exclude_targets_result[target]
+
+        if config.exclude_targets.any? { |t| target.match?(t) }
+          @exclude_targets_result[target] = true
+          return false
+        end
 
         return false unless config.targets.any? { |t| target.match?(t) }
 

lib/app_profiler/sampler/config.rb

@@ -5,7 +5,7 @@
 module AppProfiler
   module Sampler
     class Config
-      attr_reader :sample_rate, :targets, :cpu_interval, :backends_probability
+      attr_reader :sample_rate, :targets, :exclude_targets, :cpu_interval, :backends_probability
 
       SAMPLE_RATE = 0.001 # 0.1%
       TARGETS = ["/"]
@@ -18,7 +18,8 @@ def initialize(sample_rate: SAMPLE_RATE,
         backends_config: {
           stackprof: StackprofConfig.new,
         },
-        paths: nil)
+        paths: nil,
+        exclude_targets: [])
 
         if sample_rate < 0.0 || sample_rate > 1.0
           raise ArgumentError, "sample_rate must be between 0 and 1"
@@ -32,6 +33,7 @@ def initialize(sample_rate: SAMPLE_RATE,
         @targets = paths || targets
         @backends_config = backends_config
         @backends_probability = backends_probability
+      @exclude_targets = exclude_targets
       end
 
       def get_backend_config(backend_name)

test/app_profiler/sampler/sampler_test.rb

@@ -58,6 +58,19 @@ class SamplerTest < TestCase
         assert_nil(Sampler.profile_params(request, config))
       end
 
+
+      test "exclude_targets are respected" do
+        Kernel.stubs(:rand).returns(0.1)
+        config = Config.new(
+          sample_rate: 1.0,
+          targets: ["/foo"],
+          exclude_targets: ["/foo/bar"],
+        )
+
+        request = RequestParameters.new(Rack::Request.new({ "PATH_INFO" => "/foo/bar" }))
+        assert_nil(Sampler.profile_params(request, config))
+      end
+
       test "mixed backend probabilities" do
         skip("Vernier not supported") unless AppProfiler.vernier_supported?