statsig 1.25.2 → 1.33.0

data/lib/feature_gate.rb

@@ -0,0 +1,46 @@
+class FeatureGate
+
+  attr_accessor :name
+
+  attr_accessor :value
+
+  attr_accessor :rule_id
+
+  attr_accessor :group_name
+
+  attr_accessor :id_type
+
+  attr_accessor :evaluation_details
+
+  attr_accessor :target_app_ids
+
+  def initialize(
+    name,
+    value: false,
+    rule_id: '',
+    group_name: nil,
+    id_type: '',
+    evaluation_details: nil,
+    target_app_ids: nil
+  )
+    @name = name
+    @value = value
+    @rule_id = rule_id
+    @group_name = group_name
+    @id_type = id_type
+    @evaluation_details = evaluation_details
+    @target_app_ids = target_app_ids
+  end
+
+  def self.from_config_result(res)
+    new(
+      res.name,
+      value: res.gate_value,
+      rule_id: res.rule_id,
+      group_name: res.group_name,
+      id_type: res.id_type,
+      evaluation_details: res.evaluation_details,
+      target_app_ids: res.target_app_ids
+    )
+  end
+end

data/lib/hash_utils.rb

@@ -0,0 +1,32 @@
+require 'json'
+module Statsig
+  class HashUtils
+    def self.djb2(input_str)
+      hash = 0
+      input_str.each_char.each do |c|
+        hash = (hash << 5) - hash + c.ord
+        hash &= hash
+      end
+      hash &= 0xFFFFFFFF # Convert to unsigned 32-bit integer
+      return hash.to_s
+    end
+
+    def self.djb2ForHash(input_hash)
+      return djb2(input_hash.to_json)
+    end
+
+    def self.sha256(input_str)
+      return Digest::SHA256.base64digest(input_str)
+    end
+
+    def self.sortHash(input_hash)
+      dictionary = input_hash.clone.sort_by { |key| key }.to_h;
+      input_hash.each do |key, value|
+        if value.is_a?(Hash)
+          dictionary[key] = self.sortHash(value)
+        end
+      end
+      return dictionary
+    end
+  end
+end

data/lib/id_list.rb

@@ -1,4 +1,4 @@
-# typed: true
+
 module Statsig
   class IDList
     attr_accessor :name
@@ -19,7 +19,7 @@ module Statsig
     end
 
     def self.new_empty(json)
-      self.new(json)
+      new(json)
       @size = 0
     end
 

data/lib/interfaces/data_store.rb

@@ -1,4 +1,4 @@
-# typed: true
+
 module Statsig
   module Interfaces
     class IDataStore

data/lib/interfaces/user_persistent_storage.rb

@@ -0,0 +1,12 @@
+
+module Statsig
+  module Interfaces
+    class IUserPersistentStorage
+      def load(key)
+        nil
+      end
+
+      def save(key, data) end
+    end
+  end
+end

data/lib/layer.rb

@@ -1,6 +1,3 @@
-# typed: false
-
-require 'sorbet-runtime'
 ##
 # Contains the current values from Statsig.
 # Will contain layer default values for all shared parameters in that layer.
@@ -9,23 +6,22 @@ require 'sorbet-runtime'
 #
 # Layers Documentation: https://docs.statsig.com/layers
 class Layer
-  extend T::Sig
 
-  sig { returns(String) }
   attr_accessor :name
 
-  sig { returns(String) }
   attr_accessor :rule_id
 
-  sig { params(name: String, value: T::Hash[String, T.untyped], rule_id: String, exposure_log_func: T.any(Method, Proc, NilClass)).void }
-  def initialize(name, value = {}, rule_id = '', exposure_log_func = nil)
+  attr_accessor :group_name
+
+  def initialize(name, value = {}, rule_id = '', group_name = nil, allocated_experiment = nil, exposure_log_func = nil)
     @name = name
-    @value = value
+    @value = value || {}
     @rule_id = rule_id
+    @group_name = group_name
+    @allocated_experiment = allocated_experiment
     @exposure_log_func = exposure_log_func
   end
 
-  sig { params(index: String, default_value: T.untyped).returns(T.untyped) }
   ##
   # Get the value for the given key (index), falling back to the default_value if it cannot be found.
   #
@@ -41,7 +37,6 @@ class Layer
     @value[index]
   end
 
-  sig { params(index: String, default_value: T.untyped).returns(T.untyped) }
   ##
   # Get the value for the given key (index), falling back to the default_value if it cannot be found
   # or is found to have a different type from the default_value.
@@ -58,4 +53,4 @@ class Layer
 
     @value[index]
   end
-end
+end

data/lib/network.rb

@@ -1,12 +1,11 @@
-# typed: true
-
 require 'http'
 require 'json'
 require 'securerandom'
-require 'sorbet-runtime'
+
 require 'uri_helper'
+require 'connection_pool'
 
-$retry_codes = [408, 500, 502, 503, 504, 522, 524, 599]
+RETRY_CODES = [408, 500, 502, 503, 504, 522, 524, 599].freeze
 
 module Statsig
   class NetworkError < StandardError
@@ -19,9 +18,6 @@ module Statsig
   end
 
   class Network
-    extend T::Sig
-
-    sig { params(server_secret: String, options: StatsigOptions, backoff_mult: Integer).void }
 
     def initialize(server_secret, options, backoff_mult = 10)
       super()
