Timer优化 - Lab.Lawliet

666 字

3 分钟

Timer优化

2021-12-13

2026-10-12

iOS

内存管理

前言#

NSTimer 是 iOS 开发中老生常谈的知识点了，今天再来唠叨一番。

一、 NSTimer 常规使用#

通过下面简单的 Demo 看看现象。

1
#import "NSTimerViewController.h"
2

3
@interface NSTimerViewController ()
4
@property (nonatomic, strong) NSTimer *timer;
5
@property (nonatomic, assign) NSInteger count;
6
@end
7

8
@implementation NSTimerViewController
9

10
- (void)viewDidLoad {
11
    [super viewDidLoad];
12

13
    [self selfTargetTimer];
14
}
15

16
#pragma mark - 不同 NSTimer 使用方式
17
/// 常规使用 self 作为 Target 的 NSTimer
18
- (void)selfTargetTimer {
19
    self.timer = [NSTimer scheduledTimerWithTimeInterval:1 target:self selector:@selector(countAction) userInfo:nil repeats:YES];
20
    [[NSRunLoop currentRunLoop] addTimer:self.timer forMode:NSRunLoopCommonModes];
21
}
22

23
#pragma mark - Action
24
- (void)countAction {
25
    self.count += 1;
26
    NSLog(@"%@ - %ld",self, self.count);
27
}
28

29
- (void)dealloc {
30
    [self.timer invalidate];
31
    self.timer = nil;
32
    NSLog(@"NSTimerViewController dealloc");
33
}
34

35
@end

我们反复 Push 到这个 ViewController ，出现了典型的内存泄露。

通过内存图我们能够发现，根本原因是 RunLoop -> Timer <-> Self 这样的一个持有关系。

而在该API的文档中也是有说明的。

为了解决内存泄露，我们需要打破这个循环引用。

1.1 weakSelf#

我们用 weakself 替代 Self 是否有用呢？

1
__weak typeof(self)weakSelf = self;
2
self.timer = [NSTimer scheduledTimerWithTimeInterval:1 target:weakSelf selector:@selector(countAction) userInfo:nil repeats:YES];
3
[[NSRunLoop currentRunLoop] addTimer:self.timer forMode:NSRunLoopCommonModes];

还是一样，这里是 RunLoop -> Timer <-> weakSelf -> Self 依旧没有打破。

1.2 block#

还有这样一个 API

1
+ (NSTimer *)scheduledTimerWithTimeInterval:(NSTimeInterval)interval
2
                                    repeats:(BOOL)repeats
3
                                      block:(void (^)(NSTimer *timer))block;

我们试试

1
__weak typeof(self)weakSelf = self;
2
    self.timer = [NSTimer scheduledTimerWithTimeInterval:1 repeats:YES block:^(NSTimer * _Nonnull timer) {
3
        [weakSelf countAction];
4
    }];

这个 API 就可以很好的避免循环引用了。这里的关系是 RunLoop -> Timer -> block 。这里只需要注意不要让 block 产生内存泄露就ok了。

二、自定义 Timer#

2.1 加一层间接#

通过封装，解决 Timer 和 Self 的强引用关系： RunLoop -> Timer -> ?? ~~ Self

1
#import <objc/message.h>
2

3
@interface RYTimer ()
4

5
@property (nonatomic, weak) id target;
6
@property (nonatomic, assign) SEL aSelector;
7
@property (nonatomic, strong) NSTimer *timer;
8

9
@end
10

11
@implementation RYTimer
12

13
- (instancetype)initWithTimeInterval:(NSTimeInterval)ti target:(id)aTarget selector:(SEL)aSelector userInfo:(nullable id)userInfo repeats:(BOOL)yesOrNo {
14
    if (self == [super init]) {
15
        self.target = aTarget;
16
        self.aSelector = aSelector;
17

18
        if ([self.target respondsToSelector:self.aSelector]) {
19
            Method method = class_getInstanceMethod([self.target class], aSelector);
20
            const char *type = method_getTypeEncoding(method);
21
            class_addMethod([self class], aSelector, (IMP)fireIMP, type);
22

23
            self.timer = [NSTimer scheduledTimerWithTimeInterval:ti target:self selector:aSelector userInfo:userInfo repeats:yesOrNo];
24
        }
25
    }
26
    return self;
27
}
28

29
void fireIMP(RYTimer *timer){
30
    if (timer.target) {
31
        void (*ry_msgSend)(void *,SEL, id) = (void *)objc_msgSend;
32
        ry_msgSend((__bridge void *)(timer.target), timer.aSelector,timer.timer);
33
    }else{
34
        [timer invalidate];
35
    }
36
}
37

38
- (void)invalidate{
39
    [self.timer invalidate];
40
    self.timer = nil;
41
}
42

43
- (void)dealloc{
44
    NSLog(@"%s",__func__);
45
}
46

47
@end

2.2 消息转发机制#

我们同样可以利用消息转发机制来对 Timer 进行优化，这里的思想和上面一样，都是使用了一层间接，但是具体实现方式不同。这里使用了 NSProxy 结合消息转发实现。

1
@interface RYProxy : NSProxy
2

3
+ (instancetype)proxyWithTransformObject:(id)object;
4

5
@end
6

7

8
@interface RYProxy ()
9

10
@property (nonatomic, weak) id object;
11

12
@end
13

14
@implementation RYProxy
15

16
+ (instancetype)proxyWithTransformObject:(id)object {
17
    RYProxy *proxy = [RYProxy alloc];
18
    proxy.object = object;
19
    return proxy;
20
}
21

22
-(id)forwardingTargetForSelector:(SEL)aSelector {
23
    return self.object;
24
}
25

26
- (NSMethodSignature *)methodSignatureForSelector:(SEL)sel{
27
    if (self.object) {
28
    }
29
    return [self.object methodSignatureForSelector:sel];
30

31
}
32

33
- (void)forwardInvocation:(NSInvocation *)invocation{
34
    if (self.object) {
35
        [invocation invokeWithTarget:self.object];
36
    }
37
}
38

39
@end

虽然 Target 并不是 Self 但经过消息转发，最终处理消息的还是消息的还是 Self 。

1
RYProxy *timerProxy = [RYProxy proxyWithTransformObject:self];
2
self.timer = [NSTimer scheduledTimerWithTimeInterval:1 target:timerProxy selector:@selector(countAction) userInfo:nil repeats:YES];