Swift编程小技巧

swift tips

Swift中有很多有用的小技巧，用好了能使代码更加安全，简洁，易于理解或效率更加高效，在这记录一些编写swifty code的小技巧。

1、for in 循环中的可选值解包

当使用for in循环一个包含可选值的数组时，我们可能会使用if let 或guard解包：

 let animals = ["dog", nil, "pig", "cat", nil]
 for obj in animals {
     if let animal = obj {
         print(animal)
     }
 }

上述代码完全没问题，但是我们可以在for in中使用case let来简化代码：

for case let animal? in animals {
    print(animal)
}

或者使用compactMap解包,可以参考Swift：map(), flatMap() 和 compactMap() 的区别：

for animal in animals.compactMap({$0}) {
    print(animal)
}

如上两种方法都能时代码更加简洁

2、for in 循环中使用 where 语句

我们都会遇到写类似如下代码的地方：

let items = [1, 2, 3, 4]
for item in items {
    if(item % 2 == 0) {
        print(item)
    }
}
Output: 2 4 

使用where语句可以使代码变得更简洁：

for item in items where item % 2 == 0 {
    print(item)
}
Output: 2 4

3、使用filter + forEach替代 for in + where

let items = [1, 2, 3, 4]
items.filter{$0 % 2 == 0}.forEach {
    print($0)
}

可以看出代码简洁度并没有很大的提升，但是语义更加清晰更容易理解。

4、Defer保证代码块在控制流退出前被调用

defer 所声明的代码块会在当前代码执行退出后被调用:

func test(code: Int) {
    defer {
        print("end print in defer")
    }
    if code < 0 {
        print("less than 0")
        return
    }
    if code > 10 {
        print("big than 10")
    }
}
test(code: -1)
test(code: 11)

Output:
less than 0
end print in defer
big than 10
end print in defer

Objective-C中也可以实现类似的功能，可以参考：Objective-C中实现Swift中的defer

5、布尔值取反

在任何编程语言中，布尔值都是最常用和最简单的数据类型之一。而取反操作也是非常常见的，比如：

var isSelect = true
if isSelect {
  isSelect = false
}

上面的代码应该不会有人写吧，这很难说哦，哈哈哈

if isSelect {
  isSelect = !isSelect
}

这种写法应该人多一些，也不容易出错一些，但是我觉得下面的更好：

if isSelect {
  isSelect.toggle()
}

toggle()是一个swift提供的函数，用来切换布尔变量的值。

6、数组内元素的类型转换

常见的情况在获取一个视图的所有子视图后，需要给某一类视图做一些操作，以UILabel为例

• 使用for in 循环

for subview in self.view.subviews {
    if subview is UILabel {
        (subview as! UILabel).text = "find"
    }
}

for subview in self.view.subviews where subview is UILabel {
   (subview as! UILabel).text = "find"
}

for subview in self.view.subviews.compactMap({$0 as? UILabel}) {
    subview.text = "find"
}

for case let subview as UILabel in self.view.subviews {
    subview.text = "find"
}

此处for in 循环中明显使用 case let 和 compactMap是最方遍，代码最简洁的

• compactMap的另一种使用

self.view.subviews
    .compactMap{$0 as? UILabel}
    .forEach {
        $0.text = "find"
}

个人比较推荐这一种写法，语义更加清晰

7、同时遍历数组的索引和元素

let array = ["a","b","c","d"]
for (index, element) in array.enumerated() {
    print("\(index)--\(element)")
}

利用元组在swift中遍历出索引和元素非常简洁，OC中需要使用enumerateObjectsUsingBlock来实现

[array enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
    NSLog(@"%d -- %@", idx, obj);   
}];

8、静态工厂方法

在Swift使用静态工厂方法和属性来执行对象的设置可能是一种将设置代码与实际逻辑清晰分开的好方法,具体可以参考：Swift：静态工厂方法