import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.*;
import io.netty.handler.stream.ChunkedFile;
public class FileServer {
private final int port;
public FileServer(int port) {
this.port = port;
}
public void run() throws Exception {
EventLoopGroup bossGroup = new NioEventLoopGroup();
EventLoopGroup workerGroup = new NioEventLoopGroup();
try {
ServerBootstrap b = new ServerBootstrap();
b.group(bossGroup, workerGroup)
.channel(NioServerSocketChannel.class)
.childHandler(new ChannelInitializer<SocketChannel>() {
@Override
public void initChannel(SocketChannel ch) throws Exception {
ChannelPipeline p = ch.pipeline();
// 添加HTTP编解码器
p.addLast(new HttpServerCodec());
// 添加ChunkedWriteHandler，它负责处理ChunkedInput的写入
p.addLast(new ChunkedWriteHandler());
// 添加自定义的处理器
p.addLast(new SimpleChannelInboundHandler<FullHttpRequest>
() {
@Override
protected void channelRead0(ChannelHandlerContext ctx,
FullHttpRequest request) throws Exception {
// 获取请求的路径
String uri = request.uri();
// 假设文件都在服务器的某个目录下
File file = new File("path/to/your/files" + uri);
if (file.exists()) {
// 创建ChunkedFile对象
ChunkedFile chunkedFile = new
ChunkedFile(file);
// 设置响应头信息
HttpResponse response = new
DefaultHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.OK);
HttpHeaders headers = response.headers();
headers.set(HttpHeaderNames.CONTENT_TYPE,
"application/octet-stream");
headers.set(HttpHeaderNames.CONTENT_LENGTH,
file.length());
// 写入响应
ctx.write(response);
// 写入文件内容
ctx.writeAndFlush(chunkedFile);
} else {
// 文件不存在时的处理
HttpResponse response = new
DefaultHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.NOT_FOUND);
ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE);
}
}
});
}
});
// 绑定端口并开始接收传入的连接
ChannelFuture f = b.bind(port).sync();
// 等待服务器套接字关闭
f.channel().closeFuture().sync();
} finally {
workerGroup.shutdownGracefully();
bossGroup.shutdownGracefully();
}
}
public static void main(String[] args) throws Exception {
int port = 8080;
if (args.length > 0) {
port = Integer.parseInt(args[0]);
}
new FileServer(port).run();
}
}