1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67


content / browser / worker_host / dedicated_worker_hosts_for_document.h [blame]

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_HOSTS_FOR_DOCUMENT_H_
#define CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_HOSTS_FOR_DOCUMENT_H_

#include "base/containers/flat_set.h"
#include "base/memory/safe_ref.h"
#include "content/browser/worker_host/dedicated_worker_host.h"
#include "content/common/content_export.h"
#include "content/public/browser/document_user_data.h"
#include "third_party/blink/public/common/scheduler/web_scheduler_tracked_feature.h"

namespace content {

// Manages the set of dedicated workers whose ancestor is this document. This
// class is exported for testing.
class CONTENT_EXPORT DedicatedWorkerHostsForDocument
    : public DocumentUserData<DedicatedWorkerHostsForDocument> {
 public:
  ~DedicatedWorkerHostsForDocument() override;

  // Adds an associated dedicated worker to this document.
  void Add(base::SafeRef<DedicatedWorkerHost> dedicated_worker_host);

  // Removes an associated dedicated worker to this document. If the given
  // `dedicated_worker_host` is not associated to this document, Remove is a no-
  // op.
  void Remove(base::SafeRef<DedicatedWorkerHost> dedicated_worker_host);

  // Returns the union of the feature sets that disable back-forward cache.
  blink::scheduler::WebSchedulerTrackedFeatures
  GetBackForwardCacheDisablingFeatures() const;

  DedicatedWorkerHost::BackForwardCacheBlockingDetails
  GetBackForwardCacheBlockingDetails() const;

  // Called when the page is put into back/forward cache.
  void OnEnterBackForwardCache();

  // Called when the page is restored from back/forward cache.
  void OnRestoreFromBackForwardCache();

  // Called when subresource loader factories should be updated,
  // for example when DevTools interception is enabled/disabled.
  void UpdateSubresourceLoaderFactories();

 private:
  explicit DedicatedWorkerHostsForDocument(RenderFrameHost* rfh);

  using DedicatedWorkerHostComparator =
      bool (*)(const base::SafeRef<DedicatedWorkerHost>&,
               const base::SafeRef<DedicatedWorkerHost>&);

  // A set of dedicated workers whose ancestor is this document.
  base::flat_set<base::SafeRef<DedicatedWorkerHost>,
                 DedicatedWorkerHostComparator>
      dedicated_workers_;

  friend DocumentUserData;
  DOCUMENT_USER_DATA_KEY_DECL();
};

}  // namespace content

#endif  // CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_HOSTS_FOR_DOCUMENT_H_