Java StepExecution.getPersistentUserData方法代码示例

import javax.batch.runtime.StepExecution; //导入方法依赖的package包/类
@Test
public void checkpointAlgorithm() throws Exception {
	logger.info("starting checkpoint test");
	CountDownLatch latch = new CountDownLatch(1);
	JobOperator jobOperator = getJobOperator();
	Properties props = new Properties();
	long executionId = jobOperator.start(JOB_NAME, props);
	latch.await(10, SECONDS);

	JobInstance jobInstance = jobInstance(jobOperator.getJobInstance(executionId), JOB_NAME);
	JobExecution jobExecution = jobOperator.getJobExecutions(jobInstance).get(0);
	Properties jobProperties = jobExecution.getJobParameters();
	assertEquals("properties are: ", 0, jobProperties.size());
	assertEquals("batch failed because a NoRollbackException is throwed", FAILED, jobExecution.getBatchStatus());
	StepExecution stepExecution = jobOperator.getStepExecutions(executionId).get(0);
	PersistentCheckpointUserData persistentCheckpointUserData = (PersistentCheckpointUserData) stepExecution
			.getPersistentUserData();
	assertNull("persistent user data after step", persistentCheckpointUserData.getStartedAfterStep());
	Metric[] stepMetrics = stepExecution.getMetrics();
	stepMetrics(stepMetrics, 3, 10, 12, 2, 1);
}
 

import javax.batch.runtime.StepExecution; //导入方法依赖的package包/类
private String coreExitStatus(StepExecution stepExec) {		
	String action = (String)stepExec.getPersistentUserData();
	String currentExitStatus = stepExec.getExitStatus();
	
	// "Normally" we just pass set 'normalExitStatus' as exit status.
	if (currentExitStatus.equals(GOOD_STEP_EXIT_STATUS)) {
		return action;			
	// But if it's the magic number then we return our "SpecialExitStatus".
	} else {
		return specialExitStatus;
	}		
}
 

import javax.batch.runtime.StepExecution; //导入方法依赖的package包/类
@Test
public void testPartitionPropertyResolverForCollector() throws Exception {
       final JobOperator op = BatchRuntime.getJobOperator();
       Properties jobParams = new Properties();
       jobParams.setProperty(STEP_PROP, STEP_PROP_VAL);
	final long id = op.start("partition-propertyResolver", jobParams);
       Batches.waitForEnd(op, id);

       StepExecution stepExecution = op.getStepExecutions(id).iterator().next();

	assertEquals(op.getJobExecution(id).getBatchStatus(), BatchStatus.COMPLETED);
       assertNotNull(op.getJobExecution(id).getJobParameters());
       assertEquals(jobParams, op.getJobExecution(id).getJobParameters());
       for (final JobExecution exec: op.getJobExecutions(new JobInstanceImpl(id))) {
           assertNotNull(exec.getJobParameters());
           assertEquals(jobParams, exec.getJobParameters());
       }

	ArrayList<String> data = (ArrayList<String>)stepExecution.getPersistentUserData();

	for (String nextStringifiedProps : data) {
		Properties props = PropertyHelper.stringToProperties(nextStringifiedProps);
		String valFromStepProp = props.getProperty(STEP_CONTEXT_PROPERTY);
		String valFromSubstitution = props.getProperty(SUBSTITUTION_PROPERTY);
		assertEquals(valFromStepProp, STEP_PROP_VAL, "Compare values from step-level property with param used in substitution");
		assertEquals(valFromSubstitution, STEP_PROP_VAL, "Compare values from step-level property with a collector-property using this step-level property via a 'jobProperties' substitution.");
	}
}
 

import javax.batch.runtime.StepExecution; //导入方法依赖的package包/类
@Test
@org.junit.Test
public void testUserDataIsPersistedAfterCheckpoint() throws Exception {

    String METHOD = "testChunkRestartItemCount10";
    try {
        Reporter.log("Create job parameters for execution #1:<p>");
        Properties jobParams = new Properties();
        Reporter.log("execution.number=1<p>");
        Reporter.log("readrecord.fail=12<p>");
        Reporter.log("app.arraysize=30<p>");
        Reporter.log("app.writepoints=0,5,10,15,20,25,30<p>");
        Reporter.log("app.next.writepoints=0,5,10,15,20,25,30<p>");
        jobParams.put("execution.number", "1");
        jobParams.put("readrecord.fail", "12");
        jobParams.put("app.arraysize", "30");
        jobParams.put("app.checkpoint.position" , "0");
        jobParams.put("app.writepoints", "0,10,20,30");
        jobParams.put("app.next.writepoints", "10,20,30");

        Reporter.log("Locate job XML file: chunkrestartCheckpt10.xml<p>");

        Reporter.log("Invoke startJobAndWaitForResult for execution #1<p>");
        TCKJobExecutionWrapper execution1 = jobOp.startJobAndWaitForResult("chunkrestartCheckpt10", jobParams);
        Reporter.log("execution #1 JobExecution getBatchStatus()=" + execution1.getBatchStatus() + "<p>");
        Reporter.log("execution #1 JobExecution getExitStatus()=" + execution1.getExitStatus() + "<p>");
        assertWithMessage("Testing execution #1", BatchStatus.FAILED, execution1.getBatchStatus());
        assertWithMessage("Testing execution #1", "FAILED", execution1.getExitStatus());
        
        List<StepExecution> stepExecs = jobOp.getStepExecutions(execution1.getExecutionId());
        
        MyPersistentRestartUserData persistedStepData = null;
        for (StepExecution stepExec : stepExecs) {
            if (stepExec.getStepName().equals("step1")) {
                persistedStepData = (MyPersistentRestartUserData) stepExec.getPersistentUserData();
                break;
            }
        }
        
        assertWithMessage("Testing execution #1", 1, persistedStepData.getExecutionNumber());
        

    } catch (Exception e) {
        handleException(METHOD, e);
    }

}