class ActiveRecord::AsynchronousQueriesTracker::Session

def active?
  @active.true?
end

def finalize(wait = false)
  @active.make_false
  if wait
    # Wait until all thread with a read lock are done
    @lock.with_write_lock { }
  end
end

def initialize
  @active = Concurrent::AtomicBoolean.new(true)
  @lock = Concurrent::ReadWriteLock.new
end

def synchronize(&block)
  @lock.with_read_lock(&block)
end