mirror of https://github.com/grpc/grpc.git
Matthew Stevenson
6 years ago
commit
4b7eb25607
143 changed files with 3610 additions and 2037 deletions
@ -0,0 +1,136 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* This is the legacy interface of this gRPC library. This API is deprecated and users should use |
||||
* the API in GRPCCall.h. This API exists solely for the purpose of backwards compatibility. |
||||
*/ |
||||
|
||||
#import <RxLibrary/GRXWriter.h> |
||||
#import "GRPCTypes.h" |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wnullability-completeness" |
||||
|
||||
/**
|
||||
* This interface is deprecated. Please use \a GRPCCall2. |
||||
* |
||||
* Represents a single gRPC remote call. |
||||
*/ |
||||
@interface GRPCCall : GRXWriter |
||||
|
||||
- (instancetype)init NS_UNAVAILABLE; |
||||
|
||||
/**
|
||||
* The container of the request headers of an RPC conforms to this protocol, which is a subset of |
||||
* NSMutableDictionary's interface. It will become a NSMutableDictionary later on. |
||||
* The keys of this container are the header names, which per the HTTP standard are case- |
||||
* insensitive. They are stored in lowercase (which is how HTTP/2 mandates them on the wire), and |
||||
* can only consist of ASCII characters. |
||||
* A header value is a NSString object (with only ASCII characters), unless the header name has the |
||||
* suffix "-bin", in which case the value has to be a NSData object. |
||||
*/ |
||||
/**
|
||||
* These HTTP headers will be passed to the server as part of this call. Each HTTP header is a |
||||
* name-value pair with string names and either string or binary values. |
||||
* |
||||
* The passed dictionary has to use NSString keys, corresponding to the header names. The value |
||||
* associated to each can be a NSString object or a NSData object. E.g.: |
||||
* |
||||
* call.requestHeaders = @{@"authorization": @"Bearer ..."}; |
||||
* |
||||
* call.requestHeaders[@"my-header-bin"] = someData; |
||||
* |
||||
* After the call is started, trying to modify this property is an error. |
||||
* |
||||
* The property is initialized to an empty NSMutableDictionary. |
||||
*/ |
||||
@property(atomic, readonly) NSMutableDictionary *requestHeaders; |
||||
|
||||
/**
|
||||
* This dictionary is populated with the HTTP headers received from the server. This happens before |
||||
* any response message is received from the server. It has the same structure as the request |
||||
* headers dictionary: Keys are NSString header names; names ending with the suffix "-bin" have a |
||||
* NSData value; the others have a NSString value. |
||||
* |
||||
* The value of this property is nil until all response headers are received, and will change before |
||||
* any of -writeValue: or -writesFinishedWithError: are sent to the writeable. |
||||
*/ |
||||
@property(atomic, readonly) NSDictionary *responseHeaders; |
||||
|
||||
/**
|
||||
* Same as responseHeaders, but populated with the HTTP trailers received from the server before the |
||||
* call finishes. |
||||
* |
||||
* The value of this property is nil until all response trailers are received, and will change |
||||
* before -writesFinishedWithError: is sent to the writeable. |
||||
*/ |
||||
@property(atomic, readonly) NSDictionary *responseTrailers; |
||||
|
||||
/**
|
||||
* The request writer has to write NSData objects into the provided Writeable. The server will |
||||
* receive each of those separately and in order as distinct messages. |
||||
* A gRPC call might not complete until the request writer finishes. On the other hand, the request |
||||
* finishing doesn't necessarily make the call to finish, as the server might continue sending |
||||
* messages to the response side of the call indefinitely (depending on the semantics of the |
||||
* specific remote method called). |
||||
* To finish a call right away, invoke cancel. |
||||
* host parameter should not contain the scheme (http:// or https://), only the name or IP addr
|
||||
* and the port number, for example @"localhost:5050". |
||||
*/ |
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
requestsWriter:(GRXWriter *)requestWriter; |
||||
|
||||
/**
|
||||
* Finishes the request side of this call, notifies the server that the RPC should be cancelled, and |
||||
* finishes the response side of the call with an error of code CANCELED. |
||||
*/ |
||||
- (void)cancel; |
||||
|
||||
/**
|
||||
* The following methods are deprecated. |
||||
*/ |
||||
+ (void)setCallSafety:(GRPCCallSafety)callSafety host:(NSString *)host path:(NSString *)path; |
||||
@property(atomic, copy, readwrite) NSString *serverName; |
||||
@property NSTimeInterval timeout; |
||||
- (void)setResponseDispatchQueue:(dispatch_queue_t)queue; |
||||
|
||||
@end |
||||
|
||||
#pragma mark Backwards compatibiity |
||||
|
||||
/** This protocol is kept for backwards compatibility with existing code. */ |
||||
DEPRECATED_MSG_ATTRIBUTE("Use NSDictionary or NSMutableDictionary instead.") |
||||
@protocol GRPCRequestHeaders<NSObject> |
||||
@property(nonatomic, readonly) NSUInteger count; |
||||
|
||||
- (id)objectForKeyedSubscript:(id)key; |
||||
- (void)setObject:(id)obj forKeyedSubscript:(id)key; |
||||
|
||||
- (void)removeAllObjects; |
||||
- (void)removeObjectForKey:(id)key; |
||||
@end |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wdeprecated" |
||||
/** This is only needed for backwards-compatibility. */ |
||||
@interface NSMutableDictionary (GRPCRequestHeaders)<GRPCRequestHeaders> |
||||
@end |
||||
#pragma clang diagnostic pop |
||||
#pragma clang diagnostic pop |
||||
@ -0,0 +1,677 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCCallLegacy.h" |
||||
|
||||
#import "GRPCCall+OAuth2.h" |
||||
#import "GRPCCallOptions.h" |
||||
#import "GRPCTypes.h" |
||||
|
||||
#import "private/GRPCCore/GRPCChannelPool.h" |
||||
#import "private/GRPCCore/GRPCCompletionQueue.h" |
||||
#import "private/GRPCCore/GRPCHost.h" |
||||
#import "private/GRPCCore/GRPCWrappedCall.h" |
||||
#import "private/GRPCCore/NSData+GRPC.h" |
||||
|
||||
#import <RxLibrary/GRXBufferedPipe.h> |
||||
#import <RxLibrary/GRXConcurrentWriteable.h> |
||||
#import <RxLibrary/GRXImmediateSingleWriter.h> |
||||
#import <RxLibrary/GRXWriter+Immediate.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
|
||||
const char *kCFStreamVarName = "grpc_cfstream"; |
||||
static NSMutableDictionary *callFlags; |
||||
|
||||
// At most 6 ops can be in an op batch for a client: SEND_INITIAL_METADATA, |
||||
// SEND_MESSAGE, SEND_CLOSE_FROM_CLIENT, RECV_INITIAL_METADATA, RECV_MESSAGE, |
||||
// and RECV_STATUS_ON_CLIENT. |
||||
NSInteger kMaxClientBatch = 6; |
||||
|
||||
static NSString *const kAuthorizationHeader = @"authorization"; |
||||
static NSString *const kBearerPrefix = @"Bearer "; |
||||
|
||||
@interface GRPCCall ()<GRXWriteable> |
||||
// Make them read-write. |
||||
@property(atomic, strong) NSDictionary *responseHeaders; |
||||
@property(atomic, strong) NSDictionary *responseTrailers; |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages; |
||||
|
||||
@end |
||||
|
||||
// The following methods of a C gRPC call object aren't reentrant, and thus |
||||
// calls to them must be serialized: |
||||
// - start_batch |
||||
// - destroy |
||||
// |
||||
// start_batch with a SEND_MESSAGE argument can only be called after the |
||||
// OP_COMPLETE event for any previous write is received. This is achieved by |
||||
// pausing the requests writer immediately every time it writes a value, and |
||||
// resuming it again when OP_COMPLETE is received. |
||||
// |
||||
// Similarly, start_batch with a RECV_MESSAGE argument can only be called after |
||||
// the OP_COMPLETE event for any previous read is received.This is easier to |
||||
// enforce, as we're writing the received messages into the writeable: |
||||
// start_batch is enqueued once upon receiving the OP_COMPLETE event for the |
||||
// RECV_METADATA batch, and then once after receiving each OP_COMPLETE event for |
||||
// each RECV_MESSAGE batch. |
||||
@implementation GRPCCall { |
||||
dispatch_queue_t _callQueue; |
||||
|
||||
NSString *_host; |
||||
NSString *_path; |
||||
GRPCCallSafety _callSafety; |
||||
GRPCCallOptions *_callOptions; |
||||
GRPCWrappedCall *_wrappedCall; |
||||
|
||||
// The C gRPC library has less guarantees on the ordering of events than we |
||||
// do. Particularly, in the face of errors, there's no ordering guarantee at |
||||
// all. This wrapper over our actual writeable ensures thread-safety and |
||||
// correct ordering. |
||||
GRXConcurrentWriteable *_responseWriteable; |
||||
|
||||
// The network thread wants the requestWriter to resume (when the server is ready for more input), |
||||
// or to stop (on errors), concurrently with user threads that want to start it, pause it or stop |
||||
// it. Because a writer isn't thread-safe, we'll synchronize those operations on it. |
||||
// We don't use a dispatch queue for that purpose, because the writer can call writeValue: or |
||||
// writesFinishedWithError: on this GRPCCall as part of those operations. We want to be able to |
||||
// pause the writer immediately on writeValue:, so we need our locking to be recursive. |
||||
GRXWriter *_requestWriter; |
||||
|
||||
// To create a retain cycle when a call is started, up until it finishes. See |
||||
// |startWithWriteable:| and |finishWithError:|. This saves users from having to retain a |
||||
// reference to the call object if all they're interested in is the handler being executed when |
||||
// the response arrives. |
||||
GRPCCall *_retainSelf; |
||||
|
||||
GRPCRequestHeaders *_requestHeaders; |
||||
|
||||
// In the case that the call is a unary call (i.e. the writer to GRPCCall is of type |
||||
// GRXImmediateSingleWriter), GRPCCall will delay sending ops (not send them to C core |
||||
// immediately) and buffer them into a batch _unaryOpBatch. The batch is sent to C core when |
||||
// the SendClose op is added. |
||||
BOOL _unaryCall; |
||||
NSMutableArray *_unaryOpBatch; |
||||
|
||||
// The dispatch queue to be used for enqueuing responses to user. Defaulted to the main dispatch |
||||
// queue |
||||
dispatch_queue_t _responseQueue; |
||||
|
||||
// The OAuth2 token fetched from a token provider. |
||||
NSString *_fetchedOauth2AccessToken; |
||||
|
||||
// The callback to be called when a write message op is done. |
||||
void (^_writeDone)(void); |
||||
|
||||
// Indicate a read request to core is pending. |
||||
BOOL _pendingCoreRead; |
||||
|
||||
// Indicate pending read message request from user. |
||||
NSUInteger _pendingReceiveNextMessages; |
||||
} |
||||
|
||||
@synthesize state = _state; |
||||
|
||||
+ (void)initialize { |
||||
// Guarantees the code in {} block is invoked only once. See ref at: |
||||
// https://developer.apple.com/documentation/objectivec/nsobject/1418639-initialize?language=objc |
||||
if (self == [GRPCCall self]) { |
||||
grpc_init(); |
||||
callFlags = [NSMutableDictionary dictionary]; |
||||
} |
||||
} |
||||
|
||||
+ (void)setCallSafety:(GRPCCallSafety)callSafety host:(NSString *)host path:(NSString *)path { |
||||
if (host.length == 0 || path.length == 0) { |
||||
return; |
||||
} |
||||
NSString *hostAndPath = [NSString stringWithFormat:@"%@/%@", host, path]; |
||||
@synchronized(callFlags) { |
||||
switch (callSafety) { |
||||
case GRPCCallSafetyDefault: |
||||
callFlags[hostAndPath] = @0; |
||||
break; |
||||
case GRPCCallSafetyIdempotentRequest: |
||||
callFlags[hostAndPath] = @GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST; |
||||
break; |
||||
case GRPCCallSafetyCacheableRequest: |
||||
callFlags[hostAndPath] = @GRPC_INITIAL_METADATA_CACHEABLE_REQUEST; |
||||
break; |
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
+ (uint32_t)callFlagsForHost:(NSString *)host path:(NSString *)path { |
||||
NSString *hostAndPath = [NSString stringWithFormat:@"%@/%@", host, path]; |
||||
@synchronized(callFlags) { |
||||
return [callFlags[hostAndPath] intValue]; |
||||
} |
||||
} |
||||
|
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
requestsWriter:(GRXWriter *)requestWriter { |
||||
return [self initWithHost:host |
||||
path:path |
||||
callSafety:GRPCCallSafetyDefault |
||||
requestsWriter:requestWriter |
||||
callOptions:nil |
||||
writeDone:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
callSafety:(GRPCCallSafety)safety |
||||
requestsWriter:(GRXWriter *)requestsWriter |
||||
callOptions:(GRPCCallOptions *)callOptions |
||||
writeDone:(void (^)(void))writeDone { |
||||
// Purposely using pointer rather than length (host.length == 0) for backwards compatibility. |
||||
NSAssert(host != nil && path != nil, @"Neither host nor path can be nil."); |
||||
NSAssert(safety <= GRPCCallSafetyCacheableRequest, @"Invalid call safety value."); |
||||
NSAssert(requestsWriter.state == GRXWriterStateNotStarted, |
||||
@"The requests writer can't be already started."); |
||||
if (!host || !path) { |
||||
return nil; |
||||
} |
||||
if (safety > GRPCCallSafetyCacheableRequest) { |
||||
return nil; |
||||
} |
||||
if (requestsWriter.state != GRXWriterStateNotStarted) { |
||||
return nil; |
||||
} |
||||
|
||||
if ((self = [super init])) { |
||||
_host = [host copy]; |
||||
_path = [path copy]; |
||||
_callSafety = safety; |
||||
_callOptions = [callOptions copy]; |
||||
|
||||
// Serial queue to invoke the non-reentrant methods of the grpc_call object. |
||||
_callQueue = dispatch_queue_create("io.grpc.call", DISPATCH_QUEUE_SERIAL); |
||||
|
||||
_requestWriter = requestsWriter; |
||||
_requestHeaders = [[GRPCRequestHeaders alloc] initWithCall:self]; |
||||
_writeDone = writeDone; |
||||
|
||||
if ([requestsWriter isKindOfClass:[GRXImmediateSingleWriter class]]) { |
||||
_unaryCall = YES; |
||||
_unaryOpBatch = [NSMutableArray arrayWithCapacity:kMaxClientBatch]; |
||||
} |
||||
|
||||
_responseQueue = dispatch_get_main_queue(); |
||||
|
||||
// do not start a read until initial metadata is received |
||||
_pendingReceiveNextMessages = 0; |
||||
_pendingCoreRead = YES; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)setResponseDispatchQueue:(dispatch_queue_t)queue { |
||||
@synchronized(self) { |
||||
if (_state != GRXWriterStateNotStarted) { |
||||
return; |
||||
} |
||||
_responseQueue = queue; |
||||
} |
||||
} |
||||
|
||||
#pragma mark Finish |
||||
|
||||
// This function should support being called within a @synchronized(self) block in another function |
||||
// Should not manipulate _requestWriter for deadlock prevention. |
||||
- (void)finishWithError:(NSError *)errorOrNil { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
_state = GRXWriterStateFinished; |
||||
|
||||
if (errorOrNil) { |
||||
[_responseWriteable cancelWithError:errorOrNil]; |
||||
} else { |
||||
[_responseWriteable enqueueSuccessfulCompletion]; |
||||
} |
||||
|
||||
// If the call isn't retained anywhere else, it can be deallocated now. |
||||
_retainSelf = nil; |
||||
} |
||||
} |
||||
|
||||
- (void)cancel { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
[self finishWithError:[NSError |
||||
errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeCancelled |
||||
userInfo:@{NSLocalizedDescriptionKey : @"Canceled by app"}]]; |
||||
[_wrappedCall cancel]; |
||||
} |
||||
_requestWriter.state = GRXWriterStateFinished; |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
__block GRPCWrappedCall *wrappedCall = _wrappedCall; |
||||
dispatch_async(_callQueue, ^{ |
||||
wrappedCall = nil; |
||||
}); |
||||
} |
||||
|
||||
#pragma mark Read messages |
||||
|
||||
// Only called from the call queue. |
||||
// The handler will be called from the network queue. |
||||
- (void)startReadWithHandler:(void (^)(grpc_byte_buffer *))handler { |
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
[_wrappedCall startBatchWithOperations:@[ [[GRPCOpRecvMessage alloc] initWithHandler:handler] ]]; |
||||
} |
||||
|
||||
// Called initially from the network queue once response headers are received, |
||||
// then "recursively" from the responseWriteable queue after each response from the |
||||
// server has been written. |
||||
// If the call is currently paused, this is a noop. Restarting the call will invoke this |
||||
// method. |
||||
// TODO(jcanizales): Rename to readResponseIfNotPaused. |
||||
- (void)maybeStartNextRead { |
||||
@synchronized(self) { |
||||
if (_state != GRXWriterStateStarted) { |
||||
return; |
||||
} |
||||
if (_callOptions.flowControlEnabled && (_pendingCoreRead || _pendingReceiveNextMessages == 0)) { |
||||
return; |
||||
} |
||||
_pendingCoreRead = YES; |
||||
_pendingReceiveNextMessages--; |
||||
} |
||||
|
||||
dispatch_async(_callQueue, ^{ |
||||
__weak GRPCCall *weakSelf = self; |
||||
[self startReadWithHandler:^(grpc_byte_buffer *message) { |
||||
if (message == NULL) { |
||||
// No more messages from the server |
||||
return; |
||||
} |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf == nil) { |
||||
grpc_byte_buffer_destroy(message); |
||||
return; |
||||
} |
||||
NSData *data = [NSData grpc_dataWithByteBuffer:message]; |
||||
grpc_byte_buffer_destroy(message); |
||||
if (!data) { |
||||
// The app doesn't have enough memory to hold the server response. We |
||||
// don't want to throw, because the app shouldn't crash for a behavior |
||||
// that's on the hands of any server to have. Instead we finish and ask |
||||
// the server to cancel. |
||||
@synchronized(strongSelf) { |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf |
||||
finishWithError:[NSError errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeResourceExhausted |
||||
userInfo:@{ |
||||
NSLocalizedDescriptionKey : |
||||
@"Client does not have enough memory to " |
||||
@"hold the server response." |
||||
}]]; |
||||
[strongSelf->_wrappedCall cancel]; |
||||
} |
||||
strongSelf->_requestWriter.state = GRXWriterStateFinished; |
||||
} else { |
||||
@synchronized(strongSelf) { |
||||
[strongSelf->_responseWriteable enqueueValue:data |
||||
completionHandler:^{ |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
@synchronized(strongSelf) { |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf maybeStartNextRead]; |
||||
} |
||||
} |
||||
}]; |
||||
} |
||||
} |
||||
}]; |
||||
}); |
||||
} |
||||
|
||||
#pragma mark Send headers |
||||
|
||||
- (void)sendHeaders { |
||||
// TODO (mxyan): Remove after deprecated methods are removed |
||||
uint32_t callSafetyFlags = 0; |
||||
switch (_callSafety) { |
||||
case GRPCCallSafetyDefault: |
||||
callSafetyFlags = 0; |
||||
break; |
||||
case GRPCCallSafetyIdempotentRequest: |
||||
callSafetyFlags = GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST; |
||||
break; |
||||
case GRPCCallSafetyCacheableRequest: |
||||
callSafetyFlags = GRPC_INITIAL_METADATA_CACHEABLE_REQUEST; |
||||
break; |
||||
} |
||||
|
||||
NSMutableDictionary *headers = [_requestHeaders mutableCopy]; |
||||
NSString *fetchedOauth2AccessToken; |
||||
@synchronized(self) { |
||||
fetchedOauth2AccessToken = _fetchedOauth2AccessToken; |
||||
} |
||||
if (fetchedOauth2AccessToken != nil) { |
||||
headers[@"authorization"] = [kBearerPrefix stringByAppendingString:fetchedOauth2AccessToken]; |
||||
} else if (_callOptions.oauth2AccessToken != nil) { |
||||
headers[@"authorization"] = |
||||
[kBearerPrefix stringByAppendingString:_callOptions.oauth2AccessToken]; |
||||
} |
||||
|
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
GRPCOpSendMetadata *op = [[GRPCOpSendMetadata alloc] |
||||
initWithMetadata:headers |
||||
flags:callSafetyFlags |
||||
handler:nil]; // No clean-up needed after SEND_INITIAL_METADATA |
||||
dispatch_async(_callQueue, ^{ |
||||
if (!self->_unaryCall) { |
||||
[self->_wrappedCall startBatchWithOperations:@[ op ]]; |
||||
} else { |
||||
[self->_unaryOpBatch addObject:op]; |
||||
} |
||||
}); |
||||
} |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages { |
||||
if (numberOfMessages == 0) { |
||||
return; |
||||
} |
||||
@synchronized(self) { |
||||
_pendingReceiveNextMessages += numberOfMessages; |
||||
|
||||
if (_state != GRXWriterStateStarted || !_callOptions.flowControlEnabled) { |
||||
return; |
||||
} |
||||
[self maybeStartNextRead]; |
||||
} |
||||
} |
||||
|
||||
#pragma mark GRXWriteable implementation |
||||
|
||||
// Only called from the call queue. The error handler will be called from the |
||||
// network queue if the write didn't succeed. |
||||
// If the call is a unary call, parameter \a errorHandler will be ignored and |
||||
// the error handler of GRPCOpSendClose will be executed in case of error. |
||||
- (void)writeMessage:(NSData *)message withErrorHandler:(void (^)(void))errorHandler { |
||||
__weak GRPCCall *weakSelf = self; |
||||
void (^resumingHandler)(void) = ^{ |
||||
// Resume the request writer. |
||||
GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
strongSelf->_requestWriter.state = GRXWriterStateStarted; |
||||
if (strongSelf->_writeDone) { |
||||
strongSelf->_writeDone(); |
||||
} |
||||
} |
||||
}; |
||||
GRPCOpSendMessage *op = |
||||
[[GRPCOpSendMessage alloc] initWithMessage:message handler:resumingHandler]; |
||||
if (!_unaryCall) { |
||||
[_wrappedCall startBatchWithOperations:@[ op ] errorHandler:errorHandler]; |
||||
} else { |
||||
// Ignored errorHandler since it is the same as the one for GRPCOpSendClose. |
||||
// TODO (mxyan): unify the error handlers of all Ops into a single closure. |
||||
[_unaryOpBatch addObject:op]; |
||||
} |
||||
} |
||||
|
||||
- (void)writeValue:(id)value { |
||||
NSAssert([value isKindOfClass:[NSData class]], @"value must be of type NSData"); |
||||
|
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
} |
||||
|
||||
// Pause the input and only resume it when the C layer notifies us that writes |
||||
// can proceed. |
||||
_requestWriter.state = GRXWriterStatePaused; |
||||
|
||||
dispatch_async(_callQueue, ^{ |
||||
// Write error is not processed here. It is handled by op batch of GRPC_OP_RECV_STATUS_ON_CLIENT |
||||
[self writeMessage:value withErrorHandler:nil]; |
||||
}); |
||||
} |
||||
|
||||
// Only called from the call queue. The error handler will be called from the |
||||
// network queue if the requests stream couldn't be closed successfully. |
||||
- (void)finishRequestWithErrorHandler:(void (^)(void))errorHandler { |
||||
if (!_unaryCall) { |
||||
[_wrappedCall startBatchWithOperations:@[ [[GRPCOpSendClose alloc] init] ] |
||||
errorHandler:errorHandler]; |
||||
} else { |
||||
[_unaryOpBatch addObject:[[GRPCOpSendClose alloc] init]]; |
||||
[_wrappedCall startBatchWithOperations:_unaryOpBatch errorHandler:errorHandler]; |
||||
} |
||||
} |
||||
|
||||
- (void)writesFinishedWithError:(NSError *)errorOrNil { |
||||
if (errorOrNil) { |
||||
[self cancel]; |
||||
} else { |
||||
dispatch_async(_callQueue, ^{ |
||||
// EOS error is not processed here. It is handled by op batch of GRPC_OP_RECV_STATUS_ON_CLIENT |
||||
[self finishRequestWithErrorHandler:nil]; |
||||
}); |
||||
} |
||||
} |
||||
|
||||
#pragma mark Invoke |
||||
|
||||
// Both handlers will eventually be called, from the network queue. Writes can start immediately |
||||
// after this. |
||||
// The first one (headersHandler), when the response headers are received. |
||||
// The second one (completionHandler), whenever the RPC finishes for any reason. |
||||
- (void)invokeCallWithHeadersHandler:(void (^)(NSDictionary *))headersHandler |
||||
completionHandler:(void (^)(NSError *, NSDictionary *))completionHandler { |
||||
dispatch_async(_callQueue, ^{ |
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
[self->_wrappedCall |
||||
startBatchWithOperations:@[ [[GRPCOpRecvMetadata alloc] initWithHandler:headersHandler] ]]; |
||||
[self->_wrappedCall |
||||
startBatchWithOperations:@[ [[GRPCOpRecvStatus alloc] initWithHandler:completionHandler] ]]; |
||||
}); |
||||
} |
||||
|
||||
- (void)invokeCall { |
||||
__weak GRPCCall *weakSelf = self; |
||||
[self invokeCallWithHeadersHandler:^(NSDictionary *headers) { |
||||
// Response headers received. |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
@synchronized(strongSelf) { |
||||
// it is ok to set nil because headers are only received once |
||||
strongSelf.responseHeaders = nil; |
||||
// copy the header so that the GRPCOpRecvMetadata object may be dealloc'ed |
||||
NSDictionary *copiedHeaders = |
||||
[[NSDictionary alloc] initWithDictionary:headers copyItems:YES]; |
||||
strongSelf.responseHeaders = copiedHeaders; |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf maybeStartNextRead]; |
||||
} |
||||
} |
||||
} |
||||
completionHandler:^(NSError *error, NSDictionary *trailers) { |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
strongSelf.responseTrailers = trailers; |
||||
|
||||
if (error) { |
||||
NSMutableDictionary *userInfo = [NSMutableDictionary dictionary]; |
||||
if (error.userInfo) { |
||||
[userInfo addEntriesFromDictionary:error.userInfo]; |
||||
} |
||||
userInfo[kGRPCTrailersKey] = strongSelf.responseTrailers; |
||||
// Since gRPC core does not guarantee the headers block being called before this block, |
||||
// responseHeaders might be nil. |
||||
userInfo[kGRPCHeadersKey] = strongSelf.responseHeaders; |
||||
error = [NSError errorWithDomain:error.domain code:error.code userInfo:userInfo]; |
||||
} |
||||
[strongSelf finishWithError:error]; |
||||
strongSelf->_requestWriter.state = GRXWriterStateFinished; |
||||
} |
||||
}]; |
||||
} |
||||
|
||||
#pragma mark GRXWriter implementation |
||||
|
||||
// Lock acquired inside startWithWriteable: |
||||
- (void)startCallWithWriteable:(id<GRXWriteable>)writeable { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
|
||||
_responseWriteable = |
||||
[[GRXConcurrentWriteable alloc] initWithWriteable:writeable dispatchQueue:_responseQueue]; |
||||
|
||||
GRPCPooledChannel *channel = |
||||
[[GRPCChannelPool sharedInstance] channelWithHost:_host callOptions:_callOptions]; |
||||
_wrappedCall = [channel wrappedCallWithPath:_path |
||||
completionQueue:[GRPCCompletionQueue completionQueue] |
||||
callOptions:_callOptions]; |
||||
|
||||
if (_wrappedCall == nil) { |
||||
[self finishWithError:[NSError errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeUnavailable |
||||
userInfo:@{ |
||||
NSLocalizedDescriptionKey : |
||||
@"Failed to create call or channel." |
||||
}]]; |
||||
return; |
||||
} |
||||
|
||||
[self sendHeaders]; |
||||
[self invokeCall]; |
||||
} |
||||
|
||||
// Now that the RPC has been initiated, request writes can start. |
||||
[_requestWriter startWithWriteable:self]; |
||||
} |
||||
|
||||
- (void)startWithWriteable:(id<GRXWriteable>)writeable { |
||||
id<GRPCAuthorizationProtocol> tokenProvider = nil; |
||||
@synchronized(self) { |
||||
_state = GRXWriterStateStarted; |
||||
|
||||
// Create a retain cycle so that this instance lives until the RPC finishes (or is cancelled). |
||||
// This makes RPCs in which the call isn't externally retained possible (as long as it is |
||||
// started before being autoreleased). Care is taken not to retain self strongly in any of the |
||||
// blocks used in this implementation, so that the life of the instance is determined by this |
||||
// retain cycle. |
||||
_retainSelf = self; |
||||
|
||||
// If _callOptions is nil, people must be using the deprecated v1 interface. In this case, |
||||
// generate the call options from the corresponding GRPCHost configs and apply other options |
||||
// that are not covered by GRPCHost. |
||||
if (_callOptions == nil) { |
||||
GRPCMutableCallOptions *callOptions = [[GRPCHost callOptionsForHost:_host] mutableCopy]; |
||||
if (_serverName.length != 0) { |
||||
callOptions.serverAuthority = _serverName; |
||||
} |
||||
if (_timeout > 0) { |
||||
callOptions.timeout = _timeout; |
||||
} |
||||
uint32_t callFlags = [GRPCCall callFlagsForHost:_host path:_path]; |
||||
if (callFlags != 0) { |
||||
if (callFlags == GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST) { |
||||
_callSafety = GRPCCallSafetyIdempotentRequest; |
||||
} else if (callFlags == GRPC_INITIAL_METADATA_CACHEABLE_REQUEST) { |
||||
_callSafety = GRPCCallSafetyCacheableRequest; |
||||
} |
||||
} |
||||
|
||||
id<GRPCAuthorizationProtocol> tokenProvider = self.tokenProvider; |
||||
if (tokenProvider != nil) { |
||||
callOptions.authTokenProvider = tokenProvider; |
||||
} |
||||
_callOptions = callOptions; |
||||
} |
||||
|
||||
NSAssert(_callOptions.authTokenProvider == nil || _callOptions.oauth2AccessToken == nil, |
||||
@"authTokenProvider and oauth2AccessToken cannot be set at the same time"); |
||||
|
||||
tokenProvider = _callOptions.authTokenProvider; |
||||
} |
||||
|
||||
if (tokenProvider != nil) { |
||||
__weak typeof(self) weakSelf = self; |
||||
[tokenProvider getTokenWithHandler:^(NSString *token) { |
||||
__strong typeof(self) strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
BOOL startCall = NO; |
||||
@synchronized(strongSelf) { |
||||
if (strongSelf->_state != GRXWriterStateFinished) { |
||||
startCall = YES; |
||||
if (token) { |
||||
strongSelf->_fetchedOauth2AccessToken = [token copy]; |
||||
} |
||||
} |
||||
} |
||||
if (startCall) { |
||||
[strongSelf startCallWithWriteable:writeable]; |
||||
} |
||||
} |
||||
}]; |
||||
} else { |
||||
[self startCallWithWriteable:writeable]; |
||||
} |
||||
} |
||||
|
||||
- (void)setState:(GRXWriterState)newState { |
||||
@synchronized(self) { |
||||
// Manual transitions are only allowed from the started or paused states. |
||||
if (_state == GRXWriterStateNotStarted || _state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
|
||||
switch (newState) { |
||||
case GRXWriterStateFinished: |
||||
_state = newState; |
||||
// Per GRXWriter's contract, setting the state to Finished manually |
||||
// means one doesn't wish the writeable to be messaged anymore. |
||||
[_responseWriteable cancelSilently]; |
||||
_responseWriteable = nil; |
||||
return; |
||||
case GRXWriterStatePaused: |
||||
_state = newState; |
||||
return; |
||||
case GRXWriterStateStarted: |
||||
if (_state == GRXWriterStatePaused) { |
||||
_state = newState; |
||||
[self maybeStartNextRead]; |
||||
} |
||||
return; |
||||
case GRXWriterStateNotStarted: |
||||
return; |
||||
} |
||||
} |
||||
} |
||||
|
||||
@end |
||||
@ -0,0 +1,30 @@ |
||||
|
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* An object that processes its methods with a dispatch queue. |
||||
*/ |
||||
@protocol GRPCDispatchable |
||||
|
||||
/**
|
||||
* The dispatch queue where the object's methods should be run on. |
||||
*/ |
||||
@property(atomic, readonly) dispatch_queue_t dispatchQueue; |
||||
|
||||
@end |
||||
@ -0,0 +1,82 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
// The interface for a transport implementation
|
||||
|
||||
#import "GRPCInterceptor.h" |
||||
|
||||
NS_ASSUME_NONNULL_BEGIN |
||||
|
||||
#pragma mark Transport ID |
||||
|
||||
/**
|
||||
* The default transport implementations available in gRPC. These implementations will be provided |
||||
* by gRPC by default unless explicitly excluded. |
||||
*/ |
||||
extern const struct GRPCDefaultTransportImplList { |
||||
const GRPCTransportId core_secure; |
||||
const GRPCTransportId core_insecure; |
||||
} GRPCDefaultTransportImplList; |
||||
|
||||
/** Returns whether two transport id's are identical. */ |
||||
BOOL TransportIdIsEqual(GRPCTransportId lhs, GRPCTransportId rhs); |
||||
|
||||
/** Returns the hash value of a transport id. */ |
||||
NSUInteger TransportIdHash(GRPCTransportId); |
||||
|
||||
#pragma mark Transport and factory |
||||
|
||||
@protocol GRPCInterceptorInterface; |
||||
@protocol GRPCResponseHandler; |
||||
@class GRPCTransportManager; |
||||
@class GRPCRequestOptions; |
||||
@class GRPCCallOptions; |
||||
@class GRPCTransport; |
||||
|
||||
/** The factory method to create a transport. */ |
||||
@protocol GRPCTransportFactory<NSObject> |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager; |
||||
|
||||
@end |
||||
|
||||
/** The registry of transport implementations. */ |
||||
@interface GRPCTransportRegistry : NSObject |
||||
|
||||
+ (instancetype)sharedInstance; |
||||
|
||||
/**
|
||||
* Register a transport implementation with the registry. All transport implementations to be used |
||||
* in a process must register with the registry on process start-up in its +load: class method. |
||||
* Parameter \a transportId is the identifier of the implementation, and \a factory is the factory |
||||
* object to create the corresponding transport instance. |
||||
*/ |
||||
- (void)registerTransportWithId:(GRPCTransportId)transportId |
||||
factory:(id<GRPCTransportFactory>)factory; |
||||
|
||||
@end |
||||
|
||||
/**
|
||||
* Base class for transport implementations. All transport implementation should inherit from this |
||||
* class. |
||||
*/ |
||||
@interface GRPCTransport : NSObject<GRPCInterceptorInterface> |
||||
|
||||
@end |
||||
|
||||
NS_ASSUME_NONNULL_END |
||||
@ -0,0 +1,142 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCTransport.h" |
||||
|
||||
static const GRPCTransportId gGRPCCoreSecureId = "io.grpc.transport.core.secure"; |
||||
static const GRPCTransportId gGRPCCoreInsecureId = "io.grpc.transport.core.insecure"; |
||||
|
||||
const struct GRPCDefaultTransportImplList GRPCDefaultTransportImplList = { |
||||
.core_secure = gGRPCCoreSecureId, .core_insecure = gGRPCCoreInsecureId}; |
||||
|
||||
static const GRPCTransportId gDefaultTransportId = gGRPCCoreSecureId; |
||||
|
||||
static GRPCTransportRegistry *gTransportRegistry = nil; |
||||
static dispatch_once_t initTransportRegistry; |
||||
|
||||
BOOL TransportIdIsEqual(GRPCTransportId lhs, GRPCTransportId rhs) { |
||||
// Directly comparing pointers works because we require users to use the id provided by each |
||||
// implementation, not coming up with their own string. |
||||
return lhs == rhs; |
||||
} |
||||
|
||||
NSUInteger TransportIdHash(GRPCTransportId transportId) { |
||||
if (transportId == NULL) { |
||||
transportId = gDefaultTransportId; |
||||
} |
||||
return [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding].hash; |
||||
} |
||||
|
||||
@implementation GRPCTransportRegistry { |
||||
NSMutableDictionary<NSString *, id<GRPCTransportFactory>> *_registry; |
||||
id<GRPCTransportFactory> _defaultFactory; |
||||
} |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&initTransportRegistry, ^{ |
||||
gTransportRegistry = [[GRPCTransportRegistry alloc] init]; |
||||
NSAssert(gTransportRegistry != nil, @"Unable to initialize transport registry."); |
||||
if (gTransportRegistry == nil) { |
||||
NSLog(@"Unable to initialize transport registry."); |
||||
[NSException raise:NSGenericException format:@"Unable to initialize transport registry."]; |
||||
} |
||||
}); |
||||
return gTransportRegistry; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
if ((self = [super init])) { |
||||
_registry = [NSMutableDictionary dictionary]; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)registerTransportWithId:(GRPCTransportId)transportId |
||||
factory:(id<GRPCTransportFactory>)factory { |
||||
NSString *nsTransportId = [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding]; |
||||
NSAssert(_registry[nsTransportId] == nil, @"The transport %@ has already been registered.", |
||||
nsTransportId); |
||||
if (_registry[nsTransportId] != nil) { |
||||
NSLog(@"The transport %@ has already been registered.", nsTransportId); |
||||
return; |
||||
} |
||||
_registry[nsTransportId] = factory; |
||||
|
||||
// if the default transport is registered, mark it. |
||||
if (0 == strcmp(transportId, gDefaultTransportId)) { |
||||
_defaultFactory = factory; |
||||
} |
||||
} |
||||
|
||||
- (id<GRPCTransportFactory>)getTransportFactoryWithId:(GRPCTransportId)transportId { |
||||
if (transportId == NULL) { |
||||
if (_defaultFactory == nil) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Unable to get default transport factory"]; |
||||
return nil; |
||||
} |
||||
return _defaultFactory; |
||||
} |
||||
NSString *nsTransportId = [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding]; |
||||
id<GRPCTransportFactory> transportFactory = _registry[nsTransportId]; |
||||
if (transportFactory == nil) { |
||||
// User named a transport id that was not registered with the registry. |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Unable to get transport factory with id %s", transportId]; |
||||
return nil; |
||||
} |
||||
return transportFactory; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCTransport |
||||
|
||||
- (dispatch_queue_t)dispatchQueue { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the dispatch queue"]; |
||||
return nil; |
||||
} |
||||
|
||||
- (void)startWithRequestOptions:(nonnull GRPCRequestOptions *)requestOptions |
||||
callOptions:(nonnull GRPCCallOptions *)callOptions { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)writeData:(nonnull id)data { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)cancel { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)finish { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
@end |
||||
@ -0,0 +1,187 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* gRPC error codes. |
||||
* Note that a few of these are never produced by the gRPC libraries, but are of |
||||
* general utility for server applications to produce. |
||||
*/ |
||||
typedef NS_ENUM(NSUInteger, GRPCErrorCode) { |
||||
/** The operation was cancelled (typically by the caller). */ |
||||
GRPCErrorCodeCancelled = 1, |
||||
|
||||
/**
|
||||
* Unknown error. Errors raised by APIs that do not return enough error |
||||
* information may be converted to this error. |
||||
*/ |
||||
GRPCErrorCodeUnknown = 2, |
||||
|
||||
/**
|
||||
* The client specified an invalid argument. Note that this differs from |
||||
* FAILED_PRECONDITION. INVALID_ARGUMENT indicates arguments that are |
||||
* problematic regardless of the state of the server (e.g., a malformed file |
||||
* name). |
||||
*/ |
||||
GRPCErrorCodeInvalidArgument = 3, |
||||
|
||||
/**
|
||||
* Deadline expired before operation could complete. For operations that |
||||
* change the state of the server, this error may be returned even if the |
||||
* operation has completed successfully. For example, a successful response |
||||
* from the server could have been delayed long enough for the deadline to |
||||
* expire. |
||||
*/ |
||||
GRPCErrorCodeDeadlineExceeded = 4, |
||||
|
||||
/** Some requested entity (e.g., file or directory) was not found. */ |
||||
GRPCErrorCodeNotFound = 5, |
||||
|
||||
/** Some entity that we attempted to create (e.g., file or directory) already
|
||||
exists. */ |
||||
GRPCErrorCodeAlreadyExists = 6, |
||||
|
||||
/**
|
||||
* The caller does not have permission to execute the specified operation. |
||||
* PERMISSION_DENIED isn't used for rejections caused by exhausting some |
||||
* resource (RESOURCE_EXHAUSTED is used instead for those errors). |
||||
* PERMISSION_DENIED doesn't indicate a failure to identify the caller |
||||
* (UNAUTHENTICATED is used instead for those errors). |
||||
*/ |
||||
GRPCErrorCodePermissionDenied = 7, |
||||
|
||||
/**
|
||||
* The request does not have valid authentication credentials for the |
||||
* operation (e.g. the caller's identity can't be verified). |
||||
*/ |
||||
GRPCErrorCodeUnauthenticated = 16, |
||||
|
||||
/** Some resource has been exhausted, perhaps a per-user quota. */ |
||||
GRPCErrorCodeResourceExhausted = 8, |
||||
|
||||
/**
|
||||
* The RPC was rejected because the server is not in a state required for the |
||||
* procedure's execution. For example, a directory to be deleted may be |
||||
* non-empty, etc. The client should not retry until the server state has been |
||||
* explicitly fixed (e.g. by performing another RPC). The details depend on |
||||
* the service being called, and should be found in the NSError's userInfo. |
||||
*/ |
||||
GRPCErrorCodeFailedPrecondition = 9, |
||||
|
||||
/**
|
||||
* The RPC was aborted, typically due to a concurrency issue like sequencer |
||||
* check failures, transaction aborts, etc. The client should retry at a |
||||
* higher-level (e.g., restarting a read- modify-write sequence). |
||||
*/ |
||||
GRPCErrorCodeAborted = 10, |
||||
|
||||
/**
|
||||
* The RPC was attempted past the valid range. E.g., enumerating past the end |
||||
* of a list. Unlike INVALID_ARGUMENT, this error indicates a problem that may |
||||
* be fixed if the system state changes. For example, an RPC to get elements |
||||
* of a list will generate INVALID_ARGUMENT if asked to return the element at |
||||
* a negative index, but it will generate OUT_OF_RANGE if asked to return the |
||||
* element at an index past the current size of the list. |
||||
*/ |
||||
GRPCErrorCodeOutOfRange = 11, |
||||
|
||||
/** The procedure is not implemented or not supported/enabled in this server.
|
||||
*/ |
||||
GRPCErrorCodeUnimplemented = 12, |
||||
|
||||
/**
|
||||
* Internal error. Means some invariant expected by the server application or |
||||
* the gRPC library has been broken. |
||||
*/ |
||||
GRPCErrorCodeInternal = 13, |
||||
|
||||
/**
|
||||
* The server is currently unavailable. This is most likely a transient |
||||
* condition and may be corrected by retrying with a backoff. Note that it is |
||||
* not always safe to retry non-idempotent operations. |
||||
*/ |
||||
GRPCErrorCodeUnavailable = 14, |
||||
|
||||
/** Unrecoverable data loss or corruption. */ |
||||
GRPCErrorCodeDataLoss = 15, |
||||
}; |
||||
|
||||
/**
|
||||
* Safety remark of a gRPC method as defined in RFC 2616 Section 9.1 |
||||
*/ |
||||
typedef NS_ENUM(NSUInteger, GRPCCallSafety) { |
||||
/**
|
||||
* Signal that there is no guarantees on how the call affects the server |
||||
* state. |
||||
*/ |
||||
GRPCCallSafetyDefault = 0, |
||||
/** Signal that the call is idempotent. gRPC is free to use PUT verb. */ |
||||
GRPCCallSafetyIdempotentRequest = 1, |
||||
/**
|
||||
* Signal that the call is cacheable and will not affect server state. gRPC is |
||||
* free to use GET verb. |
||||
*/ |
||||
GRPCCallSafetyCacheableRequest = 2, |
||||
}; |
||||
|
||||
// Compression algorithm to be used by a gRPC call
|
||||
typedef NS_ENUM(NSUInteger, GRPCCompressionAlgorithm) { |
||||
GRPCCompressNone = 0, |
||||
GRPCCompressDeflate, |
||||
GRPCCompressGzip, |
||||
GRPCStreamCompressGzip, |
||||
}; |
||||
|
||||
// GRPCCompressAlgorithm is deprecated; use GRPCCompressionAlgorithm
|
||||
typedef GRPCCompressionAlgorithm GRPCCompressAlgorithm; |
||||
|
||||
/** The transport to be used by a gRPC call */ |
||||
typedef NS_ENUM(NSUInteger, GRPCTransportType) { |
||||
GRPCTransportTypeDefault = 0, |
||||
/** gRPC internal HTTP/2 stack with BoringSSL */ |
||||
GRPCTransportTypeChttp2BoringSSL = 0, |
||||
/** Cronet stack */ |
||||
GRPCTransportTypeCronet, |
||||
/** Insecure channel. FOR TEST ONLY! */ |
||||
GRPCTransportTypeInsecure, |
||||
}; |
||||
|
||||
/** Domain of NSError objects produced by gRPC. */ |
||||
extern NSString* _Nonnull const kGRPCErrorDomain; |
||||
|
||||
/**
|
||||
* Keys used in |NSError|'s |userInfo| dictionary to store the response headers |
||||
* and trailers sent by the server. |
||||
*/ |
||||
extern NSString* _Nonnull const kGRPCHeadersKey; |
||||
extern NSString* _Nonnull const kGRPCTrailersKey; |
||||
|
||||
/** The id of a transport implementation. */ |
||||
typedef char* _Nonnull GRPCTransportId; |
||||
|
||||
/**
|
||||
* Implement this protocol to provide a token to gRPC when a call is initiated. |
||||
*/ |
||||
@protocol GRPCAuthorizationProtocol |
||||
|
||||
/**
|
||||
* This method is called when gRPC is about to start the call. When OAuth token is acquired, |
||||
* \a handler is expected to be called with \a token being the new token to be used for this call. |
||||
*/ |
||||
- (void)getTokenWithHandler:(void (^_Nonnull)(NSString* _Nullable token))handler; |
||||
|
||||
@end |
||||
@ -0,0 +1,32 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "../GRPCCoreFactory.h" |
||||
|
||||
/**
|
||||
* The factory for gRPC Core + Cronet transport implementation. The |
||||
* implementation is not part of the default transports of gRPC and is for |
||||
* testing purpose only on Github. |
||||
* |
||||
* To use this transport, a user must include the GRPCCoreCronet module as a |
||||
* dependency of the project and use gGRPCCoreCronetId in call options to |
||||
* specify that this is the transport to be used for a call. |
||||
*/ |
||||
@interface GRPCCoreCronetFactory : NSObject<GRPCCoreTransportFactory> |
||||
|
||||
@end |
||||
@ -0,0 +1,54 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCCoreCronetFactory.h" |
||||
|
||||
#import <GRPCClient/GRPCCall+Cronet.h> |
||||
#import <GRPCClient/GRPCTransport.h> |
||||
|
||||
#import "../GRPCCallInternal.h" |
||||
#import "../GRPCCoreFactory.h" |
||||
#import "GRPCCronetChannelFactory.h" |
||||
|
||||
static GRPCCoreCronetFactory *gGRPCCoreCronetFactory = nil; |
||||
static dispatch_once_t gInitGRPCCoreCronetFactory; |
||||
|
||||
@implementation GRPCCoreCronetFactory |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&gInitGRPCCoreCronetFactory, ^{ |
||||
gGRPCCoreCronetFactory = [[GRPCCoreCronetFactory alloc] init]; |
||||
}); |
||||
return gGRPCCoreCronetFactory; |
||||
} |
||||
|
||||
+ (void)load { |
||||
[[GRPCTransportRegistry sharedInstance] |
||||
registerTransportWithId:gGRPCCoreCronetId |
||||
factory:[GRPCCoreCronetFactory sharedInstance]]; |
||||
} |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager { |
||||
return [[GRPCCall2Internal alloc] initWithTransportManager:transportManager]; |
||||
} |
||||
|
||||
- (id<GRPCChannelFactory>)createCoreChannelFactoryWithCallOptions:(GRPCCallOptions *)callOptions { |
||||
return [GRPCCronetChannelFactory sharedInstance]; |
||||
} |
||||
|
||||
@end |
||||
@ -0,0 +1,45 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import <GRPCClient/GRPCTransport.h> |
||||
|
||||
NS_ASSUME_NONNULL_BEGIN |
||||
|
||||
@protocol GRPCChannelFactory; |
||||
@protocol GRPCCallOptions; |
||||
|
||||
/** The interface for transport implementations that are based on Core. */ |
||||
@protocol GRPCCoreTransportFactory<GRPCTransportFactory> |
||||
|
||||
/** Get the channel factory for GRPCChannel from call options. */ |
||||
- (nullable id<GRPCChannelFactory>)createCoreChannelFactoryWithCallOptions: |
||||
(GRPCCallOptions *)callOptions; |
||||
|
||||
@end |
||||
|
||||
/** The factory for gRPC Core + CFStream + TLS secure channel transport implementation. */ |
||||
@interface GRPCCoreSecureFactory : NSObject<GRPCCoreTransportFactory> |
||||
|
||||
@end |
||||
|
||||
/** The factory for gRPC Core + CFStream + insecure channel transport implementation. */ |
||||
@interface GRPCCoreInsecureFactory : NSObject<GRPCCoreTransportFactory> |
||||
|
||||
@end |
||||
|
||||
NS_ASSUME_NONNULL_END |
||||
@ -0,0 +1,90 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCCoreFactory.h" |
||||
|
||||
#import <GRPCClient/GRPCTransport.h> |
||||
|
||||
#import "GRPCCallInternal.h" |
||||
#import "GRPCInsecureChannelFactory.h" |
||||
#import "GRPCSecureChannelFactory.h" |
||||
|
||||
static GRPCCoreSecureFactory *gGRPCCoreSecureFactory = nil; |
||||
static GRPCCoreInsecureFactory *gGRPCCoreInsecureFactory = nil; |
||||
static dispatch_once_t gInitGRPCCoreSecureFactory; |
||||
static dispatch_once_t gInitGRPCCoreInsecureFactory; |
||||
|
||||
@implementation GRPCCoreSecureFactory |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&gInitGRPCCoreSecureFactory, ^{ |
||||
gGRPCCoreSecureFactory = [[GRPCCoreSecureFactory alloc] init]; |
||||
}); |
||||
return gGRPCCoreSecureFactory; |
||||
} |
||||
|
||||
+ (void)load { |
||||
[[GRPCTransportRegistry sharedInstance] |
||||
registerTransportWithId:GRPCDefaultTransportImplList.core_secure |
||||
factory:[self sharedInstance]]; |
||||
} |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager { |
||||
return [[GRPCCall2Internal alloc] initWithTransportManager:transportManager]; |
||||
} |
||||
|
||||
- (id<GRPCChannelFactory>)createCoreChannelFactoryWithCallOptions:(GRPCCallOptions *)callOptions { |
||||
NSError *error; |
||||
id<GRPCChannelFactory> factory = |
||||
[GRPCSecureChannelFactory factoryWithPEMRootCertificates:callOptions.PEMRootCertificates |
||||
privateKey:callOptions.PEMPrivateKey |
||||
certChain:callOptions.PEMCertificateChain |
||||
error:&error]; |
||||
if (error != nil) { |
||||
NSLog(@"Unable to create secure channel factory"); |
||||
return nil; |
||||
} |
||||
return factory; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCCoreInsecureFactory |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&gInitGRPCCoreInsecureFactory, ^{ |
||||
gGRPCCoreInsecureFactory = [[GRPCCoreInsecureFactory alloc] init]; |
||||
}); |
||||
return gGRPCCoreInsecureFactory; |
||||
} |
||||
|
||||
+ (void)load { |
||||
[[GRPCTransportRegistry sharedInstance] |
||||
registerTransportWithId:GRPCDefaultTransportImplList.core_insecure |
||||
factory:[self sharedInstance]]; |
||||
} |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager { |
||||
return [[GRPCCall2Internal alloc] initWithTransportManager:transportManager]; |
||||
} |
||||
|
||||
- (id<GRPCChannelFactory>)createCoreChannelFactoryWithCallOptions:(GRPCCallOptions *)callOptions { |
||||
return [GRPCInsecureChannelFactory sharedInstance]; |
||||
} |
||||
|
||||
@end |
||||
@ -0,0 +1,65 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import <GRPCClient/GRPCInterceptor.h> |
||||
#import <GRPCClient/GRPCTransport.h> |
||||
|
||||
NS_ASSUME_NONNULL_BEGIN |
||||
|
||||
/**
|
||||
* Private interfaces of the transport registry. |
||||
*/ |
||||
@interface GRPCTransportRegistry (Private) |
||||
|
||||
/**
|
||||
* Get a transport implementation's factory by its transport id. If the transport id was not |
||||
* registered with the registry, the default transport factory (core + secure) is returned. If the |
||||
* default transport does not exist, an exception is thrown. |
||||
*/ |
||||
- (id<GRPCTransportFactory>)getTransportFactoryWithId:(GRPCTransportId)transportId; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCTransportManager : NSObject<GRPCInterceptorInterface> |
||||
|
||||
- (instancetype)initWithTransportId:(GRPCTransportId)transportId |
||||
previousInterceptor:(id<GRPCResponseHandler>)previousInterceptor; |
||||
|
||||
/**
|
||||
* Notify the manager that the transport has shut down and the manager should release references to |
||||
* its response handler and stop forwarding requests/responses. |
||||
*/ |
||||
- (void)shutDown; |
||||
|
||||
/** Forward initial metadata to the previous interceptor in the interceptor chain */ |
||||
- (void)forwardPreviousInterceptorWithInitialMetadata:(nullable NSDictionary *)initialMetadata; |
||||
|
||||
/** Forward a received message to the previous interceptor in the interceptor chain */ |
||||
- (void)forwardPreviousInterceptorWithData:(nullable id)data; |
||||
|
||||
/** Forward call close and trailing metadata to the previous interceptor in the interceptor chain */ |
||||
- (void)forwardPreviousInterceptorCloseWithTrailingMetadata: |
||||
(nullable NSDictionary *)trailingMetadata |
||||
error:(nullable NSError *)error; |
||||
|
||||
/** Forward write completion to the previous interceptor in the interceptor chain */ |
||||
- (void)forwardPreviousInterceptorDidWriteData; |
||||
|
||||
@end |
||||
|
||||
NS_ASSUME_NONNULL_END |
||||
@ -0,0 +1,131 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCTransport+Private.h" |
||||
|
||||
#import <GRPCClient/GRPCTransport.h> |
||||
|
||||
@implementation GRPCTransportManager { |
||||
GRPCTransportId _transportId; |
||||
GRPCTransport *_transport; |
||||
id<GRPCResponseHandler> _previousInterceptor; |
||||
dispatch_queue_t _dispatchQueue; |
||||
} |
||||
|
||||
- (instancetype)initWithTransportId:(GRPCTransportId)transportId |
||||
previousInterceptor:(id<GRPCResponseHandler>)previousInterceptor { |
||||
if ((self = [super init])) { |
||||
id<GRPCTransportFactory> factory = |
||||
[[GRPCTransportRegistry sharedInstance] getTransportFactoryWithId:transportId]; |
||||
|
||||
_transport = [factory createTransportWithManager:self]; |
||||
NSAssert(_transport != nil, @"Failed to create transport with id: %s", transportId); |
||||
if (_transport == nil) { |
||||
NSLog(@"Failed to create transport with id: %s", transportId); |
||||
return nil; |
||||
} |
||||
_previousInterceptor = previousInterceptor; |
||||
_dispatchQueue = _transport.dispatchQueue; |
||||
_transportId = transportId; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)shutDown { |
||||
dispatch_async(_dispatchQueue, ^{ |
||||
self->_transport = nil; |
||||
self->_previousInterceptor = nil; |
||||
}); |
||||
} |
||||
|
||||
- (dispatch_queue_t)dispatchQueue { |
||||
return _dispatchQueue; |
||||
} |
||||
|
||||
- (void)startWithRequestOptions:(GRPCRequestOptions *)requestOptions |
||||
callOptions:(GRPCCallOptions *)callOptions { |
||||
if (_transportId != callOptions.transport) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Interceptors cannot change the call option 'transport'"]; |
||||
return; |
||||
} |
||||
[_transport startWithRequestOptions:requestOptions callOptions:callOptions]; |
||||
} |
||||
|
||||
- (void)writeData:(id)data { |
||||
[_transport writeData:data]; |
||||
} |
||||
|
||||
- (void)finish { |
||||
[_transport finish]; |
||||
} |
||||
|
||||
- (void)cancel { |
||||
[_transport cancel]; |
||||
} |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages { |
||||
[_transport receiveNextMessages:numberOfMessages]; |
||||
} |
||||
|
||||
/** Forward initial metadata to the previous interceptor in the chain */ |
||||
- (void)forwardPreviousInterceptorWithInitialMetadata:(NSDictionary *)initialMetadata { |
||||
if (initialMetadata == nil || _previousInterceptor == nil) { |
||||
return; |
||||
} |
||||
id<GRPCResponseHandler> copiedPreviousInterceptor = _previousInterceptor; |
||||
dispatch_async(copiedPreviousInterceptor.dispatchQueue, ^{ |
||||
[copiedPreviousInterceptor didReceiveInitialMetadata:initialMetadata]; |
||||
}); |
||||
} |
||||
|
||||
/** Forward a received message to the previous interceptor in the chain */ |
||||
- (void)forwardPreviousInterceptorWithData:(id)data { |
||||
if (data == nil || _previousInterceptor == nil) { |
||||
return; |
||||
} |
||||
id<GRPCResponseHandler> copiedPreviousInterceptor = _previousInterceptor; |
||||
dispatch_async(copiedPreviousInterceptor.dispatchQueue, ^{ |
||||
[copiedPreviousInterceptor didReceiveData:data]; |
||||
}); |
||||
} |
||||
|
||||
/** Forward call close and trailing metadata to the previous interceptor in the chain */ |
||||
- (void)forwardPreviousInterceptorCloseWithTrailingMetadata:(NSDictionary *)trailingMetadata |
||||
error:(NSError *)error { |
||||
if (_previousInterceptor == nil) { |
||||
return; |
||||
} |
||||
id<GRPCResponseHandler> copiedPreviousInterceptor = _previousInterceptor; |
||||
dispatch_async(copiedPreviousInterceptor.dispatchQueue, ^{ |
||||
[copiedPreviousInterceptor didCloseWithTrailingMetadata:trailingMetadata error:error]; |
||||
}); |
||||
} |
||||
|
||||
/** Forward write completion to the previous interceptor in the chain */ |
||||
- (void)forwardPreviousInterceptorDidWriteData { |
||||
if (_previousInterceptor == nil) { |
||||
return; |
||||
} |
||||
id<GRPCResponseHandler> copiedPreviousInterceptor = _previousInterceptor; |
||||
dispatch_async(copiedPreviousInterceptor.dispatchQueue, ^{ |
||||
[copiedPreviousInterceptor didWriteData]; |
||||
}); |
||||
} |
||||
|
||||
@end |
||||
@ -0,0 +1,67 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import <GRPCClient/GRPCCallLegacy.h> |
||||
|
||||
// Import category headers for Swift build
|
||||
#import <GRPCClient/GRPCCall+ChannelArg.h> |
||||
#import <GRPCClient/GRPCCall+ChannelCredentials.h> |
||||
#import <GRPCClient/GRPCCall+Cronet.h> |
||||
#import <GRPCClient/GRPCCall+OAuth2.h> |
||||
#import <GRPCClient/GRPCCall+Tests.h> |
||||
#import <RxLibrary/GRXWriteable.h> |
||||
|
||||
@class GRPCProtoMethod; |
||||
@class GRXWriter; |
||||
@protocol GRXWriteable; |
||||
|
||||
__attribute__((deprecated("Please use GRPCProtoCall."))) @interface ProtoRPC |
||||
: GRPCCall |
||||
|
||||
/**
|
||||
* host parameter should not contain the scheme (http:// or https://), only the name or IP
|
||||
* addr and the port number, for example @"localhost:5050". |
||||
*/ |
||||
- |
||||
(instancetype)initWithHost : (NSString *)host method |
||||
: (GRPCProtoMethod *)method requestsWriter : (GRXWriter *)requestsWriter responseClass |
||||
: (Class)responseClass responsesWriteable |
||||
: (id<GRXWriteable>)responsesWriteable NS_DESIGNATED_INITIALIZER; |
||||
|
||||
- (void)start; |
||||
|
||||
@end |
||||
|
||||
/**
|
||||
* This subclass is empty now. Eventually we'll remove ProtoRPC class |
||||
* to avoid potential naming conflict |
||||
*/ |
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wdeprecated-declarations" |
||||
@interface GRPCProtoCall |
||||
: ProtoRPC |
||||
#pragma clang diagnostic pop |
||||
|
||||
@end |
||||
|
||||
/**
|
||||
* Generate an NSError object that represents a failure in parsing a proto class. For gRPC |
||||
* internal use only. |
||||
*/ |
||||
NSError * |
||||
ErrorForBadProto(id proto, Class expectedClass, NSError *parsingError); |
||||
@ -0,0 +1,121 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "ProtoRPCLegacy.h" |
||||
|
||||
#import "ProtoMethod.h" |
||||
|
||||
#if GPB_USE_PROTOBUF_FRAMEWORK_IMPORTS |
||||
#import <Protobuf/GPBProtocolBuffers.h> |
||||
#else |
||||
#import <GPBProtocolBuffers.h> |
||||
#endif |
||||
#import <GRPCClient/GRPCCall.h> |
||||
#import <RxLibrary/GRXWriteable.h> |
||||
#import <RxLibrary/GRXWriter+Transformations.h> |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wdeprecated-implementations" |
||||
@implementation ProtoRPC { |
||||
#pragma clang diagnostic pop |
||||
id<GRXWriteable> _responseWriteable; |
||||
} |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wobjc-designated-initializers" |
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
requestsWriter:(GRXWriter *)requestsWriter { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Please use ProtoRPC's designated initializer instead."]; |
||||
return nil; |
||||
} |
||||
#pragma clang diagnostic pop |
||||
|
||||
// Designated initializer |
||||
- (instancetype)initWithHost:(NSString *)host |
||||
method:(GRPCProtoMethod *)method |
||||
requestsWriter:(GRXWriter *)requestsWriter |
||||
responseClass:(Class)responseClass |
||||
responsesWriteable:(id<GRXWriteable>)responsesWriteable { |
||||
// Because we can't tell the type system to constrain the class, we need to check at runtime: |
||||
if (![responseClass respondsToSelector:@selector(parseFromData:error:)]) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"A protobuf class to parse the responses must be provided."]; |
||||
} |
||||
// A writer that serializes the proto messages to send. |
||||
GRXWriter *bytesWriter = [requestsWriter map:^id(GPBMessage *proto) { |
||||
if (![proto isKindOfClass:[GPBMessage class]]) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Request must be a proto message: %@", proto]; |
||||
} |
||||
return [proto data]; |
||||
}]; |
||||
if ((self = [super initWithHost:host path:method.HTTPPath requestsWriter:bytesWriter])) { |
||||
__weak ProtoRPC *weakSelf = self; |
||||
|
||||
// A writeable that parses the proto messages received. |
||||
_responseWriteable = [[GRXWriteable alloc] initWithValueHandler:^(NSData *value) { |
||||
// TODO(jcanizales): This is done in the main thread, and needs to happen in another thread. |
||||
NSError *error = nil; |
||||
id parsed = [responseClass parseFromData:value error:&error]; |
||||
if (parsed) { |
||||
[responsesWriteable writeValue:parsed]; |
||||
} else { |
||||
[weakSelf finishWithError:ErrorForBadProto(value, responseClass, error)]; |
||||
} |
||||
} |
||||
completionHandler:^(NSError *errorOrNil) { |
||||
[responsesWriteable writesFinishedWithError:errorOrNil]; |
||||
}]; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)start { |
||||
[self startWithWriteable:_responseWriteable]; |
||||
} |
||||
|
||||
- (void)startWithWriteable:(id<GRXWriteable>)writeable { |
||||
[super startWithWriteable:writeable]; |
||||
// Break retain cycles. |
||||
_responseWriteable = nil; |
||||
} |
||||
@end |
||||
|
||||
@implementation GRPCProtoCall |
||||
|
||||
@end |
||||
|
||||
/** |
||||
* Generate an NSError object that represents a failure in parsing a proto class. |
||||
*/ |
||||
NSError *ErrorForBadProto(id proto, Class expectedClass, NSError *parsingError) { |
||||
NSDictionary *info = @{ |
||||
NSLocalizedDescriptionKey : @"Unable to parse response from the server", |
||||
NSLocalizedRecoverySuggestionErrorKey : |
||||
@"If this RPC is idempotent, retry " |
||||
@"with exponential backoff. Otherwise, query the server status before " |
||||
@"retrying.", |
||||
NSUnderlyingErrorKey : parsingError, |
||||
@"Expected class" : expectedClass, |
||||
@"Received value" : proto, |
||||
}; |
||||
// TODO(jcanizales): Use kGRPCErrorDomain and GRPCErrorCodeInternal when they're public. |
||||
return [NSError errorWithDomain:@"io.grpc" code:13 userInfo:info]; |
||||
} |
||||
@ -0,0 +1,23 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import "ProtoService.h" |
||||
|
||||
@class GRPCProtoCall; |
||||
@class GRXWriter; |
||||
@protocol GRXWriteable; |
||||
@ -0,0 +1,71 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#import <objc/runtime.h> |
||||
|
||||
#import "ProtoMethod.h" |
||||
#import "ProtoRPCLegacy.h" |
||||
#import "ProtoService.h" |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wdeprecated-implementations" |
||||
@implementation ProtoService (Legacy) |
||||
#pragma clang diagnostic pop |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wobjc-designated-initializers" |
||||
// Do not call designated initializer here due to nullability incompatibility. This method is from |
||||
// old API and does not assert on nullability of the parameters. |
||||
|
||||
- (instancetype)initWithHost:(NSString *)host |
||||
packageName:(NSString *)packageName |
||||
serviceName:(NSString *)serviceName { |
||||
if ((self = [super init])) { |
||||
Ivar hostIvar = class_getInstanceVariable([ProtoService class], "_host"); |
||||
Ivar packageNameIvar = class_getInstanceVariable([ProtoService class], "_packageName"); |
||||
Ivar serviceNameIvar = class_getInstanceVariable([ProtoService class], "_serviceName"); |
||||
|
||||
object_setIvar(self, hostIvar, [host copy]); |
||||
object_setIvar(self, packageNameIvar, [packageName copy]); |
||||
object_setIvar(self, serviceNameIvar, [serviceName copy]); |
||||
} |
||||
return self; |
||||
} |
||||
#pragma clang diagnostic pop |
||||
|
||||
- (GRPCProtoCall *)RPCToMethod:(NSString *)method |
||||
requestsWriter:(GRXWriter *)requestsWriter |
||||
responseClass:(Class)responseClass |
||||
responsesWriteable:(id<GRXWriteable>)responsesWriteable { |
||||
Ivar hostIvar = class_getInstanceVariable([ProtoService class], "_host"); |
||||
Ivar packageNameIvar = class_getInstanceVariable([ProtoService class], "_packageName"); |
||||
Ivar serviceNameIvar = class_getInstanceVariable([ProtoService class], "_serviceName"); |
||||
NSString *host = object_getIvar(self, hostIvar); |
||||
NSString *packageName = object_getIvar(self, packageNameIvar); |
||||
NSString *serviceName = object_getIvar(self, serviceNameIvar); |
||||
|
||||
GRPCProtoMethod *methodName = |
||||
[[GRPCProtoMethod alloc] initWithPackage:packageName service:serviceName method:method]; |
||||
return [[GRPCProtoCall alloc] initWithHost:host |
||||
method:methodName |
||||
requestsWriter:requestsWriter |
||||
responseClass:responseClass |
||||
responsesWriteable:responsesWriteable]; |
||||
} |
||||
|
||||
@end |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue