CTMediator 原理详解：揭秘幕后关联机制

2024-02-12 14:45:17

人机协同撰写博客：揭秘 CTMediator 的运作机制

在 iOS 开发中，CTMediator 作为一种强大的通信机制，被广泛应用于模块解耦和跨模块调用场景。它通过一种独特的方式将 SEL（方法选择器）与 IMP（实现指针）进行关联，实现了模块之间灵活高效的交互。本文将深入剖析 CTMediator 的工作原理，为你揭开它背后的运作机制。

CTMediator 的结构

CTMediator 的核心是一个简单的结构体，它包含两个成员变量：SEL 和 IMP。

typedef struct {
    SEL sel;
    IMP imp;
} CTMediatorEntry;

SEL 是一个方法选择器，用于标识方法。IMP 是一个实现指针，指向方法的实现代码。通过将 SEL 和 IMP 关联起来，CTMediator 建立了一个映射，允许我们通过 SEL 来调用对应的 IMP 实现。

消息转发机制

CTMediator 通过消息转发机制实现方法调用。当一个对象收到一个它没有实现的方法调用时，它会触发消息转发机制。

在 CTMediator 中，消息转发机制被重写，以便它可以拦截未实现的方法调用并将其重定向到正确的 IMP 实现。

SEL 和 IMP 的关联

CTMediator 通过使用一个 NSSet 来存储所有注册的 SEL。当一个模块想要注册一个方法时，它会将该方法的 SEL 添加到 NSSet 中。

- (void)registerTarget:(id)target action:(SEL)action;
{
    // ...

    CTMediatorEntry entry = {
        .sel = action,
        .imp = imp,
    };

    [self.mediatorEntries addObject:[NSValue value:&entry withObjCType:@encode(CTMediatorEntry)]];

    // ...
}

当另一个模块想要调用这个方法时，它会向 CTMediator 发送一个消息，其中包含要调用的 SEL。CTMediator 会在 NSSet 中查找该 SEL，并返回关联的 IMP。

- (id)performTarget:(id)target action:(SEL)action parameters:(NSArray *)parameters;
{
    // ...

    CTMediatorEntry entry;
    if ([self.mediatorEntries containsObject:[NSValue value:&entry withObjCType:@encode(CTMediatorEntry)]]) {
        // ...

        return ((id (*)(id, SEL, ...))entry.imp)(target, action, parameters);

        // ...
    }

    // ...
}