如何结束线程处理套接字连接？

Question

我有一个线程处理的套接字连接：

BufferedReader socketInput = new BufferedReader(new InputStreamReader(mySocket.getInputStream())); 
while (true) 
{ 
    String line = socketInput.readLine(); 
    // do stuff 
} 

正如我在本网站上几个答案读过，推荐的解决方案是使用一个标志，一个线程组和我（插座处理）螺纹当该标志改变状态时检查并终止自己。喜欢的东西：

while (!done) 
{ 
    String line = socketInput.readLine(); 
    // do stuff 
} 

但是，这可能会被卡住时readLine()仍在等待输入。我想我可以设置超时：

mySocket.setSoTimeout(100); 
while (!done) 
{ 
    String line = socketInput.readLine(); 
    // do stuff 
} 

这可能会工作，但我还是希望我的线程之前得到一个100毫秒的延迟“实现”标志的状态改变。

线程是否有办法“马上”意识到它应该结束？如果不是，我的解决方案（超时和标志done）是否正确？

编辑：我澄清，socketInput是BufferedReader类型（或者我考虑Scanner）的。

Answer 1

知道何时完成套接字连接的最佳方式是尝试读取某些内容。如果read方法返回-1，就可以结束threadling套接字连接

byte[] data = new byte[2048]; 
while (!done) { 
    int count = input.read(data); 
    if (count <= 0) { 
     if (count < 0) 
      done = true; 
     continue; 
    } 
    String request = new String(data, 0, count); 
    //do stuff 
} 

我们尝试读取输入的东西，如果计数== -1，套接字客户端断开连接，现在我们可以结束循环，通过改变价值完成。

Answer 2

解决方案是正确的，它将在完成设置为true时退出。 是的，readLine会一直等待100ms，如果你不想等待你可能会通过调用thread.interrupt（）来中断线程，但它不是很干净的方式。

Answer 3

处理此问题的最常见方法是从另一个线程关闭套接字。这将导致读取端解除阻塞并退出，同时关闭套接字的（预期的）错误。根据您可用的套接字API，也可能仅关闭阅读侧。从简短的看JDK shutdownInput()可能工作。

如果你想在这些obvisouly后面继续从套接字读取将无法正常工作。你的解决方案应该在那里工作，但是由于你基本上轮询所有100ms的套接字，所以在性能和反应性方面显然更差。

Answer 4

1. 创建一个选择
2. 配置您的socket.getChannel()到non-blocking和register it与SelectionKey.OP_READ
3. 的选择打电话给你的选择select()方法时，有一些数据，将返回到阅读，所以你可以调用readLine()（即select()返回> 0）

无论何时您想结束插槽处理，请将您的done标志并致电您的选择器wakeup()方法。这将使select()立即返回（如果有活动，可能为0或1）。然后你可以检查你的done标志并优雅地结束你的线程。

这是一个快速实施。注意我通过BufferedReader作为参数，就好像你在线程中打开它一样，你也应该关闭它，这也会关闭套接字，所以它必须在外面完成。有两种方法对信号线平稳地停止处理输入和一个发送数据：

public class SocketHandler extends Thread { 

    private Socket sok; 
    private BufferedReader socketInput; 

    private Selector sel; 
    private SocketChannel chan; 
    private boolean done; 

    public SocketHandler(Socket sok, BufferedReader socketInput) throws IOException { 
     this.sok = sok; 
     chan = sok.getChannel(); 
     chan.configureBlocking(false); 
     sel = Selector.open(); 
     chan.register(sel, SelectionKey.OP_READ); 
     this.socketInput = socketInput; 
     done = false; 
    } 

    @Override 
    public void run() { 
     while (!done) { 
      try { 
       if (sel.select() == 0) 
        continue; 
      } catch (IOException e) { 
       e.printStackTrace(); 
      } 

      // Only one channel is registered on only one operation so we know exactly what happened. 
      sel.selectedKeys().clear(); 
      doRead(); 
      // Otherwise: loop through sel.selectedKeys(), check for readability and clear the set 
     } 
     try { 
      sel.close(); 
     } catch (IOException e) { 
      e.printStackTrace(); 
     } 
    } 

    private void doRead() { 
     try { 
      String line = socketInput.readLine(); 
      // TODO: process 'line' 
     } catch (IOException e) { 
      e.printStackTrace(); 
     } 
    } 

    public void signalStop() { 
     done = true; 
     if (sel != null) 
      sel.wakeup(); // Get out of sel.select() 
    } 

    public void doWrite(byte[] buffer) throws IOException { // Or "String message" 
     sok.getOutputStream().write(buffer); // Or anything else 
    } 

}