public boolean cancelRequest(String sessionId, long executionId) throws TeiidComponentException { RequestID requestID = new RequestID(sessionId, executionId); return cancelRequest(requestID, "by an admin operation"); //$NON-NLS-1$ }
public AtomicRequestMessage(RequestMessage requestMessage, DQPWorkContext parent, int nodeId){ this.workContext = parent; this.requestID = new RequestID(parent.getSessionId(), requestMessage.getExecutionId()); this.atomicRequestId = new AtomicRequestID(this.requestID, nodeId, EXECUTION_COUNT.getAndIncrement()); }
public AtomicRequestMessage(RequestMessage requestMessage, DQPWorkContext parent, int nodeId){ this.workContext = parent; this.requestID = new RequestID(parent.getSessionId(), requestMessage.getExecutionId()); this.atomicRequestId = new AtomicRequestID(this.requestID, nodeId, EXECUTION_COUNT.getAndIncrement()); }
public AtomicRequestMessage(RequestMessage requestMessage, DQPWorkContext parent, int nodeId){ this.workContext = parent; this.requestID = new RequestID(parent.getSessionId(), requestMessage.getExecutionId()); this.atomicRequestId = new AtomicRequestID(this.requestID, nodeId, EXECUTION_COUNT.getAndIncrement()); }
public ExecutionContextImpl(String vdbName, Object vdbVersion, Serializable executionPayload, String originalConnectionID, String connectorName, long requestId, String partId, String execCount) { commandContext = new CommandContext(); commandContext.setVdbName(vdbName); commandContext.setVdbVersion(vdbVersion.toString()); commandContext.setCommandPayload(executionPayload); commandContext.setConnectionID(originalConnectionID); commandContext.setRequestId(new RequestID(originalConnectionID, requestId)); this.connectorName = connectorName; this.partID = partId; this.executeCount = execCount; }
public ExecutionContextImpl(String vdbName, Object vdbVersion, Serializable executionPayload, String originalConnectionID, String connectorName, long requestId, String partId, String execCount) { commandContext = new CommandContext(); commandContext.setVdbName(vdbName); commandContext.setVdbVersion(vdbVersion.toString()); commandContext.setCommandPayload(executionPayload); commandContext.setConnectionID(originalConnectionID); commandContext.setRequestId(new RequestID(originalConnectionID, requestId)); this.connectorName = connectorName; this.partID = partId; this.executeCount = execCount; }
public ExecutionContextImpl(String vdbName, Object vdbVersion, Serializable executionPayload, String originalConnectionID, String connectorName, long requestId, String partId, String execCount) { commandContext = new CommandContext(); commandContext.setVdbName(vdbName); commandContext.setVdbVersion(vdbVersion.toString()); commandContext.setCommandPayload(executionPayload); commandContext.setConnectionID(originalConnectionID); commandContext.setRequestId(new RequestID(originalConnectionID, requestId)); this.connectorName = connectorName; this.partID = partId; this.executeCount = execCount; }
private void compareReqInfos(Collection<RequestID> reqs1, Collection<RequestMetadata> reqs2) { Set<RequestID> reqIDs2 = new HashSet<RequestID>(); for (RequestMetadata requestInfo : reqs2) { reqIDs2.add(new RequestID(requestInfo.getSessionId(), requestInfo.getExecutionId())); } assertEquals("Collections of request infos are not the same: ", new HashSet<RequestID>(reqs1), reqIDs2); //$NON-NLS-1$ }
private RequestID addRequest(DQPCore rm, String sessionId, int executionId) { RequestMessage r0 = new RequestMessage("test command"); //$NON-NLS-1$ RequestID id = new RequestID(sessionId, executionId); addRequest(rm, r0, id, null, null); return id; }
@Test public void testRemoveUnknownRequestState() throws Exception { helpAssureOneState(); csm.removeState(new AtomicRequestID(new RequestID("ZZZZ", 3210), 5, 5)); //$NON-NLS-1$ assertEquals("Expected size of 1", 1, csm.size()); //$NON-NLS-1$ }
public void testAddRequest() { DQPCore rm = new DQPCore(); rm.setTransactionService(new FakeTransactionService()); RequestMessage r0 = new RequestMessage("foo"); //$NON-NLS-1$ RequestID requestID = new RequestID(SESSION_STRING, 1); addRequest(rm, r0, requestID, null, null); }
public void testGetters1() { String connID = "100"; //$NON-NLS-1$ long executionID = 200; RequestID r = new RequestID(connID, executionID); assertEquals("Lost connectionID", connID, r.getConnectionID()); //$NON-NLS-1$ assertEquals("Lost executionID", executionID, r.getExecutionID()); //$NON-NLS-1$ assertEquals("Wrong string representation", "100.200", r.toString()); //$NON-NLS-1$ //$NON-NLS-2$ }
public void testGetters2() { long executionID = 200; RequestID r = new RequestID(executionID); assertEquals("Lost connectionID", null, r.getConnectionID()); //$NON-NLS-1$ assertEquals("Lost executionID", executionID, r.getExecutionID()); //$NON-NLS-1$ assertEquals("Wrong string representation", "C.200", r.toString()); //$NON-NLS-1$ //$NON-NLS-2$ }
public void testSerialize2() throws Exception { RequestID copy = UnitTestUtil.helpSerialize(new RequestID(100)); assertEquals(null, copy.getConnectionID()); assertEquals(100, copy.getExecutionID()); assertEquals("C.100", copy.toString()); //$NON-NLS-1$ }
public void testSerialize1() throws Exception { RequestID copy = UnitTestUtil.helpSerialize(new RequestID("1", 100)); //$NON-NLS-1$ assertEquals("1", copy.getConnectionID()); //$NON-NLS-1$ assertEquals(100, copy.getExecutionID()); assertEquals("1.100", copy.toString()); //$NON-NLS-1$ }
public void testGetConnectorInfo() { DQPCore rm = new DQPCore(); rm.setTransactionService(new FakeTransactionService()); RequestMessage r0 = new RequestMessage("foo"); //$NON-NLS-1$ RequestID requestID = new RequestID(SESSION_STRING, 1); RequestWorkItem workItem = addRequest(rm, r0, requestID, null, null); AtomicRequestMessage atomicReq = new AtomicRequestMessage(workItem.requestMsg, workItem.getDqpWorkContext(), 1); DataTierTupleSource info = Mockito.mock(DataTierTupleSource.class); workItem.addConnectorRequest(atomicReq.getAtomicRequestID(), info); DataTierTupleSource arInfo = workItem.getConnectorRequest(atomicReq.getAtomicRequestID()); assertTrue(arInfo == info); }