[Swift] 안전하게 index 접근하기

 

Array에서 값을 꺼내올 때는 .first나 .last로 꺼내올 수도 있지만 특정 인덱스를 꺼낼 때에는 Array[index], 이런 형태로 값에 접근하게 된다. 

 

하지만 이 때 .first나 .last는 옵셔널 타입으로 만약 값이 없다면 nil이 나오게 되지만 [index]의 형태로 값을 접근했을 때 해당 인덱스에 해당하는 값이 존재하지 않을 경우 아예 fatal error가 발생하게 된다. 

(Fatal error: Index out of range)

 

즉, 값에 잘못 접근하게 될 경우 아예 앱 자체가 죽어버리는 것이다.

 

그럼 [index]의 형태로 값에 안전하게 접근할 방법은 없을까??

 

 

이를 위해선 직접 Array의 extension으로 subscript를 정의해주면 된다. 

extension Array {
    
    subscript(safe index: Int) -> Element? {
        return indices ~= index ? self[index] : nil
    }
    
}

 

이런 식으로 따로 파라미터에 뭐가 들어갈 지 정해줄 경우 값에 접근할 때 Array[safe: 2] 이런 식으로 접근할 수 있다. 

 

여기서 indices는 iOS 8.0 이상부터 사용이 가능한 유효한 인덱스들을 오름차순으로 모아둔 것이다. 따라서 ~= 연산자를 통해 작성한 index 값이 indices에 포함되어 있는지 확인하고 만약 있을 경우 해당 값을 반환하고, 없을 경우 .first나 .last처럼 nil을 반환할 수 있도록 하는 것이다. 

 

물론 이렇게 처리해줄 경우 옵셔널을 따로 처리해줘야 하긴 하지만 앱이 죽어버리는 불상사는 피할 수 있다.