YARN-2264. Fixed a race condition in DrainDispatcher which may cause random test failures. Contributed by Li Lu
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/trunk@1611126 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
43fe48d9e2
commit
40e1bb9d31
@ -59,6 +59,9 @@ Release 2.6.0 - UNRELEASED
|
||||
YARN-2260. Fixed ResourceManager's RMNode to correctly remember containers
|
||||
when nodes resync during work-preserving RM restart. (Jian He via vinodkv)
|
||||
|
||||
YARN-2264. Fixed a race condition in DrainDispatcher which may cause random
|
||||
test failures. (Li Lu via jianhe)
|
||||
|
||||
Release 2.5.0 - UNRELEASED
|
||||
|
||||
INCOMPATIBLE CHANGES
|
||||
|
@ -28,6 +28,7 @@ public class DrainDispatcher extends AsyncDispatcher {
|
||||
// and similar grotesqueries
|
||||
private volatile boolean drained = false;
|
||||
private final BlockingQueue<Event> queue;
|
||||
final Object mutex;
|
||||
|
||||
public DrainDispatcher() {
|
||||
this(new LinkedBlockingQueue<Event>());
|
||||
@ -36,6 +37,7 @@ public DrainDispatcher() {
|
||||
private DrainDispatcher(BlockingQueue<Event> eventQueue) {
|
||||
super(eventQueue);
|
||||
this.queue = eventQueue;
|
||||
this.mutex = this;
|
||||
}
|
||||
|
||||
/**
|
||||
@ -53,8 +55,10 @@ Runnable createThread() {
|
||||
@Override
|
||||
public void run() {
|
||||
while (!Thread.currentThread().isInterrupted()) {
|
||||
// !drained if dispatch queued new events on this dispatcher
|
||||
drained = queue.isEmpty();
|
||||
synchronized (mutex) {
|
||||
// !drained if dispatch queued new events on this dispatcher
|
||||
drained = queue.isEmpty();
|
||||
}
|
||||
Event event;
|
||||
try {
|
||||
event = queue.take();
|
||||
@ -75,8 +79,10 @@ public EventHandler getEventHandler() {
|
||||
return new EventHandler() {
|
||||
@Override
|
||||
public void handle(Event event) {
|
||||
drained = false;
|
||||
actual.handle(event);
|
||||
synchronized (mutex) {
|
||||
actual.handle(event);
|
||||
drained = false;
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user