mirror of https://github.com/grpc/grpc.git
commit
ccc5265e8a
40 changed files with 1820 additions and 160 deletions
@ -0,0 +1,61 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2017, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifdef GRPC_TEST_OBJC |
||||||
|
|
||||||
|
#import "../GRPCCall.h" |
||||||
|
|
||||||
|
/**
|
||||||
|
* Methods used for gRPC internal tests. DO NOT USE. |
||||||
|
*/ |
||||||
|
@interface GRPCCall (InternalTests) |
||||||
|
|
||||||
|
/**
|
||||||
|
* Enables logging of op batches. Memory consumption increases as more ops are logged. |
||||||
|
* |
||||||
|
* This function is for internal testing of gRPC only. It is not part of gRPC's public interface. |
||||||
|
* Do not use in production. To enable, set the preprocessor flag GRPC_TEST_OBJC. |
||||||
|
*/ |
||||||
|
+ (void)enableOpBatchLog:(BOOL)enabled; |
||||||
|
|
||||||
|
/**
|
||||||
|
* Obtain the logged op batches. Invoking this method will clean the log. |
||||||
|
* |
||||||
|
* This function is for internal testing of gRPC only. It is not part of gRPC's public interface. |
||||||
|
* Do not use in production. To enable, set the preprocessor flag GRPC_TEST_OBJC. |
||||||
|
*/ |
||||||
|
+ (NSArray *)obtainAndCleanOpBatchLog; |
||||||
|
|
||||||
|
@end |
||||||
|
|
||||||
|
#endif |
@ -0,0 +1,52 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2017, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifdef GRPC_TEST_OBJC |
||||||
|
|
||||||
|
#import "GRPCCall+InternalTests.h" |
||||||
|
|
||||||
|
#import "../private/GRPCOpBatchLog.h" |
||||||
|
|
||||||
|
@implementation GRPCCall (InternalTests) |
||||||
|
|
||||||
|
+ (void)enableOpBatchLog:(BOOL)enabled { |
||||||
|
[GRPCOpBatchLog enableOpBatchLog:enabled]; |
||||||
|
} |
||||||
|
|
||||||
|
+ (NSArray *)obtainAndCleanOpBatchLog { |
||||||
|
return [GRPCOpBatchLog obtainAndCleanOpBatchLog]; |
||||||
|
} |
||||||
|
|
||||||
|
@end |
||||||
|
|
||||||
|
#endif |
@ -0,0 +1,59 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
|
||||||
|
#ifdef GRPC_TEST_OBJC |
||||||
|
|
||||||
|
/**
|
||||||
|
* Logs the op batches of a client. Used for testing. |
||||||
|
*/ |
||||||
|
@interface GRPCOpBatchLog : NSObject |
||||||
|
|
||||||
|
/**
|
||||||
|
* Enables logging of op batches. Memory consumption increases as more ops are logged. |
||||||
|
*/ |
||||||
|
+ (void)enableOpBatchLog:(BOOL)enabled; |
||||||
|
|
||||||
|
/**
|
||||||
|
* Add an op batch to log. |
||||||
|
*/ |
||||||
|
+ (void)addOpBatchToLog:(NSArray *)batch; |
||||||
|
|
||||||
|
/**
|
||||||
|
* Obtain the logged op batches. Invoking this method will clean the log. |
||||||
|
*/ |
||||||
|
+ (NSArray *)obtainAndCleanOpBatchLog; |
||||||
|
|
||||||
|
@end |
||||||
|
|
||||||
|
#endif |
@ -0,0 +1,72 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2016, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifdef GRPC_TEST_OBJC |
||||||
|
|
||||||
|
#import "GRPCOpBatchLog.h" |
||||||
|
|
||||||
|
static NSMutableArray *opBatchLog = nil; |
||||||
|
|
||||||
|
@implementation GRPCOpBatchLog |
||||||
|
|
||||||
|
+ (void)enableOpBatchLog:(BOOL)enabled { |
||||||
|
@synchronized (opBatchLog) { |
||||||
|
if (enabled) { |
||||||
|
if (!opBatchLog) { |
||||||
|
opBatchLog = [NSMutableArray array]; |
||||||
|
} |
||||||
|
} else { |
||||||
|
if (opBatchLog) { |
||||||
|
opBatchLog = nil; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
+ (void)addOpBatchToLog:(NSArray *)batch { |
||||||
|
@synchronized (opBatchLog) { |
||||||
|
[opBatchLog addObject:batch]; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
+ (NSArray *)obtainAndCleanOpBatchLog { |
||||||
|
@synchronized (opBatchLog) { |
||||||
|
NSArray *out = opBatchLog; |
||||||
|
opBatchLog = [NSMutableArray array]; |
||||||
|
return out; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@end |
||||||
|
|
||||||
|
#endif |
@ -0,0 +1,50 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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> |
||||||
|
|
||||||
|
#import "GRXImmediateWriter.h" |
||||||
|
|
||||||
|
/**
|
||||||
|
* Utility to construct GRXWriter instances from values that are immediately available when |
||||||
|
* required. |
||||||
|
*/ |
||||||
|
@interface GRXImmediateSingleWriter : GRXImmediateWriter |
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a writer that sends the passed value to its writeable and then finishes (releasing the |
||||||
|
* value). |
||||||
|
*/ |
||||||
|
+ (GRXWriter *)writerWithValue:(id)value; |
||||||
|
|
||||||
|
@end |
@ -0,0 +1,89 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2016, 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 "GRXImmediateSingleWriter.h" |
||||||
|
|
||||||
|
@implementation GRXImmediateSingleWriter { |
||||||
|
id _value; |
||||||
|
id<GRXWriteable> _writeable; |
||||||
|
} |
||||||
|
|
||||||
|
@synthesize state = _state; |
||||||
|
|
||||||
|
- (instancetype)initWithValue:(id)value { |
||||||
|
if (self = [super init]) { |
||||||
|
_value = value; |
||||||
|
_state = GRXWriterStateNotStarted; |
||||||
|
} |
||||||
|
return self; |
||||||
|
} |
||||||
|
|
||||||
|
+ (GRXWriter *)writerWithValue:(id)value { |
||||||
|
return [[self alloc] initWithValue:value]; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)startWithWriteable:(id<GRXWriteable>)writeable { |
||||||
|
_state = GRXWriterStateStarted; |
||||||
|
_writeable = writeable; |
||||||
|
[writeable writeValue:_value]; |
||||||
|
[self finish]; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)finish { |
||||||
|
_state = GRXWriterStateFinished; |
||||||
|
_value = nil; |
||||||
|
id<GRXWriteable> writeable = _writeable; |
||||||
|
_writeable = nil; |
||||||
|
[writeable writesFinishedWithError:nil]; |
||||||
|
} |
||||||
|
|
||||||
|
// Overwrite the setter to disallow manual state transition. The getter |
||||||
|
// of _state is synthesized. |
||||||
|
- (void)setState:(GRXWriterState)newState { |
||||||
|
// Manual state transition is not allowed |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// Overrides [requestWriter(Transformations):map:] for Protocol Buffers |
||||||
|
// encoding. |
||||||
|
// We need the return value of this map to be a GRXImmediateSingleWriter but |
||||||
|
// the original \a map function returns a new Writer of another type. So we |
||||||
|
// need to override this function here. |
||||||
|
- (GRXWriter *)map:(id (^)(id))map { |
||||||
|
// Since _value is available when creating the object, we can simply |
||||||
|
// apply the map and store the output. |
||||||
|
_value = map(_value); |
||||||
|
return self; |
||||||
|
} |
||||||
|
|
||||||
|
@end |
Binary file not shown.
Loading…
Reference in new issue