Spring WebFlux：从控制器提供文件

Question

来自。NET和Node的我真的很难弄清楚如何将这个阻塞的MVC控制器转移到一个非阻塞的WebFlux注释控制器？我已经理解了这些概念，但是没有找到合适的异步Java IO方法（我希望返回一个Flux或Mono）。

@RestController
@RequestMapping("/files")
public class FileController {

    @GetMapping("/{fileName}")
    public void getFile(@PathVariable String fileName, HttpServletResponse response) {
        try {
            File file = new File(fileName);
            InputStream in = new java.io.FileInputStream(file);
            FileCopyUtils.copy(in, response.getOutputStream());
            response.flushBuffer();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

Answer 1

首先，使用Spring MVC实现这一目标的方法应该如下所示：

@RestController
@RequestMapping("/files")
public class FileController {

    @GetMapping("/{fileName}")
    public Resource getFile(@PathVariable String fileName) {
        Resource resource = new FileSystemResource(fileName);        
        return resource;
    }
}

而且，如果您只是在没有附加逻辑的情况下服务器这些资源，您就可以使用Spring MVC的静态资源支持。使用Spring Boot，Spring.resources.static-locations可以帮助您自定义位置。

现在，使用Spring WebFlux，您还可以配置相同的Spring.resources.static-locations配置属性来为静态资源服务。

它的WebFlux版本看起来完全相同。如果需要执行一些涉及I/O的逻辑，可以返回一个 而不是resource，如下所示：

@RestController
@RequestMapping("/files")
public class FileController {

    @GetMapping("/{fileName}")
    public Mono<Resource> getFile(@PathVariable String fileName) {
        return fileRepository.findByName(fileName)
                 .map(name -> new FileSystemResource(name));
    }
}

注意，对于WebFlux，如果返回的资源实际上是磁盘上的一个文件，我们将利用零拷贝机制来提高效率。