mirror of https://github.com/grpc/grpc.git
Yang Gao
10 years ago
commit
1ac95abbf3
27 changed files with 765 additions and 216 deletions
@ -0,0 +1,97 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#import <Foundation/Foundation.h> |
||||
#include <grpc/grpc.h> |
||||
#import "GRPCChannel.h" |
||||
|
||||
typedef void(^GRPCCompletionHandler)(NSDictionary *); |
||||
|
||||
@protocol GRPCOp <NSObject> |
||||
|
||||
- (void)getOp:(grpc_op *)op; |
||||
|
||||
- (void)finish; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpSendMetadata : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithMetadata:(NSDictionary *)metadata |
||||
handler:(void(^)(void))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpSendMessage : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithMessage:(NSData *)message |
||||
handler:(void(^)(void))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpSendClose : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithHandler:(void(^)(void))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpRecvMetadata : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithHandler:(void(^)(NSDictionary *))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpRecvMessage : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithHandler:(void(^)(grpc_byte_buffer *))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCOpRecvStatus : NSObject <GRPCOp> |
||||
|
||||
- (instancetype)initWithHandler:(void(^)(NSError *))handler NS_DESIGNATED_INITIALIZER; |
||||
|
||||
@end |
||||
|
||||
@interface GRPCWrappedCall : NSObject |
||||
|
||||
- (instancetype)initWithChannel:(GRPCChannel *)channel |
||||
method:(NSString *)method |
||||
host:(NSString *)host NS_DESIGNATED_INITIALIZER; |
||||
|
||||
- (void)startBatchWithOperations:(NSArray *)ops errorHandler:(void(^)())errorHandler; |
||||
|
||||
- (void)startBatchWithOperations:(NSArray *)ops; |
||||
|
||||
- (void)cancel; |
||||
@end |
||||
@ -0,0 +1,326 @@ |
||||
/* |
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCWrappedCall.h" |
||||
#import <Foundation/Foundation.h> |
||||
#include <grpc/grpc.h> |
||||
#include <grpc/byte_buffer.h> |
||||
#include <grpc/support/alloc.h> |
||||
#import "GRPCCompletionQueue.h" |
||||
#import "NSDictionary+GRPC.h" |
||||
#import "NSData+GRPC.h" |
||||
#import "NSError+GRPC.h" |
||||
|
||||
@implementation GRPCOpSendMetadata{ |
||||
void(^_handler)(void); |
||||
grpc_metadata *_sendMetadata; |
||||
size_t _count; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
return [self initWithMetadata:nil handler:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithMetadata:(NSDictionary *)metadata handler:(void (^)(void))handler { |
||||
if (self = [super init]) { |
||||
_sendMetadata = [metadata grpc_metadataArray]; |
||||
_count = metadata.count; |
||||
_handler = handler; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_SEND_INITIAL_METADATA; |
||||
op->data.send_initial_metadata.count = _count; |
||||
op->data.send_initial_metadata.metadata = _sendMetadata; |
||||
} |
||||
|
||||
- (void)finish { |
||||
if (_handler) { |
||||
_handler(); |
||||
} |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
gpr_free(_sendMetadata); |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCOpSendMessage{ |
||||
void(^_handler)(void); |
||||
grpc_byte_buffer *_byteBuffer; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
return [self initWithMessage:nil handler:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithMessage:(NSData *)message handler:(void (^)(void))handler { |
||||
if (!message) { |
||||
[NSException raise:NSInvalidArgumentException format:@"message cannot be nil"]; |
||||
} |
||||
if (self = [super init]) { |
||||
_byteBuffer = [message grpc_byteBuffer]; |
||||
_handler = handler; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_SEND_MESSAGE; |
||||
op->data.send_message = _byteBuffer; |
||||
} |
||||
|
||||
- (void)finish { |
||||
if (_handler) { |
||||
_handler(); |
||||
} |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
gpr_free(_byteBuffer); |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCOpSendClose{ |
||||
void(^_handler)(void); |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
return [self initWithHandler:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithHandler:(void (^)(void))handler { |
||||
if (self = [super init]) { |
||||
_handler = handler; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
||||
} |
||||
|
||||
- (void)finish { |
||||
if (_handler) { |
||||
_handler(); |
||||
} |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCOpRecvMetadata{ |
||||
void(^_handler)(NSDictionary *); |
||||
grpc_metadata_array _recvInitialMetadata; |
||||
} |
||||
|
||||
- (instancetype) init { |
||||
return [self initWithHandler:nil]; |
||||
} |
||||
|
||||
- (instancetype) initWithHandler:(void (^)(NSDictionary *))handler { |
||||
if (self = [super init]) { |
||||
_handler = handler; |
||||
grpc_metadata_array_init(&_recvInitialMetadata); |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_RECV_INITIAL_METADATA; |
||||
op->data.recv_initial_metadata = &_recvInitialMetadata; |
||||
} |
||||
|
||||
- (void)finish { |
||||
NSDictionary *metadata = [NSDictionary |
||||
grpc_dictionaryFromMetadata:_recvInitialMetadata.metadata |
||||
count:_recvInitialMetadata.count]; |
||||
if (_handler) { |
||||
_handler(metadata); |
||||
} |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
grpc_metadata_array_destroy(&_recvInitialMetadata); |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCOpRecvMessage{ |
||||
void(^_handler)(grpc_byte_buffer *); |
||||
grpc_byte_buffer *_recvMessage; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
return [self initWithHandler:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithHandler:(void (^)(grpc_byte_buffer *))handler { |
||||
if (self = [super init]) { |
||||
_handler = handler; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_RECV_MESSAGE; |
||||
op->data.recv_message = &_recvMessage; |
||||
} |
||||
|
||||
- (void)finish { |
||||
if (_handler) { |
||||
_handler(_recvMessage); |
||||
} |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCOpRecvStatus{ |
||||
void(^_handler)(NSError *); |
||||
size_t _detailsCapacity; |
||||
grpc_status _status; |
||||
} |
||||
|
||||
- (instancetype) init { |
||||
return [self initWithHandler:nil]; |
||||
} |
||||
|
||||
- (instancetype) initWithHandler:(void (^)(NSError *))handler { |
||||
if (self = [super init]) { |
||||
_handler = handler; |
||||
grpc_metadata_array_init(&_status.metadata); |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)getOp:(grpc_op *)op { |
||||
op->op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
||||
op->data.recv_status_on_client.status = &_status.status; |
||||
op->data.recv_status_on_client.status_details = &_status.details; |
||||
op->data.recv_status_on_client.status_details_capacity = &_detailsCapacity; |
||||
op->data.recv_status_on_client.trailing_metadata = &_status.metadata; |
||||
} |
||||
|
||||
- (void)finish { |
||||
if (_handler) { |
||||
NSError *error = [NSError grpc_errorFromStatus:&_status]; |
||||
_handler(error); |
||||
} |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
grpc_metadata_array_destroy(&_status.metadata); |
||||
gpr_free(_status.details); |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCWrappedCall{ |
||||
grpc_call *_call; |
||||
GRPCCompletionQueue *_queue; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
return [self initWithChannel:nil method:nil host:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithChannel:(GRPCChannel *)channel |
||||
method:(NSString *)method |
||||
host:(NSString *)host { |
||||
if (!channel || !method || !host) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"channel, method, and host cannot be nil."]; |
||||
} |
||||
|
||||
if (self = [super init]) { |
||||
static dispatch_once_t initialization; |
||||
dispatch_once(&initialization, ^{ |
||||
grpc_init(); |
||||
}); |
||||
|
||||
_queue = [GRPCCompletionQueue completionQueue]; |
||||
if (!_queue) { |
||||
return nil; |
||||
} |
||||
_call = grpc_channel_create_call(channel.unmanagedChannel, _queue.unmanagedQueue, |
||||
method.UTF8String, host.UTF8String, gpr_inf_future); |
||||
if (_call == NULL) { |
||||
return nil; |
||||
} |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)startBatchWithOperations:(NSArray *)operations { |
||||
[self startBatchWithOperations:operations errorHandler:nil]; |
||||
} |
||||
|
||||
- (void)startBatchWithOperations:(NSArray *)operations errorHandler:(void (^)())errorHandler { |
||||
size_t nops = operations.count; |
||||
grpc_op *ops_array = gpr_malloc(nops * sizeof(grpc_op)); |
||||
size_t i = 0; |
||||
for (id op in operations) { |
||||
[op getOp:&ops_array[i++]]; |
||||
} |
||||
grpc_call_error error = grpc_call_start_batch(_call, ops_array, nops, |
||||
(__bridge_retained void *)(^(grpc_op_error error){ |
||||
if (error != GRPC_OP_OK) { |
||||
if (errorHandler) { |
||||
errorHandler(); |
||||
} else { |
||||
return; |
||||
} |
||||
} |
||||
for (id<GRPCOp> operation in operations) { |
||||
[operation finish]; |
||||
} |
||||
})); |
||||
|
||||
if (error != GRPC_CALL_OK) { |
||||
[NSException raise:NSInternalInconsistencyException |
||||
format:@"A precondition for calling grpc_call_start_batch wasn't met"]; |
||||
} |
||||
} |
||||
|
||||
- (void)cancel { |
||||
grpc_call_cancel(_call); |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
grpc_call_destroy(_call); |
||||
} |
||||
|
||||
@end |
||||
Loading…
Reference in new issue