提出此问题的动机 我正在运行一个在非常昂贵的硬件上运行的大型产品。出于测试目的而关闭它是不可能的,也不可能在生产环境中放置一个坏的 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() ;
最佳答案
您无法向已关闭的 ExecutorService 提交任务。如果您想停止执行任务,cancel它。除此之外,您的取消任务将在 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增加了很多不确定性。
关于java - ScheduledExecutorService 的使用和流内锁定套接字的清理,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/16371012/