Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-259: Expose the ZK exception code in the event of a KeeperException within the ZooKeeperStateProvider.

This commit is contained in:
Aldrin Piri 2016-02-01 15:50:37 -05:00
parent 447e401912
commit a931e72787
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/main/java/org/apache/nifi/controller/state/providers/zookeeper/ZooKeeperStateProvider.java
View File

@ -244,16 +244,17 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
ZKUtil.deleteRecursive(getZooKeeper(), getComponentPath(componentId));
} catch (final KeeperException ke) {
// Node doesn't exist so just ignore
if (Code.NONODE == ke.code()) {
final Code exceptionCode = ke.code();
if (Code.NONODE == exceptionCode) {
return;
}
if (Code.SESSIONEXPIRED == ke.code()) {
if (Code.SESSIONEXPIRED == exceptionCode) {
invalidateClient();
onComponentRemoved(componentId);
return;
}
throw new IOException("Unable to remove state for component with ID '" + componentId + "' from ZooKeeper", ke);
throw new IOException("Unable to remove state for component with ID '" + componentId + " with exception code " + exceptionCode, ke);
} catch (final InterruptedException e) {
Thread.currentThread().interrupt();
throw new IOException("Failed to remove state for component with ID '" + componentId + "' from ZooKeeper due to being interrupted", e);
@ -331,7 +332,8 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
try {
keeper.setData(path, data, version);
} catch (final KeeperException ke) {
if (ke.code() == Code.NONODE) {
final Code exceptionCode = ke.code();
if (exceptionCode == Code.NONODE) {
createNode(path, data);
return;
} else {
@ -342,17 +344,18 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
Thread.currentThread().interrupt();
throw new IOException("Failed to set cluster-wide state in ZooKeeper for component with ID " + componentId + " due to interruption", e);
} catch (final KeeperException ke) {
if (Code.SESSIONEXPIRED == ke.code()) {
final Code exceptionCode = ke.code();
if (Code.SESSIONEXPIRED == exceptionCode) {
invalidateClient();
setState(stateValues, version, componentId);
return;
}
if (Code.NODEEXISTS == ke.code()) {
if (Code.NODEEXISTS == exceptionCode) {
setState(stateValues, version, componentId);
return;
}
throw new IOException("Failed to set cluster-wide state in ZooKeeper for component with ID " + componentId, ke);
throw new IOException("Failed to set cluster-wide state in ZooKeeper for component with ID " + componentId + " with exception code " + exceptionCode, ke);
} catch (final IOException ioe) {
throw new IOException("Failed to set cluster-wide state in ZooKeeper for component with ID " + componentId, ioe);
}
@ -365,20 +368,21 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
} catch (final InterruptedException ie) {
throw new IOException("Failed to update cluster-wide state due to interruption", ie);
} catch (final KeeperException ke) {
if (ke.code() == Code.NONODE) {
final Code exceptionCode = ke.code();
if (Code.NONODE == exceptionCode) {
final String parentPath = StringUtils.substringBeforeLast(path, "/");
createNode(parentPath, null);
createNode(path, data);
return;
}
if (Code.SESSIONEXPIRED == ke.code()) {
if (Code.SESSIONEXPIRED == exceptionCode) {
invalidateClient();
createNode(path, data);
return;
}
// Node already exists. Node must have been created by "someone else". Just set the data.
if (ke.code() == Code.NODEEXISTS) {
if (Code.NODEEXISTS == exceptionCode) {
try {
getZooKeeper().setData(path, data, -1);
return;
@ -392,8 +396,6 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
throw new IOException("Failed to update cluster-wide state due to interruption", ie);
}
}
throw ke;
}
}
@ -413,15 +415,16 @@ public class ZooKeeperStateProvider extends AbstractStateProvider {
Thread.currentThread().interrupt();
throw new IOException("Failed to obtain value from ZooKeeper for component with ID " + componentId + ", due to interruption", e);
} catch (final KeeperException ke) {
if (ke.code() == Code.NONODE) {
final Code exceptionCode = ke.code();
if (Code.NONODE == exceptionCode) {
return new StandardStateMap(null, -1L);
}
if (Code.SESSIONEXPIRED == ke.code()) {
if (Code.SESSIONEXPIRED == exceptionCode) {
invalidateClient();
return getState(componentId);
}
throw new IOException("Failed to obtain value from ZooKeeper for component with ID " + componentId, ke);
throw new IOException("Failed to obtain value from ZooKeeper for component with ID " + componentId + " with exception code " + exceptionCode, ke);
} catch (final IOException ioe) {
// provide more context in the error message
throw new IOException("Failed to obtain value from ZooKeeper for component with ID " + componentId, ioe);