技术深潜：iOS 代理的一对多通知实现

IOS

2023-11-10 00:46:54

iOS 代理机制为松耦合通信模式提供了强大、灵活的解决方案。虽然一对一代理实现非常简单，但一对多通知有时也会带来便利。然而，由于与 NSNotificationCenter 相关的潜在内存泄漏和耦合问题，iOS 并不原生支持一对多代理。

本文将探讨使用代理机制实现一对多通知的巧妙方法，同时解决与 NSNotificationCenter 相关的问题。我们将深入剖析代理设计模式，了解其原理和优势。

代理设计模式的原理

代理设计模式是一种行为型设计模式，允许对象通过代理对象间接与另一个对象交互。当主体对象需要将请求委托给另一个对象时，代理对象充当中间人，控制对目标对象的访问。

代理设计的优势在于：

松散耦合： 主体和代理对象彼此不直接依赖，因此可以独立更改。
可扩展性： 代理对象可以根据需要添加或移除，而无需修改主体对象。
灵活性： 代理对象可以增强或修改目标对象的行为，提供额外的功能或安全性措施。

iOS 代理机制

在 iOS 中，代理机制基于 Objective-C 的协议。协议定义了一组方法，充当代理对象和委托对象之间通信的接口。委托对象是使用代理对象的主体对象。

代理对象遵循协议中的方法，并处理委托对象发送的任何请求。通过这种方式，代理对象可以对委托对象的行为进行中介和定制。

一对多代理实现

为了实现一对多代理通知，我们将使用一个自定义委托类，该类将充当所有代理对象的中央枢纽。这个委托类将：

包含所有代理对象的数组： 这将允许我们存储和管理所有参与通知的对象。
实现协议中的方法： 该委托类将实现所有代理对象应响应的方法。
向所有代理对象转发请求： 当收到委托对象请求时，委托类将遍历代理对象数组，并向每个对象转发请求。

解决 NSNotificationCenter 的问题

NSNotificationCenter 是一个强大的通知机制，但它存在内存泄漏的风险。如果一个对象注册了通知，但没有注销，它可能会在注销后仍然收到通知，从而导致内存泄漏。

为了解决这个问题，我们将使用 weak 引用来持有代理对象。weak 引用是一种指向对象的指针，当对象不再被其他对象引用时，它会自动将其值设置为 nil。通过使用 weak 引用，当代理对象被注销时，委托类将自动将其从数组中移除，防止内存泄漏。

示例实现

以下是使用自定义委托类实现一对多代理通知的示例代码：

@interface MulticastDelegate : NSObject

@property (nonatomic, strong) NSMutableArray *delegates;

- (void)addDelegate:(id<MyProtocol>)delegate;
- (void)removeDelegate:(id<MyProtocol>)delegate;
- (void)forwardInvocation:(NSInvocation *)invocation;

@end

@implementation MulticastDelegate

- (instancetype)init {
    self = [super init];
    if (self) {
        _delegates = [NSMutableArray array];
    }
    return self;
}

- (void)addDelegate:(id<MyProtocol>)delegate {
    [self.delegates addObject:delegate];
}

- (void)removeDelegate:(id<MyProtocol>)delegate {
    [self.delegates removeObject:delegate];
}

- (void)forwardInvocation:(NSInvocation *)invocation {
    for (id<MyProtocol> delegate in self.delegates) {
        if ([delegate respondsToSelector:invocation.selector]) {
            [invocation invokeWithTarget:delegate];
        }
    }
}

@end