| Index: trunk/src/athena/system/device_socket_listener.cc
|
| ===================================================================
|
| --- trunk/src/athena/system/device_socket_listener.cc (revision 291393)
|
| +++ trunk/src/athena/system/device_socket_listener.cc (working copy)
|
| @@ -12,6 +12,7 @@
|
|
|
| #include "base/bind.h"
|
| #include "base/files/file_path.h"
|
| +#include "base/memory/singleton.h"
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/observer_list.h"
|
| #include "base/stl_util.h"
|
| @@ -54,22 +55,17 @@
|
| // A singleton instance for managing all connections to sockets.
|
| class DeviceSocketManager {
|
| public:
|
| - static void Create(scoped_refptr<base::TaskRunner> file_task_runner) {
|
| - device_socket_manager_instance_ = new DeviceSocketManager(file_task_runner);
|
| + static void Create(scoped_refptr<base::TaskRunner> io_task_runner) {
|
| + device_socket_manager_instance_ =
|
| + new DeviceSocketManager(io_task_runner);
|
| }
|
|
|
| static void Shutdown() {
|
| CHECK(device_socket_manager_instance_);
|
| - device_socket_manager_instance_->ScheduleDelete();
|
| - // Once scheduled to be deleted, no-one should be
|
| - // able to access it.
|
| + delete device_socket_manager_instance_;
|
| device_socket_manager_instance_ = NULL;
|
| }
|
|
|
| - static DeviceSocketManager* GetInstanceUnsafe() {
|
| - return device_socket_manager_instance_;
|
| - }
|
| -
|
| static DeviceSocketManager* GetInstance() {
|
| CHECK(device_socket_manager_instance_);
|
| return device_socket_manager_instance_;
|
| @@ -98,6 +94,8 @@
|
| void OnEOF(const std::string& socket_path);
|
|
|
| private:
|
| + friend struct DefaultSingletonTraits<DeviceSocketManager>;
|
| +
|
| struct SocketData {
|
| SocketData()
|
| : fd(-1) {
|
| @@ -109,29 +107,26 @@
|
| scoped_ptr<DeviceSocketReader> watcher;
|
| };
|
|
|
| - static void DeleteOnFILE(DeviceSocketManager* manager) { delete manager; }
|
| + DeviceSocketManager(scoped_refptr<base::TaskRunner> io_task_runner)
|
| + : io_task_runner_(io_task_runner) {
|
| + }
|
|
|
| - DeviceSocketManager(scoped_refptr<base::TaskRunner> file_task_runner)
|
| - : file_task_runner_(file_task_runner) {}
|
| -
|
| ~DeviceSocketManager() {
|
| STLDeleteContainerPairSecondPointers(socket_data_.begin(),
|
| socket_data_.end());
|
| }
|
|
|
| - void ScheduleDelete();
|
| + void StartListeningOnIO(const std::string& socket_path,
|
| + size_t data_size,
|
| + DeviceSocketListener* listener);
|
|
|
| - void StartListeningOnFILE(const std::string& socket_path,
|
| - size_t data_size,
|
| - DeviceSocketListener* listener);
|
| + void StopListeningOnIO(const std::string& socket_path,
|
| + DeviceSocketListener* listener);
|
|
|
| - void StopListeningOnFILE(const std::string& socket_path,
|
| - DeviceSocketListener* listener);
|
| -
|
| void CloseSocket(const std::string& socket_path);
|
|
|
| std::map<std::string, SocketData*> socket_data_;
|
| - scoped_refptr<base::TaskRunner> file_task_runner_;
|
| + scoped_refptr<base::TaskRunner> io_task_runner_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(DeviceSocketManager);
|
| };
|
| @@ -167,31 +162,23 @@
|
| void DeviceSocketManager::StartListening(const std::string& socket_path,
|
| size_t data_size,
|
| DeviceSocketListener* listener) {
|
| - file_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&DeviceSocketManager::StartListeningOnFILE,
|
| - base::Unretained(this),
|
| - socket_path,
|
| - data_size,
|
| - listener));
|
| + io_task_runner_->PostTask(FROM_HERE,
|
| + base::Bind(&DeviceSocketManager::StartListeningOnIO,
|
| + base::Unretained(this), socket_path, data_size, listener));
|
| }
|
|
|
| void DeviceSocketManager::StopListening(const std::string& socket_path,
|
| DeviceSocketListener* listener) {
|
| - file_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&DeviceSocketManager::StopListeningOnFILE,
|
| - base::Unretained(this),
|
| - socket_path,
|
| - listener));
|
| + io_task_runner_->PostTask(FROM_HERE,
|
| + base::Bind(&DeviceSocketManager::StopListeningOnIO,
|
| + base::Unretained(this), socket_path, listener));
|
| }
|
|
|
| void DeviceSocketManager::OnDataAvailable(const std::string& socket_path,
|
| const void* data) {
|
| CHECK_GT(socket_data_.count(socket_path), 0UL);
|
| DeviceSocketListeners& listeners = socket_data_[socket_path]->observers;
|
| - FOR_EACH_OBSERVER(
|
| - DeviceSocketListener, listeners, OnDataAvailableOnFILE(data));
|
| + FOR_EACH_OBSERVER(DeviceSocketListener, listeners, OnDataAvailableOnIO(data));
|
| }
|
|
|
| void DeviceSocketManager::CloseSocket(const std::string& socket_path) {
|
| @@ -215,10 +202,10 @@
|
| CloseSocket(socket_path);
|
| }
|
|
|
| -void DeviceSocketManager::StartListeningOnFILE(const std::string& socket_path,
|
| - size_t data_size,
|
| - DeviceSocketListener* listener) {
|
| - CHECK(file_task_runner_->RunsTasksOnCurrentThread());
|
| +void DeviceSocketManager::StartListeningOnIO(const std::string& socket_path,
|
| + size_t data_size,
|
| + DeviceSocketListener* listener) {
|
| + CHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| SocketData* socket_data = NULL;
|
| if (!socket_data_.count(socket_path)) {
|
| int socket_fd = -1;
|
| @@ -250,12 +237,12 @@
|
| socket_data->observers.AddObserver(listener);
|
| }
|
|
|
| -void DeviceSocketManager::StopListeningOnFILE(const std::string& socket_path,
|
| - DeviceSocketListener* listener) {
|
| +void DeviceSocketManager::StopListeningOnIO(const std::string& socket_path,
|
| + DeviceSocketListener* listener) {
|
| if (!socket_data_.count(socket_path))
|
| return; // Happens if unable to create a socket.
|
|
|
| - CHECK(file_task_runner_->RunsTasksOnCurrentThread());
|
| + CHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| DeviceSocketListeners& listeners = socket_data_[socket_path]->observers;
|
| listeners.RemoveObserver(listener);
|
| if (!listeners.might_have_observers()) {
|
| @@ -264,14 +251,6 @@
|
| }
|
| }
|
|
|
| -void DeviceSocketManager::ScheduleDelete() {
|
| - // Schedule a task to delete on FILE thread because
|
| - // there may be a task scheduled on |file_task_runner_|.
|
| - file_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&DeleteOnFILE, base::Unretained(this)));
|
| -}
|
| -
|
| } // namespace
|
|
|
| DeviceSocketListener::DeviceSocketListener(const std::string& socket_path,
|
| @@ -286,8 +265,8 @@
|
|
|
| // static
|
| void DeviceSocketListener::CreateSocketManager(
|
| - scoped_refptr<base::TaskRunner> file_task_runner) {
|
| - DeviceSocketManager::Create(file_task_runner);
|
| + scoped_refptr<base::TaskRunner> io_task_runner) {
|
| + DeviceSocketManager::Create(io_task_runner);
|
| }
|
|
|
| // static
|
| @@ -302,9 +281,7 @@
|
| }
|
|
|
| void DeviceSocketListener::StopListening() {
|
| - DeviceSocketManager* instance = DeviceSocketManager::GetInstanceUnsafe();
|
| - if (instance)
|
| - instance->StopListening(socket_path_, this);
|
| + DeviceSocketManager::GetInstance()->StopListening(socket_path_, this);
|
| }
|
|
|
| } // namespace athena
|
|
|