@Override public int hashCode() { int result = (int) (markedTime ^ (markedTime >>> 32)); result = 31 * result + operatorId.hashCode(); result = 31 * result + subtaskIndex; return result; }
private static JobGraph getWorkingJobGraph() { final JobVertex jobVertex = new JobVertex("Working job vertex."); jobVertex.setInvokableClass(NoOpInvokable.class); return new JobGraph("Working testing job", jobVertex); }
@Override public OperatorID getOperatorID() { return new OperatorID(); } }
/** * Verifies that each {@link JobVertexID} of the {@link JobGraph} is contained in the given map * and mapped to the same vertex name. */ private void verifyIdsEqual(JobGraph jobGraph, Map<JobVertexID, String> ids) { // Verify same number of vertices assertEquals(jobGraph.getNumberOfVertices(), ids.size()); // Verify that all IDs->name mappings are identical for (JobVertex vertex : jobGraph.getVertices()) { String expectedName = ids.get(vertex.getID()); assertNotNull(expectedName); assertEquals(expectedName, vertex.getName()); } }
@Override public JobSubmissionResult submitJob(JobGraph jobGraph, ClassLoader classLoader) throws ProgramInvocationException { if (isDetached()) { if (newlyCreatedCluster) { stopAfterJob(jobGraph.getJobID()); } return super.runDetached(jobGraph, classLoader); } else { return super.run(jobGraph, classLoader); } }
/** * Returns a {@link JobVertexID} to vertex name mapping for the given graph. */ private Map<JobVertexID, String> rememberIds(JobGraph jobGraph) { final Map<JobVertexID, String> ids = new HashMap<>(); for (JobVertex vertex : jobGraph.getVertices()) { ids.put(vertex.getID(), vertex.getName()); } return ids; }
/** * Verifies that no {@link JobVertexID} of the {@link JobGraph} is contained in the given map. */ private void verifyIdsNotEqual(JobGraph jobGraph, Map<JobVertexID, String> ids) { // Verify same number of vertices assertEquals(jobGraph.getNumberOfVertices(), ids.size()); // Verify that all IDs->name mappings are identical for (JobVertex vertex : jobGraph.getVertices()) { assertFalse(ids.containsKey(vertex.getID())); } }
@Override JobGraph validateGraph() { JobGraph jobGraph = super.validateGraph(); final SavepointRestoreSettings savepointRestoreSettings = jobGraph.getSavepointRestoreSettings(); Assert.assertTrue(savepointRestoreSettings.allowNonRestoredState()); Assert.assertEquals(RESTORE_PATH, savepointRestoreSettings.getRestorePath()); return jobGraph; } }
@Override public JobVertexID getJobVertexId() { return new JobVertexID(new byte[16]); }
private CompletableFuture<JobGraph> getJobGraphAsync( JarHandlerContext context, final SavepointRestoreSettings savepointRestoreSettings) { return CompletableFuture.supplyAsync(() -> { final JobGraph jobGraph = context.toJobGraph(configuration); jobGraph.setSavepointRestoreSettings(savepointRestoreSettings); return jobGraph; }, executor); } }
public JobSubmissionResult run( FlinkPlan compiledPlan, List<URL> libraries, List<URL> classpaths, ClassLoader classLoader) throws ProgramInvocationException { return run(compiledPlan, libraries, classpaths, classLoader, SavepointRestoreSettings.none()); }
public void setOperatorID(OperatorID operatorID) { this.config.setBytes(OPERATOR_ID, operatorID.getBytes()); }
@Override JobGraph validateDefaultGraph() { JobGraph jobGraph = super.validateDefaultGraph(); final SavepointRestoreSettings savepointRestoreSettings = jobGraph.getSavepointRestoreSettings(); Assert.assertFalse(savepointRestoreSettings.allowNonRestoredState()); Assert.assertNull(savepointRestoreSettings.getRestorePath()); return jobGraph; }
public AbstractStreamOperatorTestHarness( StreamOperator<OUT> operator, int maxParallelism, int parallelism, int subtaskIndex) throws Exception { this( operator, maxParallelism, parallelism, subtaskIndex, new OperatorID()); }
public JobSubmissionResult run(JobWithJars program, int parallelism) throws ProgramInvocationException { return run(program, parallelism, SavepointRestoreSettings.none()); }
@Override public OperatorID getOperatorID() { return new OperatorID(); } }
public OneInputStreamOperatorTestHarness( OneInputStreamOperator<IN, OUT> operator, int maxParallelism, int parallelism, int subtaskIndex) throws Exception { this(operator, maxParallelism, parallelism, subtaskIndex, new OperatorID()); }
@Override public OperatorID getOperatorID() { return new OperatorID(); }