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

third_party/WebKit/Source/modules/fetch/CompositeDataConsumerHandle.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/CompositeDataConsumerHandle.h"
 
 #include "platform/CrossThreadFunctional.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebTaskRunner.h"
 #include "public/platform/WebThread.h"
@@ skipping 19 matching lines @@
 private:
     RefPtr<Context> m_context;
 };
 
 class CompositeDataConsumerHandle::Context final : public ThreadSafeRefCounted<Context> {
 public:
     using Token = unsigned;
     static PassRefPtr<Context> create(std::unique_ptr<WebDataConsumerHandle> handle) { return adoptRef(new Context(std::move(handle))); }
     ~Context()
     {
-        DCHECK(!m_readerThread);
         DCHECK(!m_reader);
         DCHECK(!m_client);
     }
     std::unique_ptr<ReaderImpl> obtainReader(Client* client)
     {
         MutexLocker locker(m_mutex);
-        DCHECK(!m_readerThread);
+        DCHECK(!m_readerTaskRunner);
         DCHECK(!m_reader);
         DCHECK(!m_client);
         ++m_token;
         m_client = client;
-        m_readerThread = Platform::current()->currentThread();
+        if (client)
+            m_readerTaskRunner = client->getTaskRunner()->clone();
+        else
+            m_readerTaskRunner = Platform::current()->currentThread()->getWebTaskRunner()->clone();
         m_reader = m_handle->obtainReader(m_client);
         return wrapUnique(new ReaderImpl(this));
     }
     void detachReader()
     {
         MutexLocker locker(m_mutex);
-        DCHECK(m_readerThread);
-        DCHECK(m_readerThread->isCurrentThread());
+        DCHECK(m_readerTaskRunner);
+        DCHECK(m_readerTaskRunner->runsTasksOnCurrentThread());
         DCHECK(m_reader);
         DCHECK(!m_isInTwoPhaseRead);
         DCHECK(!m_isUpdateWaitingForEndRead);
         ++m_token;
         m_reader = nullptr;
-        m_readerThread = nullptr;
+        m_readerTaskRunner = nullptr;
         m_client = nullptr;
     }
     void update(std::unique_ptr<WebDataConsumerHandle> handle)
     {
         MutexLocker locker(m_mutex);
         m_handle = std::move(handle);
-        if (!m_readerThread) {
+        if (!m_readerTaskRunner) {
             // There is no reader.
             return;
         }
         ++m_token;
         updateReaderNoLock(m_token);
     }
 
     Result read(void* data, size_t size, Flags flags, size_t* readSize)
     {
-        DCHECK(m_readerThread && m_readerThread->isCurrentThread());
+        DCHECK(m_readerTaskRunner);
+        DCHECK(m_readerTaskRunner->runsTasksOnCurrentThread());
         return m_reader->read(data, size, flags, readSize);
     }
     Result beginRead(const void** buffer, Flags flags, size_t* available)
     {
-        DCHECK(m_readerThread && m_readerThread->isCurrentThread());
+        DCHECK(m_readerTaskRunner);
+        DCHECK(m_readerTaskRunner->runsTasksOnCurrentThread());
         DCHECK(!m_isInTwoPhaseRead);
         Result r = m_reader->beginRead(buffer, flags, available);
         m_isInTwoPhaseRead = (r == Ok);
         return r;
     }
     Result endRead(size_t readSize)
     {
-        DCHECK(m_readerThread && m_readerThread->isCurrentThread());
+        DCHECK(m_readerTaskRunner);
+        DCHECK(m_readerTaskRunner->runsTasksOnCurrentThread());
         DCHECK(m_isInTwoPhaseRead);
         Result r = m_reader->endRead(readSize);
         m_isInTwoPhaseRead = false;
         if (m_isUpdateWaitingForEndRead) {
             // We need this lock to access |m_handle|.
             MutexLocker locker(m_mutex);
             m_reader = nullptr;
             m_reader = m_handle->obtainReader(m_client);
             m_isUpdateWaitingForEndRead = false;
         }
         return r;
     }
 
 private:
-    explicit Context(std::unique_ptr<WebDataConsumerHandle> handle)
+    Context(std::unique_ptr<WebDataConsumerHandle> handle)
         : m_handle(std::move(handle))
-        , m_readerThread(nullptr)
+        , m_readerTaskRunner(nullptr)
         , m_client(nullptr)
         , m_token(0)
         , m_isUpdateWaitingForEndRead(false)
         , m_isInTwoPhaseRead(false)
     {
     }
     void updateReader(Token token)
     {
         MutexLocker locker(m_mutex);
         updateReaderNoLock(token);
     }
     void updateReaderNoLock(Token token)
     {
         if (token != m_token) {
             // This request is not fresh. Ignore it.
             return;
         }
-        DCHECK(m_readerThread);
+        DCHECK(m_readerTaskRunner);
         DCHECK(m_reader);
-        if (m_readerThread->isCurrentThread()) {
+        if (m_readerTaskRunner->runsTasksOnCurrentThread()) {
             if (m_isInTwoPhaseRead) {
                 // We are waiting for the two-phase read completion.
                 m_isUpdateWaitingForEndRead = true;
                 return;
             }
             // Unregister the old one, then register the new one.
             m_reader = nullptr;
             m_reader = m_handle->obtainReader(m_client);
             return;
         }
         ++m_token;
-        m_readerThread->getWebTaskRunner()->postTask(BLINK_FROM_HERE, crossThreadBind(&Context::updateReader, wrapPassRefPtr(this), m_token));
+        m_readerTaskRunner->postTask(BLINK_FROM_HERE, crossThreadBind(&Context::updateReader, wrapPassRefPtr(this), m_token));
     }
 
     std::unique_ptr<Reader> m_reader;
     std::unique_ptr<WebDataConsumerHandle> m_handle;
-    // 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;
+    std::unique_ptr<WebTaskRunner> m_readerTaskRunner;
     Client* m_client;
     Token m_token;
     // These boolean values are bound to the reader thread.
     bool m_isUpdateWaitingForEndRead;
     bool m_isInTwoPhaseRead;
     Mutex m_mutex;
 };
 
 CompositeDataConsumerHandle::ReaderImpl::ReaderImpl(PassRefPtr<Context> context) : m_context(context) { }
 
@@ skipping 43 matching lines @@
 }
 
 CompositeDataConsumerHandle::~CompositeDataConsumerHandle() { }
 
 std::unique_ptr<WebDataConsumerHandle::Reader> CompositeDataConsumerHandle::obtainReader(Client* client)
 {
     return m_context->obtainReader(client);
 }
 
 } // namespace blink