Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(159)

Side by Side Diff: content/browser/loader/stream_writer.cc

Issue 625993002: Split up streams logic to prepare for PlzNavigate RDH changes. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: zork comments Created 6 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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 "content/browser/loader/stream_writer.h"
6
7 #include "base/guid.h"
8 #include "content/browser/streams/stream.h"
9 #include "content/browser/streams/stream_registry.h"
10 #include "content/public/browser/resource_controller.h"
11 #include "net/base/io_buffer.h"
12 #include "url/gurl.h"
13 #include "url/url_constants.h"
14
15 namespace content {
16
17 StreamWriter::StreamWriter() : controller_(nullptr) {
18 }
19
20 StreamWriter::~StreamWriter() {
21 if (stream_.get())
22 Finalize();
23 }
24
25 void StreamWriter::InitializeStream(StreamRegistry* registry,
26 const GURL& origin) {
27 DCHECK(!stream_.get());
28
29 // TODO(tyoshino): Find a way to share this with the blob URL creation in
30 // WebKit.
31 GURL url(std::string(url::kBlobScheme) + ":" + origin.spec() +
32 base::GenerateGUID());
33 stream_ = new Stream(registry, this, url);
34 }
35
36 void StreamWriter::OnWillRead(scoped_refptr<net::IOBuffer>* buf,
37 int* buf_size,
38 int min_size) {
39 static const int kReadBufSize = 32768;
40
41 DCHECK(buf);
mmenke 2014/10/15 20:05:44 include base/logging.h for DCHECK
42 DCHECK(buf_size);
43 DCHECK_LE(min_size, kReadBufSize);
44 if (!read_buffer_.get())
45 read_buffer_ = new net::IOBuffer(kReadBufSize);
46 *buf = read_buffer_.get();
47 *buf_size = kReadBufSize;
48 }
49
50 void StreamWriter::OnReadCompleted(int bytes_read, bool* defer) {
51 if (!bytes_read)
52 return;
53
54 // We have more data to read.
mmenke 2014/10/15 20:05:44 While you're here, shouldn't use "we" in comments.
55 DCHECK(read_buffer_.get());
56
57 // Release the ownership of the buffer, and store a reference
58 // to it. A new one will be allocated in OnWillRead().
59 scoped_refptr<net::IOBuffer> buffer;
60 read_buffer_.swap(buffer);
61 stream_->AddData(buffer, bytes_read);
62
63 if (!stream_->can_add_data())
64 *defer = true;
65 }
66
67 void StreamWriter::Finalize() {
68 DCHECK(stream_.get());
69 stream_->Finalize();
70 stream_->RemoveWriteObserver(this);
71 stream_ = nullptr;
72 }
73
74 void StreamWriter::OnSpaceAvailable(Stream* stream) {
75 controller_->Resume();
76 }
77
78 void StreamWriter::OnClose(Stream* stream) {
79 controller_->Cancel();
80 }
81
82 } // namespace content
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698