Module: Mongoid::Threaded

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

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

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

def database_override
  get(DATABASE_OVERRIDE_KEY)
end

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

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

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

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

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

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

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

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

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

def autosaves
  get(AUTOSAVES_KEY) { {} }
end

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

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

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

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

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

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

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

def begin_touch_merged(document)
  touch_merged_for(document.class).push(document._id)
end

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

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

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

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

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

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

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

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

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

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 326

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

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

def delete(key)
  storage.delete(key)
end

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

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

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

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

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

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

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

def exit_touch_merged(document)
  touch_merged_for(document.class).delete_one(document._id)
end

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

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

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

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

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

def get(key, &default)
  result = storage[key]

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

  result
end

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

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

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

def has?(key)
  storage.key?(key)
end

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

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

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

def reset!
  case Config.real_isolation_level
  when :thread
    Thread.current.thread_variable_set(STORAGE_KEY, nil)
  when :fiber
    Fiber[STORAGE_KEY] = {}
    Fiber[STORAGE_OWNER_KEY] = Fiber.current.object_id
  else
    raise "Unknown isolation level: #{Config.real_isolation_level.inspect}"
  end
end

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

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

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

def set(key, value)
  storage[key] = value
end

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

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 470

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

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

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

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

def storage
  case Config.real_isolation_level
  when :thread
    storage_hash = Thread.current.thread_variable_get(STORAGE_KEY)

    unless storage_hash
      storage_hash = {}
      Thread.current.thread_variable_set(STORAGE_KEY, storage_hash)
    end

    storage_hash

  when :fiber
    # Fiber[] storage is inherited by child fibers, so multiple sibling
    # fibers would otherwise share a single mutable hash. We detect
    # inheritance by comparing the stored owner ID to the current fiber;
    # on first access by a new fiber we copy the inherited state so each
    # fiber starts with a snapshot of its parent's state rather than a
    # shared reference.
    if Fiber[STORAGE_OWNER_KEY] != Fiber.current.object_id
      Fiber[STORAGE_KEY] = (Fiber[STORAGE_KEY] || {}).dup
      Fiber[STORAGE_OWNER_KEY] = Fiber.current.object_id
    end
    Fiber[STORAGE_KEY]

  else
    raise "Unknown isolation level: #{Config.real_isolation_level.inspect}"
  end
end

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

def touch_merged
  get(TOUCH_MERGED_KEY) { {} }
end

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

def touch_merged?(document)
  touch_merged_for(document.class).include?(document._id)
end

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

def touch_merged_for(klass)
  touch_merged[klass] ||= []
end

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

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 380

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

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

def validations
  get(VALIDATIONS_KEY) { {} }
end

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

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

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

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