| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "config.h" | |
| 6 #include "modules/serviceworkers/BodyStreamBuffer.h" | |
| 7 | |
| 8 #include "core/dom/DOMArrayBuffer.h" | |
| 9 | |
| 10 namespace blink { | |
| 11 | |
| 12 namespace { | |
| 13 | |
| 14 class BlobCreator final : public BodyStreamBuffer::Observer { | |
| 15 public: | |
| 16 BlobCreator(BodyStreamBuffer* buffer, const String& contentType, BodyStreamB
uffer::BlobHandleCreatorClient* client) | |
| 17 : m_buffer(buffer) | |
| 18 , m_client(client) | |
| 19 , m_blobData(BlobData::create()) | |
| 20 { | |
| 21 m_blobData->setContentType(contentType); | |
| 22 } | |
| 23 ~BlobCreator() override { } | |
| 24 void trace(Visitor* visitor) override | |
| 25 { | |
| 26 visitor->trace(m_buffer); | |
| 27 visitor->trace(m_client); | |
| 28 BodyStreamBuffer::Observer::trace(visitor); | |
| 29 } | |
| 30 void onWrite() override | |
| 31 { | |
| 32 ASSERT(m_buffer); | |
| 33 while (RefPtr<DOMArrayBuffer> buf = m_buffer->read()) { | |
| 34 m_blobData->appendBytes(buf->data(), buf->byteLength()); | |
| 35 } | |
| 36 } | |
| 37 void onClose() override | |
| 38 { | |
| 39 ASSERT(m_buffer); | |
| 40 const long long size = m_blobData->length(); | |
| 41 m_client->didCreateBlobHandle(BlobDataHandle::create(m_blobData.release(
), size)); | |
| 42 cleanup(); | |
| 43 } | |
| 44 void onError() override | |
| 45 { | |
| 46 ASSERT(m_buffer); | |
| 47 m_client->didFail(m_buffer->exception()); | |
| 48 cleanup(); | |
| 49 } | |
| 50 void start() | |
| 51 { | |
| 52 ASSERT(!m_buffer->isObserverRegistered()); | |
| 53 m_buffer->registerObserver(this); | |
| 54 onWrite(); | |
| 55 if (m_buffer->hasError()) { | |
| 56 return onError(); | |
| 57 } | |
| 58 if (m_buffer->isClosed()) | |
| 59 return onClose(); | |
| 60 } | |
| 61 void cleanup() | |
| 62 { | |
| 63 m_buffer->unregisterObserver(); | |
| 64 m_buffer.clear(); | |
| 65 m_client.clear(); | |
| 66 m_blobData.clear(); | |
| 67 } | |
| 68 private: | |
| 69 Member<BodyStreamBuffer> m_buffer; | |
| 70 Member<BodyStreamBuffer::BlobHandleCreatorClient> m_client; | |
| 71 OwnPtr<BlobData> m_blobData; | |
| 72 }; | |
| 73 | |
| 74 } // namespace | |
| 75 | |
| 76 PassRefPtr<DOMArrayBuffer> BodyStreamBuffer::read() | |
| 77 { | |
| 78 if (m_queue.isEmpty()) | |
| 79 return PassRefPtr<DOMArrayBuffer>(); | |
| 80 return m_queue.takeFirst(); | |
| 81 } | |
| 82 | |
| 83 void BodyStreamBuffer::write(PassRefPtr<DOMArrayBuffer> chunk) | |
| 84 { | |
| 85 ASSERT(!m_isClosed); | |
| 86 ASSERT(!m_exception); | |
| 87 ASSERT(chunk); | |
| 88 m_queue.append(chunk); | |
| 89 if (m_observer) | |
| 90 m_observer->onWrite(); | |
| 91 } | |
| 92 | |
| 93 void BodyStreamBuffer::close() | |
| 94 { | |
| 95 ASSERT(!m_isClosed); | |
| 96 ASSERT(!m_exception); | |
| 97 m_isClosed = true; | |
| 98 if (m_observer) | |
| 99 m_observer->onClose(); | |
| 100 } | |
| 101 | |
| 102 void BodyStreamBuffer::error(PassRefPtrWillBeRawPtr<DOMException> exception) | |
| 103 { | |
| 104 ASSERT(exception); | |
| 105 ASSERT(!m_isClosed); | |
| 106 ASSERT(!m_exception); | |
| 107 m_exception = exception; | |
| 108 if (m_observer) | |
| 109 m_observer->onError(); | |
| 110 } | |
| 111 | |
| 112 bool BodyStreamBuffer::readAllAndCreateBlobHandle(const String& contentType, Blo
bHandleCreatorClient* client) | |
| 113 { | |
| 114 if (m_observer) | |
| 115 return false; | |
| 116 BlobCreator* blobCreator = new BlobCreator(this, contentType, client); | |
| 117 blobCreator->start(); | |
| 118 return true; | |
| 119 } | |
| 120 | |
| 121 bool BodyStreamBuffer::registerObserver(Observer* observer) | |
| 122 { | |
| 123 if (m_observer) | |
| 124 return false; | |
| 125 ASSERT(observer); | |
| 126 m_observer = observer; | |
| 127 return true; | |
| 128 } | |
| 129 | |
| 130 void BodyStreamBuffer::unregisterObserver() | |
| 131 { | |
| 132 m_observer.clear(); | |
| 133 } | |
| 134 | |
| 135 void BodyStreamBuffer::trace(Visitor* visitor) | |
| 136 { | |
| 137 visitor->trace(m_exception); | |
| 138 visitor->trace(m_observer); | |
| 139 } | |
| 140 | |
| 141 BodyStreamBuffer::BodyStreamBuffer() | |
| 142 : m_isClosed(false) | |
| 143 { | |
| 144 } | |
| 145 | |
| 146 } // namespace blink | |
| OLD | NEW |