Index: mojo/public/java/bindings/src/org/chromium/mojo/bindings/RouterImpl.java |
diff --git a/mojo/public/java/bindings/src/org/chromium/mojo/bindings/RouterImpl.java b/mojo/public/java/bindings/src/org/chromium/mojo/bindings/RouterImpl.java |
index 25337fbaced02dfea554d252a2ae7eabe6ff3a5d..e65fcae7a9469ddf21eebea14d7d34d7ad1bff24 100644 |
--- a/mojo/public/java/bindings/src/org/chromium/mojo/bindings/RouterImpl.java |
+++ b/mojo/public/java/bindings/src/org/chromium/mojo/bindings/RouterImpl.java |
@@ -21,10 +21,10 @@ public class RouterImpl implements Router { |
private class ResponderThunk implements MessageReceiver { |
/** |
- * @see MessageReceiver#accept(MessageWithHeader) |
+ * @see MessageReceiver#accept(Message) |
*/ |
@Override |
- public boolean accept(MessageWithHeader message) { |
+ public boolean accept(Message message) { |
return handleIncomingMessage(message); |
} |
@@ -97,21 +97,23 @@ public class RouterImpl implements Router { |
} |
/** |
- * @see MessageReceiver#accept(MessageWithHeader) |
+ * @see MessageReceiver#accept(Message) |
*/ |
@Override |
- public boolean accept(MessageWithHeader message) { |
+ public boolean accept(Message message) { |
// A message without responder is directly forwarded to the connector. |
return mConnector.accept(message); |
} |
/** |
- * @see MessageReceiverWithResponder#acceptWithResponder(MessageWithHeader, MessageReceiver) |
+ * @see MessageReceiverWithResponder#acceptWithResponder(Message, MessageReceiver) |
*/ |
@Override |
- public boolean acceptWithResponder(MessageWithHeader message, MessageReceiver responder) { |
+ public boolean acceptWithResponder(Message message, MessageReceiver responder) { |
+ // The message must have a header. |
+ MessageWithHeader messageWithHeader = message.asMojoMessage(); |
// Checking the message expects a response. |
- assert message.getHeader().hasFlag(MessageHeader.MESSAGE_EXPECTS_RESPONSE_FLAG); |
+ assert messageWithHeader.getHeader().hasFlag(MessageHeader.MESSAGE_EXPECTS_RESPONSE_FLAG); |
// Compute a request id for being able to route the response. |
long requestId = mNextRequestId++; |
@@ -122,8 +124,8 @@ public class RouterImpl implements Router { |
if (mResponders.containsKey(requestId)) { |
throw new IllegalStateException("Unable to find a new request identifier."); |
} |
- message.setRequestId(requestId); |
- if (!mConnector.accept(message)) { |
+ messageWithHeader.setRequestId(requestId); |
+ if (!mConnector.accept(messageWithHeader)) { |
return false; |
} |
// Only keep the responder is the message has been accepted. |
@@ -158,8 +160,8 @@ public class RouterImpl implements Router { |
/** |
* Receive a message from the connector. Returns |true| if the message has been handled. |
*/ |
- private boolean handleIncomingMessage(MessageWithHeader message) { |
- MessageHeader header = message.getHeader(); |
+ private boolean handleIncomingMessage(Message message) { |
+ MessageHeader header = message.asMojoMessage().getHeader(); |
if (header.hasFlag(MessageHeader.MESSAGE_EXPECTS_RESPONSE_FLAG)) { |
if (mIncomingMessageReceiver != null) { |
return mIncomingMessageReceiver.acceptWithResponder(message, this); |