diff options
| -rw-r--r-- | src/objective-c/GRPCClient/GRPCCall.m | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/src/objective-c/GRPCClient/GRPCCall.m b/src/objective-c/GRPCClient/GRPCCall.m index 0a10322367..1a8fc2e2ef 100644 --- a/src/objective-c/GRPCClient/GRPCCall.m +++ b/src/objective-c/GRPCClient/GRPCCall.m @@ -164,8 +164,8 @@ static NSMutableDictionary *callFlags; _requestHeaders = [[GRPCRequestHeaders alloc] initWithCall:self]; if ([requestWriter isKindOfClass:[GRXImmediateSingleWriter class]]) { - _unaryCall = true; - _unaryOpBatch = [[NSMutableArray alloc] init]; + _unaryCall = YES; + _unaryOpBatch = [NSMutableArray arrayWithCapacity:6]; } } return self; @@ -267,14 +267,13 @@ static NSMutableDictionary *callFlags; - (void)sendHeaders:(NSDictionary *)headers { // TODO(jcanizales): Add error handlers for async failures + GRPCOpSendMetadata *op = [[GRPCOpSendMetadata alloc] initWithMetadata:headers + flags:[GRPCCall callFlagsForHost:_host path:_path] + handler:nil]; if (!_unaryCall) { - [_wrappedCall startBatchWithOperations:@[[[GRPCOpSendMetadata alloc] initWithMetadata:headers - flags:[GRPCCall callFlagsForHost:_host path:_path] - handler:nil]]]; + [_wrappedCall startBatchWithOperations:@[op]]; } else { - [_unaryOpBatch addObject:[[GRPCOpSendMetadata alloc] initWithMetadata:headers - flags:[GRPCCall callFlagsForHost:_host path:_path] - handler:nil]]; + [_unaryOpBatch addObject:op]; } } @@ -294,13 +293,14 @@ static NSMutableDictionary *callFlags; } } }; + + GRPCOpSendMessage *op = [[GRPCOpSendMessage alloc] initWithMessage:message + handler:resumingHandler]; if (!_unaryCall) { - [_wrappedCall startBatchWithOperations:@[[[GRPCOpSendMessage alloc] initWithMessage:message - handler:resumingHandler]] + [_wrappedCall startBatchWithOperations:@[op] errorHandler:errorHandler]; } else { - [_unaryOpBatch addObject:[[GRPCOpSendMessage alloc] initWithMessage:message - handler:resumingHandler]]; + [_unaryOpBatch addObject:op]; } } |