YARN-5663. Small refactor in ZKRMStateStore. Contributed by Oleksii Dymytrov.

(cherry picked from commit 14a696f369)
(cherry picked from commit 74f2df16a9)
This commit is contained in:
Akira Ajisaka 2016-09-26 15:00:01 +09:00
parent 4c2b20ca3e
commit ece3ca0cba
1 changed files with 3 additions and 6 deletions

View File

@ -761,17 +761,14 @@ public class ZKRMStateStore extends RMStateStore {
String nodeCreatePath = String nodeCreatePath =
getNodePath(dtMasterKeysRootPath, DELEGATION_KEY_PREFIX getNodePath(dtMasterKeysRootPath, DELEGATION_KEY_PREFIX
+ delegationKey.getKeyId()); + delegationKey.getKeyId());
ByteArrayOutputStream os = new ByteArrayOutputStream();
DataOutputStream fsOut = new DataOutputStream(os);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("Storing RMDelegationKey_" + delegationKey.getKeyId()); LOG.debug("Storing RMDelegationKey_" + delegationKey.getKeyId());
} }
delegationKey.write(fsOut); ByteArrayOutputStream os = new ByteArrayOutputStream();
try { try(DataOutputStream fsOut = new DataOutputStream(os)) {
delegationKey.write(fsOut);
safeCreate(nodeCreatePath, os.toByteArray(), zkAcl, safeCreate(nodeCreatePath, os.toByteArray(), zkAcl,
CreateMode.PERSISTENT); CreateMode.PERSISTENT);
} finally {
os.close();
} }
} }