Tải lên tệp bằng cách sử dụng Akka HTTP

Question

Tôi đang cố triển khai chức năng tải lên tệp trong ứng dụng của mình bằng cách sử dụng Akka HTTP. Tôi đang sử dụng akka-stream phiên bản 2.4.4.

Đây là mã (sửa đổi từ akka-doc)

path("fileupload") { 
    post { 
     extractRequestContext { 
     ctx => { 
      implicit val materializer = ctx.materializer 
      implicit val ec = ctx.executionContext 
      fileUpload("fileUpload") { 
      case (metadata, byteSource) => 
       val location = FileUtil.getUploadPath(metadata) 
       val updatedFileName = metadata.fileName.replaceAll(" ", "").replaceAll("\"", "") 
       val uniqFileName = uniqueFileId.concat(updatedFileName) 
       val fullPath = location + File.separator + uniqFileName 
       val writer = new FileOutputStream(fullPath) 
       val bufferedWriter = new BufferedOutputStream(writer) 

       val result = byteSource.map(s => { 
       bufferedWriter.write(s.toArray) 
       }).runWith(Sink.ignore) 

       val result1 = byteSource.runWith(Sink.foreach(s=>bufferedWriter.write(s.toArray))) 
       Await.result(result1, 5.seconds) 
       bufferedWriter.flush() 
       bufferedWriter.close() 
       complete(uniqFileName) 
      /*onSuccess(result) { x => 
       bufferedWriter.flush() 
       bufferedWriter.close() 
       complete("hello world") 
      }*/ 
      } 
     } 
     } 
    } 
    } 

Mã này đang làm việc tốt và được tải lên các tập tin vào con đường đã chọn. Tôi đang tạo tên tệp mới bằng cách thêm UUID để đảm bảo rằng tên tệp là duy nhất. Vì vậy, tôi cần phải trả lại tên tập tin mới cho người gọi. Tuy nhiên, phương pháp này không trả lại tên tệp luôn. Đôi khi, nó kết thúc với Response has no content.

Có ai cho tôi biết tôi đang làm gì sai ở đây không?

Answer 1

Không có nhu cầu sử dụng các tiêu chuẩn chặn suối khi bạn có con suối phản ứng cho mục đích đó:

path("fileUpload") { 
    post { 
     fileUpload("fileUpload") { 
     case (fileInfo, fileStream) => 
      val sink = FileIO.toPath(Paths.get("/tmp") resolve fileInfo.fileName) 
      val writeResult = fileStream.runWith(sink) 
      onSuccess(writeResult) { result => 
      result.status match { 
       case Success(_) => complete(s"Successfully written ${result.count} bytes") 
       case Failure(e) => throw e 
      } 
      } 
     } 
    } 
    } 

Mã này sẽ tải lên fileUpload lĩnh vực nhiều phần dữ liệu vào một tập tin bên trong thư mục /tmp. Nó chỉ đổ nội dung của nguồn đầu vào vào tệp chìm tương ứng, trả về một thông báo sau khi hoàn thành thao tác ghi.

Bạn cũng có thể muốn điều chỉnh bộ điều phối được sử dụng cho FileIO nguồn và bồn rửa, như được mô tả trong their scaladocs.