statsig 2.8.3 → 2.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a17857d17823d2f8d7f6a13d01c4f3d117e4187233d377441c4e7be7a6499ef1
-  data.tar.gz: 34bb6a509d1fca2ab2973edf7ce0f76814440c5197649a68af8c573130ef13d5
+  metadata.gz: fbf7ba0eaa7ee393b1c4060244eeef4e2dfc83d8554532511409825a22bee21b
+  data.tar.gz: f3509f7c05fa8d90c8b38b99b2366eded563e4cbeca91f13b01924ea3c66d35f
 SHA512:
-  metadata.gz: eef23c96fefa5aff93eec2ce8562a45ccb59434bfb68d264f39b50df987689c402286895103295b701ba6f71db970f43bd8832f608a2da0622a5bde1717d9521
-  data.tar.gz: 7df3f01db686cf387e30b7d45be967be4d9376340d64506359c0df6a0a9b169f45ef310a7b727b34f87cb9c4ddb97f7ba0a80daf7e79d8102f256f83f32369cd
+  metadata.gz: 4fc3b066283642cee6990b2ca8a12f21a3355adec4c09d2cc25fabeed97ace70ebd23f565d56d053c7d77946dbd76cc565dccb94ab7d17868ba6904f84d433e1
+  data.tar.gz: 2d8064a954dab46521ac9034cca3a8d920266d408d9337149d1fce92973338220f82b47781c35292a8c963868e2f5d866b08df251cb1c63701ee12cc5b798b12

data/lib/network.rb

@@ -28,34 +28,19 @@ 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
+      @connection_pools = {}
+      @connection_pools_mutex = Mutex.new
     end
 
-    def download_config_specs(since_time)
+    def download_config_specs(since_time, context)
       url = @options.download_config_specs_url
       dcs_url = "#{url}#{@server_secret}.json"
       if since_time.positive?
         dcs_url += "?sinceTime=#{since_time}"
       end
+      if context == 'initialize'
+        return get(dcs_url, @options.initialize_retry_limit)
+      end
       get(dcs_url)
     end
 
@@ -66,10 +51,7 @@ module Statsig
       gzip = Zlib::GzipWriter.new(StringIO.new)
       gzip << json_body
 
-      response, e = post(url, gzip.close.string, @post_logs_retry_limit, 1, true, event_count)
-
-      # Consume response body to ensure connection can be closed.
-      response&.flush
+      _response, e = post(url, gzip.close.string, @post_logs_retry_limit, 1, true, event_count)
 
       unless e == nil
         message = "Failed to log #{event_count} events after #{@post_logs_retry_limit} retries"
@@ -81,11 +63,18 @@ module Statsig
 
     end
 
-    def get_id_lists
+    def get_id_lists(context)
       url = @options.get_id_lists_url
+      if context == 'initialize'
+        return post(url, JSON.generate({ 'statsigMetadata' => Statsig.get_statsig_metadata }), @options.initialize_retry_limit)
+      end
       post(url, JSON.generate({ 'statsigMetadata' => Statsig.get_statsig_metadata }))
     end
 
+    def download_id_list(url, start_byte = 0)
+      request(:GET, url, nil, 0, 1, false, 0, { 'Range' => "bytes=#{start_byte}-" }, false)
+    end
+
     def get(url, retries = 0, backoff = 1)
       request(:GET, url, nil, retries, backoff)
     end
@@ -94,7 +83,21 @@ module Statsig
       request(:POST, url, body, retries, backoff, zipped, event_count)
     end
 
-    def request(method, url, body, retries = 0, backoff = 1, zipped = false, event_count = 0)
+    def shutdown
+      pools = @connection_pools_mutex.synchronize do
+        pools = @connection_pools.values
+        @connection_pools = {}
+        pools
+      end
+
+      pools.each do |pool|
+        pool.shutdown do |conn|
+          conn.close if conn.respond_to?(:close)
+        end
+      end
+    end
+
+    def request(method, url, body, retries = 0, backoff = 1, zipped = false, event_count = 0, extra_headers = {}, use_statsig_headers = true)
       if @local_mode
         return nil, nil
       end
