update

Author: swiftuiux

README.md

@@ -21,9 +21,9 @@
 
 ### 3. Call ViewModel method start() within async environment
 If task will be canceled the streaming stops automatically
-
+I would recomend to use .task modifire it manages cancelation if you desided to use Task and keep it in @State don't forget to cancel() when the time has come
 ```
- Task{
+ .task{
        do{
              try await viewModel.start()
          }catch{

Sources/d3-async-location/LocationManagerAsync.swift

@@ -7,11 +7,14 @@
 
 import CoreLocation
 
+fileprivate typealias Termination = AsyncThrowingStream<CLLocation, Error>.Continuation.Termination
+
 ///Location manager streaming data asynchronously via instance of ``AsyncStream`` returning from ``start`` asking permission in advance if it's not determined.
 @available(iOS 15.0, watchOS 7.0, *)
 public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, ILocationManagerAsync{
+   
 
-    private typealias StreamType = AsyncStream<CLLocation>.Continuation
+    private typealias StreamType = AsyncThrowingStream<CLLocation, Error>.Continuation
 
     // MARK: - Private properties
 
@@ -21,7 +24,7 @@ public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, IL
     // Streaming locations
 
     /// Async stream of locations
-    private var locations : AsyncStream<CLLocation>{
+    private var locations : AsyncThrowingStream<CLLocation, Error>{
         .init(CLLocation.self) { continuation in
                 streaming(with: continuation)
             }
@@ -30,7 +33,10 @@ public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, IL
     /// Continuation asynchronously passing location data
     private var stream: StreamType?{
         didSet {
-            stream?.onTermination = { @Sendable _ in self.stop() }
+            stream?.onTermination = { @Sendable termination in
+                self.onTermination(termination)
+                
+            }
         }
     }
 
@@ -73,9 +79,12 @@ public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, IL
     // MARK: - API
 
     /// Check status and get stream of async data Throw an error ``LocationManagerErrors`` if permission is not granted
-    public var start : AsyncStream<CLLocation>{
+    public var start : AsyncThrowingStream<CLLocation, Error>{
         get async throws {
             if await getPermission{
+                #if DEBUG
+                print("start")
+                #endif
                 return locations
             }
             throw AsyncLocationErrors.accessIsNotAuthorized
@@ -84,8 +93,10 @@ public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, IL
 
     /// Stop streaming
     public func stop(){
+        
         stream = nil
         manager.stopUpdatingLocation()
+        
         #if DEBUG
         print("stop updating")
         #endif
@@ -149,6 +160,19 @@ public final class LocationManagerAsync: NSObject, CLLocationManagerDelegate, IL
         manager.allowsBackgroundLocationUpdates = backgroundUpdates
     }
 
+    /// Precess termination
+    /// - Parameter termination: A type that indicates how the stream terminated.
+    private func onTermination(_ termination: Termination){
+         switch termination {
+             case .finished:
+             stop()
+             case .cancelled:
+             stream?.finish(throwing: AsyncLocationErrors.streamCancelled)
+         @unknown default:
+             stream?.finish(throwing: AsyncLocationErrors.unknownTermination)
+         }
+     }
+    
     // MARK: - Delegate
 
     /// Pass locations into the async stream

Sources/d3-async-location/enum/AsyncLocationErrors.swift

@@ -17,4 +17,10 @@ public enum AsyncLocationErrors: Error{
     /// Attempt to launch streaming while it's been already started
     /// Subscribe different Views to LMViewModel.locations publisher to feed them
     case streamingProcessHasAlreadyStarted
+    
+    /// Stream was cancelled
+    case streamCancelled
+    
+    /// Unknown termination
+    case unknownTermination
 }

Sources/d3-async-location/protocol/ILocationManagerAsync.swift

@@ -11,7 +11,7 @@ import CoreLocation
 protocol ILocationManagerAsync{
 
     /// Check status and get stream of async data
-    var start : AsyncStream<CLLocation> { get async throws }
+    var start : AsyncThrowingStream<CLLocation, Error> { get async throws }
 
     /// Stop streaming
     func stop()

Sources/d3-async-location/viewmodel/LMViewModel.swift

@@ -13,7 +13,7 @@ import CoreLocation
 /// Add or inject LMViewModel into a View ```@EnvironmentObject var model: LMViewModel```
 /// Call method start() within async environment to start async stream of locations
 @available(iOS 15.0, watchOS 7.0, *)
-public actor LMViewModel: ILocationManagerViewModel{
+public final class LMViewModel: ILocationManagerViewModel{
 
     // MARK: - Public
 
@@ -60,23 +60,41 @@ public actor LMViewModel: ILocationManagerViewModel{
         state = .streaming
 
         do {
-            for await coordinate in try await manager.start{
+            for try await coordinate in try await manager.start{
                 await add(coordinate)
             }
         }catch{
+            
             state = .idle
+            
+            if isStreamCancelled(with: error){ stop() }
+            
             throw error
         }
     }
 
     /// Stop streaming locations
     public func stop(){
-        manager.stop()
-        state = .idle
+  
+            manager.stop()
+            state = .idle
+
+            #if DEBUG
+            print("stop viewmodel")
+            #endif
     }
 
     // MARK: - Private
 
+    func isStreamCancelled(with error : Error) -> Bool{
+        
+        if let e = error as? AsyncLocationErrors{
+            return [AsyncLocationErrors.streamCancelled, .unknownTermination].contains(e)
+        }
+        
+        return false
+    }
+    
     @MainActor
     private func add(_ coordinate : CLLocation) {
         locations.append(coordinate)