NSTimer本没有现成API支持添加Block方法 但是有些变态面试题废除这种恶心的东西,so 搞了一下当练手啦
#import <Foundation/Foundation.h>
@interface NSTimer (BlockMethod)
/**
* 无参数无返回值Block
*/
typedef void (^HYBVoidBlock)(void);
/**
* 创建Timer---Block版本
*
* @param interval 每隔interval秒就回调一次callback
* @param repeats 是否重复
* @param callback 回调block
*
* @return NSTimer对象
*/
+ (NSTimer *)scheduledTimerWithTimeInterval:(NSTimeInterval)interval
repeats:(BOOL)repeats
callback:(HYBVoidBlock)callback;
/**
* 创建Timer---Block版本
*
* @param interval 每隔interval秒就回调一次callback
* @param count 回调多少次后自动暂停,如果count <= 0,则表示无限次,否则表示具体的次数
* @param callback 回调block
*
* @return NSTimer对象
*/
+ (NSTimer *)scheduledTimerWithTimeInterval:(NSTimeInterval)interval
count:(NSInteger)count
callback:(HYBVoidBlock)callback;
/**
* 开始启动定时器
*/
- (void)fireTimer;
/**
* 暂停定时器
*/
- (void)unfireTimer;
@end
//
// NSTimer+LGBlockMethod.m
// nsoperationQueueDemo
//
// Created by 张传奇 on 2018/3/20.
// Copyright © 2018年 张传奇. All rights reserved.
//
#import "NSTimer+LGBlockMethod.h"
@implementation NSTimer (LGBlockMethod)
+ (NSTimer *)scheduledTimerWithTimeInterval:(NSTimeInterval)interval
repeats:(BOOL)repeats
callback:(LGVoidBlock)callback {
return [NSTimer scheduledTimerWithTimeInterval:interval
target:self
selector:@selector(onTimerUpdateBlock:)
userInfo:[callback copy]
repeats:repeats];
}
+ (NSTimer *)scheduledTimerWithTimeInterval:(NSTimeInterval)interval
count:(NSInteger)count
callback:(LGVoidBlock)callback {
NSDictionary *userInfo = @{@"callback" : [callback copy],
@"count" : @(count)};
return [NSTimer scheduledTimerWithTimeInterval:interval
target:self
selector:@selector(onTimerUpdateCountBlock:)
userInfo:userInfo
repeats:YES];
}
+ (void)onTimerUpdateBlock:(NSTimer *)timer {
LGVoidBlock block = timer.userInfo;
if (block) {
block();
}
}
+ (void)onTimerUpdateCountBlock:(NSTimer *)timer {
static NSUInteger currentCount = 0;
NSDictionary *userInfo = timer.userInfo;
LGVoidBlock callback = userInfo[@"callback"];
NSNumber *count = userInfo[@"count"];
if (count.integerValue <= 0) {
if (callback) {
callback();
}
} else {
if (currentCount < count.integerValue) {
currentCount++;
if (callback) {
callback();
}
} else {
currentCount = 0;
[timer stopTimer];
}
}
}
- (void)startTimer {
[self setFireDate:[NSDate distantPast]];
}
- (void)stopTimer {
[self setFireDate:[NSDate distantFuture]];
}
@end