@@ -108,18 +111,25 @@ module Statsig
       end
 
       begin
-        res = @connection_pool.with do |conn|
-          request = conn.headers(
-            'STATSIG-CLIENT-TIME' => (Time.now.to_f * 1000).to_i.to_s,
-            'CONTENT-ENCODING' => zipped ? 'gzip' : nil,
-            'STATSIG-EVENT-COUNT' => event_count == 0 ? nil : event_count.to_s
-          )
-
-          case method
-          when :GET
-            request.get(url)
-          when :POST
-            request.post(url, body: body)
+        pool = connection_pool_for(url)
+        res = pool.with do |conn|
+          begin
+            request_headers = build_request_headers(zipped, event_count, extra_headers, use_statsig_headers)
+            request = conn.headers(request_headers)
+
+            response = case method
+                       when :GET
+                         request.get(url)
+                       when :POST
+                         request.post(url, body: body)
+                       end
+
+            # Fully drain the response before the client returns to the pool.
+            response.body.to_s
+            response
+          rescue StandardError
+            pool.discard_current_connection(&:close)
+            raise
           end
         end
       rescue StandardError => e
@@ -127,7 +137,7 @@ module Statsig
         return nil, e unless retries.positive?
 
         sleep backoff_adjusted
-        return request(method, url, body, retries - 1, backoff * @backoff_multiplier, zipped, event_count)
+        return request(method, url, body, retries - 1, backoff * @backoff_multiplier, zipped, event_count, extra_headers, use_statsig_headers)
       end
       return res, nil if res.status.success?
 
@@ -138,7 +148,50 @@ module Statsig
 
       ## status code retry
       sleep backoff_adjusted
-      request(method, url, body, retries - 1, backoff * @backoff_multiplier, zipped, event_count)
+      request(method, url, body, retries - 1, backoff * @backoff_multiplier, zipped, event_count, extra_headers, use_statsig_headers)
+    end
+
+    private
+
+    def connection_pool_for(url)
+      origin = HTTP::URI.parse(url).origin
+      @connection_pools_mutex.synchronize do
+        @connection_pools[origin] ||= ConnectionPool.new(size: 3) do
+          build_persistent_client(origin)
+        end
+      end
+    end
+
+    def build_persistent_client(origin)
+      client = HTTP.use(:auto_inflate)
+      client = client.timeout(@timeout) if @timeout
+      client.persistent(origin)
+    end
+
+    def build_request_headers(zipped, event_count, extra_headers, use_statsig_headers)
+      headers = {}
+
+      if use_statsig_headers
+        meta = Statsig.get_statsig_metadata
+        headers.merge!(
+          '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' => 'application/json',
+          'Accept-Encoding' => 'gzip',
+          'STATSIG-CLIENT-TIME' => (Time.now.to_f * 1000).to_i.to_s
+        )
+      end
+
+      headers['CONTENT-ENCODING'] = 'gzip' if zipped
+      headers['STATSIG-EVENT-COUNT'] = event_count.to_s unless event_count == 0
+
+      headers.merge!(extra_headers)
+      headers.delete_if { |_key, value| value.nil? }
+      headers
     end
   end
 end

data/lib/spec_store.rb

@@ -1,6 +1,4 @@
 require 'concurrent-ruby'
-require 'net/http'
-require 'uri'
 require_relative 'api_config'
 require_relative 'evaluation_details'
 require_relative 'hash_utils'
@@ -291,7 +289,7 @@ module Statsig
       error = nil
       failure_details = nil
       begin
-        response, e = @network.download_config_specs(@last_config_sync_time)
+        response, e = @network.download_config_specs(@last_config_sync_time, context)
         code = response&.status.to_i
         if e.is_a? NetworkError
           code = e.http_code
@@ -302,14 +300,15 @@ module Statsig
         if e.nil?
           unless response.nil?
             tracker = @diagnostics.track(context, 'download_config_specs', 'process')
-            failure_details = process_specs(response.body.to_s)
+            body = response.body.to_s
+            failure_details = process_specs(body)
             if failure_details.nil?
               @init_reason = EvaluationReason::NETWORK
             end
             tracker.end(success: @init_reason == EvaluationReason::NETWORK)
 
             unless response.body.nil? or @rules_updated_callback.nil?
