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

Fixed a bug where per-channel memory limitation didn't work correctly

Browse Source
This commit is contained in:
Trustin Lee 2008-09-30 01:49:22 +00:00
parent 8c15102bc3
commit 85dc8f93f5
2 changed files with 30 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/main/java/org/jboss/netty/handler/execution/MemoryAwareThreadPoolExecutor.java
View File

@ -279,8 +279,12 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
decreaseCounter(r);
}
private boolean increaseCounter(Runnable task) {
if (isInterestOpsEvent(task)) {
@Override
protected void afterExecute(Runnable r, Throwable e) {
super.afterExecute(r, e);
}
protected boolean increaseCounter(Runnable task) {
if (!shouldCount(task)) {
return false;
}
@ -296,8 +300,10 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
eventTask.estimatedSize = increment;
Channel channel = eventTask.getEvent().getChannel();
long channelCounter = getChannelCounter(channel).addAndGet(increment);
//System.out.println("IC: " + channelCounter + ", " + increment);
if (maxChannelMemorySize != 0 && channelCounter >= maxChannelMemorySize && channel.isOpen()) {
if (channel.isReadable()) {
//System.out.println("UNREADABLE");
channel.setReadable(false);
}
}
@ -307,8 +313,8 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
return maxTotalMemorySize != 0 && totalCounter >= maxTotalMemorySize;
}
private void decreaseCounter(Runnable task) {
if (isInterestOpsEvent(task)) {
protected void decreaseCounter(Runnable task) {
if (!shouldCount(task)) {
return;
}
@ -334,8 +340,10 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
if (task instanceof ChannelEventRunnable) {
Channel channel = ((ChannelEventRunnable) task).getEvent().getChannel();
long channelCounter = getChannelCounter(channel).addAndGet(-increment);
if (maxChannelMemorySize != 0 && channelCounter + increment >= maxChannelMemorySize && channel.isOpen()) {
//System.out.println("DC: " + channelCounter + ", " + increment);
if (maxChannelMemorySize != 0 && channelCounter < maxChannelMemorySize && channel.isOpen()) {
if (!channel.isReadable()) {
//System.out.println("READABLE");
channel.setReadable(true);
}
}
@ -359,17 +367,21 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
return counter;
}
private static boolean isInterestOpsEvent(Runnable task) {
private static boolean shouldCount(Runnable task) {
if (task instanceof Executor) {
return false;
}
if (task instanceof ChannelEventRunnable) {
ChannelEventRunnable r = (ChannelEventRunnable) task;
if (r.getEvent() instanceof ChannelStateEvent) {
ChannelStateEvent e = (ChannelStateEvent) r.getEvent();
if (e.getState() == ChannelState.INTEREST_OPS) {
return true;
return false;
}
}
}
return false;
return true;
}
private static class Settings {

10
src/main/java/org/jboss/netty/handler/execution/OrderedMemoryAwareThreadPoolExecutor.java
View File

@ -188,14 +188,24 @@ public class OrderedMemoryAwareThreadPoolExecutor extends
}
public void run() {
Thread thread = Thread.currentThread();
for (;;) {
final Runnable task;
synchronized (tasks) {
task = tasks.getFirst();
}
boolean ran = false;
OrderedMemoryAwareThreadPoolExecutor.this.beforeExecute(thread, task);
try {
task.run();
ran = true;
OrderedMemoryAwareThreadPoolExecutor.this.afterExecute(task, null);
} catch (RuntimeException e) {
if (!ran) {
OrderedMemoryAwareThreadPoolExecutor.this.afterExecute(task, e);
}
throw e;
} finally {
synchronized (tasks) {
tasks.removeFirst();