mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-08-08 08:31:13 +00:00
65 lines
1.5 KiB
Objective-C
65 lines
1.5 KiB
Objective-C
#import "SDisposableSet.h"
|
|
|
|
#import <libkern/OSAtomic.h>
|
|
|
|
@interface SDisposableSet ()
|
|
{
|
|
OSSpinLock _lock;
|
|
id<SDisposable> _singleDisposable;
|
|
NSArray *_multipleDisposables;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation SDisposableSet
|
|
|
|
- (void)add:(id<SDisposable>)disposable
|
|
{
|
|
if (disposable == nil)
|
|
return;
|
|
|
|
OSSpinLockLock(&_lock);
|
|
if (_multipleDisposables != nil)
|
|
{
|
|
NSMutableArray *multipleDisposables = [[NSMutableArray alloc] initWithArray:_multipleDisposables];
|
|
[multipleDisposables addObject:disposable];
|
|
_multipleDisposables = multipleDisposables;
|
|
}
|
|
else if (_singleDisposable != nil)
|
|
{
|
|
NSMutableArray *multipleDisposables = [[NSMutableArray alloc] initWithObjects:_singleDisposable, disposable, nil];
|
|
_multipleDisposables = multipleDisposables;
|
|
_singleDisposable = nil;
|
|
}
|
|
else
|
|
{
|
|
_singleDisposable = disposable;
|
|
}
|
|
OSSpinLockUnlock(&_lock);
|
|
}
|
|
|
|
- (void)dispose
|
|
{
|
|
id<SDisposable> singleDisposable = nil;
|
|
NSArray *multipleDisposables = nil;
|
|
|
|
OSSpinLockLock(&_lock);
|
|
singleDisposable = _singleDisposable;
|
|
multipleDisposables = _multipleDisposables;
|
|
_singleDisposable = nil;
|
|
_multipleDisposables = nil;
|
|
OSSpinLockUnlock(&_lock);
|
|
|
|
if (singleDisposable != nil)
|
|
[singleDisposable dispose];
|
|
if (multipleDisposables != nil)
|
|
{
|
|
for (id<SDisposable> disposable in multipleDisposables)
|
|
{
|
|
[disposable dispose];
|
|
}
|
|
}
|
|
}
|
|
|
|
@end
|