如何设置阻止呼叫的超时时间?

时间:2019-12-22 09:54:57

标签: scala sockets serversocket blocking

我有一个与多个客户端一起工作的多线程聊天服务器(每个客户端都在新线程中处理)。

如果客户端断开连接,服务器上的相关线程将挂在inputstream.readLine()上,这是一个阻塞调用。

如何设置某种超时时间,以便我可以在2000ms后关闭套接字并释放线程而没有响应?

class Handler(socket: Socket) extends Runnable {
  def run() : Unit = {
    val inputstream = new BufferedReader(new InputStreamReader(socket.getInputStream()))
    val outputstream = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()))
    var break = false
    while(break != true){
      val input = inputstream.readLine() // blocking call
// how do i time out after 2000ms?
      input match {
        case "update" =>
          outputstream.write("message")
          outputstream.newLine()
          outputstream.flush()
        case _ => // any other input, break and close socket
          break = true
      }
    }
    socket.close()
  }
}

object ChatServer extends App {
  val server = new ServerSocket(10000)
  while(true) {
    val socket = server.accept // blocking call
    (new Thread(new Handler(socket))).start()
    // some condition{
    //   server.close()
    // }
  }
}

1 个答案:

答案 0 :(得分:5)

您可以使用BufferedReader.ready()

  

如果保证下一个read()不会阻塞输入,则为true,否则为false   除此以外。请注意,返回false不能保证下一个   读取将阻止。

class Handler(socket: Socket) extends Runnable {
  def run(): Unit = {
    val inputstream = new BufferedReader(new InputStreamReader(socket.getInputStream))
    val outputstream = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream))

    var break = false
    val startTime = System.currentTimeMillis()

    while (!break) {
      if (inputstream.ready()) {
        val input = inputstream.readLine()
        input match {
          case "update" =>
            outputstream.write("message")
            outputstream.newLine()
            outputstream.flush()
          case _ => // any other input, break and close socket
            break = true
        }
      } else if (System.currentTimeMillis() - startTime >= 2000) {
        break = true
      }
    }
    socket.close()
  }
}