NSTimer内存泄漏分析与绝对完美解决方法

2023-11-16 15:38:20

NSTimer是一个功能强大且在iOS应用开发中使用广泛的定时器类，但它也是内存泄漏的常见根源。本文将深入分析循环引用的产生与解决方法，并最终给出一种完美的解决方案，确保NSTimer定时器在不再需要时能够立即销毁，不留任何可能产生泄漏的隐患。

问题所在：循环引用

循环引用是指两个或多个对象相互引用，导致它们都无法被释放。在NSTimer的情况下，循环引用通常发生在NSTimer本身和持有它的ViewController之间。

ViewController持有NSTimer的强引用，以便能够在需要时使用它。而NSTimer也持有ViewController的强引用，以便能够在触发时向它发送消息。这样就形成了一个循环引用，导致NSTimer和ViewController都无法被释放。

解决方法一：使用弱引用

一种常见的解决方法是使用弱引用。弱引用是指一种不会增加被引用对象引用计数的引用。这意味着当一个弱引用的对象被销毁时，它的引用计数不会减少，也不会触发它的dealloc方法。

在NSTimer的情况下，可以在ViewController中使用弱引用来持有NSTimer。这样，当ViewController被销毁时，NSTimer的引用计数不会减少，它也不会被销毁。但当NSTimer不再需要时，它仍然会自动销毁，从而避免了内存泄漏。

__weak NSTimer *timer;

- (void)viewDidLoad {
    [super viewDidLoad];

    timer = [NSTimer scheduledTimerWithTimeInterval:1.0 target:self selector:@selector(onTimer:) userInfo:nil repeats:YES];
}

- (void)onTimer:(NSTimer *)timer {
    // Do something...
}

- (void)dealloc {
    [timer invalidate];
}

解决方法二：使用`NSProxy`

另一种解决方法是使用NSProxy。NSProxy是一个可以转发消息给另一个对象的类。这意味着可以使用NSProxy来持有NSTimer，而无需直接持有ViewController。

@interface TimerProxy : NSProxy

@property (nonatomic, weak) ViewController *viewController;

- (instancetype)initWithViewController:(ViewController *)viewController;

@end

@implementation TimerProxy

- (instancetype)initWithViewController:(ViewController *)viewController {
    self = [super init];
    if (self) {
        _viewController = viewController;
    }
    return self;
}

- (void)forwardInvocation:(NSInvocation *)invocation {
    if (_viewController) {
        [invocation invokeWithTarget:_viewController];
    }
}

- (NSMethodSignature *)methodSignatureForSelector:(SEL)sel {
    return [_viewController methodSignatureForSelector:sel];
}

@end

- (void)viewDidLoad {
    [super viewDidLoad];

    TimerProxy *proxy = [[TimerProxy alloc] initWithViewController:self];

    NSTimer *timer = [NSTimer scheduledTimerWithTimeInterval:1.0 target:proxy selector:@selector(onTimer:) userInfo:nil repeats:YES];
}

- (void)onTimer:(NSTimer *)timer {
    // Do something...
}

完美解决方案：使用闭包

上述两种解决方法都存在一些局限性。例如，使用弱引用需要在销毁NSTimer之前手动调用invalidate方法，否则NSTimer仍然会继续运行，造成不必要的资源浪费。而使用NSProxy则需要创建一个额外的类，增加了代码的复杂性。

为了解决这些问题，这里给出一种完美的解决方案：使用闭包。闭包是指一段可以被存储在变量中并可以像普通函数一样调用的代码块。

__block NSTimer *timer;

- (void)viewDidLoad {
    [super viewDidLoad];

    timer = [NSTimer scheduledTimerWithTimeInterval:1.0 block:^(NSTimer *timer) {
        // Do something...
    } repeats:YES];
}

- (void)dealloc {
    [timer invalidate];
}

使用闭包的好处在于，它可以自动释放NSTimer。当ViewController被销毁时，闭包也会被销毁，从而自动调用invalidate方法并释放NSTimer。这样就避免了内存泄漏，而且代码也更加简洁。