0

这个问题的动机 我正在运行一个在非常昂贵的硬件上运行的巨大产品。出于测试目的而关闭它是不可能的，也不可能在生产环境中放置一个坏 jar。我需要尽可能确定，几乎可以确保我不会把生产环境搞得一团糟。

在我在登台设置上运行它之前，我需要检查下面的代码是否存在明显的问题（这同样昂贵）。

问题 我有一个基于套接字的应用程序，有时客户端不会显式发送 CloseConnection 请求。有时 IOException 不会发生，因为阻塞readObject调用阻塞了线程。

我需要通过在超时后关闭连接来关闭这个线程。如果我从服务器收到新请求，则会刷新超时。

所以你会在下面看到 3 个部分

  • 初始化
  • while(true) 循环中的 readObject 调用，以及计划的服务重置
  • 内流的实际关闭

代码

有人建议我使用 ScheduledExecutorService 而不是 Timer/TimerTask。

class StreamManager {
    ....
    private ScheduledExecutorService activityTimeOut = Executors
            .newSingleThreadScheduledExecutor();
    private CloseConnectionOnTimeOut closeOnTimeOut = new CloseConnectionOnTimeOut();
    ....

    public void initialize(Socket newClientSocket, ObjectInputStream newInputStream,
            ObjectOutputStream newOutputStream, ThreadMonitor newThreadMonitor) {
        ....
        closeOnTimeOut.setInputStream(myInputStream);
        activityTimeOut.scheduleAtFixedRate(closeOnTimeOut, 0, Globals.INACTIVITY_TIME_OUT,
                TimeUnit.MILLISECONDS);
    }

    public void run() {
    ....
    while (true) {
            try {
                AMessageStrategy incomingCommand = (AMessageStrategy) myInputStream
                        .readObject();
                activityTimeOut.shutdown();
                activityTimeOut.scheduleAtFixedRate(closeOnTimeOut, 0,
                        Globals.INACTIVITY_TIME_OUT, TimeUnit.MILLISECONDS);
        ....
     }
    ....
 }
 class CloseConnectionOnTimeOut implements Runnable {
        private ObjectInputStream myInputStream;

        public CloseConnectionOnTimeOut() {

        }

        public void setInputStream(ObjectInputStream myInputStream) {
            this.myInputStream = myInputStream;
        }

        public void run() {
            try {
                myInputStream.close();
                myOutputStream.close();
                clientSocket.close();
                log.info("Time out occured for client, closed connection forcefully.") ;
            } catch (IOException e) {
                e.printStackTrace();
                log.fatal("Time out has occured, yet unable to clean up client connection. Keep a watch out on \"Size of clientStreamQ\"");
            }
        }
    }

编辑：刚刚测试了一个较小的应用程序，它似乎工作。我仍然需要你的反馈。

再次编辑：

我根据建议修改了下面的代码。

初始化

private ScheduledExecutorService activityTimeOut = Executors
            .newSingleThreadScheduledExecutor();
    private Future<Void> timeoutTask ;
    private CloseConnectionOnTimeOut closeOnTimeOut = new CloseConnectionOnTimeOut();

删除了此代码

closeOnTimeOut.setInputStream(myInputStream);
activityTimeOut.scheduleAtFixedRate(closeOnTimeOut, 0, Globals.INACTIVITY_TIME_OUT,
                TimeUnit.MILLISECONDS);

在 readObject 前后替换

timeoutTask = (Future<Void>) activityTimeOut.scheduleAtFixedRate(
                        closeOnTimeOut.setInputStream(myInputStream), 0,
                        Globals.INACTIVITY_TIME_OUT, TimeUnit.MILLISECONDS);
AMessageStrategy incomingCommand = (AMessageStrategy) myInputStream
                        .readObject();
timeoutTask.cancel(true) ;

清理时

activityTimeOut.shutdown() ;
4

1 回答 1

1

您不能将任务提交给已关闭的 ExecutorService。如果要停止执行任务，请取消它。除此之外，您的取消任务将被安排在初始化后立即运行StreamManager- 如果初始化和运行之间存在间隙，您可能会遇到麻烦。我建议在尝试从套接字读取之前创建并安排一个新任务，并在读取成功后取消它：

while (true) {
   ...
   Future<Void> timeoutTask = activityTimeOut.schedule(new CloseConnection(/*init with streams*/), Globals.INACTIVITY_TIME_OUT, TimeUnit.MILLISECONDS);
   try {
      AMessageStrategy incomingCommand = (AMessageStrategy) myInputStream.readObject();
   } finally {
      timeoutTask.cancel(false);
   }
   ...
}

在清理方法StreamManager或结束时run()你应该关闭使用的ScheduledExecutorService.

如果您的软件是关键任务，我会在本地对其进行彻底测试。编写单元测试和可能的小型集成测试来验证取消是否有效。但恐怕这个解决方案相当脆弱。多线程和 IO 增加了很多不确定性。

于 2013-05-04T06:53:19.017 回答