Flow and Cancellation/onCompletion Callback not working

[frankois944]

Hello, I'm using Flow on SwiftUI,

let flow: SkieSwiftStateFlow<F>
@Binding var binding: F
.task {
  for await flowValue in flow {
      binding = flowValue
   }
}

On the kotlin side, i'm listening the events .onStart {} and .onCompletion {} of the StateFlow bound to the view.

private val _mainScreenUIState = MutableSharedFlow<MainScreenUIState>()
public val mainScreenUIState: StateFlow<MainScreenUIState> = _mainScreenUIState
.onStart {
    d("STARTED MainScreenViewModel") // call with SwiftUI and Compose
}
.onCompletion {
    d("CLEARED MainScreenViewModel") // call with Compose but NOT on SwiftUI
}
.stateIn(viewModelScope, SharingStarted.Lazily, MainScreenUIState.Loading)

When the view is destroy, the flow must be cancelled, right?

onStartis correctly called but not onCompletion on the SwiftUI.

Do I miss someting?

[frankois944]

This is the following code I'm using on swiftUI

Even I'm forcing the task to cancel, on the kotlin side, onCompletion is not called!

class MyMainScreenViewModel: ObservableObject {
    let viewmodel: MainScreenViewModel
    
    var jobs = Set<Task<(), Never>>()

    @Published
    private(set) var mainScreenUIState: MainScreenUIState = .Loading()
    
    init(viewmodel: MainScreenViewModel = MainScreenViewModel(param1: nil)) {
        self.viewmodel = viewmodel
        jobs.insert(Task { @MainActor [weak self] in
            for await data in viewmodel.mainScreenUIState {
                self?.mainScreenUIState = data
            }
        })
    }
    
    deinit {
        print("DESTROYED \(self)")
        jobs.forEach {
            print("CANCEL \($0)")
            $0.cancel()
        }
        jobs.removeAll()
    }
}