Chromium Code Reviews| Index: blimp/common/net/message_dispatcher.cc |
| diff --git a/blimp/common/net/message_dispatcher.cc b/blimp/common/net/message_dispatcher.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..5a90d510e6a34051850415cb9b932fe1959983c1 |
| --- /dev/null |
| +++ b/blimp/common/net/message_dispatcher.cc |
| @@ -0,0 +1,53 @@ |
| +// Copyright 2015 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 "blimp/common/net/message_dispatcher.h" |
| + |
| +#include <string> |
| + |
| +#include "base/strings/stringprintf.h" |
| + |
| +namespace blimp { |
| +namespace { |
| + |
| +std::string BlimpMessageToString(const BlimpMessage& message) { |
|
nyquist
2015/09/08 17:30:12
Nit: Could we make it clear that this is meant to
Kevin M
2015/09/08 20:50:19
Done.
|
| + return base::StringPrintf("<message type=%d, tab=%d>", message.type(), |
| + message.target_tab_id()); |
| +} |
| + |
| +} // namespace |
| + |
| +MessageDispatcher::MessageDispatcher() {} |
| + |
| +MessageDispatcher::~MessageDispatcher() { |
| + DCHECK(thread_checker_.CalledOnValidThread()); |
| +} |
| + |
| +void MessageDispatcher::AddHandler(BlimpMessage::Type type, Handler* handler) { |
| + DCHECK(thread_checker_.CalledOnValidThread()); |
| + if (feature_handler_map_.find(type) == feature_handler_map_.end()) { |
|
nyquist
2015/09/08 17:30:12
Nit: Should we log an error if there already exist
Kevin M
2015/09/08 20:50:19
Done.
|
| + feature_handler_map_.insert(std::make_pair(type, handler)); |
| + } |
| +} |
| + |
| +void MessageDispatcher::Dispatch(const BlimpMessage& message) const { |
| + DCHECK(thread_checker_.CalledOnValidThread()); |
| + |
| + auto handler_iter = feature_handler_map_.find(message.type()); |
| + if (handler_iter == feature_handler_map_.end()) { |
| + LOG(ERROR) << "No registered handler for " << BlimpMessageToString(message) |
| + << "."; |
| + return; |
| + } |
| + |
| + DCHECK(handler_iter->second); |
| + if (!handler_iter->second->Validate(message)) { |
| + LOG(WARNING) << BlimpMessageToString(message) << " rejected by handler."; |
| + return; |
| + } |
| + |
| + handler_iter->second->OnMessage(message); |
| +} |
| + |
| +} // namespace blimp |