一定時間、処理を非同期で待つメソッド

通信処理など時間が掛かる処理を行いたいが、一瞬で完了してはUI的に困る。最低xx秒は処理の完了を待ってほしいときに以下のように呼び出す。
下の場合、A点の処理が一瞬で完了したとしても、最低0.8秒はB点の処理の実行は待機される。

sender.enabled = NO;
[NSObject perform:^(NSNotifyHandler done) {
	//A 通信などの処理
	[self hogeWithCompletion:^{
		done(^{
			//B 完了した時に行う動作
			NSLog(@"done");
			sender.enabled = YES;
		});
	}];
} waitUntil:0.8];

実装

実装には以前こちらで紹介されていたSTDeferredライブラリを利用させて頂きました。

NSObject+AsyncWaitUntil.h

typedef void(^NSNotifyHandler)(void (^completion)());

@interface NSObject (AsyncWaitUntil)
+ (void)perform:(void (^)(NSNotifyHandler done))task waitUntil:(NSTimeInterval)timeInterval;
@end

NSObject+AsyncWaitUntil.m

#import "NSObject+AsyncWaitUntil.h"

#import <STDeferred/STDeferred.h>

@implementation NSObject (AsyncWaitUntil)

+ (void)perform:(void (^)(NSNotifyHandler))task waitUntil:(NSTimeInterval)timeInterval
{
	NSAssert(task, @"NSNotifyHandler must be non-nil.");
	STDeferredBlock runTask = ^{
		STDeferred *deferred = [STDeferred deferred];
		NSNotifyHandler notify = ^(void (^completion)()) {
			[deferred resolve:completion];
		};
		task(notify);
		return deferred;
	};
	
	STDeferredBlock waitUntil = ^{
		STDeferred *deferred = [STDeferred deferred];
		int64_t delayInMilliSeconds = timeInterval * 1000;
		dispatch_time_t popTime = dispatch_time(DISPATCH_TIME_NOW, delayInMilliSeconds * NSEC_PER_MSEC);
		dispatch_after(popTime, dispatch_get_current_queue(), ^(void){
			[deferred resolve:nil];
		});
		return deferred;
	};
	
	[[STDeferred when:runTask, waitUntil, nil] then:^(id resultObject) {
		id block = [resultObject objectAtIndex:0];
		if (![block isKindOfClass:[NSNull class]]) {
			void (^completion)() = block;
			completion();
		}
	}];
}