本文介绍如何通过自定义 LimitedInputStream 包装原始输入流，使其在读取指定字节数后自动返回 EOF，从而安全、高效地将大流分块用于 HTTP 请求（如 BodyPublisher）等场景。
本文介绍如何通过自定义 `limitedinputstream` 包装原始输入流，使其在读取指定字节数后自动返回 eof，从而安全、高效地将大流分块用于 http 请求（如 `bodypublisher`）等场景。
在 Java 网络编程或 HTTP 客户端开发中，常需将大型数据流（如文件、网络响应体）按固定大小分块发送，例如配合 HttpClient 的 BodyPublisher.ofInputStream() 实现流式分片上传。但标准 InputStream 不支持“仅读取 N 字节后停止”的语义——一旦关闭或耗尽，无法复用；而反复创建新流又可能破坏底层连接或状态。
此时，一个轻量、可组合的装饰器流是理想解法：LimitedInputStream 继承 FilterInputStream，对底层流进行字节级读取限制。它不缓冲数据，不改变原始流生命周期，仅在逻辑层面拦截并截断读取行为。
以下是完整实现（已适配 Java 11+，含关键方法重写与边界处理）：
import java.io.*;
public class LimitedInputStream extends FilterInputStream {
private int remaining;
public LimitedInputStream(InputStream in, int limit) {
super(in);
if (in == null) throw new NullPointerException("in");
if (limit < 0) throw new IllegalArgumentException("limit must be non-negative");
this.remaining = limit;
}
@Override
public int read() throws IOException {
if (remaining <= 0) return -1;
int b = in.read();
if (b != -1) remaining--;
return b;
}
@Override
public int read(byte[] b, int off, int len) throws IOException {
if (remaining <= 0) return -1;
int lim = Math.min(len, remaining);
int r = in.read(b, off, lim);
if (r > 0) remaining -= r;
return r;
}
@Override
public long skip(long n) throws IOException {
long toSkip = Math.min(n, remaining);
long skipped = in.skip(toSkip);
if (skipped > 0) remaining -= (int) skipped;
return skipped;
}
@Override
public int available() throws IOException {
return Math.min(in.available(), remaining);
}
@Override
public void close() throws IOException {
// 不关闭底层流：由调用方统一管理资源
// 若需自动关闭，可取消注释下一行（谨慎使用）
// in.close();
}
public int getRemaining() { return remaining; }
}关键设计说明：
- ✅ 非负限制支持：允许
limit = 0（立即 EOF），便于空块处理； - ✅ 线程安全无关：该类不保证并发安全，应确保单线程使用或外部同步；
- ✅ 资源责任清晰：
close()默认不关闭底层流，避免意外中断上游资源（如FileInputStream或网络连接），调用方需自行管理原始流生命周期； - ✅ 兼容所有
read()变体：包括单字节、字节数组、带偏移/长度的重载，以及skip()和available()，行为一致可靠。
典型使用示例（分块 POST）：
InputStream original = Files.newInputStream(Paths.get("large-file.bin"));
long totalSize = Files.size(Paths.get("large-file.bin"));
int chunkSize = 8192;
long offset = 0;
while (offset < totalSize) {
int currentChunk = (int) Math.min(chunkSize, totalSize - offset);
InputStream chunkStream = new LimitedInputStream(original, currentChunk);
HttpRequest request = HttpRequest.newBuilder()
.uri(URI.create("https://api.example.com/upload"))
.header("Content-Type", "application/octet-stream")
.POST(HttpRequest.BodyPublishers.ofInputStream(() -> chunkStream))
.build();
HttpResponse<Void> response = HttpClient.newHttpClient()
.send(request, HttpResponse.BodyHandlers.discarding());
offset += currentChunk;
System.out.printf("Uploaded %d bytes (%.1f%%)%n", offset, 100.0 * offset / totalSize);
}
// 最后显式关闭 original
original.close();⚠️ 注意事项：
-
LimitedInputStream是一次性流：一旦remaining归零，后续所有读取均返回-1，不可重置或重用； - 若底层流本身不支持
skip()或available()，其行为仍以底层实现为准，LimitedInputStream仅做上层限制； - 在
BodyPublisher中使用时，务必确保InputStream的 lambda 表达式每次返回新实例（如示例中在循环内新建），否则多个请求会共享同一受限流，导致后续请求读取为空。
通过该方案，你可在不加载全量数据到内存的前提下，精准控制每批次传输字节数，兼顾性能、可控性与资源安全性。
