| Index: extensions/browser/api/cast_channel/cast_socket_service.cc
|
| diff --git a/extensions/browser/api/cast_channel/cast_socket_service.cc b/extensions/browser/api/cast_channel/cast_socket_service.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..bbf054220b66db044c0a6fd39517e632616580a9
|
| --- /dev/null
|
| +++ b/extensions/browser/api/cast_channel/cast_socket_service.cc
|
| @@ -0,0 +1,62 @@
|
| +// Copyright 2017 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 "extensions/browser/api/cast_channel/cast_socket_service.h"
|
| +
|
| +#include "base/memory/ptr_util.h"
|
| +
|
| +namespace extensions {
|
| +namespace api {
|
| +namespace cast_channel {
|
| +
|
| +int CastSocketRegistry::last_channel_id_ = 0;
|
| +
|
| +CastSocketRegistry::CastSocketRegistry() = default;
|
| +
|
| +CastSocketRegistry::~CastSocketRegistry() {
|
| + DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
|
| +}
|
| +
|
| +int CastSocketRegistry::AddSocket(std::unique_ptr<CastSocket> socket) {
|
| + DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
|
| + DCHECK(socket);
|
| + int id = ++last_channel_id_;
|
| + socket->set_id(id);
|
| + sockets_.insert(std::make_pair(id, std::move(socket)));
|
| + return id;
|
| +}
|
| +
|
| +std::unique_ptr<CastSocket> CastSocketRegistry::RemoveSocket(int channel_id) {
|
| + DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
|
| + DCHECK(channel_id > 0);
|
| + auto socket_it = sockets_.find(channel_id);
|
| +
|
| + std::unique_ptr<CastSocket> socket;
|
| + if (socket_it != sockets_.end()) {
|
| + socket = std::move(socket_it->second);
|
| + sockets_.erase(socket_it);
|
| + }
|
| + return socket;
|
| +}
|
| +
|
| +CastSocket* CastSocketRegistry::GetSocket(int channel_id) const {
|
| + DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
|
| + DCHECK(channel_id > 0);
|
| + const auto& socket_it = sockets_.find(channel_id);
|
| + return socket_it == sockets_.end() ? nullptr : socket_it->second.get();
|
| +}
|
| +
|
| +CastSocketService::CastSocketService() = default;
|
| +CastSocketService::~CastSocketService() = default;
|
| +
|
| +CastSocketRegistry* CastSocketService::GetOrCreateSocketRegistry() {
|
| + DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
|
| + if (!sockets_)
|
| + sockets_.reset(new CastSocketRegistry());
|
| + return sockets_.get();
|
| +}
|
| +
|
| +} // namespace cast_channel
|
| +} // namespace api
|
| +} // namespace extensions
|
|
|