如何在Spring5MVC中将FilePart转换为byte[]

Question

我有一个从web表单接收和上传文件的控制器方法。如何从FilePart中提取字节数组并将其保存到DB？

我可以通过使用FilePart将FilePart保存到一个文件来实现。但这似乎又慢又难看。有更好的方法吗？

import org.springframework.http.codec.multipart.FilePart;
import org.springframework.web.bind.annotation.*;


 Mono<UploadResult> uploadFile(@RequestParam("files") FilePart file){

    byte[] fileAsByteArray = convertFilePartToByteArray(file);

    fileService.saveByteArrayToDB(fileAsByteArray);

    /* Rest of the method */
 }

Answer 1

你可以这样做：

file.content()
.map { it -> it.asInputStream().readAllBytes() }
.map { it -> fileService.saveByteArrayToDB(it) } // it is Byte array

Answer 2

你是说界面组织吗。springframework。http。编解码器。多部分。文件部分？

请参阅如何正确读取Flux

Answer 3

您可以利用内部dataBuffer并将其转换为字节[]。

辅助函数：

suspend fun FilePart.toBytes(): ByteArray {
    val bytesList: List<ByteArray> = this.content()
            .flatMap { dataBuffer -> Flux.just(dataBuffer.asByteBuffer().array()) }
            .collectList()
            .awaitFirst()

    // concat ByteArrays
    val byteStream = ByteArrayOutputStream()
    bytesList.forEach { bytes -> byteStream.write(bytes) }
    return byteStream.toByteArray()
}

控制器：

@PostMapping("/upload", consumes = [MediaType.MULTIPART_FORM_DATA_VALUE])
suspend fun upload(@RequestPart("file") file: Mono<FilePart>) {
    val bytes = file.awaitFirst().toBytes()
    myService.handle(bytes) // do your business stuff
}