Index: services/user/user_shell_client.cc |
diff --git a/services/user/user_shell_client.cc b/services/user/user_shell_client.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..06bb96bbfc17aa5f510225dfe06c89dddd556edf |
--- /dev/null |
+++ b/services/user/user_shell_client.cc |
@@ -0,0 +1,125 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "services/user/user_shell_client.h" |
+ |
+#include "base/bind.h" |
+#include "base/memory/weak_ptr.h" |
+#include "components/filesystem/lock_table.h" |
+#include "components/leveldb/leveldb_service_impl.h" |
+#include "mojo/public/cpp/bindings/callback.h" |
+#include "services/shell/public/cpp/connection.h" |
+#include "services/user/user_id_map.h" |
+#include "services/user/user_service.h" |
+ |
+namespace user_service { |
+ |
+class UserShellClient::UserServiceObjects |
+ : public base::SupportsWeakPtr<UserServiceObjects> { |
+ public: |
+ // Created on the main thread. |
+ UserServiceObjects(base::FilePath user_dir) : user_dir_(user_dir) {} |
+ |
+ // Destroyed on the |user_service_runner_|. |
+ ~UserServiceObjects() {} |
+ |
+ // Called on the |user_service_runner_|. |
+ void OnUserServiceRequest(mojo::Connection* connection, |
+ mojom::UserServiceRequest request) { |
+ if (!lock_table_) |
+ lock_table_ = new filesystem::LockTable; |
+ user_service_bindings_.AddBinding(new UserService(user_dir_, lock_table_), |
+ std::move(request)); |
+ } |
+ |
+ private: |
+ mojo::BindingSet<mojom::UserService> user_service_bindings_; |
+ scoped_refptr<filesystem::LockTable> lock_table_; |
+ base::FilePath user_dir_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(UserServiceObjects); |
+}; |
+ |
+class UserShellClient::LevelDBServiceObjects |
+ : public base::SupportsWeakPtr<LevelDBServiceObjects> { |
+ public: |
+ // Created on the main thread. |
+ LevelDBServiceObjects(scoped_refptr<base::SingleThreadTaskRunner> task_runner) |
+ : task_runner_(std::move(task_runner)) {} |
+ |
+ // Destroyed on the |leveldb_service_runner_|. |
+ ~LevelDBServiceObjects() {} |
+ |
+ // Called on the |leveldb_service_runner_|. |
+ void OnLevelDBServiceRequest(mojo::Connection* connection, |
+ leveldb::LevelDBServiceRequest request) { |
+ if (!leveldb_service_) |
+ leveldb_service_.reset(new leveldb::LevelDBServiceImpl(task_runner_)); |
+ leveldb_bindings_.AddBinding(leveldb_service_.get(), std::move(request)); |
+ } |
+ |
+ private: |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
+ |
+ // Variables that are only accessible on the |leveldb_service_runner_| thread. |
+ scoped_ptr<leveldb::LevelDBService> leveldb_service_; |
+ mojo::BindingSet<leveldb::LevelDBService> leveldb_bindings_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(LevelDBServiceObjects); |
+}; |
+ |
+scoped_ptr<mojo::ShellClient> CreateUserShellClient( |
+ scoped_refptr<base::SingleThreadTaskRunner> user_service_runner, |
+ scoped_refptr<base::SingleThreadTaskRunner> leveldb_service_runner) { |
+ return make_scoped_ptr(new UserShellClient(std::move(user_service_runner), |
+ std::move(leveldb_service_runner))); |
+} |
+ |
+UserShellClient::UserShellClient( |
+ scoped_refptr<base::SingleThreadTaskRunner> user_service_runner, |
+ scoped_refptr<base::SingleThreadTaskRunner> leveldb_service_runner) |
+ : user_service_runner_(std::move(user_service_runner)), |
+ leveldb_service_runner_(std::move(leveldb_service_runner)) {} |
+ |
+UserShellClient::~UserShellClient() { |
+ user_service_runner_->DeleteSoon(FROM_HERE, user_objects_.release()); |
+ leveldb_service_runner_->DeleteSoon(FROM_HERE, leveldb_objects_.release()); |
+} |
+ |
+void UserShellClient::Initialize(mojo::Connector* connector, |
+ const mojo::Identity& identity, |
+ uint32_t id) { |
+ tracing_.Initialize(connector, identity.name()); |
+ user_objects_.reset(new UserShellClient::UserServiceObjects( |
+ GetUserDirForUserID(identity.user_id()))); |
+ leveldb_objects_.reset( |
+ new UserShellClient::LevelDBServiceObjects(leveldb_service_runner_)); |
+} |
+ |
+bool UserShellClient::AcceptConnection(mojo::Connection* connection) { |
+ connection->AddInterface<leveldb::LevelDBService>(this); |
+ connection->AddInterface<mojom::UserService>(this); |
+ return true; |
+} |
+ |
+void UserShellClient::Create(mojo::Connection* connection, |
+ mojom::UserServiceRequest request) { |
+ user_service_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind(&UserShellClient::UserServiceObjects::OnUserServiceRequest, |
+ user_objects_->AsWeakPtr(), connection, |
+ base::Passed(&request))); |
+} |
+ |
+void UserShellClient::Create(mojo::Connection* connection, |
+ leveldb::LevelDBServiceRequest request) { |
+ leveldb_service_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind( |
+ &UserShellClient::LevelDBServiceObjects::OnLevelDBServiceRequest, |
+ leveldb_objects_->AsWeakPtr(), connection, |
+ base::Passed(&request))); |
+} |
+ |
+} // namespace user_service |