status-im
/
react-native-firebase
mirror of
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

[perf][ios] Performance ios integration

This commit is contained in:
Elliot Hesp 2018-07-18 16:05:59 +01:00
parent a018810124
commit 89d8a0a92a
2 changed files with 223 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ios/RNFirebase/perf/RNFirebasePerformance.h
View File

@ -10,6 +10,7 @@
}
@property NSMutableDictionary *traces;
@property NSMutableDictionary *httpMetrics;
@end

232
ios/RNFirebase/perf/RNFirebasePerformance.m
View File

@ -2,6 +2,7 @@
#if __has_include(<FirebasePerformance/FIRPerformance.h>)
#import <FirebasePerformance/FIRPerformance.h>
#import <FirebasePerformance/FIRHTTPMetric.h>
@implementation RNFirebasePerformance
RCT_EXPORT_MODULE();
@ -9,10 +10,25 @@ RCT_EXPORT_MODULE();
self = [super init];
if (self != nil) {
_traces = [[NSMutableDictionary alloc] init];
_httpMetrics = [[NSMutableDictionary alloc] init];
}
return self;
}
- (FIRHTTPMethod) mapStringToMethod:(NSString *) value {
if ([value compare:@"get" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodGET;
if ([value compare:@"put" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodPUT;
if ([value compare:@"post" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodPUT;
if ([value compare:@"delete" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodDELETE;
if ([value compare:@"head" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodHEAD;
if ([value compare:@"patch" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodPATCH;
if ([value compare:@"options" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodOPTIONS;
if ([value compare:@"trace" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodTRACE;
if ([value compare:@"connect" options:NSCaseInsensitiveSearch] == NSOrderedSame) return FIRHTTPMethodCONNECT;
return FIRHTTPMethodGET;
}
- (FIRTrace *)getOrCreateTrace:(NSString *)identifier {
if (_traces[identifier]) {
return _traces[identifier];
@ -22,27 +38,223 @@ RCT_EXPORT_MODULE();
return trace;
}
- (FIRHTTPMetric *)getOrCreateHttpMetric:(NSString *)url httpMethod:(NSString *) httpMethod {
NSString *identifier = [NSString stringWithFormat:@"%@%@", url, httpMethod];
if (_httpMetrics[identifier]) {
return _httpMetrics[identifier];
}
NSURL * toURL = [NSURL URLWithString:url];
FIRHTTPMethod method = [self mapStringToMethod:httpMethod];
FIRHTTPMetric *httpMetric = [[FIRHTTPMetric alloc] initWithURL:toURL HTTPMethod:method];
_httpMetrics[identifier] = httpMetric;
return httpMetric;
}
RCT_EXPORT_METHOD(setPerformanceCollectionEnabled:
(BOOL *) enabled) {
[FIRPerformance sharedInstance].dataCollectionEnabled = (BOOL) enabled;
}
RCT_EXPORT_METHOD(start:
(NSString *) identifier) {
/**
* Trace
*/
RCT_EXPORT_METHOD(startTrace:
(NSString *) identifier
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateTrace:identifier] start];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(stop:
(NSString *) identifier) {
RCT_EXPORT_METHOD(stopTrace:
(NSString *) identifier
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateTrace:identifier] stop];
_traces[identifier] = nil;
[_traces removeObjectForKey:identifier];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(incrementCounter:
(NSString *) identifier
event:
(NSString *) event) {
[[self getOrCreateTrace:identifier] incrementCounterNamed:event];
RCT_EXPORT_METHOD(getTraceAttribute:
(NSString *) identifier
attribute:(NSString *) attribute
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
NSString *value = [[self getOrCreateTrace:identifier] valueForAttribute:attribute];
resolve(value ? value : [NSNull null]);
}
RCT_EXPORT_METHOD(getTraceAttributes:
(NSString *) identifier
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
resolve([[self getOrCreateTrace:identifier] attributes]);
}
RCT_EXPORT_METHOD(getTraceLongMetric:
(NSString *) identifier
metricName:(NSString *) metricName
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
int64_t value = [[self getOrCreateTrace:identifier] valueForIntMetric:metricName];
resolve(@(value));
}
RCT_EXPORT_METHOD(incrementTraceMetric:
(NSString *) identifier
metricName:(NSString *) metricName
incrementBy:(NSNumber *) incrementBy
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
int64_t byInt = [incrementBy intValue];
[[self getOrCreateTrace:identifier] incrementMetric:metricName byInt:byInt];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(putTraceAttribute:
(NSString *) identifier
attribute:(NSString *) attribute
value:(NSString *) value
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
FIRTrace * trace = [self getOrCreateTrace:identifier];
[trace setValue:value forAttribute:attribute];
if (trace.attributes[attribute] != nil) {
resolve(@(YES));
} else {
resolve(@(NO));
}
}
RCT_EXPORT_METHOD(putTraceMetric:
(NSString *) identifier
attribute:(NSString *) attribute
value:(NSNumber *) value
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
int64_t byInt = [value intValue];
[[self getOrCreateTrace:identifier] setIntValue:byInt forMetric:attribute];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(removeTraceAttribute:
(NSString *) identifier
attribute:(NSString *) attribute
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateTrace:identifier] removeAttribute:attribute];
resolve([NSNull null]);
}
/**
* HTTP Metric
*/
RCT_EXPORT_METHOD(startHttpMetric:
(NSString *) url
httpMethod:(NSString *) httpMethod
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] start];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(stopHttpMetric:
(NSString *) url
httpMethod:(NSString *) httpMethod
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] stop];
[_httpMetrics removeObjectForKey:[NSString stringWithFormat:@"%@%@", url, httpMethod]];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(getHttpMetricAttribute:
(NSString *) url
httpMethod:(NSString *) httpMethod
attribute:(NSString *) attribute
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
NSString *value = [[self getOrCreateHttpMetric:url httpMethod:httpMethod] valueForAttribute:attribute];
resolve(value ? value : [NSNull null]);
}
RCT_EXPORT_METHOD(getHttpMetricAttributes:
(NSString *) url
httpMethod:(NSString *) httpMethod
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
resolve([[self getOrCreateHttpMetric:url httpMethod:httpMethod] attributes]);
}
RCT_EXPORT_METHOD(putHttpMetricAttribute:
(NSString *) url
httpMethod:(NSString *) httpMethod
attribute:(NSString *) attribute
value:(NSString *) value
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
FIRHTTPMetric * httpMetric = [self getOrCreateHttpMetric:url httpMethod:httpMethod];
[httpMetric setValue:value forAttribute:attribute];
if (httpMetric.attributes[attribute] != nil) {
resolve(@(YES));
} else {
resolve(@(NO));
}
}
RCT_EXPORT_METHOD(removeHttpMetricAttribute:
(NSString *) url
httpMethod:(NSString *) httpMethod
attribute:(NSString *) attribute
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] removeAttribute:attribute];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(setHttpMetricResponseCode:
(NSString *) url
httpMethod:(NSString *) httpMethod
code:(NSNumber *) code
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] setResponseCode:[code integerValue]];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(setHttpMetricRequestPayloadSize:
(NSString *) url
httpMethod:(NSString *) httpMethod
bytes:(NSNumber *) bytes
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] setRequestPayloadSize:[bytes longLongValue]];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(setHttpMetricResponseContentType:
(NSString *) url
httpMethod:(NSString *) httpMethod
type:(NSString *) type
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] setResponseContentType:type];
resolve([NSNull null]);
}
RCT_EXPORT_METHOD(setHttpMetricResponsePayloadSize:
(NSString *) url
httpMethod:(NSString *) httpMethod
bytes:(NSNumber *) bytes
resolver:(RCTPromiseResolveBlock)resolve
rejecter:(RCTPromiseRejectBlock)reject) {
[[self getOrCreateHttpMetric:url httpMethod:httpMethod] setResponsePayloadSize:[bytes longLongValue]];
resolve([NSNull null]);
}
+ (BOOL)requiresMainQueueSetup