基于GCD自定义定时器

话不多说,直接上代码

//
//  SFJTimer.h
//  定时器-GCD
//
//  Created by 宋法键 on 2019/8/14.
//  Copyright © 2019 WZSG. All rights reserved.
//

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@interface SFJTimer : NSObject

+ (NSString *)executeTask:(void(^)(void))task start:(NSTimeInterval)start interval:(NSTimeInterval)interval repeat:(BOOL)repeat async:(BOOL)async;

+ (NSString *)executeTask:(id)target selector:(SEL)selector start:(NSTimeInterval)start interval:(NSTimeInterval)interval repeat:(BOOL)repeat async:(BOOL)async;

+ (void)cancelTask:(NSString *)taskSign;

@end

NS_ASSUME_NONNULL_END
//
//  SFJTimer.m
//  定时器-GCD
//
//  Created by 宋法键 on 2019/8/14.
//  Copyright © 2019 WZSG. All rights reserved.
//

#import "SFJTimer.h"

//多线程操作,需要加锁
static NSMutableDictionary *timers_;
static dispatch_semaphore_t semaphore_t_;

@implementation SFJTimer

+ (void)initialize {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        timers_ = [NSMutableDictionary dictionary];
        semaphore_t_ = dispatch_semaphore_create(1);
    });
    
    
}

+ (NSString *)executeTask:(void (^)(void))task start:(NSTimeInterval)start interval:(NSTimeInterval)interval repeat:(BOOL)repeat async:(BOOL)async {
    
    if (!task || start < 0 || (interval <= 0 && repeat)) return nil;
    
    //获取队列
    //    dispatch_queue_t mainQueue = dispatch_get_main_queue();
    dispatch_queue_t queue = async ? dispatch_queue_create("serialQueue", DISPATCH_QUEUE_SERIAL) : dispatch_get_main_queue();
    
    //创建定时器
    dispatch_source_t timer_t = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, queue);
    
    //设置时间
    dispatch_source_set_timer(timer_t, dispatch_time(DISPATCH_TIME_NOW, start * NSEC_PER_SEC), interval * NSEC_PER_SEC, 0);
    
    //加锁
    dispatch_semaphore_wait(semaphore_t_, DISPATCH_TIME_FOREVER);
    //定时器唯一标识
    NSString *taskSign = [NSString stringWithFormat:@"%lu", (unsigned long)timers_.count];
    timers_[taskSign] = timer_t;
    //解锁
    dispatch_semaphore_signal(semaphore_t_);
    
    //设置回调
    //两个方法同时写,只会有一个执行,而且是最后那个
//    dispatch_source_set_event_handler_f(timer_t, timerFunction);
    dispatch_source_set_event_handler(timer_t, ^{
        task();
        if (!repeat) {
            [self cancelTask:taskSign];
        }
    });
    
    //启动定时
    dispatch_resume(timer_t);
    
    return taskSign;
}

+ (NSString *)executeTask:(id)target selector:(SEL)aSelector start:(NSTimeInterval)start interval:(NSTimeInterval)interval repeat:(BOOL)repeat async:(BOOL)async {
    
    if (!target || !aSelector) return nil;
    
    return [self executeTask:^{
        if ([target respondsToSelector:aSelector]) {
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
            [target performSelector:aSelector];
#pragma clang diagnostic pop
        }
    } start:start interval:interval repeat:repeat async:async];
}

+ (void)cancelTask:(NSString *)taskSign {
    if (!taskSign.length) return;
    
    //加锁
    dispatch_semaphore_wait(semaphore_t_, DISPATCH_TIME_FOREVER);
    dispatch_source_t timer_t = timers_[taskSign];
    if (!timer_t) return;
    
    dispatch_source_cancel(timers_[taskSign]);
    [timers_ removeObjectForKey:taskSign];
    //解锁
    dispatch_semaphore_signal(semaphore_t_);
}

@end

使用示例

//
//  ViewController.m
//  定时器-GCD
//
//  Created by 宋法键 on 2019/8/14.
//  Copyright © 2019 WZSG. All rights reserved.
//

#import "ViewController.h"
#import "SFJTimer.h"

@interface ViewController ()

@property (strong, nonatomic) dispatch_source_t timer_t;
@property (copy, nonatomic) NSString *taskSign1;
@property (copy, nonatomic) NSString *taskSign2;

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    _taskSign1 = [SFJTimer executeTask:^{
        NSLog(@"-----aaaaa-----%@", [NSThread currentThread]);
    } start:2.0f interval:1.0f repeat:NO async:NO];
    
    _taskSign2 = [SFJTimer executeTask:self selector:@selector(doTask) start:2.0f interval:1.0f repeat:YES async:NO];
    
//    self.timer_t = timer_t;
}

- (void)doTask {
    NSLog(@"%s : %@", __func__, [NSThread currentThread]);
}

- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event {
    [SFJTimer cancelTask:_taskSign1];
}

@end

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Swift1> Swift和OC的区别1.1> Swift没有地址/指针的概念1.2> 泛型1.3> 类型严谨 对...
    cosWriter阅读 11,148评论 1 32
  • 喜欢的小伙伴请帮我点赞,你的支持会让我更加努力创作,比心❤️❤️❤️
    江子衿阅读 228评论 2 2
  • 我是在飞机上读完了这本书,不得不说这本书非常吸引人,以致于我一口气把这本书看完了。但是因为里面时空的转换搞...
    吴姐姐的简书阅读 337评论 0 0
  • 感恩今天回老家一路顺利,安全抵达。 感恩在老家去办事的时候老舅亲自开车,不用担心路不熟,有老舅这位老司机带路方便多...
    毛毛细雨mmxy阅读 187评论 0 0
  • 今天休息,睡了2个小时觉,快到点了去托辅接女儿,回来女儿说妈妈我想吃武大郎烧饼了,给女儿买了一个,女儿说给弟弟也买...
    海浪花_2642阅读 140评论 0 0