Colin's Blog
[swift] RxCocoa의 리소스 관리 및 메모리 관련 이슈

RxCocoa는 ReactiveX를 Swift 언어에 맞게 구현한 라이브러리로, iOS 및 macOS 앱의 리액티브 프로그래밍을 지원합니다. 하지만 RxCocoa를 사용할 때 주의해야 할 몇 가지 리소스 관리와 메모리 관련 이슈가 있습니다. 이번 포스트에서는 이러한 이슈들을 알아보고 어떻게 해결할 수 있는지 살펴보겠습니다.

1. DisposeBag

RxCocoa를 사용하면 Observable과 Disposable 객체를 관리해야 합니다. Observable은 이벤트 스트림을 생성하고, Disposable은 구독을 취소하거나 정리하는 데 사용됩니다. 이 때 DisposeBag을 사용하면 간단하게 여러 Disposable을 관리할 수 있습니다.

import RxCocoa
import RxSwift

class MyViewController: UIViewController {
    
    let disposeBag = DisposeBag()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let button = UIButton()
        
        Observable.just("Hello, RxCocoa!")
            .subscribe(onNext: { text in
                print(text)
            })
            .disposed(by: disposeBag)
        
        button.rx.tap
            .subscribe(onNext: {
                // 버튼이 클릭되었을 때의 동작
            })
            .disposed(by: disposeBag)
    }
}

위의 예제에서는 disposeBag을 생성하고, Observable과 버튼 탭 이벤트에 대한 Disposable을 disposeBag에 추가하였습니다. 이렇게 하면 뷰 컨트롤러가 해제될 때 disposeBag에 추가된 Disposable들이 자동으로 정리되어 메모리 누수를 막을 수 있습니다.

2. Retain Cycle

RxCocoa를 사용할 때 주의해야 할 리소스 관리 이슈 중 하나는 Retain Cycle입니다. Retain Cycle이란 서로 참조를 하고 있는 객체들이 메모리에서 해제되지 않고 남아있는 상황을 의미합니다. 이는 적절한 리소스 정리가 이루어지지 않았을 때 발생할 수 있습니다.

RxCocoa에서 Retain Cycle을 방지하기 위해 주로 [weak self] 또는 [unowned self]를 사용합니다. 이를 통해 클로저 내에서 self를 약한 참조로 캡처하거나, 강한 참조가 아닌 비소유 참조로 캡처할 수 있습니다.

import RxCocoa
import RxSwift

class MyViewController: UIViewController {
    
    let disposeBag = DisposeBag()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        fetchData()
            .subscribe(onNext: { [weak self] result in
                // 데이터 처리
                self?.updateUI(with: result)
            })
            .disposed(by: disposeBag)
    }
    
    func fetchData() -> Observable<String> {
        // 데이터 가져오기
        ...
    }
    
    func updateUI(with result: String) {
        // UI 업데이트
        ...
    }
}

위의 예제에서는 fetchData 함수를 호출하여 비동기로 데이터를 가져오고, 가져온 데이터를 updateUI 함수를 통해 UI를 업데이트합니다. 이때 [weak self]를 통해 self를 약한 참조로 캡처하므로, MyViewController가 해제될 때 메모리에서 정리됩니다. 이렇게 함으로써 Retain Cycle을 방지할 수 있습니다.

3. 메모리 누수 확인

RxCocoa를 사용할 때 메모리 누수가 발생할 수 있으므로 이를 확인하기 위해 앱 실행 중에 메모리 프로파일링을 진행하는 것이 좋습니다. Xcode의 Instruments를 사용하여 앱 실행 시 메모리 사용량을 분석하고, 메모리 누수가 있는지 여부를 확인할 수 있습니다.

또한 Observable 객체가 계속해서 이벤트를 발생시켜 리소스를 소모하는 경우, takeUntil 연산자를 사용하여 수명 주기를 제어할 수 있습니다. 이를 통해 필요한 시점에 Observable의 수명을 종료시키고 리소스 낭비를 방지할 수 있습니다.

import RxCocoa
import RxSwift

class MyViewController: UIViewController {
    
    let disposeBag = DisposeBag()
    let button = UIButton()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        button.rx.tap
            .takeUntil(self.rx.deallocating) // 뷰 컨트롤러가 해제될 때까지만 이벤트 발생
            .subscribe(onNext: {
                // 버튼이 클릭되었을 때의 동작
            })
            .disposed(by: disposeBag)
    }
}

위의 예제에서는 takeUntil 연산자를 통해 rx.deallocating을 전달하여 뷰 컨트롤러가 해제될 때까지만 버튼 탭 이벤트가 발생하도록 설정하였습니다. 이렇게 함으로써 더 이상 필요하지 않은 이벤트 발생을 막을 수 있습니다.

위에서 살펴본 DisposeBag, Retain Cycle, 메모리 누수 등의 이슈들을 주의하고 적절하게 관리한다면 RxCocoa를 사용하면서 발생할 수 있는 문제를 줄일 수 있습니다. 따라서 앱의 안정성과 성능 향상을 위해 이러한 이슈에 대해 항상 주의해야 합니다.

참고 자료

본 내용은 RxSwift 및 RxCocoa의 문서 및 예제 코드를 기반으로 작성되었습니다.

© 2018 Colin. All rights reserved.