Blimp: create MessageDispatcher class.

blimp/common/net/message_dispatcher.cc

+// 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 BlimpMessageToDebugString(const BlimpMessage& message) {
+  return base::StringPrintf("<message type=%d>", message.type());
+}
+
+}  // namespace
+
+MessageDispatcher::MessageDispatcher() {}
+
+MessageDispatcher::~MessageDispatcher() {}
+
+void MessageDispatcher::AddHandler(BlimpMessage::Type type, Handler* handler) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(handler);
+  if (feature_handler_map_.find(type) == feature_handler_map_.end()) {
+    feature_handler_map_.insert(std::make_pair(type, handler));
+  } else {
+    DCHECK(false) << "Handler already registered for type=" << type << ".";
+  }
+}
+
+bool 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()) {
+    VLOG(0) << "No registered handler for "
+            << BlimpMessageToDebugString(message) << ".";
+    return false;
+  }
+
+  if (!handler_iter->second->Validate(message)) {
+    VLOG(0) << BlimpMessageToDebugString(message) << " rejected by handler.";
+    return false;
+  }
+
+  handler_iter->second->OnMessage(message);
+  return true;
+}
+
+}  // namespace blimp