-
Notifications
You must be signed in to change notification settings - Fork 511
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Extract KWBackgroundTask into its own file
- Loading branch information
Adam Sharp
committed
Apr 30, 2015
1 parent
e488490
commit 009e3ba
Showing
4 changed files
with
146 additions
and
117 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,38 @@ | ||
| // | ||
| // Licensed under the terms in License.txt | ||
| // | ||
| // Copyright 2010 Allen Ding. All rights reserved. | ||
| // | ||
|
|
||
| #import <Foundation/Foundation.h> | ||
|
|
||
| // Used to suppress compiler warnings by | ||
| // casting receivers to this protocol | ||
| @protocol NSTask_KWWarningSuppressor | ||
|
|
||
| - (void)setLaunchPath:(NSString *)path; | ||
| - (void)setArguments:(NSArray *)arguments; | ||
| - (void)setEnvironment:(NSDictionary *)dict; | ||
| - (void)setStandardOutput:(id)output; | ||
| - (void)setStandardError:(id)output; | ||
| - (void)launch; | ||
| - (void)waitUntilExit; | ||
|
|
||
| @property (readonly) int terminationStatus; | ||
|
|
||
| @end | ||
|
|
||
| @interface KWBackgroundTask : NSObject | ||
|
|
||
| @property (nonatomic, readonly) id<NSTask_KWWarningSuppressor> task; | ||
| @property (nonatomic, readonly) NSPipe *standardOutput; | ||
| @property (nonatomic, readonly) NSPipe *standardError; | ||
| @property (nonatomic, readonly) NSString *command; | ||
| @property (nonatomic, readonly) NSArray *arguments; | ||
| @property (nonatomic, readonly) NSData *output; | ||
|
|
||
| - (instancetype)initWithCommand:(NSString *)command arguments:(NSArray *)arguments; | ||
|
|
||
| - (void)launchAndWaitForExit; | ||
|
|
||
| @end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| // | ||
| // Licensed under the terms in License.txt | ||
| // | ||
| // Copyright 2010 Allen Ding. All rights reserved. | ||
| // | ||
|
|
||
| #import "KWBackgroundTask.h" | ||
|
|
||
| NSString *const NSTaskDidTerminateNotification; | ||
|
|
||
| static NSString *const KWTaskDidTerminateNotification = @"KWTaskDidTerminateNotification"; | ||
|
|
||
| static NSString *const KWBackgroundTaskException = @"KWBackgroundTaskException"; | ||
|
|
||
| @implementation KWBackgroundTask | ||
|
|
||
| - (instancetype)initWithCommand:(NSString *)command arguments:(NSArray *)arguments { | ||
| if (self = [super init]) { | ||
| _command = command; | ||
| _arguments = arguments; | ||
| } | ||
| return self; | ||
| } | ||
|
|
||
| - (void)dealloc { | ||
| [[NSNotificationCenter defaultCenter] removeObserver:self name:NSTaskDidTerminateNotification object:nil]; | ||
| } | ||
|
|
||
| - (NSString *)description { | ||
| return [NSString stringWithFormat:@"%@ `%@ %@`", [super description], self.command, [self.arguments componentsJoinedByString:@" "]]; | ||
| } | ||
|
|
||
| // Run this task for 10 seconds | ||
| // if it times out raise an exception | ||
| - (void)launchAndWaitForExit { | ||
| CFRunLoopRef runLoop = [NSRunLoop currentRunLoop].getCFRunLoop; | ||
| __weak KWBackgroundTask *weakSelf = self; | ||
| CFRunLoopTimerRef timer = CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + 10.0, 0, 0, 0, ^(CFRunLoopTimerRef timer) { | ||
| [NSException raise:KWBackgroundTaskException format:@"Task %@ timed out", weakSelf]; | ||
| CFRunLoopStop(runLoop); | ||
| }); | ||
| CFRunLoopAddTimer(runLoop, timer, kCFRunLoopDefaultMode); | ||
|
|
||
| id taskObserver = [[NSNotificationCenter defaultCenter] addObserverForName:KWTaskDidTerminateNotification object:self queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification *note) { | ||
| CFRunLoopStop(runLoop); | ||
| }]; | ||
|
|
||
| [NSThread detachNewThreadSelector:@selector(launch) toTarget:self withObject:nil]; | ||
| CFRunLoopRun(); | ||
| CFRunLoopRemoveTimer(runLoop, timer, kCFRunLoopDefaultMode); | ||
|
|
||
| [[NSNotificationCenter defaultCenter] removeObserver:taskObserver]; | ||
| } | ||
|
|
||
| #pragma mark - Private | ||
|
|
||
| - (void)launch { | ||
| __block id<NSTask_KWWarningSuppressor> task = [[NSClassFromString(@"NSTask") alloc] init]; | ||
| [task setEnvironment:[NSDictionary dictionary]]; | ||
| [task setLaunchPath:_command]; | ||
| [task setArguments:_arguments]; | ||
|
|
||
| NSPipe *standardOutput = [NSPipe pipe]; | ||
| [task setStandardOutput:standardOutput]; | ||
|
|
||
| // Consume standard error but don't use it | ||
| NSPipe *standardError = [NSPipe pipe]; | ||
| [task setStandardError:standardError]; | ||
|
|
||
| _task = task; | ||
| _standardError = standardError; | ||
| _standardOutput = standardOutput; | ||
|
|
||
| [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(taskDidTerminate:) name:NSTaskDidTerminateNotification object:task]; | ||
|
|
||
| @try { | ||
| [_task launch]; | ||
| } @catch (NSException *exception) { | ||
| [NSException raise:KWBackgroundTaskException format:@"Task %@ failed to launch", self]; | ||
| } | ||
| CFRunLoopRun(); | ||
| } | ||
|
|
||
| - (void)taskDidTerminate:(NSNotification *)note { | ||
| if ([_task terminationStatus] != 0) { | ||
| [NSException raise:KWBackgroundTaskException format:@"Task %@ terminated with non 0 exit code", self]; | ||
| } else { | ||
| _output = [[_standardOutput fileHandleForReading] readDataToEndOfFile]; | ||
| } | ||
|
|
||
| [[NSNotificationCenter defaultCenter] postNotificationName:KWTaskDidTerminateNotification object:self]; | ||
| [NSThread exit]; | ||
| } | ||
|
|
||
| @end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters