From 2d903f4732ee21c5c319c88aa738671b56b2e729 Mon Sep 17 00:00:00 2001
From: Muxi Yan <mxyan@google.com>
Date: Fri, 19 Oct 2018 10:15:02 -0700
Subject: [PATCH] More specific typing in response handlers

---
 src/objective-c/GRPCClient/GRPCCall.h   | 2 +-
 src/objective-c/ProtoRPC/ProtoRPC.h     | 2 +-
 src/objective-c/ProtoRPC/ProtoRPC.m     | 2 +-
 src/objective-c/tests/GRPCClientTests.m | 2 +-
 src/objective-c/tests/InteropTests.m    | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/objective-c/GRPCClient/GRPCCall.h b/src/objective-c/GRPCClient/GRPCCall.h
index bd43a0a384b..d8d3e3cf629 100644
--- a/src/objective-c/GRPCClient/GRPCCall.h
+++ b/src/objective-c/GRPCClient/GRPCCall.h
@@ -160,7 +160,7 @@ extern id const kGRPCTrailersKey;
  * Issued when a message is received from the server. The message is the raw data received from the
  * server, with decompression and without proto deserialization.
  */
-- (void)receivedRawMessage:(id)message;
+- (void)receivedRawMessage:(NSData *)message;
 
 /**
  * Issued when a call finished. If the call finished successfully, \a error is nil and \a
diff --git a/src/objective-c/ProtoRPC/ProtoRPC.h b/src/objective-c/ProtoRPC/ProtoRPC.h
index 4121d4f6af2..34a519bee5d 100644
--- a/src/objective-c/ProtoRPC/ProtoRPC.h
+++ b/src/objective-c/ProtoRPC/ProtoRPC.h
@@ -34,7 +34,7 @@
 /**
  * Issued when a message is received from the server. The message is the deserialized proto object.
  */
-- (void)receivedProtoMessage:(id)message;
+- (void)receivedProtoMessage:(GPBMessage *)message;
 
 /**
  * Issued when a call finished. If the call finished successfully, \a error is nil and \a
diff --git a/src/objective-c/ProtoRPC/ProtoRPC.m b/src/objective-c/ProtoRPC/ProtoRPC.m
index b04b6aca67b..44e9bfde0ab 100644
--- a/src/objective-c/ProtoRPC/ProtoRPC.m
+++ b/src/objective-c/ProtoRPC/ProtoRPC.m
@@ -168,7 +168,7 @@
   if (_handler) {
     id<GRPCProtoResponseHandler> handler = _handler;
     NSError *error = nil;
-    id parsed = [_responseClass parseFromData:message error:&error];
+    GPBMessage *parsed = [_responseClass parseFromData:message error:&error];
     if (parsed) {
       if ([handler respondsToSelector:@selector(receivedProtoMessage:)]) {
         dispatch_async(handler.dispatchQueue, ^{
diff --git a/src/objective-c/tests/GRPCClientTests.m b/src/objective-c/tests/GRPCClientTests.m
index 387fcab7e9f..985e105b816 100644
--- a/src/objective-c/tests/GRPCClientTests.m
+++ b/src/objective-c/tests/GRPCClientTests.m
@@ -120,7 +120,7 @@ static GRPCProtoMethod *kFullDuplexCallMethod;
   }
 }
 
-- (void)receivedProtoMessage:(id)message {
+- (void)receivedProtoMessage:(GPBMessage *)message {
   if (_messageCallback) {
     _messageCallback(message);
   }
diff --git a/src/objective-c/tests/InteropTests.m b/src/objective-c/tests/InteropTests.m
index d38e1e0d972..a9f33aab6f1 100644
--- a/src/objective-c/tests/InteropTests.m
+++ b/src/objective-c/tests/InteropTests.m
@@ -108,7 +108,7 @@ BOOL isRemoteInteropTest(NSString *host) {
   }
 }
 
-- (void)receivedProtoMessage:(id)message {
+- (void)receivedProtoMessage:(GPBMessage *)message {
   if (_messageCallback) {
     _messageCallback(message);
   }