moesif_rack 1.4.15 → 1.4.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9533c0d146d3b8f0b3a6f8b5edf0f9cda27e7207fb34d92f7c24f07b33a4d38
-  data.tar.gz: 233466faadd9f15e6bf22a65b3995a78c5eb9a752bf36c743440ad094486dc28
+  metadata.gz: a4dc902308676d3b9bb9a23a61f9fc166024b0848d06e026f98a68be8de336b7
+  data.tar.gz: 898963aa24db49d6772f4a037d2919650c93e989e9ba735db8ac5e0c6b65bc02
 SHA512:
-  metadata.gz: 0236b79eb83085bfe54cca228792267031bea3edc6aa71cb556adef68ff46143c4087035396c8698b6502f7b93137444dd6a5337a060c9777b5e6933f4352af2
-  data.tar.gz: '08bfc8a7fe975f9f6d5992a4bcf4b761852c361b30882577dc8e031cb066f78c5735cae1ba5315e4725623385cec5a6ebb057a48e1b67ea86dd327db25c754b8'
+  metadata.gz: dfca38cca1df0bf26113b8c7198f478dd178f0cf925c2103972b4770e6bcfa032eaa9079801079224cc7127d6166d499e73bc826062835675efe3ff642ba8fd9
+  data.tar.gz: 89d1b57bddaa065bc338aad7c92451405ad6a077bfc0e17635da7e42412681a922a22a2395b15f4a478969a4c63114fea582b693523504a87c3f1aec76862c50

data/lib/moesif_rack/app_config.rb

@@ -4,12 +4,14 @@ require 'time'
 require 'zlib'
 require 'stringio'
 require_relative './helpers.rb'
+require_relative './regex_config_helper.rb'
 
 class AppConfig
 
     def initialize debug
         @debug = debug
         @helpers = Helpers.new(debug)
+        @regex_config_helper = RegexConfigHelper.new(debug)
     end
 
     def get_config(api_controller)
@@ -55,7 +57,7 @@ class AppConfig
         end
     end
 
-    def get_sampling_percentage(config_api_response, user_id, company_id)
+    def get_sampling_percentage(event_model, config_api_response, user_id, company_id)
         # Get sampling percentage
         begin
             # Check if response body is not nil
@@ -63,6 +65,17 @@ class AppConfig
                 @helpers.log_debug("Getting sample rate for user #{user_id} company #{company_id}")
                 @helpers.log_debug(config_api_response.to_s)
 
+                # Get Regex Sampling rate
+                regex_config = config_api_response.fetch('regex_config', nil)
+
+                if !regex_config.nil? and !event_model.nil?
+                    config_mapping = @regex_config_helper.prepare_config_mapping(event_model)
+                    regex_sample_rate = @regex_config_helper.fetch_sample_rate_on_regex_match(regex_config, config_mapping)
+                    if !regex_sample_rate.nil?
+                        return regex_sample_rate
+                    end
+                end
+
                 # Get user sample rate object
                 user_sample_rate = config_api_response.fetch('user_sample_rate', nil)
 
@@ -87,6 +100,7 @@ class AppConfig
             end
         rescue => exception
             @helpers.log_debug 'Error while geting sampling percentage, assuming default behavior'
+            @helpers.log_debug exception.to_s
             return 100
         end
     end

data/lib/moesif_rack/moesif_middleware.rb

@@ -2,13 +2,13 @@ require 'moesif_api'
 require 'json'
 require 'time'
 require 'base64'
+require 'zlib'
+require 'stringio'
 require_relative './client_ip.rb'
 require_relative './app_config.rb'
 require_relative './update_user.rb'
 require_relative './update_company.rb'
 require_relative './helpers.rb'
-require 'zlib'
-require 'stringio'
 
 module MoesifRack
 
@@ -286,7 +286,7 @@ module MoesifRack
           random_percentage  = Random.rand(0.00..100.00)
 
           begin 
-            sampling_percentage = @app_config.get_sampling_percentage(@config, event_model.user_id, event_model.company_id)
+            sampling_percentage = @app_config.get_sampling_percentage(event_model, @config, event_model.user_id, event_model.company_id)
             @helpers.log_debug "Using sample rate #{sampling_percentage}"
           rescue => exception
             @helpers.log_debug 'Error while getting sampling percentage, assuming default behavior'
@@ -341,6 +341,8 @@ module MoesifRack
           @helpers.log_debug exception.to_s
           @helpers.log_debug exception.backtrace
         end
+      else
+         @helpers.log_debug "Skipped Event using should_skip configuration option."
       end
 
       [status, headers, body]

data/lib/moesif_rack/regex_config_helper.rb

