SFAnalyticsActivityTracker.m   [plain text]

/*
* Copyright (c) 2017 Apple Inc. All Rights Reserved.
*
* @APPLE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this
* file.
*
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_LICENSE_HEADER_END@
*/

#if __OBJC2__

#import "SFAnalyticsActivityTracker.h"
#import "SFAnalyticsActivityTracker+Internal.h"
#import "SFAnalytics.h"
#import <mach/mach_time.h>
#import "utilities/debugging.h"

@interface SFAnalyticsActivityTracker ()
@property (readwrite) NSNumber * measurement;
@end

@implementation SFAnalyticsActivityTracker {
dispatch_queue_t _queue;
NSString* _name;
Class _clientClass;
uint64_t _start;
BOOL _canceled;
}

@synthesize measurement = _measurement;

- (instancetype)initWithName:(NSString*)name clientClass:(Class)className {
if (![name isKindOfClass:[NSString class]] || ![className isSubclassOfClass:[SFAnalytics class]] ) {
secerror("Cannot instantiate SFActivityTracker without name and client class");
return nil;
}

if (self = [super init]) {
_queue = dispatch_queue_create("SFAnalyticsActivityTracker queue", DISPATCH_QUEUE_SERIAL_WITH_AUTORELEASE_POOL);
_name = name;
_clientClass = className;
_measurement = nil;
_canceled = NO;
_start = 0;
}
return self;
}

- (void)performAction:(void (^)(void))action
{
[self start];
dispatch_sync(_queue, ^{
action();
});
[self stop];
}

- (void)start
{
if (_canceled) {
return;
}
NSAssert(_start == 0, @"SFAnalyticsActivityTracker user called start twice");
_start = mach_absolute_time();
}

- (void)stop
{
uint64_t end = mach_absolute_time();

if (_canceled) {
_start = 0;
return;
}
NSAssert(_start != 0, @"SFAnalyticsActivityTracker user called stop w/o calling start");

static mach_timebase_info_data_t sTimebaseInfo;
if ( sTimebaseInfo.denom == 0 ) {
(void)mach_timebase_info(&sTimebaseInfo);
}

_measurement = @([_measurement doubleValue] + (1.0f * (end - _start) * (1.0f * sTimebaseInfo.numer / sTimebaseInfo.denom)));
_start = 0;
}

- (void)stopWithEvent:(NSString*)eventName
result:(NSError* _Nullable)eventResultError
{
[self stop];

[[_clientClass logger] logResultForEvent:eventName hardFailure:false result:eventResultError];
}

- (void)cancel
{
_canceled = YES;
}

- (void)dealloc
{
if (_start != 0) {
[self stop];
}
if (!_canceled && _measurement != nil) {
[[_clientClass logger] logMetric:_measurement withName:_name];
}
}

@end

#endif