Tôi nên sử dụng AsynchronousServerSocketChannel để chấp nhận kết nối như thế nào?

Question

Tôi muốn viết một máy chủ không đồng bộ sử dụng Java 7 và NIO 2.

Nhưng làm cách nào tôi nên sử dụng AsynchronousServerSocketChannel?

Ví dụ: nếu tôi bắt đầu với:

final AsynchronousServerSocketChannel server = 
    AsynchronousServerSocketChannel.open().bind(
     new InetSocketAddress(port)); 

Sau đó, khi tôi làm server.accept(), chương trình chấm dứt vì gọi đó là không đồng bộ. Và nếu tôi đặt mã đó trong một vòng lặp vô hạn, một số AcceptPendingException sẽ bị ném.

Bất kỳ đề xuất nào về cách viết một máy chủ không đồng bộ đơn giản bằng cách sử dụng AsynchronousServerSocketChannel?

Dưới đây là ví dụ của tôi đầy đủ (giống như ví dụ trong javadoc):

import java.io.IOException; 
import java.net.InetSocketAddress; 
import java.nio.channels.AsynchronousServerSocketChannel; 
import java.nio.channels.AsynchronousSocketChannel; 
import java.nio.channels.CompletionHandler; 

public class AsyncServer { 

    public static void main(String[] args) { 
     int port = 8060; 
     try { 
      final AsynchronousServerSocketChannel server = 
        AsynchronousServerSocketChannel.open().bind(
          new InetSocketAddress(port)); 

      System.out.println("Server listening on " + port); 

      server.accept("Client connection", 
        new CompletionHandler<AsynchronousSocketChannel, Object>() { 
       public void completed(AsynchronousSocketChannel ch, Object att) { 
        System.out.println("Accepted a connection"); 

        // accept the next connection 
        server.accept("Client connection", this); 

        // handle this connection 
        //TODO handle(ch); 
       } 

       public void failed(Throwable exc, Object att) { 
        System.out.println("Failed to accept connection"); 
       } 
      }); 
     } catch (IOException e) { 
      e.printStackTrace(); 
     } 
    } 
} 

Answer 1

Bạn đang đi đúng hướng, gọi chấp nhận() từ cuộc gọi lại hoàn thành để chấp nhận nhiều kết nối hơn sẽ hoạt động.

Cách đơn giản (nhưng xấu xí) để ngăn chặn chuỗi từ chấm dứt chỉ đơn giản là lặp cho đến khi chuỗi bị gián đoạn.

// yes, sleep() is evil, but sometimes I don't care 
while (true) { 
    Thread.sleep(1000); 
} 

Cách rõ ràng hơn là sử dụng AsynchronousChannelGroup. Ví dụ:

AsynchronousChannelGroup group = AsynchronousChannelGroup.withThreadPool(Executors 
      .newSingleThreadExecutor()); 
AsynchronousServerSocketChannel server = AsynchronousServerSocketChannel.open(group).bind(
      new InetSocketAddress(port)); 

// (insert server.accept() logic here) 

// wait until group.shutdown()/shutdownNow(), or the thread is interrupted: 
group.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS); 

Bạn có thể điều chỉnh cách xử lý luồng, xem AsynchronousChannelGroup API docs để biết thêm thông tin.

Answer 2

Sử dụng đồng bộ chấp nhận là hữu ích nếu bạn có một cái gì đó khác để làm trong cùng một thread. Trong trường hợp của bạn, bạn không làm điều gì khác để tôi có thể sử dụng

while(true) { 
    AsynchronousSocketChannel socket = server.accept().get(); 
    System.out.println("Accepted " + socket); 
    socket.close(); 
} 

Answer 3

Một cách khác là để phương pháp chính của bạn chờ tín hiệu trước khi trở về. Sau đó, nếu bạn có một số loại lệnh tắt máy bên ngoài, bạn chỉ cần thông báo cho tín hiệu và chủ đề chính tắt.

private static final Object shutdownSignal = new Object(); 

public static void main(String[] args) { 

    ... 

    synchronized (shutdownSignal) { 
     try { 
      shutdownSignal.wait(); 
     } 
     catch (InterruptedException e) { 
      // handle it! 
     } 
    } 
} 

Answer 4

Sử dụng đếm ngược chốt như ví dụ sau

final AsynchronousServerSocketChannel serverChannel = AsynchronousServerSocketChannel.open(); 
    InetSocketAddress address = new InetSocketAddress(port); 
    serverChannel.bind(address); 
    final CountDownLatch latch = new CountDownLatch(1); 
    serverChannel.accept(null, new CompletionHandler<AsynchronousSocketChannel, Object>() { 
@Override 
     public void completed(final AsynchronousSocketChannel channel, Object attachment) { 
      serverChannel.accept(null, this); 
         } 

}); 
try { 
     latch.await(); 
    } catch (InterruptedException e) { 
     e.printStackTrace(); 
     Thread.currentThread().interrupt(); 
    }