Gather trace data by waiting on handles with a fixed timeout

mojo/public/cpp/bindings/lib/router.h

 // Copyright 2014 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 MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_
 #define MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_
 
 #include <map>
 
 #include "mojo/public/cpp/bindings/lib/connector.h"
@@ skipping 34 matching lines @@
 
   ScopedMessagePipeHandle PassMessagePipe() {
     return connector_.PassMessagePipe();
   }
 
   // MessageReceiver implementation:
   bool Accept(Message* message) override;
   bool AcceptWithResponder(Message* message,
                            MessageReceiver* responder) override;
 
   // Blocks the current thread for the first incoming method call, i.e., either

[viettrungluu, 2015/04/02 19:54:13]

nit: update comment?

   // a call to a client method or a callback method.
-  bool WaitForIncomingMessage() { return connector_.WaitForIncomingMessage(); }
+  bool WaitForIncomingMessage(MojoDeadline deadline) {
+    return connector_.WaitForIncomingMessage(deadline);
+  }
 
   // Sets this object to testing mode.
   // In testing mode:
   // - the object is more tolerant of unrecognized response messages;
   // - the connector continues working after seeing errors from its incoming
   //   receiver.
   void EnableTestingMode();
 
+  MessagePipeHandle handle() const { return connector_.handle(); }
+
  private:
   typedef std::map<uint64_t, MessageReceiver*> ResponderMap;
 
   class HandleIncomingMessageThunk : public MessageReceiver {
    public:
     HandleIncomingMessageThunk(Router* router);
     ~HandleIncomingMessageThunk() override;
 
     // MessageReceiver implementation:
     bool Accept(Message* message) override;
@@ skipping 11 matching lines @@
   MessageReceiverWithResponderStatus* incoming_receiver_;
   ResponderMap responders_;
   uint64_t next_request_id_;
   bool testing_mode_;
 };
 
 }  // namespace internal
 }  // namespace mojo
 
 #endif  // MOJO_PUBLIC_CPP_BINDINGS_LIB_ROUTER_H_