[Swift] MEMORY LEAK 잡기

메모리 누수, 메모리 릭.. 조심해야된다…. 약한 참조 해줘야된다 … 잘 안해주면 앱 터질 수도 있다…

라는 경고를 듣기만 했지, 직접 겪어보긴 처음이었습니다 ..

 

최근 영상을 다루는 서비스를 개발하고 있었는데여

서비스 내에서 라이브 영상도 다루고, 다시보기 같은 다양한 영상을 처리하다보니

영상을 돌리다보면 메모리 문제로 인해서 앱이 터지는 문제가 발생했습니다.

😡

특히 라이브 다시보기 영상같은 경우, 3시간짜리 영상을 전달받다보니

영상을 플레이할 때마다 메모리가 팍팍 치솟고,, 최대 3번까지 플레이하면 앱이 터지더라구요..

영상을 한번.. 두번.. 세번.. 플레이했을때..

저희가 전달받을 수 있는 영상의 종류나 크기는 일단 당장 처리할 수 있는 영역이 아니라 생각되어

최대한 메모리 누수를 잡아보자 하는 생각으로 메모리를 해제해보기로 했습니다.

 

---

🚒 메모리 누수가 발생할 수 있는 경우

* RxSwift + MVVM + Clean Architecture를 채택한 프로젝트입니다.

 

1. Closures

1) Subscribe

rxswift를 사용해서 구독하는 모든 부분에 약한 참조를 통해서 메모리 해제해주었습니다.

onNext만 사용하는 경우에는 withUnretained를 사용했고,

에러 처리를 위해 onError까지 사용하는 경우에는 .subscribe(with:)을 사용해서 한번에 약한 참조를 할 수 있도록 했습니다.

// ex1
enterButton.rx.tap
            .withUnretained(self)
            .subscribe { owner, _ in
                owner.dismiss(animated: true) {
                    let userInfo = ["matchId": owner.matchId,
                                    "matchInfo": owner.viewModel.matchInfoModel as Any,
                                    "matchScore": owner.viewModel.matchScoreModel as Any]
                    owner.postObserverAction(.presentPlayVC,
                                             userInfo: userInfo)
                }
            }.disposed(by: self.disposeBag)

//ex2
output.hasDuplicatedUser
            .withUnretained(self)
            .subscribe { owner, hasUser in
                if hasUser {
                    owner.showToast(message: I18N.Play.alreadyPlaying, bottom: 100)
                    owner.renewPlayerList(reload: true)
                }
            }.disposed(by: self.disposeBag)

// ex3
matchRepository.getMatchInfo(matchId: matchId)
            .subscribe(with: self) { (owner, entity) in
                guard let entity = entity else { return }
                let model = entity.toDomain()
                owner.matchInfo.accept(model)
            } onError: { owner, error in
                owner.error.accept(error)
            }
            .disposed(by: disposeBag)

 

2) UIView.animate

animations도 @escaping closure이기 때문에 전부 약한 참조 해주었습니다.

UIView.animate는 순환참조가 일어나지 않는다고 합니다.. ! 알려주셔서 감사합니다.. 

참고 https://daeun28.github.io/%EC%9D%B4%EB%A1%A0/post29/

 

3) DiffableDataSource (cellProvider)

UITableViewDiffableDataSource를 사용했는데,

이때 cellProvider도 마찬가지로 @escaping closure이기 때문에 약한 참조를 해줬습니다.

UITableViewDiffableDataSource<TeamType, PlayerModel>(tableView: self.tableView, cellProvider: { [weak self] (tableView, indexPath, _) in
            guard let self = self else { return UITableViewCell() }
            guard let cell = tableView.dequeueReusableCell(withIdentifier: AddPlayerTVC.className, for: indexPath) as? AddPlayerTVC else { return UITableViewCell() }
            switch indexPath.section {
            case 0:
                let data = self.homePlayerList[indexPath.row]
                let isMe: Bool = data.userId == UserDefaults.standard.integer(forKey: Const.UserDefaultsKey.userId)
                cell.setData(data: data, isMe: isMe)
                if self.duplicatedUserChecked && data.isPlaying {
                    cell.setDuplicatedUI()
                }
						...
            return cell
        })

 

2. Custom UIView

플젝 내에서 네비바를 CustomNavigationBar이라는 UIView로 커스텀해서 사용했습니다.

왼쪽의 backButton을 눌렀을때 각 뷰컨에서 별다른 지정을 해주지 않아도 항상 pop되도록 하는 액션을 넣어주기 위해

CustomNavigationBar 안에 pvc라는 UIViewController을 갖고 사용하도록 했습니다.

class CustomNavigationBar: UIView {
    
    private var pvc: UIViewController?

		...
}

// 뷰컨 내에서 아래처럼 self(UIViewController)을 넣어줘서 사용
naviBar.setNavibar(self, title: I18N.Play.startPlay, type: .withCloseButton)

뷰 내의 UIView가 UIViewController를 참조하고 있기 때문에 순환 참조가 발생하면서 참조 해제가 안됩니다..

이를 방지하기 위해서 UIView 내에서 UIViewController를 약한 참조로 선언해주면

class CustomNavigationBar: UIView {
    
    private weak var pvc: UIViewController?

		...
}

customNavigationBar이라는 UIView를 갖는 상위 UIViewController가 deinit되면

하위의 UIView도 deinit되게 됩니다..

 

 

3. Protocol

delegate pattern을 사용하는 경우에도 프로토콜을 선언할때 weak로 선언해줘야 메모리에서 해제됩니다..!

class AddPlayerFooterView: UITableViewHeaderFooterView {
    
    weak var buttonDelegate: AddPlayerDelegate?
		...
}

 

 

틀린 부분이나 보충하면 좋을 부분이 있다면 언제든 .. 말씀해주십셔 .. 🙇‍♀️