@PostConstruct public void init() { initializeRegistryRoot(); initializeRegistry(); }
@Before @Override public void setUp() { super.setUp(); when(registryHelper.ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH)).thenReturn(registryRoot); pipelineExecutorRegistry = spy(new VFSPipelineExecutorRegistry(registryHelper)); ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init(); }
@Test public void testInit() throws Exception { traces = new ArrayList<>(); for (int i = 0; i < TRACES_COUNT; i++) { PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class); when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i)); traces.add(trace); } when(registryHelper.readEntries(registryRoot, VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX))).thenReturn(traces); ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init(); verify(registryHelper, times(2)).ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH); verify(registryHelper, times(2)).readEntries(registryRoot, VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX)); for (Object trace : traces) { PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(((PipelineExecutorTrace) trace).getTaskId()); assertNotNull(result); assertEquals(trace, result); } }
@Override public void deregister(final String taskId) { checkNotNull("taskId", taskId); final Path path = buildTracePath(taskId); registryHelper.deleteBatch(path); super.deregister(taskId); }
@Before @Override public void setUp() { super.setUp(); when(registryHelper.ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH)).thenReturn(registryRoot); pipelineExecutorRegistry = spy(new VFSPipelineExecutorRegistry(registryHelper)); ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init(); }
@Test public void testInit() throws Exception { traces = new ArrayList<>(); for (int i = 0; i < TRACES_COUNT; i++) { PipelineExecutorTrace trace = mock(PipelineExecutorTrace.class); when(trace.getTaskId()).thenReturn(PIPELINE_EXECUTION_ID + Integer.toString(i)); traces.add(trace); } when(registryHelper.readEntries(registryRoot, VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX))).thenReturn(traces); ((VFSPipelineExecutorRegistry) pipelineExecutorRegistry).init(); verify(registryHelper, times(2)).ensureDirectory(PIPELINE_EXECUTOR_REGISTRY_PATH); verify(registryHelper, times(2)).readEntries(registryRoot, VFSRegistryHelper.BySuffixFilter.newFilter(TRACE_SUFFIX)); for (Object trace : traces) { PipelineExecutorTrace result = pipelineExecutorRegistry.getExecutorTrace(((PipelineExecutorTrace) trace).getTaskId()); assertNotNull(result); assertEquals(trace, result); } }
@Override public void register(final PipelineExecutorTrace trace) { checkNotNull("trace", trace); final Path path = buildTracePath(trace.getTaskId()); try { registryHelper.storeEntry(path, trace); } catch (Exception e) { //uncommon error logger.error("Unexpected error was produced during trace marshalling/storing, trace: " + trace, e); throw new RuntimeException("Unexpected error was produced during trace marshalling/storing, trace: " + trace, e); } super.register(trace); }