Make HTMLParserScheduler inherit from ActiveDOMObject

third_party/WebKit/Source/core/html/parser/HTMLParserScheduler.cpp

 /*
  * Copyright (C) 2010 Google, Inc. All Rights Reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 38 matching lines @@
 SpeculationsPumpSession::~SpeculationsPumpSession() {}
 
 inline double SpeculationsPumpSession::elapsedTime() const {
   return currentTime() - m_startTime;
 }
 
 void SpeculationsPumpSession::addedElementTokens(size_t count) {
   m_processedElementTokens += count;
 }
 
-HTMLParserScheduler::HTMLParserScheduler(HTMLDocumentParser* parser,
+HTMLParserScheduler* HTMLParserScheduler::create(
+    Document& document,
+    HTMLDocumentParser* parser,
+    WebTaskRunner* loadingTaskRunner) {
+  return new HTMLParserScheduler(document, parser, loadingTaskRunner);
+}
+
+HTMLParserScheduler::HTMLParserScheduler(Document& document,
+                                         HTMLDocumentParser* parser,
                                          WebTaskRunner* loadingTaskRunner)
-    : m_parser(parser),
+    : ActiveDOMObject(&document),
+      m_parser(parser),
       m_loadingTaskRunner(loadingTaskRunner->clone()),
       m_isSuspendedWithActiveTimer(false) {}
 
 HTMLParserScheduler::~HTMLParserScheduler() {}
 
 DEFINE_TRACE(HTMLParserScheduler) {
   visitor->trace(m_parser);
+  ActiveDOMObject::trace(visitor);
 }
 
 bool HTMLParserScheduler::isScheduledForResume() const {
   return m_isSuspendedWithActiveTimer ||
          m_cancellableContinueParseTaskHandle.isActive();
 }
 
 void HTMLParserScheduler::scheduleForResume() {
+  if (getExecutionContext()->activeDOMObjectsAreSuspended()) {
+    DCHECK(!m_cancellableContinueParseTaskHandle.isActive());
+    m_isSuspendedWithActiveTimer = true;
+    return;
+  }
   DCHECK(!m_isSuspendedWithActiveTimer);
   m_cancellableContinueParseTaskHandle =
       m_loadingTaskRunner->postCancellableTask(
           BLINK_FROM_HERE, WTF::bind(&HTMLParserScheduler::continueParsing,
                                      wrapWeakPersistent(this)));
 }
 
 void HTMLParserScheduler::suspend() {
   DCHECK(!m_isSuspendedWithActiveTimer);
   if (!m_cancellableContinueParseTaskHandle.isActive())
@@ skipping 46 matching lines @@
 bool HTMLParserScheduler::yieldIfNeeded(const SpeculationsPumpSession& session,
                                         bool startingScript) {
   if (shouldYield(session, startingScript)) {
     scheduleForResume();
     return true;
   }
 
   return false;
 }
 
-void HTMLParserScheduler::forceResumeAfterYield() {
-  DCHECK(!m_cancellableContinueParseTaskHandle.isActive());
-  m_isSuspendedWithActiveTimer = true;
-}
-
 void HTMLParserScheduler::continueParsing() {
   m_parser->resumeParsingAfterYield();
 }
 
 }  // namespace blink