[SCRUM-51] StreamListener 개선 (#66)

Projects/Core/StreamListener/Interface/StreamContinuation.swift

@@ -0,0 +1,25 @@
+import Foundation
+
+public class StreamContinuation {
+  private let _yield: (any StreamType) -> Void
+  private let _finish: () -> Void
+
+  public init<T: StreamType>(_ continuation: AsyncStream<T>.Continuation) {
+    self._yield = { value in
+      guard let value = value as? T else { return }
+      continuation.yield(value)
+    }
+
+    self._finish = {
+      continuation.finish()
+    }
+  }
+
+  public func yield(_ value: any StreamType) {
+    _yield(value)
+  }
+
+  public func finish() {
+    _finish()
+  }
+}

Projects/Core/StreamListener/Interface/StreamListenerInterface.swift

@@ -10,9 +10,19 @@ import Foundation
 import Dependencies
 import DependenciesMacros
 
+// MARK: - Stream releated Protocol
+
+public typealias StreamKey = String
+
+public protocol StreamType: Hashable {
+  static var key: StreamKey { get }
+}
+
+// MARK: - StreamListener
+
 public protocol StreamListenerProtocol {
-  func send<T: StreamTypeProtocol>(_ state: T) async
-  func receive<T: StreamTypeProtocol>(_ type: T.Type) -> AsyncStream<T>
+  func send<T: StreamType>(_ state: T) async
+  func receive<T: StreamType>(_ type: T.Type) -> AsyncStream<T>
 }
 
 @DependencyClient
@@ -30,8 +40,8 @@ extension StreamListener: TestDependencyKey {
 }
 
 private struct StreamListenerTestImpl: StreamListenerProtocol {
-  func send<T: StreamTypeProtocol>(_ state: T) async {}
-  func receive<T: StreamTypeProtocol>(_ type: T.Type) -> AsyncStream<T> { .never }
+  func send<T: StreamType>(_ state: T) async {}
+  func receive<T: StreamType>(_ type: T.Type) -> AsyncStream<T> { .never }
 }
 
 /*

Projects/Core/StreamListener/Sources/StreamListener.swift

@@ -22,11 +22,11 @@ extension StreamListener: DependencyKey {
 final class StreamListenerImpl: StreamListenerProtocol {
   private let actor = StreamActor()
 
-  func send<T: StreamTypeProtocol>(_ state: T) async {
-    await actor.yield(type: T.key, value: T.self)
+  func send<T: StreamType>(_ state: T) async {
+    await actor.yield(key: T.key, value: state)
   }
 
-  func receive<T: StreamTypeProtocol>(_ type: T.Type) -> AsyncStream<T> {
+  func receive<T: StreamType>(_ type: T.Type) -> AsyncStream<T> {
     let (stream, continuation) = AsyncStream<T>.makeStream()
     Task {
       await actor.register(key: T.key, continuation: continuation)
@@ -36,18 +36,24 @@ final class StreamListenerImpl: StreamListenerProtocol {
 }
 
 private actor StreamActor {
-  private var streams: [StreamType: AnyStreamContinuation] = [:]
-
-  func register<T: StreamTypeProtocol>(key: StreamType, continuation: AsyncStream<T>.Continuation) {
-    streams[key] = AnyStreamContinuation(continuation)
+  private var streams: [StreamKey: [StreamContinuation]] = [:]
+
+  func register<T: StreamType>(key: StreamKey, continuation: AsyncStream<T>.Continuation) {
+    let newContinuation = StreamContinuation(continuation)
+    if streams[key] == nil {
+      streams[key] = [newContinuation]
+    } else {
+      streams[key]?.append(newContinuation)
+    }
   }
 
-  func yield<T: StreamTypeProtocol>(type: StreamType, value: T.Type) {
-    guard let continuation = streams[type] else { return }
-    continuation.yield(value)
+  func yield<T: StreamType>(key: StreamKey, value: T) {
+    guard let continuations = streams[key] else { return }
+    continuations.forEach { $0.yield(value) }
   }
 
-  func remove(type: StreamType) {
-    streams[type] = nil
+  func remove(key: StreamKey) {
+    streams[key]?.forEach { $0.finish() }
+    streams[key] = nil
   }
 }

Projects/Domain/AppService/Sources/ServerStateStream.swift

@@ -0,0 +1,22 @@
+//
+//  ServerStateStream.swift
+//  AppService
+//
+//  Created by 김지현 on 12/25/24.
+//  Copyright © 2024 PomoNyang. All rights reserved.
+//
+
+import StreamListenerInterface
+
+public enum ServerState: StreamType {
+  public static var key: StreamKey { "serverState" }
+  case requestStarted
+  case requestCompleted
+  case errorOccured
+  case networkDisabled
+}
+
+public struct RetryState: StreamType {
+  public static var key: StreamKey { "retryState" }
+  public var retry: Int?
+}

Projects/Feature/CatFeature/Sources/NamingCat/NamingCatCore.swift

@@ -13,6 +13,7 @@ import DesignSystem
 import UserServiceInterface
 import DatabaseClientInterface
 import StreamListenerInterface
+import AppService
 
 import ComposableArchitecture
 import RiveRuntime