Move speculations pump scheduling logic to HTMLParserScheduler.

Source/core/html/parser/HTMLParserScheduler.h

 /*
  * 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 10 matching lines @@
  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #ifndef HTMLParserScheduler_h
 #define HTMLParserScheduler_h
 
 #include "core/html/parser/NestingLevelIncrementer.h"
 #include "platform/Timer.h"
-#include "wtf/CurrentTime.h"
 #include "wtf/PassOwnPtr.h"
 #include "wtf/RefPtr.h"
 
 namespace blink {
 
 class Document;
 class HTMLDocumentParser;
 
 class ActiveParserSession {
     STACK_ALLOCATED();
 public:
     explicit ActiveParserSession(Document*);
     ~ActiveParserSession();
 
 private:
     RefPtrWillBeMember<Document> m_document;
 };
 
 class PumpSession : public NestingLevelIncrementer, public ActiveParserSession {
     STACK_ALLOCATED();
 public:
     PumpSession(unsigned& nestingLevel, Document*);
     ~PumpSession();
 };
 
+class SpeculationsPumpSession : public ActiveParserSession {
+public:
+    explicit SpeculationsPumpSession(Document*);
+    ~SpeculationsPumpSession();
+
+    double elapsedTime() const;
+
+private:
+    double m_startTime;
+};
+
 class HTMLParserScheduler {
     WTF_MAKE_NONCOPYABLE(HTMLParserScheduler); WTF_MAKE_FAST_ALLOCATED;
 public:
     static PassOwnPtr<HTMLParserScheduler> create(HTMLDocumentParser* parser)
     {
         return adoptPtr(new HTMLParserScheduler(parser));
     }
     ~HTMLParserScheduler();
 
-    void scheduleForResume();
     bool isScheduledForResume() const { return m_isSuspendedWithActiveTimer || m_continueNextChunkTimer.isActive(); }
 
+    bool yieldIfNeeded(const SpeculationsPumpSession&);
+
     void suspend();
     void resume();
 
 private:
-    HTMLParserScheduler(HTMLDocumentParser*);
+    explicit HTMLParserScheduler(HTMLDocumentParser*);
 
+    bool shouldYield(const SpeculationsPumpSession&) const;
+    void scheduleForResume();
     void continueNextChunkTimerFired(Timer<HTMLParserScheduler>*);
 
     HTMLDocumentParser* m_parser;
 
     Timer<HTMLParserScheduler> m_continueNextChunkTimer;
     bool m_isSuspendedWithActiveTimer;
 };
 
 }
 
 #endif