OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/leveldb/leveldb_service_impl.h" | 5 #include "components/leveldb/leveldb_service_impl.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
10 #include "components/leveldb/env_mojo.h" | 10 #include "components/leveldb/env_mojo.h" |
11 #include "components/leveldb/leveldb_database_impl.h" | 11 #include "components/leveldb/leveldb_database_impl.h" |
12 #include "components/leveldb/public/cpp/util.h" | 12 #include "components/leveldb/public/cpp/util.h" |
13 #include "mojo/public/cpp/bindings/strong_associated_binding.h" | 13 #include "mojo/public/cpp/bindings/strong_associated_binding.h" |
14 #include "third_party/leveldatabase/env_chromium.h" | 14 #include "third_party/leveldatabase/env_chromium.h" |
15 #include "third_party/leveldatabase/src/helpers/memenv/memenv.h" | 15 #include "third_party/leveldatabase/src/helpers/memenv/memenv.h" |
16 #include "third_party/leveldatabase/src/include/leveldb/db.h" | 16 #include "third_party/leveldatabase/src/include/leveldb/db.h" |
17 #include "third_party/leveldatabase/src/include/leveldb/env.h" | 17 #include "third_party/leveldatabase/src/include/leveldb/env.h" |
18 #include "third_party/leveldatabase/src/include/leveldb/filter_policy.h" | 18 #include "third_party/leveldatabase/src/include/leveldb/filter_policy.h" |
19 #include "third_party/leveldatabase/src/include/leveldb/slice.h" | 19 #include "third_party/leveldatabase/src/include/leveldb/slice.h" |
20 | 20 |
21 namespace leveldb { | 21 namespace leveldb { |
22 | 22 |
23 LevelDBServiceImpl::LevelDBServiceImpl( | 23 LevelDBServiceImpl::LevelDBServiceImpl(MojoEnv* env) : env_(env) {} |
24 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner) | |
25 : thread_(new LevelDBMojoProxy(std::move(file_task_runner))), | |
26 environment_name_("LevelDBEnv") {} | |
27 | 24 |
28 LevelDBServiceImpl::~LevelDBServiceImpl() {} | 25 LevelDBServiceImpl::~LevelDBServiceImpl() {} |
29 | 26 |
30 void LevelDBServiceImpl::SetEnvironmentName(const std::string& name) { | |
31 environment_name_ = name; | |
32 } | |
33 | |
34 void LevelDBServiceImpl::Open( | 27 void LevelDBServiceImpl::Open( |
35 filesystem::mojom::DirectoryPtr directory, | 28 filesystem::mojom::DirectoryPtr directory, |
36 const std::string& dbname, | 29 const std::string& dbname, |
37 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, | 30 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, |
38 const OpenCallback& callback) { | 31 const OpenCallback& callback) { |
39 OpenWithOptions(leveldb::mojom::OpenOptions::New(), std::move(directory), | 32 OpenWithOptions(leveldb::mojom::OpenOptions::New(), std::move(directory), |
40 dbname, std::move(database), callback); | 33 dbname, std::move(database), callback); |
41 } | 34 } |
42 | 35 |
43 void LevelDBServiceImpl::OpenWithOptions( | 36 void LevelDBServiceImpl::OpenWithOptions( |
44 leveldb::mojom::OpenOptionsPtr open_options, | 37 leveldb::mojom::OpenOptionsPtr open_options, |
45 filesystem::mojom::DirectoryPtr directory, | 38 filesystem::mojom::DirectoryPtr directory, |
46 const std::string& dbname, | 39 const std::string& dbname, |
47 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, | 40 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, |
48 const OpenCallback& callback) { | 41 const OpenCallback& callback) { |
49 leveldb::Options options; | 42 leveldb::Options options; |
50 options.create_if_missing = open_options->create_if_missing; | 43 options.create_if_missing = open_options->create_if_missing; |
51 options.error_if_exists = open_options->error_if_exists; | 44 options.error_if_exists = open_options->error_if_exists; |
52 options.paranoid_checks = open_options->paranoid_checks; | 45 options.paranoid_checks = open_options->paranoid_checks; |
53 options.write_buffer_size = open_options->write_buffer_size; | 46 options.write_buffer_size = open_options->write_buffer_size; |
54 options.max_open_files = open_options->max_open_files; | 47 options.max_open_files = open_options->max_open_files; |
55 | 48 |
56 options.reuse_logs = leveldb_env::kDefaultLogReuseOptionValue; | 49 options.reuse_logs = leveldb_env::kDefaultLogReuseOptionValue; |
57 options.compression = leveldb::kSnappyCompression; | 50 options.compression = leveldb::kSnappyCompression; |
58 | 51 |
59 // Register our directory with the file thread. | 52 std::unique_ptr<MojoEnv::Directory> dir = |
60 LevelDBMojoProxy::OpaqueDir* dir = | 53 env_->RegisterDirectory(std::move(directory)); |
61 thread_->RegisterDirectory(std::move(directory)); | 54 options.env = env_; |
62 | |
63 std::unique_ptr<MojoEnv> env_mojo( | |
64 new MojoEnv(environment_name_, thread_, dir)); | |
65 options.env = env_mojo.get(); | |
66 | 55 |
67 leveldb::DB* db = nullptr; | 56 leveldb::DB* db = nullptr; |
68 leveldb::Status s = leveldb::DB::Open(options, dbname, &db); | 57 leveldb::Status s = leveldb::DB::Open(options, dir->prefix() + dbname, &db); |
69 | 58 |
70 if (s.ok()) { | 59 if (s.ok()) { |
71 mojo::MakeStrongAssociatedBinding( | 60 mojo::MakeStrongAssociatedBinding(base::MakeUnique<LevelDBDatabaseImpl>( |
72 base::MakeUnique<LevelDBDatabaseImpl>(std::move(env_mojo), | 61 std::move(dir), base::WrapUnique(db)), |
73 base::WrapUnique(db)), | 62 std::move(database)); |
74 std::move(database)); | |
75 } | 63 } |
76 | 64 |
77 callback.Run(LeveldbStatusToError(s)); | 65 callback.Run(LeveldbStatusToError(s)); |
78 } | 66 } |
79 | 67 |
80 void LevelDBServiceImpl::OpenInMemory( | 68 void LevelDBServiceImpl::OpenInMemory( |
81 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, | 69 leveldb::mojom::LevelDBDatabaseAssociatedRequest database, |
82 const OpenCallback& callback) { | 70 const OpenCallback& callback) { |
83 leveldb::Options options; | 71 leveldb::Options options; |
84 options.create_if_missing = true; | 72 options.create_if_missing = true; |
(...skipping 12 matching lines...) Expand all Loading... |
97 std::move(database)); | 85 std::move(database)); |
98 } | 86 } |
99 | 87 |
100 callback.Run(LeveldbStatusToError(s)); | 88 callback.Run(LeveldbStatusToError(s)); |
101 } | 89 } |
102 | 90 |
103 void LevelDBServiceImpl::Destroy(filesystem::mojom::DirectoryPtr directory, | 91 void LevelDBServiceImpl::Destroy(filesystem::mojom::DirectoryPtr directory, |
104 const std::string& dbname, | 92 const std::string& dbname, |
105 const DestroyCallback& callback) { | 93 const DestroyCallback& callback) { |
106 leveldb::Options options; | 94 leveldb::Options options; |
107 // Register our directory with the file thread. | 95 std::unique_ptr<MojoEnv::Directory> dir = |
108 LevelDBMojoProxy::OpaqueDir* dir = | 96 env_->RegisterDirectory(std::move(directory)); |
109 thread_->RegisterDirectory(std::move(directory)); | 97 options.env = env_; |
110 std::unique_ptr<MojoEnv> env_mojo( | 98 callback.Run(LeveldbStatusToError( |
111 new MojoEnv(environment_name_, thread_, dir)); | 99 leveldb::DestroyDB(dir->prefix() + dbname, options))); |
112 options.env = env_mojo.get(); | |
113 callback.Run(LeveldbStatusToError(leveldb::DestroyDB(dbname, options))); | |
114 } | 100 } |
115 | 101 |
116 } // namespace leveldb | 102 } // namespace leveldb |
OLD | NEW |