@@ -0,0 +1,119 @@
+require 'moesif_api'
+
+require_relative './helpers.rb'
+
+class RegexConfigHelper
+
+    def initialize debug
+        @debug = debug
+    end
+
+    def prepare_config_mapping(event)
+        # Function to prepare config mapping
+        # Params:
+        #  - event: Event to be logged
+        # Return:
+        #  - regex_config: Regex config mapping
+        regex_config = {}
+
+        # Config mapping for request.verb
+        if defined? event.request.verb
+            regex_config["request.verb"] = event.request.verb
+        end
+
+        # Config mapping for request.uri
+        if defined? event.request.uri
+            extracted = /http[s]*:\/\/[^\/]+(\/[^?]+)/.match(event.request.uri)
+            if !extracted.nil?
+              route_mapping = extracted.captures[0]
+            else
+              route_mapping = '/'
+            end
+            regex_config["request.route"] = route_mapping
+        end
+
+        # Config mapping for request.ip_address
+        if defined? event.request.ip_address
+            regex_config["request.ip_address"] = event.request.ip_address
+        end
+
+        # Config mapping for response.status
+        if defined? event.response.status
+            regex_config["response.status"] = event.response.status
+        end
+
+        return regex_config
+
+    end
+
+    def regex_match(event_value, condition_value)
+        # Function to perform the regex matching with event value and condition value
+        # Params:
+        #  - event_value: Value associated with event (request)
+        #  - condition_value: Value associated with the regex config condition
+        # Return:
+        #  - regex_matched: Regex matched value to determine if the regex match was successful
+
+        extracted = Regexp.new(condition_value).match(event_value)
+        if !extracted.nil?
+          return extracted.to_s
+        end
+    end
+
+    def fetch_sample_rate_on_regex_match(regex_configs, config_mapping)
+        # Function to fetch the sample rate and determine if request needs to be block or not
+        # Args:
+        #  - regex_configs: Regex configs
+        #  - config_mapping: Config associated with the request
+        # Return:
+        #  - sample_rate: Sample rate
+
+        # Iterate through the list of regex configs
+        regex_configs.each { |regex_rule|
+            # Fetch the sample rate
+            sample_rate = regex_rule["sample_rate"]
+            # Fetch the conditions
+            conditions = regex_rule["conditions"]
+            # Bool flag to determine if the regex conditions are matched
+            regex_matched = false
+            # Create a table to hold the conditions mapping (path and value)
+            condition_table = {}
+
+            # Iterate through the regex rule conditions and map the path and value
+            conditions.each { |condition|
+                # Add condition path -> value to the condition table
+                condition_table[condition["path"]] = condition["value"]
+            }
+
+            # Iterate through conditions table and perform `and` operation between each conditions
+            condition_table.each do |path, values|
+
+                # Check if the path exists in the request config mapping
+                if !config_mapping[path].nil?
+                    # Fetch the value of the path in request config mapping
+                    event_data = config_mapping[path]
+
+                    # Perform regex matching with event value
+                    regex_matched = regex_match(event_data, values)
+                else
+                    # Path does not exists in request config mapping, so no need to match regex condition rule
+                    regex_matched = false
+                end
+
+                # If one of the rule does not match, skip the condition & avoid matching other rules for the same condition
+                if !regex_matched
+                    break
+                end
+            end
+
+            # If regex conditions matched, return sample rate
+            if regex_matched
+                return sample_rate
+            end
+        }
+
+        # If regex conditions are not matched, return sample rate as None and will use default sample rate
+        return nil
+    end
+
+end

data/moesif_capture_outgoing/httplog/http_log.rb

@@ -176,7 +176,7 @@ module MoesifCaptureOutgoing
           begin
             @random_percentage = Random.rand(0.00..100.00)
             begin 
-              @sampling_percentage = @app_config.get_sampling_percentage(@config, event_model.user_id, event_model.company_id)
+              @sampling_percentage = @app_config.get_sampling_percentage(event_model, @config, event_model.user_id, event_model.company_id)
             rescue => exception
               if @debug
                 puts 'Error while getting sampling percentage, assuming default behavior'

data/test/moesif_rack_test.rb

@@ -118,7 +118,7 @@ class MoesifRackTest < Test::Unit::TestCase
     @api_controller = @api_client.api
     @config = @app_config.get_config(@api_controller)
     @config_body, @config_etag, @last_updated_time = @app_config.parse_configuration(@config)
-    @sampling_percentage = @app_config.get_sampling_percentage(@config_body, nil, nil)
+    @sampling_percentage = @app_config.get_sampling_percentage(nil, @config_body, nil, nil)
     assert_operator 100, :>=, @sampling_percentage
   end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: moesif_rack
 version: !ruby/object:Gem::Version
-  version: 1.4.15
+  version: 1.4.16
 platform: ruby
 authors:
 - Moesif, Inc
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-03 00:00:00.000000000 Z
+date: 2022-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: test-unit
@@ -58,6 +58,7 @@ files:
 - lib/moesif_rack/client_ip.rb
 - lib/moesif_rack/helpers.rb
 - lib/moesif_rack/moesif_middleware.rb
+- lib/moesif_rack/regex_config_helper.rb
 - lib/moesif_rack/update_company.rb
 - lib/moesif_rack/update_user.rb
 - moesif_capture_outgoing/httplog.rb
@@ -90,7 +91,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
+rubygems_version: 3.3.3
 signing_key:
 specification_version: 4
 summary: moesif_rack