1 // Copyright 2017 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 "components/download/internal/background_service/blob_task_proxy.h"
6
7 #include <utility>
8
9 #include "base/bind.h"
10 #include "base/guid.h"
11 #include "base/task_runner_util.h"
12 #include "base/threading/thread_task_runner_handle.h"
13 #include "storage/browser/blob/blob_data_builder.h"
14 #include "storage/browser/blob/blob_data_handle.h"
15 #include "storage/browser/blob/blob_storage_context.h"
16
17 namespace download {
18
19 // static
Create(BlobContextGetter blob_context_getter,scoped_refptr<base::SingleThreadTaskRunner> io_task_runner)20 std::unique_ptr<BlobTaskProxy> BlobTaskProxy::Create(
21 BlobContextGetter blob_context_getter,
22 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner) {
23 return std::make_unique<BlobTaskProxy>(blob_context_getter, io_task_runner);
24 }
25
BlobTaskProxy(BlobContextGetter blob_context_getter,scoped_refptr<base::SingleThreadTaskRunner> io_task_runner)26 BlobTaskProxy::BlobTaskProxy(
27 BlobContextGetter blob_context_getter,
28 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner)
29 : main_task_runner_(base::ThreadTaskRunnerHandle::Get()),
30 io_task_runner_(io_task_runner) {
31 // Unretained the raw pointer because owner on UI thread should destroy this
32 // object on IO thread.
33 io_task_runner_->PostTask(
34 FROM_HERE, base::BindOnce(&BlobTaskProxy::InitializeOnIO,
35 base::Unretained(this), blob_context_getter));
36 }
37
~BlobTaskProxy()38 BlobTaskProxy::~BlobTaskProxy() {
39 CHECK(io_task_runner_->BelongsToCurrentThread());
40 }
41
InitializeOnIO(BlobContextGetter blob_context_getter)42 void BlobTaskProxy::InitializeOnIO(BlobContextGetter blob_context_getter) {
43 DCHECK(io_task_runner_->BelongsToCurrentThread());
44
45 blob_storage_context_ = blob_context_getter.Run();
46 }
47
SaveAsBlob(std::unique_ptr<std::string> data,BlobDataHandleCallback callback)48 void BlobTaskProxy::SaveAsBlob(std::unique_ptr<std::string> data,
49 BlobDataHandleCallback callback) {
50 // Unretained the raw pointer because owner on UI thread should destroy this
51 // object on IO thread.
52 io_task_runner_->PostTask(
53 FROM_HERE,
54 base::BindOnce(&BlobTaskProxy::SaveAsBlobOnIO, base::Unretained(this),
55 std::move(data), std::move(callback)));
56 }
57
SaveAsBlobOnIO(std::unique_ptr<std::string> data,BlobDataHandleCallback callback)58 void BlobTaskProxy::SaveAsBlobOnIO(std::unique_ptr<std::string> data,
59 BlobDataHandleCallback callback) {
60 DCHECK(io_task_runner_->BelongsToCurrentThread());
61
62 // Build blob data. This has to do a copy into blob's internal storage.
63 std::string blob_uuid = base::GenerateGUID();
64 auto builder = std::make_unique<storage::BlobDataBuilder>(blob_uuid);
65 builder->AppendData(*data);
66 blob_data_handle_ =
67 blob_storage_context_->AddFinishedBlob(std::move(builder));
68
69 // Wait for blob data construction complete.
70 auto cb = base::BindOnce(&BlobTaskProxy::BlobSavedOnIO,
71 weak_ptr_factory_.GetWeakPtr(), std::move(callback));
72 blob_data_handle_->RunOnConstructionComplete(std::move(cb));
73 }
74
BlobSavedOnIO(BlobDataHandleCallback callback,storage::BlobStatus status)75 void BlobTaskProxy::BlobSavedOnIO(BlobDataHandleCallback callback,
76 storage::BlobStatus status) {
77 DCHECK(io_task_runner_->BelongsToCurrentThread());
78
79 // Relay BlobDataHandle and |status| back to main thread.
80 auto cb =
81 base::BindOnce(std::move(callback), std::move(blob_data_handle_), status);
82 main_task_runner_->PostTask(FROM_HERE, std::move(cb));
83 }
84
85 } // namespace download
86