feat: add mapToContent function to Internal Response (#45)

ydataai · Jul 10, 2023 · 542cea8 · 542cea8
1 parent 89d2105
commit 542cea8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Sources/YData/Client/InternalResponse+Vapor.swift b/Sources/YData/Client/InternalResponse+Vapor.swift
@@ -18,8 +18,8 @@ public extension EventLoopFuture where Value: InternalResponse {
   func mapContent<NewValue>(_ callback: @escaping (ContentContainer) throws -> (NewValue))
   -> EventLoopFuture<Value> where NewValue: Content { flatMapThrowing { try $0.map(callback) } }
 
-  func mapToContent<R>() -> EventLoopFuture<R> where R: Decodable {
-    flatMapThrowing { response -> R in try response.content.decode(R.self) }
+  func mapToContent<D: Decodable>(_ type: D.Type = D.self) -> EventLoopFuture<D> {
+    flatMapThrowing { response -> D in try response.content.decode(type) }
   }
 
   func flatMapContentThrowing<NewValue>(_ callback: @escaping (ContentContainer) throws -> (NewValue))

diff --git a/Sources/YData/Client/InternalResponse.swift b/Sources/YData/Client/InternalResponse.swift
@@ -28,6 +28,11 @@ public extension InternalResponse {
 
     return Self.init(headers: headers, status: status, body: nil)
   }
+
+  @inlinable
+  func mapToContent<D: Decodable>(_ type: D.Type = D.self) throws -> D {
+    try content.decode(type)
+  }
 }
 
 public extension Internal {