[247] fix putObject remove ByteBuffer wrap, use ByteArray

zio · Aug 27, 2021 · ddbf359 · ddbf359
1 parent 4d7ce6e
commit ddbf359
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/main/scala/zio/s3/Live.scala b/src/main/scala/zio/s3/Live.scala
@@ -120,9 +120,11 @@ final class Live(unsafeClient: S3AsyncClient) extends S3.Service {
   ): ZIO[R, S3Exception, Unit] =
     content
       .mapChunks(Chunk.single)
-      .map(c => ByteBuffer.wrap(c.toArray))
-      .toPublisher
-      .flatMap(publisher =>
+      .mapError(e => S3Exception.builder().message(e.getMessage).cause(e).build())
+      .refineOrDie {
+        case e: S3Exception => e
+      }
+      .mapM { chunk =>
         execute(
           _.putObject(
             {
@@ -137,10 +139,11 @@ final class Live(unsafeClient: S3AsyncClient) extends S3.Service {
                 .fold(builder)(builder.contentType)
                 .build()
             },
-            AsyncRequestBody.fromPublisher(publisher)
+            AsyncRequestBody.fromBytes(chunk.toArray)
           )
         )
-      )
+      }
+      .runCollect
       .unit
 
   def multipartUpload[R](