public void runInContext(final Runnable runnable) { DQPWorkContext previous = DQPWorkContext.getWorkContext(); DQPWorkContext.setWorkContext(this); Object previousSecurityContext = null; if (securityHelper != null) { previousSecurityContext = securityHelper.associateSecurityContext(this.getSecurityContext()); } try { runnable.run(); } finally { if (securityHelper != null) { securityHelper.associateSecurityContext(previousSecurityContext); } DQPWorkContext.setWorkContext(previous); } }
public void runInContext(final Runnable runnable) { DQPWorkContext previous = DQPWorkContext.getWorkContext(); DQPWorkContext.setWorkContext(this); Object previousSecurityContext = null; if (securityHelper != null) { previousSecurityContext = securityHelper.associateSecurityContext(this.getSecurityContext()); } try { runnable.run(); } finally { if (securityHelper != null) { securityHelper.associateSecurityContext(previousSecurityContext); } DQPWorkContext.setWorkContext(previous); } }
public void runInContext(final Runnable runnable) { DQPWorkContext previous = DQPWorkContext.getWorkContext(); DQPWorkContext.setWorkContext(this); Object previousSecurityContext = null; if (securityHelper != null) { previousSecurityContext = securityHelper.associateSecurityContext(this.getSecurityContext()); } try { runnable.run(); } finally { if (securityHelper != null) { securityHelper.associateSecurityContext(previousSecurityContext); } DQPWorkContext.setWorkContext(previous); } }
@After public void tearDown() throws Exception { DQPWorkContext.setWorkContext(new DQPWorkContext()); core.stop(); }
@Override public Void call() throws Exception { DQPWorkContext.setWorkContext(context); RequestMessage reqMsg = exampleRequestMessage("select * FROM BQT1.SmallA"); DQPWorkContext.getWorkContext().getSession().setSessionId("1"); DQPWorkContext.getWorkContext().getSession().setUserName("a"); Future<ResultsMessage> message = null; try { message = core.executeRequest(counter.getAndIncrement(), reqMsg); } finally { submitted.countDown(); } assertNotNull(core.getClientState("1", false)); submitted.await(); //after this, both will be submitted agds.latch.countDown(); //allow the execution to proceed message.get(500000, TimeUnit.MILLISECONDS); return null; } }));
public static DQPWorkContext buildWorkContext(QueryMetadataInterface metadata, VDBMetaData vdb) { DQPWorkContext workContext = new DQPWorkContext(); SessionMetadata session = new SessionMetadata(); workContext.setSession(session); session.setVDBName(vdb.getName()); session.setVDBVersion(vdb.getVersion()); session.setSessionId(String.valueOf(1)); session.setUserName("foo"); //$NON-NLS-1$ session.setVdb(vdb); workContext.getVDB().addAttchment(QueryMetadataInterface.class, metadata); if (metadata instanceof TransformationMetadata) { workContext.getVDB().addAttchment(TransformationMetadata.class, (TransformationMetadata)metadata); } DQPWorkContext.setWorkContext(workContext); return workContext; }