Merge pull request #51 from SW-Maestro-OSS/refactor/CoinTradeStream

[CVW-039] 코인 거래내역 정보 객체 책임 리팩토링

Author: J0onYEong

Projects/App/Sources/DI/Assembly/DataAssembly.swift

@@ -40,6 +40,10 @@ public class DataAssembly: Assembly {
             DefaultExchangeRateService()
         }
 
+        // MARK: DataSource
+        container.register(CoinTradeDataSource.self) { _ in
+            BinanceCoinTradeDataSource()
+        }
 
         // MARK: Repository
         container.register(UserConfigurationRepository.self) { _ in
@@ -63,8 +67,8 @@ public class DataAssembly: Assembly {
             BinanceSingleMarketTickerRepository()
         }
 
-        container.register(TradeRepository.self) { _ in
-            BinanceTradeRepository()
+        container.register(CoinTradeRepository.self) { _ in
+            BinanceCoinTradeRepository()
         }
     }
 }

Projects/Data/DataSource/Sources/DataSource/CoinTradeDataSource.swift

@@ -0,0 +1,39 @@
+//
+//  CoinTradeDataSource.swift
+//  Data
+//
+//  Created by choijunios on 4/21/25.
+//
+
+import Combine
+
+import Foundation
+import CoreUtil
+
+public protocol CoinTradeDataSource {
+    func getTradeList(symbolPair: String, tableUpdateInterval: Double?) -> AnyPublisher<HashMap<Int64, BinanceCoinTradeDTO>, Never>
+}
+
+public final class BinanceCoinTradeDataSource: CoinTradeDataSource {
+    // Dependency
+    @Injected private var webSocketService: WebSocketService
+    
+    // State
+    private let tradeList: ThreadSafeHashMap<Int64, BinanceCoinTradeDTO> = .init()
+    
+    public init() { }
+    
+    public func getTradeList(symbolPair: String, tableUpdateInterval: Double?) -> AnyPublisher<HashMap<Int64, BinanceCoinTradeDTO>, Never>  {
+        webSocketService
+            .getMessageStream()
+            .filter { (dto: BinanceCoinTradeDTO) in
+                dto.symbol.lowercased() == symbolPair.lowercased()
+            }
+            .throttle(for: .init(floatLiteral: tableUpdateInterval ?? 0.0), scheduler: DispatchQueue.global(qos: .default), latest: true)
+            .unretained(self)
+            .asyncTransform { source, dto in
+                await source.tradeList.insert(key: dto.tradeTime, value: dto)
+                return await source.tradeList.hashMap.copy()
+            }
+    }
+}

Projects/Data/Repository/Sources/BinanceCoinTradeRepository.swift

@@ -0,0 +1,35 @@
+//
+//  BinanceTradeRepository.swift
+//  Data
+//
+//  Created by choijunios on 4/15/25.
+//
+
+import Foundation
+import Combine
+
+import DomainInterface
+import DataSource
+import CoreUtil
+
+public final class BinanceCoinTradeRepository: CoinTradeRepository {
+    // Dependency
+    @Injected private var webSocketService: WebSocketService
+    @Injected private var coinTradeDataSource: CoinTradeDataSource
+    
+    public init() { }
+    
+    public func getCoinTradeList(symbolPair: String, tableUpdateInterval: Double?) -> AnyPublisher<HashMap<Date, CoinTradeVO>, Never> {
+        coinTradeDataSource
+            .getTradeList(symbolPair: symbolPair, tableUpdateInterval: tableUpdateInterval)
+            .map { dtoList in
+                var entityList = HashMap<Date, CoinTradeVO>()
+                dtoList.values.forEach { dto in
+                    let entity = dto.toEntity()
+                    entityList[entity.tradeTime] = entity
+                }
+                return entityList
+            }
+            .eraseToAnyPublisher()
+    }
+}

Projects/Data/Repository/Sources/BinanceOrderbookRepository.swift

