Examples of org.apache.flink.core.io.InputSplitAssigner

  @Test
  public void testSetupInputSplits() {
    try {
      final InputSplit[] emptySplits = new InputSplit[0];
      
      InputSplitAssigner assigner1 = mock(InputSplitAssigner.class);
      InputSplitAssigner assigner2 = mock(InputSplitAssigner.class);
      
      @SuppressWarnings("unchecked")
      InputSplitSource<InputSplit> source1 = mock(InputSplitSource.class);
      @SuppressWarnings("unchecked")
      InputSplitSource<InputSplit> source2 = mock(InputSplitSource.class);

    if (vertex == null) {
      LOG.error("Cannot find execution vertex for vertex ID " + vertexId);
      return null;
    }


    InputSplitAssigner splitAssigner = vertex.getSplitAssigner();
    if (splitAssigner == null) {
      LOG.error("No InputSplitAssigner for vertex ID " + vertexId);
      return null;
    }
    
    // get hostname for input split assignment
    String host = null;
    Execution execution = graph.getRegisteredExecutions().get(executionAttempt);
    if(execution == null) {
      LOG.error("Can not find Execution for attempt " + executionAttempt);
    } else {
      AllocatedSlot slot = execution.getAssignedResource();
      if(slot != null) {
        host = slot.getInstance().getInstanceConnectionInfo().getHostname();
      }
    }
    
    return splitAssigner.getNextInputSplit(host);
  }

Related Classes of org.apache.flink.core.io.InputSplitAssigner

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.