1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
/* Glazed Lists (c) 2003-2006 */
/* http://publicobject.com/glazedlists/ publicobject.com,*/
/* O'Dell Engineering Ltd.*/
package ca.odell.glazedlists.impl.nio;
// NIO is used for CTP
import java.io.IOException;
import java.nio.channels.SelectionKey;
import java.nio.channels.ServerSocketChannel;
import java.util.Iterator;
import java.util.logging.Logger;
/**
* A task that gracefully shuts down the NIO daemon.
*/
class Shutdown implements Runnable {
/** logging */
private static Logger logger = Logger.getLogger(Shutdown.class.toString());
/** the I/O event queue daemon */
private NIODaemon nioDaemon = null;
/**
* Create a new NIOShutdown that shuts down a server using the specified
* NIODaemon.
*/
public Shutdown(NIODaemon nioDaemon) {
this.nioDaemon = nioDaemon;
}
/**
* Runs the specified task.
*/
public void run() {
logger.info("Cleaning up listening socket and closing " + (nioDaemon.getSelector().keys().size()-1) + " connections");
// kill all connections
for(Iterator k = nioDaemon.getSelector().keys().iterator(); k.hasNext(); ) {
SelectionKey key = (SelectionKey)k.next();
// close an invalid connection
if(!key.isValid()) {
NIOAttachment attachment = (NIOAttachment)key.attachment();
attachment.close(new IOException("Connection closed"));
// close the server socket
} else if((key.interestOps() & SelectionKey.OP_ACCEPT) != 0) {
try {
ServerSocketChannel server = (ServerSocketChannel)key.channel();
server.close();
key.cancel();
} catch(IOException e) {
logger.warning("Error closing server socket, " + e.getMessage());
}
// close a connection socket
} else {
NIOAttachment attachment = (NIOAttachment)key.attachment();
attachment.close(new ServerShutdownException());
}
}
}
}
|
