如何使用Undertow返回文件以在Java中下载

Question

我试图允许我的游戏客户端下载客户端运行所需的缓存。在我的游戏网络服务器中，我正在这样做：

@RouteManifest(template="/cache", method="GET")
public class APICacheRoute extends RouteHttpHandler<JadePreprocessor> {
    @Override
    public void handleRequest(HttpServerExchange exchange) throws Exception {
        Path path = Paths.get("./cache/Alterscape.zip");
        File file = path.toFile();
        exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "application/octet-stream");
        exchange.getResponseSender().send(file);
    }
}

但是，我收到的错误是该文件必须是字节扣。我如何返回文件下载？

我的Web服务器看起来像这样：

public static void initialize() {
    ServerController server = new ServerController("localhost", 8080, 8443);
    server.register(new APIVirtualHost());
    server.inititialize();
}

我的apivirtualhost看起来像这样：

public APIVirtualHost() {
    super(127.0.0.1);
    setDirectoryListingEnabled(false);
}

Answer 1

使用Undertow，您需要将处理程序从I/O线程中分配以使用阻止操作。您需要在需要执行阻止操作的任何时候都需要执行此操作，例如阅读A请求主体，接受文件上传流，当然也要发送文件。

在您的句柄（）方法的顶部，使用它将请求分配给阻止线程，然后将请求分配给非阻止I/O线程：

：

if (serverExchange.isInIoThread()) {
    serverExchange.dispatch(this);
    return;
}

准备好发送缓冲区时开始阻止：

serverExchange.startBlocking();

然后发送文件，您可以使用缓冲流：

serverExchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "application/octet-stream");
final File file = new File("/location/to/your/file");
final OutputStream outputStream = serverExchange.getOutputStream();
final InputStream inputStream = new FileInputStream(file);
byte[] buf = new byte[8192];
int c;
while ((c = inputStream.read(buf, 0, buf.length)) > 0) {
    outputStream.write(buf, 0, c);
    outputStream.flush();
}
outputStream.close();
inputStream.close();

ServerExchange对象上的send方法用于使用非块方法发送响应数据 - 适用于文本或JSON。ByteBuffer参数过载用于发送原始字节数据，这是undertow在发送时将字符串转换为的。这可能有些误导。

快乐的编码。

Answer 2

尝试用此io.undertow.server.handlers.BlockingHandler包装自己的处理程序。在执行阻止操作之前，它已经包含必要的检查。

public final class BlockingHandler implements HttpHandler {
    ...
    @Override
    public void handleRequest(final HttpServerExchange exchange) throws Exception {
        exchange.startBlocking();
        if (exchange.isInIoThread()) {
            exchange.dispatch(handler);
        } else {
            handler.handleRequest(exchange);
        }
    }
    ...
}