Examples of TaskOperationProtocol

org.apache.flink.runtime.protocols.TaskOperationProtocol
The task submission protocol is implemented by the task manager and allows the job manager to submit and cancel tasks, as well as to query the task manager for cached libraries and submit these if necessary.

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

    
    List<AbstractJobVertex> ordered = Arrays.asList(v1, v2);
    eg.attachJobGraph(ordered);
    
    // create a mock taskmanager that accepts deployment calls
    TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
    when(taskManager.submitTask(Matchers.any(TaskDeploymentDescriptor.class))).thenAnswer(new Answer<TaskOperationResult>() {
      @Override
      public TaskOperationResult answer(InvocationOnMock invocation) {
        final TaskDeploymentDescriptor tdd = (TaskDeploymentDescriptor) invocation.getArguments()[0];
        return new TaskOperationResult(tdd.getExecutionId(), true);
      }
    });
    when(taskManager.cancelTask(Matchers.any(ExecutionAttemptID.class))).thenAnswer(new Answer<TaskOperationResult>() {
      @Override
      public TaskOperationResult answer(InvocationOnMock invocation) {
        final ExecutionAttemptID id = (ExecutionAttemptID) invocation.getArguments()[0];
        return new TaskOperationResult(id, true);
      }

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

      setGraphStatus(graph, JobStatus.RUNNING);
      
      ExecutionJobVertex ejv = graph.getJobVertex(vid);
      
      // mock resources and mock taskmanager
      TaskOperationProtocol taskManager = getSimpleAcknowledgingTaskmanager();
      for (ExecutionVertex ee : ejv.getTaskVertices()) {
        AllocatedSlot slot = getInstance(taskManager).allocateSlot(jid);
        ee.deployToSlot(slot);
      }

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

  public void testDeployCall() {
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexNotExecuting(jid);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

  public void testDeployWithSynchronousAnswer() {
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexExecutingSynchronously(jid);
      
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
      
      when(taskManager.submitTask(Matchers.any(TaskDeploymentDescriptor.class))).thenReturn(new TaskOperationResult(vertex.getCurrentExecutionAttempt().getAttemptId(), true));
      
      assertEquals(ExecutionState.CREATED, vertex.getExecutionState());
      
      vertex.deployToSlot(slot);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

  public void testDeployWithAsynchronousAnswer() {
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      
      
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexExecutingAsynchronously(jid);
      
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
      when(taskManager.submitTask(Matchers.any(TaskDeploymentDescriptor.class))).thenReturn(new TaskOperationResult(vertex.getCurrentExecutionAttempt().getAttemptId(), true));
      
      assertEquals(ExecutionState.CREATED, vertex.getExecutionState());
      
      vertex.deployToSlot(slot);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

    
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexExecutingSynchronously(jid);
      
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
      
      when(taskManager.submitTask(Matchers.any(TaskDeploymentDescriptor.class))).thenReturn(new TaskOperationResult(vertex.getCurrentExecutionAttempt().getAttemptId(), false, ERROR_MESSAGE));
      
      assertEquals(ExecutionState.CREATED, vertex.getExecutionState());
      
      vertex.deployToSlot(slot);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

    
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      final TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexExecutingAsynchronously(jid);
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
      
      when(taskManager.submitTask(Matchers.any(TaskDeploymentDescriptor.class))).thenReturn(new TaskOperationResult(vertex.getCurrentExecutionAttempt().getAttemptId(), false, ERROR_MESSAGE));
      
      assertEquals(ExecutionState.CREATED, vertex.getExecutionState());
      
      vertex.deployToSlot(slot);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

    
    try {
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexNotExecuting(jid);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

    try {
      final ActionQueue queue = new ActionQueue();
      final JobVertexID jid = new JobVertexID();
      
      // mock taskmanager to simply accept the call
      TaskOperationProtocol taskManager = mock(TaskOperationProtocol.class);
      
      final Instance instance = getInstance(taskManager);
      final AllocatedSlot slot = instance.allocateSlot(new JobID());
      
      final ExecutionJobVertex ejv = getJobVertexExecutingTriggered(jid, queue);
      
      final ExecutionVertex vertex = new ExecutionVertex(ejv, 0, new IntermediateResult[0]);
      final ExecutionAttemptID eid = vertex.getCurrentExecutionAttempt().getAttemptId();
      
      // the deployment call succeeds regularly
      when(taskManager.submitTask(any(TaskDeploymentDescriptor.class))).thenReturn(new TaskOperationResult(eid, true));
      
      // first cancel call does not find a task, second one finds it
      when(taskManager.cancelTask(any(ExecutionAttemptID.class))).thenReturn(
          new TaskOperationResult(eid, false), new TaskOperationResult(eid, true));
      
      assertEquals(ExecutionState.CREATED, vertex.getExecutionState());
      
      vertex.deployToSlot(slot);

View Full Code Here

Examples of org.apache.flink.runtime.protocols.TaskOperationProtocol

  public TaskOperationProtocol getTaskManagerProxy() throws IOException {
    if (isDead) {
      throw new IOException("Instance has died");
    }
    
    TaskOperationProtocol tm = this.taskManager;
    
    if (tm == null) {
      synchronized (this) {
        if (this.taskManager == null) {
          this.taskManager = RPC.getProxy(TaskOperationProtocol.class,

View Full Code Here

0 1 2

TOP

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.