throw new SQLException(e); taskInfo = new TaskInfo<>( resultSet.getString("id"), DateTimes.of(resultSet.getString("created_date")),
private List<TaskInfo<Task, TaskStatus>> getNRecentlyCreatedAlreadyFinishedTaskInfo(int n, Ordering<TaskStuff> createdDateDesc) { giant.lock(); try { List<TaskStuff> list = createdDateDesc .sortedCopy(tasks.values()) .stream() .filter(taskStuff -> taskStuff.getStatus().isComplete()) .limit(n) .collect(Collectors.toList()); final ImmutableList.Builder<TaskInfo<Task, TaskStatus>> listBuilder = ImmutableList.builder(); for (final TaskStuff taskStuff : list) { String id = taskStuff.getTask().getId(); TaskInfo t = new TaskInfo<>( id, taskStuff.getCreatedDate(), taskStuff.getStatus(), taskStuff.getDataSource(), taskStuff.getTask() ); listBuilder.add(t); } return listBuilder.build(); } finally { giant.unlock(); } }
private List<TaskInfo<Task, TaskStatus>> getRecentlyCreatedAlreadyFinishedTaskInfoSince( DateTime start, Ordering<TaskStuff> createdDateDesc ) { giant.lock(); try { List<TaskStuff> list = createdDateDesc .sortedCopy(tasks.values()) .stream() .filter(taskStuff -> taskStuff.getStatus().isComplete() && taskStuff.createdDate.isAfter(start)) .collect(Collectors.toList()); final ImmutableList.Builder<TaskInfo<Task, TaskStatus>> listBuilder = ImmutableList.builder(); for (final TaskStuff taskStuff : list) { String id = taskStuff.getTask().getId(); TaskInfo t = new TaskInfo( id, taskStuff.getCreatedDate(), taskStuff.getStatus(), taskStuff.getDataSource(), taskStuff.getTask() ); listBuilder.add(t); } return listBuilder.build(); } finally { giant.unlock(); } }
@Nullable @Override public TaskInfo<Task, TaskStatus> getTaskInfo(String taskId) { giant.lock(); try { Preconditions.checkNotNull(taskId, "taskId"); final TaskStuff taskStuff = tasks.get(taskId); if (taskStuff != null) { return new TaskInfo<>( taskStuff.getTask().getId(), taskStuff.getCreatedDate(), taskStuff.getStatus(), taskStuff.getDataSource(), taskStuff.getTask() ); } else { return null; } } finally { giant.unlock(); } }
@Override public List<TaskInfo<Task, TaskStatus>> getActiveTaskInfo(@Nullable String dataSource) { giant.lock(); try { final ImmutableList.Builder<TaskInfo<Task, TaskStatus>> listBuilder = ImmutableList.builder(); for (final TaskStuff taskStuff : tasks.values()) { if (taskStuff.getStatus().isRunnable()) { TaskInfo t = new TaskInfo( taskStuff.getTask().getId(), taskStuff.getCreatedDate(), taskStuff.getStatus(), taskStuff.getDataSource(), taskStuff.getTask() ); listBuilder.add(t); } } return listBuilder.build(); } finally { giant.unlock(); } }
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, duration, null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "deny") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()),
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "deny") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()),
@Test public void testDeletePendingSegments() { final List<TaskInfo<Task, TaskStatus>> taskInfos = ImmutableList.of( new TaskInfo<>( "id1", DateTimes.of("2017-12-01"), TaskStatus.running("id1"), "dataSource", NoopTask.create("id1", 0) ), new TaskInfo<>( "id1", DateTimes.of("2017-12-02"), TaskStatus.running("id2"), "dataSource", NoopTask.create("id2", 0) ) ); EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("dataSource")).andReturn(taskInfos); final Interval deleteInterval = Intervals.of("2017-01-01/2017-12-01"); EasyMock .expect( indexerMetadataStorageCoordinator.deletePendingSegments(EasyMock.anyString(), EasyMock.eq(deleteInterval)) ) .andReturn(10); EasyMock.replay(taskStorageQueryAdapter, indexerMetadataStorageCoordinator); Assert.assertEquals(10, indexerMetadataStorageAdapter.deletePendingSegments("dataSource", deleteInterval)); }
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("allow")).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_3", "allow") ), new TaskInfo( "id_4", DateTime.now(ISOChronology.getInstanceUTC()),
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "deny") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_3", "allow") ), new TaskInfo( "id_4", DateTime.now(ISOChronology.getInstanceUTC()),
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), null ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "deny") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()),
).anyTimes(); EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("datasource")).andStubReturn(ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "datasource") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()),
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_3", "deny") ), new TaskInfo( "id_4", DateTime.now(ISOChronology.getInstanceUTC()),
new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo( "id_3", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_3", "deny") ), new TaskInfo( "id_4", DateTime.now(ISOChronology.getInstanceUTC()),
new TaskInfo( "id_5", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_5", "allow") ), new TaskInfo( "id_6", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_6", "allow") ), new TaskInfo( "id_7", DateTime.now(ISOChronology.getInstanceUTC()), new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo(
new TaskInfo( "id_5", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_5", "deny") ), new TaskInfo( "id_6", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_6", "allow") ), new TaskInfo( "id_7", DateTime.now(ISOChronology.getInstanceUTC()), new TaskInfo( "id_1", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_1", "allow") ), new TaskInfo( "id_2", DateTime.now(ISOChronology.getInstanceUTC()), getTaskWithIdAndDatasource("id_2", "allow") ), new TaskInfo(
.andReturn(new TaskInfo<>(task.getId(), DateTimes.of("2018-01-01"), status, task.getDataSource(), task));