andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Less restriction in shutdown hook modification

Browse Source
This commit is contained in:
Trustin Lee 2012-05-30 04:27:19 -07:00
parent 078a502c5f
commit 4154f42520
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
transport/src/main/java/io/netty/channel/SingleThreadEventLoop.java
View File

@ -221,19 +221,29 @@ public abstract class SingleThreadEventLoop extends AbstractExecutorService impl
return Thread.currentThread() == thread;
}
public void addShutdownHook(Runnable task) {
ensureShutdownHookAccess();
shutdownHooks.add(task);
public void addShutdownHook(final Runnable task) {
if (inEventLoop()) {
shutdownHooks.add(task);
} else {
execute(new Runnable() {
@Override
public void run() {
shutdownHooks.add(task);
}
});
}
}
public void removeShutdownHook(Runnable task) {
ensureShutdownHookAccess();
shutdownHooks.remove(task);
}
private void ensureShutdownHookAccess() {
if (!inEventLoop()) {
throw new IllegalStateException("must be called from the event loop");
public void removeShutdownHook(final Runnable task) {
if (inEventLoop()) {
shutdownHooks.remove(task);
} else {
execute(new Runnable() {
@Override
public void run() {
shutdownHooks.remove(task);
}
});
}
}