iOS -- 代码优化

1. filter 优化

原代码

 if var vcs = vcs {for vc in vcs {if vc is XXLoginViewController {if let index = vcs.firstIndex(of: vc) {vcs.remove(at: index)}}
}

修改后

vcs = vcs?.filter { !($0 is XXLoginViewController) }

或者

vcs?.removeAll(where: { $0 is XXLoginViewController })

filter, map, reduce的用法详见
https://juejin.cn/post/6844903574506307592
https://www.jianshu.com/p/74854e641d87

2. return 优化

案例1

func checkExpire(time: TimeInterval) -> Bool {if Date().timeIntervalSince1970 - time > 0 {return false}return true
}

优化后

func checkExpire(time: TimeInterval) -> Bool {return Date().timeIntervalSince1970 < time
}

是不是更简洁了呢！

案例2

if (isInCache) {return result;
} else {return [DataXXManager.shareManager isStickUpWithXxId:xxId];
}

可以通过移除不必要的else语句来优化代码，因为在if条件满足时，return会直接返回，后续代码不会执行。以下是优化后的代码：

优化后

if (isInCache) {return result;
}
return [DataXXManager.shareManager isStickUpWithXxId:xxId];

3. for循环优化

案例1

for index in 0..<array.count {guard let item = array[index] as? XXX else {continue}// do other
}

优化后

for (index, header) in array.enumerated() {guard let item = header as? XXX else {continue}// do other
}

案例2

优化前

var index = 0
var tempItems: [[ObjectProtocol]] = [ObjectProtocol]
for item in self.items {if index >= 3 {break}if let f = self.items.last as? [ObjectProtocol] {tempItems.append(f)}index += 1
}
self.items = tempItems

优化后

var tempItems: [[ObjectProtocol]] = []for item in self.items.prefix(3) {if let f = item as? [ObjectProtocol] {tempItems.append(f)}
}self.items = tempItems

4. 枚举优化

例如有如下的需求，像数据库中添加obj字段，obj的order字段如果是auto，则根据head和tail字段来表示自动变化order，如果非auto，则使用obj自己的order字段内容。
直观的代码如下：

- (void)addSomethingToDB(NSObject *)obj auto:(BOOL)auto isAddToHead:(BOOL)head isAddToTail:(BOOL)tail {if (auto) {if (head) {obj.order = --index} else if (tail) {obj.order = ++index}}
}

上面的代码，用了auto, head和tail来表示，添加的顺序。是不是繁琐。那怎么简化呢？
可以用枚举来优化。

typedef NS_ENUM(NSInteger, XXSortOrder) {XXSortOrderNo, // 不自动排序XXSortOrderAddToHead, // 自动排序，在头插入XXSortOrderAddTail, // 自动排序，在尾插入
}- (void)addSomethingToDB(NSObject *)obj sortType:(XXSortOrder)sortType {if (XXSortOrder == XXSortOrderAddToHead) {obj.order = --index} else if (XXSortOrder == XXSortOrderAddToHead) {obj.order = ++index}
}