HADOOP-10404. Commit correct version of patch (cmccabe)

(cherry picked from commit 687d83c9e1)
This commit is contained in:
Colin Patrick Mccabe 2014-10-06 14:44:13 -07:00
parent c97134d054
commit b5158c8d24
1 changed files with 12 additions and 11 deletions

View File

@ -101,7 +101,7 @@ public final class DomainSocketWatcher implements Closeable {
*/ */
private class NotificationHandler implements Handler { private class NotificationHandler implements Handler {
public boolean handle(DomainSocket sock) { public boolean handle(DomainSocket sock) {
lock.lock(); assert(lock.isHeldByCurrentThread());
try { try {
if (LOG.isTraceEnabled()) { if (LOG.isTraceEnabled()) {
LOG.trace(this + ": NotificationHandler: doing a read on " + LOG.trace(this + ": NotificationHandler: doing a read on " +
@ -125,8 +125,6 @@ public final class DomainSocketWatcher implements Closeable {
} }
closed = true; closed = true;
return true; return true;
} finally {
lock.unlock();
} }
} }
} }
@ -349,15 +347,13 @@ public final class DomainSocketWatcher implements Closeable {
* Wake up the DomainSocketWatcher thread. * Wake up the DomainSocketWatcher thread.
*/ */
private void kick() { private void kick() {
lock.lock(); assert(lock.isHeldByCurrentThread());
try { try {
notificationSockets[0].getOutputStream().write(0); notificationSockets[0].getOutputStream().write(0);
} catch (IOException e) { } catch (IOException e) {
if (!closed) { if (!closed) {
LOG.error(this + ": error writing to notificationSockets[0]", e); LOG.error(this + ": error writing to notificationSockets[0]", e);
} }
} finally {
lock.unlock();
} }
} }
@ -467,12 +463,17 @@ public final class DomainSocketWatcher implements Closeable {
} catch (IOException e) { } catch (IOException e) {
LOG.error(toString() + " terminating on IOException", e); LOG.error(toString() + " terminating on IOException", e);
} finally { } finally {
lock.lock();
try {
kick(); // allow the handler for notificationSockets[0] to read a byte kick(); // allow the handler for notificationSockets[0] to read a byte
for (Entry entry : entries.values()) { for (Entry entry : entries.values()) {
sendCallback("close", entries, fdSet, entry.getDomainSocket().fd); sendCallback("close", entries, fdSet, entry.getDomainSocket().fd);
} }
entries.clear(); entries.clear();
fdSet.close(); fdSet.close();
} finally {
lock.unlock();
}
} }
} }
}); });