@@ -18,8 +18,8 @@ public final class BinanceOrderbookRepository: OrderbookRepository {
     @Injected private var httpService: HTTPService
 
     // Store
-    private let bidOrderbookStore = ThreadSafeOrderbookHashMap()
-    private let askOrderbookStore = ThreadSafeOrderbookHashMap()
+    private let bidOrderbookStore = ThreadSafeHashMap<CVNumber, CVNumber>()
+    private let askOrderbookStore = ThreadSafeHashMap<CVNumber, CVNumber>()
 
     public init() { }
 }
@@ -79,7 +79,7 @@ private extension BinanceOrderbookRepository {
 
     enum StoreType { case bid, ask }
     func adaptToStore(orderbooks: [[String]], storeType: StoreType) async {
-        var store: ThreadSafeOrderbookHashMap
+        var store: ThreadSafeHashMap<CVNumber, CVNumber>
         switch storeType {
         case .bid:
             store = bidOrderbookStore

Projects/Data/Repository/Sources/BinanceTradeRepository.swift

@@ -15,7 +15,7 @@ final public class DefaultCoinDetailPageUseCase: CoinDetailPageUseCase {
     // Dependency
     @Injected private var orderbookRepository: OrderbookRepository
     @Injected private var singleTickerRepository: SingleMarketTickerRepository
-    @Injected private var coinTradeRepository: TradeRepository
+    @Injected private var coinTradeRepository: CoinTradeRepository
     @Injected private var webSocketHelper: WebSocketManagementHelper
 
     public init() { }
@@ -49,8 +49,18 @@ public extension DefaultCoinDetailPageUseCase {
         singleTickerRepository.request24hTickerChange(pairSymbol: symbolPair)
     }
 
-    func getRecentTrade(symbolPair: String) -> AsyncStream<CoinTradeVO> {
-        coinTradeRepository.getSingleTrade(symbolPair: symbolPair)
+    func getRecentTrade(symbolPair: String, maxRowCount: UInt) -> AnyPublisher<[CoinTradeVO], Never> {
+        coinTradeRepository
+            .getCoinTradeList(symbolPair: symbolPair, tableUpdateInterval: 0.5)
+            .map { entity in
+                // 최신으로 maxRowCount개수 만큼
+                let slicedList = entity
+                    .keys(order: .DESC, maxCount: maxRowCount)
+                    .compactMap { key in entity[key] }
+                    .prefix(Int(maxRowCount))
+                return Array(slicedList)
+            }
+            .eraseToAnyPublisher()
     }
 
     func getOrderbookTable(symbolPair: String, rowCount: UInt) -> AnyPublisher<OrderbookTableVO, Error> {

Projects/Data/Repository/Sources/Util/ThreadSafeOrderbookHashMap.swift

@@ -0,0 +1,15 @@
+//
+//  CoinTradeRepository.swift
+//  Domain
+//
+//  Created by choijunios on 4/15/25.
+//
+
+import Combine
+import Foundation
+
+import CoreUtil
+
+public protocol CoinTradeRepository {
+    func getCoinTradeList(symbolPair: String, tableUpdateInterval: Double?) -> AnyPublisher<HashMap<Date, CoinTradeVO>, Never>
+}

Projects/Domain/Concrete/UseCase/DefaultCoinDetailPageUseCase.swift

@@ -14,6 +14,6 @@ public protocol CoinDetailPageUseCase {
     func disconnectToStreams(symbolPair: String)
 
     func getOrderbookTable(symbolPair: String, rowCount: UInt) -> AnyPublisher<OrderbookTableVO, Error>
+    func getRecentTrade(symbolPair: String, maxRowCount: UInt) -> AnyPublisher<[CoinTradeVO], Never>
     func get24hTickerChange(symbolPair: String) -> AsyncStream<Twenty4HourTickerForSymbolVO>
-    func getRecentTrade(symbolPair: String) -> AsyncStream<CoinTradeVO>
 }