@Override public TwillController apply(TwillController controller) { return new ImpersonatedTwillController(controller, impersonator, programId); } });
@Override public TwillController apply(TwillController controller) { return new ImpersonatedTwillController(controller, impersonator, programId); } });
private Iterable<TwillController> wrapControllers(Iterable<TwillController> controllers, String applicationName) { if (isMasterService(applicationName)) { return controllers; } try { final ProgramId programId = TwillAppNames.fromTwillAppName(applicationName); return Iterables.transform(controllers, new Function<TwillController, TwillController>() { @Override public TwillController apply(TwillController controller) { return new ImpersonatedTwillController(controller, impersonator, programId); } }); } catch (IllegalArgumentException e) { // If the conversion from twill app name to programId failed, don't wrap return controllers; } }
private Iterable<TwillController> wrapControllers(Iterable<TwillController> controllers, String applicationName) { if (isMasterService(applicationName)) { return controllers; } try { final ProgramId programId = TwillAppNames.fromTwillAppName(applicationName); return Iterables.transform(controllers, new Function<TwillController, TwillController>() { @Override public TwillController apply(TwillController controller) { return new ImpersonatedTwillController(controller, impersonator, programId); } }); } catch (IllegalArgumentException e) { // If the conversion from twill app name to programId failed, don't wrap return controllers; } }
@Override public TwillController lookup(String applicationName, RunId runId) { TwillController controller = delegate.lookup(applicationName, runId); if (isMasterService(applicationName)) { return controller; } try { return new ImpersonatedTwillController(controller, impersonator, TwillAppNames.fromTwillAppName(applicationName)); } catch (IllegalArgumentException e) { // If the conversion from twill app name to programId failed, don't wrap return controller; } }
@Override public TwillController lookup(String applicationName, RunId runId) { TwillController controller = delegate.lookup(applicationName, runId); if (isMasterService(applicationName)) { return controller; } try { return new ImpersonatedTwillController(controller, impersonator, TwillAppNames.fromTwillAppName(applicationName)); } catch (IllegalArgumentException e) { // If the conversion from twill app name to programId failed, don't wrap return controller; } }
@Override public TwillController start(final long timeout, final TimeUnit timeoutUnit) { try { return impersonator.doAs(programId, () -> { // Add secure tokens if (User.isHBaseSecurityEnabled(hConf) || UserGroupInformation.isSecurityEnabled()) { addSecureStore(YarnSecureStore.create(secureStoreRenewer.createCredentials())); } return new ImpersonatedTwillController(delegate.start(timeout, timeoutUnit), impersonator, programId); }); } catch (Exception e) { throw Throwables.propagate(e); } } }
@Override public TwillController start(final long timeout, final TimeUnit timeoutUnit) { try { return impersonator.doAs(programId, () -> { // Add secure tokens if (User.isHBaseSecurityEnabled(hConf) || UserGroupInformation.isSecurityEnabled()) { addSecureStore(YarnSecureStore.create(secureStoreRenewer.createCredentials())); } return new ImpersonatedTwillController(delegate.start(timeout, timeoutUnit), impersonator, programId); }); } catch (Exception e) { throw Throwables.propagate(e); } } }