Use per-frame TaskRunner instead of thread's default in DataConsumerHandle

third_party/WebKit/Source/modules/fetch/DataConsumerTee.cpp

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "modules/fetch/DataConsumerTee.h"
 
 #include "core/dom/ActiveDOMObject.h"
 #include "core/dom/ExecutionContext.h"
+#include "core/dom/TaskRunnerHelper.h"
 #include "modules/fetch/DataConsumerHandleUtil.h"
 #include "modules/fetch/FetchBlobDataConsumerHandle.h"
 #include "platform/CrossThreadFunctional.h"
 #include "platform/heap/Handle.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebTaskRunner.h"
 #include "public/platform/WebThread.h"
 #include "public/platform/WebTraceLocation.h"
 #include "wtf/Deque.h"
 #include "wtf/Functional.h"
@@ skipping 157 matching lines @@
     }
 
     void notify()
     {
         {
             MutexLocker locker(m_mutex);
             if (!m_client) {
                 // No client is registered.
                 return;
             }
-            DCHECK(m_readerThread);
-            if (!m_readerThread->isCurrentThread()) {
-                m_readerThread->getWebTaskRunner()->postTask(BLINK_FROM_HERE, crossThreadBind(&DestinationContext::notify, wrapPassRefPtr(this)));
+            if (!m_client->getTaskRunner()->runsTasksOnCurrentThread()) {
+                m_client->getTaskRunner()->postTask(BLINK_FROM_HERE, crossThreadBind(&DestinationContext::notify, wrapPassRefPtr(this)));
                 return;
             }
         }
         // The reading thread is the current thread.
         if (m_client)
             m_client->didGetReadable();
     }
 
     Mutex& mutex() { return m_mutex; }
 
     // The following functions don't use lock. They should be protected by the
     // caller.
     void attachReader(WebDataConsumerHandle::Client* client)
     {
-        DCHECK(!m_readerThread);
         DCHECK(!m_client);
-        m_readerThread = Platform::current()->currentThread();
         m_client = client;
     }
     void detachReader()
     {
-        DCHECK(m_readerThread && m_readerThread->isCurrentThread());
-        m_readerThread = nullptr;
+        DCHECK(!m_client || m_client->getTaskRunner()->runsTasksOnCurrentThread());
         m_client = nullptr;
     }
     const std::unique_ptr<Vector<char>>& top() const { return m_queue.first(); }
     bool isEmpty() const { return m_queue.isEmpty(); }
     size_t offset() const { return m_offset; }
     void consume(size_t size)
     {
         const auto& top = m_queue.first();
         DCHECK(m_offset <= m_offset + size);
         DCHECK(m_offset + size <= top->size());
         if (top->size() <= m_offset + size) {
             m_offset = 0;
             m_queue.removeFirst();
         } else {
             m_offset += size;
         }
     }
     Result getResult() { return m_result; }
 
 private:
     DestinationContext()
         : m_result(WebDataConsumerHandle::ShouldWait)
-        , m_readerThread(nullptr)
         , m_client(nullptr)
         , m_offset(0)
         , m_isTwoPhaseReadInProgress(false)
     {
     }
 
     void detach()
     {
         MutexLocker locker(m_mutex);
         DCHECK(!m_client);
-        DCHECK(!m_readerThread);
         m_queue.clear();
     }
 
     Result m_result;
     Deque<std::unique_ptr<Vector<char>>> m_queue;
-    // Note: Holding a WebThread raw pointer is not generally safe, but we can
-    // do that in this case because:
-    //  1. Destructing a ReaderImpl when the bound thread ends is a user's
-    //     responsibility.
-    //  2. |m_readerThread| will never be used after the associated reader is
-    //     detached.
-    WebThread* m_readerThread;
     WebDataConsumerHandle::Client* m_client;
     size_t m_offset;
     bool m_isTwoPhaseReadInProgress;
     Mutex m_mutex;
 };
 
 class DestinationReader final : public WebDataConsumerHandle::Reader {
 public:
     DestinationReader(PassRefPtr<DestinationContext::Proxy> contextProxy, WebDataConsumerHandle::Client* client)
         : m_contextProxy(contextProxy)
     {
         MutexLocker locker(context()->mutex());
         context()->attachReader(client);
         if (client) {
-            // We need to use crossThreadBind here to retain the context. Note
-            // |context()| return value is of type DestinationContext*, not
-            // PassRefPtr<DestinationContext>.
-            Platform::current()->currentThread()->getWebTaskRunner()->postTask(BLINK_FROM_HERE, crossThreadBind(&DestinationContext::notify, wrapPassRefPtr(context())));
+            client->getTaskRunner()->postTask(BLINK_FROM_HERE, bind(&DestinationContext::notify, wrapPassRefPtr(context())));
         }
     }
     ~DestinationReader() override
     {
         MutexLocker locker(context()->mutex());
         context()->detachReader();
     }
 
     Result beginRead(const void** buffer, Flags, size_t* available) override
     {
@@ skipping 80 matching lines @@
                 break;
             m_dest1->enqueue(static_cast<const char*>(buffer), available);
             m_dest2->enqueue(static_cast<const char*>(buffer), available);
             m_reader->endRead(available);
         }
         m_dest1->setResult(r);
         m_dest2->setResult(r);
         stopInternal();
     }
 
+    WebTaskRunner* getTaskRunner() override
+    {
+        return TaskRunnerHelper::getUnthrottledTaskRunner(getExecutionContext());
+    }
+
     void stop() override
     {
         stopInternal();
         ActiveDOMObject::stop();
     }
 
     DEFINE_INLINE_VIRTUAL_TRACE()
     {
         ActiveDOMObject::trace(visitor);
     }
@@ skipping 44 matching lines @@
     }
 
     std::unique_ptr<WebDataConsumerHandle> webDest1, webDest2;
     DataConsumerTee::create(executionContext, static_cast<std::unique_ptr<WebDataConsumerHandle>>(std::move(src)), &webDest1, &webDest2);
     *dest1 = createFetchDataConsumerHandleFromWebHandle(std::move(webDest1));
     *dest2 = createFetchDataConsumerHandleFromWebHandle(std::move(webDest2));
     return;
 }
 
 } // namespace blink