Refactor IqRequest.

remoting/jingle_glue/iq_sender.cc

+// Copyright (c) 2011 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 "remoting/jingle_glue/iq_sender.h"
+
+#include "base/logging.h"
+#include "base/string_number_conversions.h"
+#include "remoting/jingle_glue/signal_strategy.h"
+#include "third_party/libjingle/source/talk/xmllite/xmlelement.h"
+#include "third_party/libjingle/source/talk/xmpp/constants.h"
+
+namespace remoting {
+
+// static
+buzz::XmlElement* IqSender::MakeIqStanza(const std::string& type,
+                                         const std::string& addressee,
+                                         buzz::XmlElement* iq_body) {
+  buzz::XmlElement* stanza = new buzz::XmlElement(buzz::QN_IQ);
+  stanza->AddAttr(buzz::QN_TYPE, type);
+  if (!addressee.empty())
+    stanza->AddAttr(buzz::QN_TO, addressee);
+  stanza->AddElement(iq_body);
+  return stanza;
+}
+
+IqSender::IqSender(SignalStrategy* signal_strategy)
+    : signal_strategy_(signal_strategy) {
+  signal_strategy_->AddListener(this);
+}
+
+IqSender::~IqSender() {
+  signal_strategy_->RemoveListener(this);
+}
+
+IqRequest* IqSender::SendIq(buzz::XmlElement* stanza,
+                            const ReplyCallback& callback) {
+  std::string id = signal_strategy_->GetNextId();
+  stanza->AddAttr(buzz::QN_ID, id);
+  if (!signal_strategy_->SendStanza(stanza)) {
+    return NULL;
+  }
+  DCHECK(requests_.find(id) == requests_.end());
+  IqRequest* request = new IqRequest(this, callback);
+  if (!callback.is_null())
+    requests_[id] = request;
+  return request;
+}
+
+IqRequest* IqSender::SendIq(const std::string& type,
+                            const std::string& addressee,
+                            buzz::XmlElement* iq_body,
+                            const ReplyCallback& callback) {
+  return SendIq(MakeIqStanza(type, addressee, iq_body), callback);
+}
+
+void IqSender::RemoveRequest(IqRequest* request) {
+  IqRequestMap::iterator it = requests_.begin();
+  while (it != requests_.end()) {
+    IqRequestMap::iterator cur = it;
+    ++it;
+    if (cur->second == request) {
+      requests_.erase(cur);
+      break;
+    }
+  }
+}
+
+bool IqSender::OnIncomingStanza(const buzz::XmlElement* stanza) {
+  if (stanza->Name() != buzz::QN_IQ) {
+    LOG(WARNING) << "Received unexpected non-IQ packet" << stanza->Str();
+    return false;
+  }
+
+  const std::string& id = stanza->Attr(buzz::QN_ID);
+  if (id.empty()) {
+    LOG(WARNING) << "IQ packet missing id" << stanza->Str();
+    return false;
+  }
+
+  IqRequestMap::iterator it = requests_.find(id);
+  if (it == requests_.end()) {
+    return false;
+  }
+
+  IqRequest* request = it->second;
+  requests_.erase(it);
+
+  request->OnResponse(stanza);
+  return true;
+}
+
+IqRequest::IqRequest(IqSender* sender, const IqSender::ReplyCallback& callback)
+    : sender_(sender),
+      callback_(callback) {
+}
+
+IqRequest::~IqRequest() {
+  sender_->RemoveRequest(this);
+}
+
+void IqRequest::OnResponse(const buzz::XmlElement* stanza) {
+  DCHECK(!callback_.is_null());
+  IqSender::ReplyCallback callback(callback_);
+  callback_.Reset();
+  callback.Run(stanza);
+}
+
+}  // namespace remoting