Refactor IqRequest.

remoting/jingle_glue/iq_sender.h

Index: remoting/jingle_glue/iq_sender.h
diff --git a/remoting/jingle_glue/iq_sender.h b/remoting/jingle_glue/iq_sender.h
new file mode 100644
index 0000000000000000000000000000000000000000..77b48ee4fe097badd0926295ef886232673176f7
--- /dev/null
+++ b/remoting/jingle_glue/iq_sender.h
@@ -0,0 +1,90 @@
+// 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.
+
+#ifndef REMOTING_JINGLE_GLUE_IQ_SENDER_H_
+#define REMOTING_JINGLE_GLUE_IQ_SENDER_H_
+
+#include <map>
+#include <string>
+
+#include "base/callback.h"
+#include "base/compiler_specific.h"
+#include "base/gtest_prod_util.h"
+#include "base/memory/scoped_ptr.h"
+#include "remoting/jingle_glue/signal_strategy.h"
+
+namespace buzz {
+class XmlElement;
+}  // namespace buzz
+
+namespace remoting {
+
+class IqRequest;
+class SignalStrategy;
+
+// IqSender handles sending iq requests and routing of responses to
+// those requests.
+class IqSender : public SignalStrategy::Listener {
+ public:
+  typedef base::Callback<void(const buzz::XmlElement*)> ReplyCallback;
+
+  explicit IqSender(SignalStrategy* signal_strategy);
+  virtual ~IqSender();
+
+  // Send an iq stanza. Returns an IqRequest object that represends
+  // the request. |callback| is called when response to |stanza| is
+  // received. Destroy the returned IqRequest to cancel the callback.
+  // Takes ownership of |stanza|. Caller must take ownership of the
+  // result. Result must be destroyed before sender is destroyed.
+  IqRequest* SendIq(buzz::XmlElement* stanza, const ReplyCallback& callback);
+
+  // Same as above, but also formats the message. Takes ownership of
+  // |iq_body|.
+  IqRequest* SendIq(const std::string& type,
+                    const std::string& addressee,
+                    buzz::XmlElement* iq_body,
+                    const ReplyCallback& callback);

[Wez, 2011/11/03 20:29:19]

Overloads make style pixie cry.

[Sergey Ulanov, 2011/11/03 20:52:12]

I think this is a case when function overload is acceptable, though I can rename
it if you disagree.

+
+  // SignalStrategy::Listener implementation.
+  virtual bool OnIncomingStanza(const buzz::XmlElement* stanza) OVERRIDE;
+
+ private:
+  typedef std::map<std::string, IqRequest*> IqRequestMap;
+  friend class IqRequest;
+
+  // Helper function used to create iq stanzas.
+  static buzz::XmlElement* MakeIqStanza(const std::string& type,
+                                        const std::string& addressee,
+                                        buzz::XmlElement* iq_body);
+
+  // Removes |request| from the list of pending requests. Called by IqRequest.
+  void RemoveRequest(IqRequest* request);
+
+  SignalStrategy* signal_strategy_;
+  IqRequestMap requests_;
+
+  DISALLOW_COPY_AND_ASSIGN(IqSender);
+};
+
+// This call must only be used on the thread it was created on.
+class IqRequest {
+ public:
+  IqRequest(IqSender* sender, const IqSender::ReplyCallback& callback);
+  ~IqRequest();
+
+ private:
+  friend class IqSender;
+
+  // Called by IqSender when a response is received.
+  void OnResponse(const buzz::XmlElement* stanza);
+
+  IqSender* sender_;
+  IqSender::ReplyCallback callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(IqRequest);
+};
+
+}  // namespace remoting
+
+#endif  // REMOTING_JINGLE_GLUE_IQ_SENDER_H_