Java NIO 服务器与客户端实现文件下载

对于Java NIO已经学习了一段时间了,周末实践了下,折腾了一天,总算对NIO的理论,有了一个感性的认识。下面的实践是:服务器与客户端都采用NIO的方式来实现文件下载。对于传统的SOCKET BIO方式,服务器端会为每个连接上的客户端分配一个Worker线程来进行doWork,而NIO SERVER却没有为每个Socket链接分配线程的必要了,避免了大量的线程所需的上下文切换,借助NIO提供的Selector机制,只需要一个或者几个线程来管理成百上千的SOCKET连接。那么下面我们就来看看吧!

文件下载辅助类

/**
 * 这个类的基本思路是,读取本地文件到缓冲区
 * 因为通道只能操作缓冲区
 */
class DownloadFileProcesser implements Closeable{
private ByteBuffer buffer = ByteBuffer.allocate(8 * 1024);
private FileChannel fileChannel ;
public DownloadFileProcesser() {
    try{
        FileInputStream fis = new FileInputStream("e:/tmp/Shell学习笔记.pdf");
        fileChannel = fis.getChannel();
    }catch(Exception e){
        e.printStackTrace();
    }
}
public int readFile2Buffer() throws IOException{
    int count = 0;
    buffer.clear();
    count = fileChannel.read(buffer);
    buffer.flip();
    return count;
}
public ByteBuffer getByteBuffer(){
    return buffer;
}
@Override
public void close() throws IOException {
    fileChannel.close();
}
 
}

服务端代码:

public class ServerMain {
 
public static void main(String[] args) throws IOException {
    ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
    serverSocketChannel.configureBlocking(false);
    serverSocketChannel.socket().bind(new InetSocketAddress(8887));
    Selector selector = Selector.open();
    serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
 
    while (true) {
        selector.select();
        Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
        while (iterator.hasNext()) {
        SelectionKey s = iterator.next();
        // 如果客户端有连接请求
        if (s.isAcceptable()) {
        System.out.println("客户端连接请求..");
        ServerSocketChannel ssc = (ServerSocketChannel) s.channel();
        SocketChannel sc = ssc.accept();
        sc.configureBlocking(false);
        sc.register(selector, SelectionKey.OP_READ);
        }
        // 如果客户端有发送数据请求
        if (s.isReadable()) {
        System.out.println("接受客户端发送过来的文本消息...");
        //这里拿出的通道就是ACCEPT上注册的SocketChannel通道
        SocketChannel sc = (SocketChannel) s.channel();
        //要读取数据先要准备好BUFFER缓冲区
        ByteBuffer buffer = ByteBuffer.allocate(8 * 1024);
        //准备BYTE数组,形成输出
        sc.read(buffer);
        byte[] clientByteInfo = new byte[buffer.position()];
        buffer.flip();
        buffer.get(clientByteInfo);
        System.out.println("服务器端收到消息:" + new String(clientByteInfo,"utf-8"));
        //CLIENT下一步的动作就是读取服务器端的文件,因此需要注册写事件
        SelectionKey selectionKey = sc.register(selector, SelectionKey.OP_WRITE);
        //在这个selectionKey上绑定一个对象,以供写操作时取出进行处理
        DownloadFileProcesser downloadFileProcesser = new DownloadFileProcesser();
        selectionKey.attach(downloadFileProcesser);
        }
   
    // 如果客户端有下载文件数据请求
    if (s.isWritable()) {
    //这里把attachment取出进行写入操作
    DownloadFileProcesser downloadFileProcesser = (DownloadFileProcesser)s.attachment();
    int count = downloadFileProcesser.readFile2Buffer();
   
    if(count <= 0){
        System.out.println("客户端下载完毕...");
        //关闭通道
        s.channel().close();
        downloadFileProcesser.close();
    }else{
        //需要注意的是我们这里并没有出现常见的while写的结构,这是为何?
        //因为client其实不断的在read操作,从而触发了SELECTOR的不断写事件!
        SocketChannel sc = (SocketChannel)s.channel();
        sc.write(downloadFileProcesser.getByteBuffer());
    }
  }
        iterator.remove();
  }
 }
 }
}

客户端代码:

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/f157866878a74ff2f4f21a8ade37450d.html