[RxSwift] 메모리 누수 일어나기 딱 좋은(?) .bind(with:onNext:) & 중첩 클로저의 객체 참조

 

 

 

📍.bind(with:onNext:)의 메모리 누수?

 

메모리 누수 방지하려고 쓰는 RxSwift의 .bind(with:onNext:) 모두 알고 계신가요?

객체에 대한 retain 없이 참조하기 위해 .bind(with:onNext:) 쓰는데 제목에 뭔 메모리 누수 일어나기 딱 좋다고 하는거냐!! 라고 할 수 있는데요. 

 

 

 

 

프로젝트를 하다가 어느 날보니 뷰컨트롤러에 deinit이 찍히지 않는 이슈가 있었습니다.

혹시 이 코드에서 이상한점이 보이시나요???

 

 

 

onNext 클로저의 파라미터 이름으로 owner 가 있는데 클로저 내부에 self…도 사용되고 있네요?

뭔가 어색하지 않나요??

 

 

 

 

 

 

 

 

우선 .bind(with:onNext:)가 뭔지 정확히 알아보기 위해 RxSwift의 .bind(with:onNext:) 구현부로 가보면 with 의 인자로는 무엇을 넣어야하는지 설명하고 있는데, 

"The object to provide an unretained reference on."

“참조가 retain되지 않도록 하고 싶은 객체를 넣어준다” 

 

 

그렇습니다. 우리가 아주 자연스럽게 with 파라미터의 인자로 self를 넣어주고 있었는데 그게 객체의 rc를 올리지 않게 해서 메모리 릭을 방지하고자 했던 것입니다.

 

 

 

 

 

 

우리가 with에 인자로 넣은 self, 즉,  retain하지 않을 객체는!

onNext 클로저에서 우리가 지정해 준 owner 같은 파라미터 이름으로 사용해 주어야 비로소 retain되지 않으며 사용할 수 있는 것입니다.

 

with 인자로 self를 전달해 주었다고 맘대로 우리가 클로저 self를 마음대로 써주면 안 되는 것이죠 ㅎㅎ

 

 

 

 

 

굉장히 사소하지만, 사소한만큼 놓치고 넘어가기 쉬운것 같고 클로저 내부에서 owner로 사용되도록 강제된 것도, self를 썼다고 컴파일 에러를 띄워주는 것도 아니기 때문에 충분히 많은 사람들이 실수할 수 있을것 같아 공유하고 싶었습니다.

 

 

 

 

 

 

---

 

 

 

📍 중첩 클로저의 메모리 누수

 

그리고 또 한 가지!

.bind(with:onNext:)클로저 내부에  중첩되는 클로저를 사용할 경우에 유의해야 할 점에 대해 말해보려 합니다.

 

 

 

 

 

만약 onNext 로 전달되는 클로저가 이렇게 중첩 클로저일 경우,

내부클로저에 [weak self]를 해줌으로써 self에 대한 참조 카운트를 올리지 않도록 시도했다고 가정하면 어떨까??

 

 

 

 

 

 

외부 클로저에서도 .bind(with:onNext:)로 객체의 강한 참조를 “방지”해주었고, 내부클로저에서도  [weak self]를 써주었기 때문에 메모리 누수가 일어나지 않을 것이라고 생각할 수 있지만, 안타깝게도(?) 메모리릭이 발생하는 코드입니다.

 

⭐️ 중첩된 클로저는 매우 중요하게 고려해주어야 할 게 있는데요,

바로 외부 클로저에서 객체를 어떻게 잡아주고 있는지가 내부클로저에도 영향을 끼친다는 것입니다.

 

 

 

 

 

 

아까 위에서 봤던 그림을 설명해 보자면,

내부클로저에서 [weak self]를 해주는 순간, 외부 클로저에서 self를 어떻게 참조하고 있는지 살피게 되는데  이렇게 되면 아래 그림과 동일하다고 볼 수 있습니다.

 

즉, 외부 클로저에서 [weak self]를 쓰지 않고 내부 클로저에서 [weak self]를 쓰는 순간

외부 클로저에서 강하게 self를 캡처하고 있는 것처럼 되니까 메모리 릭이 생기겠죠?

 

* 우리는 onNext 클로저에서 owner라는 이름으로 self에 대한 참조 계수를 올리지 않고 사용하겠다고 작성해 준 것이지, self자체는 강하게 참조하고 있는 것이니까요! (그래서 위의 첫 번째 주제에서 말했던 것처럼 클로저 내에서 self를 써주면 메모리릭이 일어나는 것이고요.)

 

 

 

 

 

 

그럼 아래 코드는 어떨까요?

외부클로저에서도 클로저를 약하게 참조하고 있기 때문에 이 클로저의 중첩으로 쓰이는 self는 모두 rc를 올리지 않고 참조하게됩니다.

메모리 릭이 생기지 않는 것이죠!

 

 

 

 

 

그렇다면 이 코드도 이해가 되시겠죠?

내부클로저에서 강하게 캡처하는 것처럼 보이더라도 외부 클로저에서 이미 약하게 참조하고 있기 때문에 self를 캡처하는 두 곳 모두 rc는 올리지는 않습니다.

 

 

 

 

 

 

 

 

 

---

 

 

이번 주제에서는 RxSwift에서 많이 쓰고 있는 .bind(with:onNext:)에 대해서 알아봤습니다. 

이번 글의 두 번째 주제로 설명했던 중첩 클로저 예시는 RX를 적용하지 않은 보통의 중첩 클로저에서도 동일하게 문제가 될 수 있기 때문에 유의해줍시다.

 

객체를 강하게 캡처함으로써 발생하는 메모리 누수를 방지하기 위해 사용하지만, 잘 사용하지 않으면 누수가 충분히 일어날 수 있는 것이 .bind(with:onNext:)이기 때문에 유의해서 사용해주어야할 것 같습니다:)