[iOS] Associated Type

associated type에 관하여

 

안녕하세요, 태끼입니다.
이번 글은 정리형태로 작성해두어 어투가 조금 딱딱할 수 있습니다.. 이해부탁드려요 🥲

이번 글은 어투도 딱딱하고 줄글 형식이다. 너무 어렵게 생각할 필요가 없는 개념이다. 단순하게 맞서보자.

 

요약

• 프로토콜에서 사용되는 제네릭 Placeholder의 느낌이다.
• Associated Type은 원래 typealias였는데, Swift 2.2부터 Associated Type이란 키워드로 바뀌었다.

 

설명

위의 두 가지 관점에서 보면 내용은 크게 어렵지 않다. 아래의 예시로 함께 이해해보겠다.

 

우선 프로젝트에 대해 잠깐 이해해보면, 프로젝트는 현재 MVVM 패턴을 따르고 있다. 다음 프로젝트에서는 비즈니스 로직을 처리하는 ViewModel을 구현해야 한다. 보통 각 ViewController마다 대응되어 각각의 ViewModel을 하나씩 구현하고, ViewController는 ViewModel을 속성으로 가지게 된다.

 

프로젝트를 구성하다보면 공통적으로 생기는 요소들, 반복되는 속성이나 메서드가 있을텐데 우리는 그것을 프로토콜의 형태로 사용할 수 있을 것이다. 프로토콜 지향 프로그래밍을 적절하게 잘 이용하게 되면 재사용성이나 확장성에 매우 용이하게 된다.

 

프로젝트에서 ViewModel이라는 것을 ViewController마다 가지게 되고, ViewModel과 View를 연결해주는 작업인 bindViewModel()을 공통적으로 수행하게 된다. 이렇게 공통되는 작업을 우리는 프로토콜로 묶어서 작업할 것이다.

 

프로토콜에는 형태만 잡아두게 된다. 직접 구현까지는 하지 않는다는 것이다. 혹시 이 부분이 이해가 되지 않는다면 프로토콜에 대해서 한 번 이해하고 오는 것이 좋을 것 같다. 계속 이어나가면 우리가 만들 ViewModelBindableType 프로토콜안에는 구현되어야 할 viewModel 프로퍼티와 bindViewModel 메서드가 존재한다.

 

그러나 생각해 볼 포인트가 하나 있다. 우리는 ViewModel을 ViewController에 따라 계속해서 새로 만들고 있기 때문에 각 ViewModel의 Type 은 모두 다르다. 그러면 어떻게 해야하지? 여기서 좋은 해결책이 하나 나온다. associatedType 을 이용하면 된다. 임의로 타입을 정해두어서 viewModel의 타입으로 사용하는 것이다. 여기서 typealias와 generic 개념이 떠오른다.

 

만약 프로퍼티의 타입이 동일하다면 우리는 그냥 이렇게 작성하면 된다. 공통적으로 사용될 name이라는 프로퍼티가 항상 String 타입일 것이라면 아래와 같이 작성하면 될 것이다.

 

protocol ViewContainer {
	var name: String { get set }
}

 

그러나 우리는 타입이 달라질 수 있는 경우를 생각해서 다음과 같이 타입을 임의로 지정해줄 것이다. 그리고 실제로 ViewController에서 ViewModelBindableType을 채택해서 사용할 때에 viewModel의 타입을 우리가 원하는 형식(MemoListViewModel, MemoDetailViewModel 등)으로 지정해줄 수 있다.

 

// ViewModelBindableType.swift (프로토콜)

protocol ViewModelBindableType {
    associatedtype ViewModelType
    
    var viewModel: ViewModelType! { get set }
    func bindViewModel()
}

 

// MemoListViewController.swift (뷰 컨트롤러)

import UIKit

class MemoListViewController: UIViewController, ViewModelBindableType {
    
    var viewModel: MemoListViewModel!
    
    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    func bindViewModel() {
        
    }
}

// MemoDetailViewController.swift (뷰 컨트롤러)

import UIKit

class MemoDetailViewController: UIViewController, ViewModelBindableType {

    var viewModel: MemoDetailViewModel!
    
    override func viewDidLoad() {
        super.viewDidLoad()

        // Do any additional setup after loading the view.
    }
    
    func bindViewModel() {
        
    }
}

// 위의 있는 코드에 typealias를 추가한 것

import UIKit

class MemoDetailViewController: UIViewController, ViewModelBindableType {
    
    typealias ViewModelType = MemoDetailViewModel

    var viewModel: ViewModelType!
    
    override func viewDidLoad() {
        super.viewDidLoad()

        // Do any additional setup after loading the view.
    }
    
    func bindViewModel() {
        
    }
}

 

마무리

어렵게 생각할 필요없다. 다른 유형도 들어올 수 있으므로 유형을 미리부터 정하지 않는 것이다. 임의의 타입을 정해두고 프로토콜이 적용되는 순간 실제 타입을 지정하는 것이다. 프로토콜에서 사용되는 제네릭이라고 보면 간단할 것 같다.

 

레퍼런스

https://zeddios.tistory.com/382

https://woongsios.tistory.com/97