技术专栏 | 浅析iOS开发中的实例方法和类方法“优化”方案

2024-01-29 16:27:45

在iOS开发的世界里，实例方法和类方法是两大基本概念。实例方法属于特定类的实例，而类方法则属于类本身。通常情况下，实例方法用于处理实例特定的数据，而类方法用于执行与类本身相关的任务。

然而，在某些情况下，您可能希望根据运行时的某些条件动态地修改实例方法或类方法的行为。这就是resolveInstanceMethod和resolveClassMethod方法的用武之地。

本文将深入探讨resolveInstanceMethod和resolveClassMethod方法，并演示如何使用它们来挽救看似无用的实例方法和类方法。

实例方法的救赎

假设您有一个名为Person的类，其中包含一个名为getName的实例方法，该方法返回该人的姓名。现在，您希望根据特定的条件修改getName方法的行为，例如，当调用该方法时，该方法返回Unknown。

要实现此目的，您可以重写resolveInstanceMethod方法，如下所示：

- (BOOL)resolveInstanceMethod:(SEL)sel {
    if (sel == @selector(getName)) {
        class_addMethod(self.class, sel, (IMP)customGetName, "v@:");
        return YES;
    }
    return [super resolveInstanceMethod:sel];
}

static void customGetName(id self, SEL _cmd) {
    NSLog(@"Custom getName implementation");
    NSLog(@"Returning Unknown");
}

在上面的示例中，我们检查sel是否等于@selector(getName)。如果是，我们使用class_addMethod函数动态地将自定义getName实现添加到类中。自定义实现将打印日志消息并返回Unknown。

类方法的复生

同样，您可以使用resolveClassMethod方法来修改类方法的行为。例如，假设您有一个名为DataManager的类，其中包含一个名为fetchData的类方法，该方法从数据库中检索数据。您希望根据特定条件（例如，当用户已登录时）修改fetchData方法的行为。

要做到这一点，您可以重写resolveClassMethod方法，如下所示：

+ (BOOL)resolveClassMethod:(SEL)sel {
    if (sel == @selector(fetchData)) {
        class_addMethod(self, sel, (IMP)customFetchData, "v@:");
        return YES;
    }
    return [super resolveClassMethod:sel];
}

static void customFetchData(id self, SEL _cmd) {
    if ([User isLoggedIn]) {
        NSLog(@"Custom fetchData implementation");
        NSLog(@"Fetching data from database");
    } else {
        NSLog(@"User is not logged in");
    }
}