GJCollection/GJCollection.m at master · GJGroup/GJCollection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71


#import "GJCollection.h"

@implementation NSArray(GJRecursiveDeepCopy)

- (id)gjw_recursiveDeepCopy {
    NSMutableArray *copyArr = [NSMutableArray arrayWithCapacity:self.count];
    [self enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        id copyValue;
        if ([obj respondsToSelector:@selector(gjw_recursiveDeepCopy)]) { //Collection对象进行递归拷贝
            copyValue = [obj gjw_recursiveDeepCopy];
        } else if ([obj conformsToProtocol:@protocol(NSMutableCopying)]) { //非Collection对象的NSObject对象
            copyValue = [obj mutableCopy];
        } else if ([obj conformsToProtocol:@protocol(NSCopying)]) { //自定义的NSObject对象，自己实现了拷贝
            copyValue = [obj copy];
        }
        if (copyValue) {
            [copyArr addObject:copyValue];
        }
    }];
    return copyArr;
}

@end


@implementation NSDictionary(GJRecursiveDeepCopy)

- (id)gjw_recursiveDeepCopy {
    NSMutableDictionary *copyDic = [NSMutableDictionary dictionaryWithCapacity:self.count];
    [self enumerateKeysAndObjectsUsingBlock:^(id  _Nonnull key, id  _Nonnull obj, BOOL * _Nonnull stop) {
        id copyValue;
        if ([obj respondsToSelector:@selector(gjw_recursiveDeepCopy)]) { //Collection对象进行递归拷贝
            copyValue = [obj gjw_recursiveDeepCopy];
        } else if ([obj conformsToProtocol:@protocol(NSMutableCopying)]) { //非Collection对象的NSObject对象
            copyValue = [obj mutableCopy];
        } else if ([obj conformsToProtocol:@protocol(NSCopying)]) { //自定义的NSObject对象，自己实现了拷贝
            copyValue = [obj copy];
        }
        if (copyValue) {
            copyDic[key] = copyValue;
        }
    }];
    return copyDic;
}

@end


@implementation NSSet(GJRecursiveDeepCopy)

- (id)gjw_recursiveDeepCopy {
    NSMutableSet *copySet = [NSMutableSet setWithCapacity:self.count];
    [self enumerateObjectsUsingBlock:^(id  _Nonnull obj, BOOL * _Nonnull stop) {
        id copyValue;
        if ([obj respondsToSelector:@selector(gjw_recursiveDeepCopy)]) { //Collection对象进行递归拷贝
            copyValue = [obj gjw_recursiveDeepCopy];
        } else if ([obj conformsToProtocol:@protocol(NSMutableCopying)]) { //非Collection对象的NSObject对象
            copyValue = [obj mutableCopy];
        } else if ([obj conformsToProtocol:@protocol(NSCopying)]) { //自定义的NSObject对象，自己实现了拷贝
            copyValue = [obj copy];
        }
        if (copyValue) {
            [copySet addObject:copyValue];
        }
    }];
    return copySet;
}

@end