@@ -33,29 +29,44 @@ module Statsig
       @post_logs_retry_backoff = options.post_logs_retry_backoff
       @post_logs_retry_limit = options.post_logs_retry_limit
       @session_id = SecureRandom.uuid
+      @connection_pool = ConnectionPool.new(size: 3) do
+        meta = Statsig.get_statsig_metadata
+        client = HTTP.use(:auto_inflate).headers(
+          {
+            'STATSIG-API-KEY' => @server_secret,
+            'STATSIG-SERVER-SESSION-ID' => @session_id,
+            'Content-Type' => 'application/json; charset=UTF-8',
+            'STATSIG-SDK-TYPE' => meta['sdkType'],
+            'STATSIG-SDK-VERSION' => meta['sdkVersion'],
+            'STATSIG-SDK-LANGUAGE-VERSION' => meta['languageVersion'],
+            'Accept-Encoding' => 'gzip'
+          }
+        ).accept(:json)
+        if @timeout
+          client = client.timeout(@timeout)
+        end
+
+        client
+      end
     end
 
-    sig { params(endpoint: String, body: String, retries: Integer, backoff: Integer)
-            .returns([T.any(HTTP::Response, NilClass), T.any(StandardError, NilClass)]) }
+    def download_config_specs(since_time)
+      get("download_config_specs/#{@server_secret}.json?sinceTime=#{since_time}")
+    end
 
-    def post_helper(endpoint, body, retries = 0, backoff = 1)
+    def get(endpoint, retries = 0, backoff = 1)
+      request(:GET, endpoint, nil, retries, backoff)
+    end
+
+    def post(endpoint, body, retries = 0, backoff = 1)
+      request(:POST, endpoint, body, retries, backoff)
+    end
+
+    def request(method, endpoint, body, retries = 0, backoff = 1)
       if @local_mode
         return nil, nil
       end
 
-      meta = Statsig.get_statsig_metadata
-      http = HTTP.headers(
-        {
-          "STATSIG-API-KEY" => @server_secret,
-          "STATSIG-CLIENT-TIME" => (Time.now.to_f * 1000).to_i.to_s,
-          "STATSIG-SERVER-SESSION-ID" => @session_id,
-          "Content-Type" => "application/json; charset=UTF-8",
-          "STATSIG-SDK-TYPE" => meta['sdkType'],
-          "STATSIG-SDK-VERSION" => meta['sdkVersion'],
-        }).accept(:json)
-      if @timeout
-        http = http.timeout(@timeout)
-      end
       backoff_adjusted = backoff > 10 ? backoff += Random.rand(10) : backoff # to deter overlap
       if @post_logs_retry_backoff
         if @post_logs_retry_backoff.is_a? Integer
@@ -66,48 +77,39 @@ module Statsig
       end
       url = URIHelper.build_url(endpoint)
       begin
-        res = http.post(url, body: body)
+        res = @connection_pool.with do |conn|
+          request = conn.headers('STATSIG-CLIENT-TIME' => (Time.now.to_f * 1000).to_i.to_s)
+          case method
+          when :GET
+            request.get(url)
+          when :POST
+            request.post(url, body: body)
+          end
+        end
       rescue StandardError => e
         ## network error retry
-        return nil, e unless retries > 0
+        return nil, e unless retries.positive?
+
         sleep backoff_adjusted
-        return post_helper(endpoint, body, retries - 1, backoff * @backoff_multiplier)
+        return request(method, endpoint, body, retries - 1, backoff * @backoff_multiplier)
       end
       return res, nil if res.status.success?
-      return nil, NetworkError.new("Got an exception when making request to #{url}: #{res.to_s}", res.status.to_i) unless retries > 0 && $retry_codes.include?(res.code)
-      ## status code retry
-      sleep backoff_adjusted
-      post_helper(endpoint, body, retries - 1, backoff * @backoff_multiplier)
-    end
 
-    def check_gate(user, gate_name)
-      begin
-        request_body = JSON.generate({ 'user' => user&.serialize(false), 'gateName' => gate_name })
-        response, _ = post_helper('check_gate', request_body)
-        return JSON.parse(response.body) unless response.nil?
-        false
-      rescue
-        return false
+      unless retries.positive? && RETRY_CODES.include?(res.code)
+        return res, NetworkError.new("Got an exception when making request to #{url}: #{res.to_s}",
+                                     res.status.to_i)
       end
-    end
 
-    def get_config(user, dynamic_config_name)
-      begin
-        request_body = JSON.generate({ 'user' => user&.serialize(false), 'configName' => dynamic_config_name })
-        response, _ = post_helper('get_config', request_body)
-        return JSON.parse(response.body) unless response.nil?
-        nil
-      rescue
-        return nil
-      end
+      ## status code retry
+      sleep backoff_adjusted
+      request(method, endpoint, body, retries - 1, backoff * @backoff_multiplier)
     end
 
     def post_logs(events)
-      begin
-        json_body = JSON.generate({ 'events' => events, 'statsigMetadata' => Statsig.get_statsig_metadata })
-        post_helper('log_event', json_body, @post_logs_retry_limit)
-      rescue
-      end
+      json_body = JSON.generate({ events: events, statsigMetadata: Statsig.get_statsig_metadata })
+      post('log_event', json_body, @post_logs_retry_limit)
+    rescue StandardError
+
     end
   end
-end
+end