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 |