-              @rules_updated_callback.call(response.body.to_s,
+              @rules_updated_callback.call(body,
                                            @last_config_sync_time)
             end
           end
@@ -401,7 +400,7 @@ module Statsig
 
     def get_id_lists_from_network(context)
       tracker = @diagnostics.track(context, 'get_id_list_sources', 'network_request')
-      response, e = @network.get_id_lists
+      response, e = @network.get_id_lists(context)
       code = response&.status.to_i
       if e.is_a? NetworkError
         code = e.http_code
@@ -413,9 +412,10 @@ module Statsig
       end
 
       begin
-        server_id_lists = JSON.parse(response)
+        body = response.body.to_s
+        server_id_lists = JSON.parse(body)
         process_id_lists(server_id_lists, context)
-        save_id_lists_to_adapter(response.body.to_s)
+        save_id_lists_to_adapter(body)
       rescue StandardError
         # Ignored, will try again
       end
@@ -509,14 +509,18 @@ module Statsig
 
     def download_single_id_list(list, context)
       nil unless list.is_a? IDList
-      http = HTTP.headers({ 'Range' => "bytes=#{list&.size || 0}-" }).accept(:json)
       tracker = @diagnostics.track(context, 'get_id_list', 'network_request', { url: list.url })
       begin
-        res = http.get(list.url)
-        tracker.end(statusCode: res.status.code, success: res.status.success?)
-        nil unless res.status.success?
+        res, e = @network.download_id_list(list.url, list&.size || 0)
+        code = res&.status&.to_i
+        if e.is_a? NetworkError
+          code = e.http_code
+        end
+        success = e.nil? && !res.nil?
+        tracker.end(statusCode: code, success: success)
+        return nil unless success
         content_length = Integer(res['content-length'])
-        nil if content_length.nil? || content_length <= 0
+        return nil if content_length.nil? || content_length <= 0
         content = res.body.to_s
         success = process_single_id_list(list, context, content, content_length)
         save_single_id_list_to_adapter(list.name, content) unless success.nil? || !success

data/lib/statsig.rb

@@ -410,7 +410,7 @@ module Statsig
   def self.get_statsig_metadata
     {
       'sdkType' => 'ruby-server',
-      'sdkVersion' => '2.8.3',
+      'sdkVersion' => '2.9.0',
       'languageVersion' => RUBY_VERSION
     }
   end

data/lib/statsig_driver.rb

@@ -291,8 +291,12 @@ class StatsigDriver
   def shutdown
     @err_boundary.capture(caller: __method__) do
       @shutdown = true
-      @logger.shutdown
-      @evaluator.shutdown
+      begin
+        @logger.shutdown
+        @evaluator.shutdown
+      ensure
+        @net.shutdown
+      end
     end
   end
 

data/lib/statsig_options.rb

@@ -92,6 +92,10 @@ class StatsigOptions
   # default: false
   attr_accessor :disable_evaluation_memoization
 
+  # Number of times to retry fetching rulesets and id lists on initialization.
+  # default: 0
+  attr_accessor :initialize_retry_limit
+
   def initialize(
     environment = nil,
     download_config_specs_url: nil,
@@ -115,7 +119,8 @@ class StatsigOptions
     post_logs_retry_limit: 3,
     post_logs_retry_backoff: nil,
     user_persistent_storage: nil,
-    disable_evaluation_memoization: false
+    disable_evaluation_memoization: false,
+    initialize_retry_limit: 0
   )
     @environment = environment.is_a?(Hash) ? environment : nil
 
@@ -146,5 +151,6 @@ class StatsigOptions
     @post_logs_retry_backoff = post_logs_retry_backoff
     @user_persistent_storage = user_persistent_storage
     @disable_evaluation_memoization = disable_evaluation_memoization
+    @initialize_retry_limit = initialize_retry_limit
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: statsig
 version: !ruby/object:Gem::Version
-  version: 2.8.3
+  version: 2.9.0
 platform: ruby
 authors:
 - Statsig, Inc
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-01-26 00:00:00.000000000 Z
+date: 2026-04-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler