Module: Mongoid::Threaded

# File 'lib/mongoid/threaded.rb', line 261

def client_override=(name)
  set(CLIENT_OVERRIDE_KEY, name)
end

# File 'lib/mongoid/threaded.rb', line 127

def database_override
  get(DATABASE_OVERRIDE_KEY)
end

# File 'lib/mongoid/threaded.rb', line 139

def database_override=(name)
  set(DATABASE_OVERRIDE_KEY, name)
end

# File 'lib/mongoid/threaded.rb', line 474

def execute_callbacks=(flag)
  set(EXECUTE_CALLBACKS, flag)
end

# File 'lib/mongoid/threaded.rb', line 460

def execute_callbacks?
  if has?(EXECUTE_CALLBACKS)
    get(EXECUTE_CALLBACKS)
  else
    true
  end
end

# File 'lib/mongoid/threaded.rb', line 334

def autosaved?(document)
  autosaves_for(document.class).include?(document._id)
end

# File 'lib/mongoid/threaded.rb', line 356

def autosaves
  get(AUTOSAVES_KEY) { {} }
end

# File 'lib/mongoid/threaded.rb', line 378

def autosaves_for(klass)
  autosaves[klass] ||= []
end

# File 'lib/mongoid/threaded.rb', line 185

def begin_autosave(document)
  autosaves_for(document.class).push(document._id)
end

# File 'lib/mongoid/threaded.rb', line 117

def begin_execution(name)
  stack(name).push(true)
end

# File 'lib/mongoid/threaded.rb', line 195

def begin_validate(document)
  validations_for(document.class).push(document._id)
end

# File 'lib/mongoid/threaded.rb', line 227

def begin_without_default_scope(klass)
  stack(:without_default_scope).push(klass)
end

# File 'lib/mongoid/threaded.rb', line 448

def clear_modified_documents(session)
  modified_documents.delete(session) || []
end

# File 'lib/mongoid/threaded.rb', line 425

def clear_session(client: nil)
  sessions.delete(client)&.end_session
end

# File 'lib/mongoid/threaded.rb', line 274

def current_scope(klass = nil)
  current_scope = get(CURRENT_SCOPE_KEY)

  if klass && current_scope.respond_to?(:keys)
    current_scope[current_scope.keys.find { |k| k <= klass }]
  else
    current_scope
  end
end

# File 'lib/mongoid/threaded.rb', line 292

def current_scope=(scope)
  set(CURRENT_SCOPE_KEY, scope)
end

# File 'lib/mongoid/threaded.rb', line 84

def delete(key)
  set(key, nil)
end

# File 'lib/mongoid/threaded.rb', line 151

def executing?(name)
  !stack(name).empty?
end

# File 'lib/mongoid/threaded.rb', line 205

def exit_autosave(document)
  autosaves_for(document.class).delete_one(document._id)
end

# File 'lib/mongoid/threaded.rb', line 163

def exit_execution(name)
  stack(name).pop
end

# File 'lib/mongoid/threaded.rb', line 215

def exit_validate(document)
  validations_for(document.class).delete_one(document._id)
end

# File 'lib/mongoid/threaded.rb', line 239

def exit_without_default_scope(klass)
  stack(:without_default_scope).delete(klass)
end

# File 'lib/mongoid/threaded.rb', line 59

def get(key, &default)
  result = Thread.current.thread_variable_get(key)

  if result.nil? && default
    result = yield
    set(key, result)
  end

  result
end

# File 'lib/mongoid/threaded.rb', line 413

def get_session(client: nil)
  sessions[client]
end

# File 'lib/mongoid/threaded.rb', line 93

def has?(key)
  # Here we have a classic example of JRuby not behaving like MRI. In
  # MRI, if you set a thread variable to nil, it removes it from the list
  # and subsequent calls to thread_variable?(key) will return false. Not
  # so with JRuby. Once set, you cannot unset the thread variable.
  #
  # However, because setting a variable to nil is supposed to remove it,
  # we can assume a nil-valued variable doesn't actually exist.

  # So, instead of this:
  # Thread.current.thread_variable?(key)

  # We have to do this:
  !get(key).nil?
end

# File 'lib/mongoid/threaded.rb', line 493

def modified_documents
  get(MODIFIED_DOCUMENTS_KEY) { Hash.new { |h, k| h[k] = Set.new } }
end

# File 'lib/mongoid/threaded.rb', line 483

def sessions
  get(SESSIONS_KEY) { {}.compare_by_identity }
end

# File 'lib/mongoid/threaded.rb', line 77

def set(key, value)
  Thread.current.thread_variable_set(key, value)
end

# File 'lib/mongoid/threaded.rb', line 305

def set_current_scope(scope, klass)
  if scope.nil?
    unset_current_scope(klass)
  else
    current_scope = get(CURRENT_SCOPE_KEY) { {} }
    current_scope[klass] = scope
  end
end

# File 'lib/mongoid/threaded.rb', line 401

def set_session(session, client: nil)
  sessions[client] = session
end

# File 'lib/mongoid/threaded.rb', line 175

def stack(name)
  get(STACK_KEYS[name]) { [] }
end

# File 'lib/mongoid/threaded.rb', line 503

def unset_current_scope(klass)
  return unless has?(CURRENT_SCOPE_KEY)

  scope = get(CURRENT_SCOPE_KEY)
  scope.delete(klass)

  delete(CURRENT_SCOPE_KEY) if scope.empty?
end

# File 'lib/mongoid/threaded.rb', line 346

def validated?(document)
  validations_for(document.class).include?(document._id)
end

# File 'lib/mongoid/threaded.rb', line 366

def validations
  get(VALIDATIONS_KEY) { {} }
end

# File 'lib/mongoid/threaded.rb', line 390

def validations_for(klass)
  validations[klass] ||= []
end

# File 'lib/mongoid/threaded.rb', line 322

def without_default_scope?(klass)
  stack(:without_default_scope).include?(klass)
end