Sha256: 3c49d35a55c1d44093d203329107c355984b322c92b92cfd1086015416ef26dd
Contents?: true
Size: 1.53 KB
Versions: 37
Compression:
Stored size: 1.53 KB
Contents
# frozen_string_literal: true
# encoding: utf-8
require 'singleton'
module Mrss
def self.patch_mongo_for_session_registry
Mongo::Client.class_eval do
alias :get_session_without_tracking :get_session
def get_session(options = {})
get_session_without_tracking(options).tap do |session|
SessionRegistry.instance.register(session) if session&.materialized?
end
end
end
Mongo::Session.class_eval do
alias :end_session_without_tracking :end_session
def end_session
SessionRegistry.instance.unregister(self)
end_session_without_tracking
end
alias :materialize_if_needed_without_tracking :materialize_if_needed
def materialize_if_needed
materialize_if_needed_without_tracking.tap do
SessionRegistry.instance.register(self)
end
end
end
end
end
module Mrss
class SessionRegistry
include Singleton
def initialize
@registry = {}
end
def register(session)
@registry[session.session_id] = session if session
end
def unregister(session)
return if session.ended? || !session.materialized?
@registry.delete(session.session_id)
end
def verify_sessions_ended!
@registry.delete_if { |_, session| session.ended? }
unless @registry.empty?
sessions = @registry.map { |_, session| session }
raise "Session registry contains live sessions: #{sessions.join(', ')}"
end
end
def clear_registry
@registry = {}
end
end
end
Version data entries
37 entries across 37 versions & 4 rubygems