Examples with DataSource org.motechproject.tasks.domain.mds.task.DataSource used on opensource projects

Example 6 with DataSource

use of org.motechproject.tasks.domain.mds.task.DataSource in project motech by motech.

the class TaskInitializer method evalConfigSteps.

/**
 * Executes all config steps (loading data from data sources, checking filters) defined for this task.
 *
 * @param dataProviders  the map of data providers, not null or empty
 * @return  true if all steps were executed, false otherwise
 * @throws TaskHandlerException if there were error while handling task
 */
@Transactional
public boolean evalConfigSteps(Map<String, DataProvider> dataProviders) throws TaskHandlerException {
    LOGGER.info("Executing all config steps for task: {}", taskContext.getTask().getName());
    Iterator<TaskConfigStep> iterator = taskContext.getTask().getTaskConfig().getSteps().iterator();
    boolean result = true;
    TaskFilterExecutor taskFilterExecutor = new TaskFilterExecutor();
    while (result && iterator.hasNext()) {
        TaskConfigStep step = iterator.next();
        if (step instanceof DataSource) {
            DataSource ds = (DataSource) step;
            taskContext.addDataSourceObject(ds.getObjectId().toString(), getDataSourceObject(ds, dataProviders), ds.isFailIfDataNotFound());
            LOGGER.info("Task data source: {} for task: {} added", ds.getName(), taskContext.getTask().getName());
        } else if (step instanceof FilterSet && !isActionFilter((FilterSet) step)) {
            try {
                FilterSet filterSet = (FilterSet) step;
                result = taskFilterExecutor.checkFilters(filterSet.getFilters(), filterSet.getOperator(), taskContext);
            } catch (RuntimeException e) {
                throw new TaskHandlerException(FILTER, "task.error.filterError", e);
            }
        }
    }
    return result;
}

Example 7 with DataSource

use of org.motechproject.tasks.domain.mds.task.DataSource in project motech by motech.

the class TaskServiceImpl method validateDataSources.

private Set<TaskError> validateDataSources(Task task) {
    LOGGER.debug("Validating task data sources in task: {} with ID: {}", task.getName(), task.getId());
    Set<TaskError> errors = new HashSet<>();
    Map<Long, TaskDataProvider> availableDataProviders = new HashMap<>();
    for (DataSource dataSource : task.getTaskConfig().getDataSources()) {
        TaskDataProvider provider = providerService.getProvider(dataSource.getProviderName());
        errors.addAll(validateProvider(provider, dataSource, task, availableDataProviders));
        if (provider != null) {
            availableDataProviders.put(provider.getId(), provider);
        }
    }
    logResultOfValidation("task data sources", task.getName(), errors);
    return errors;
}

Example 8 with DataSource

use of org.motechproject.tasks.domain.mds.task.DataSource in project motech by motech.

the class ProviderNameTaskMigrator method migrateDataSources.

private void migrateDataSources(Task task) {
    // replace data source references in other data source lookups
    for (DataSource dataSource : task.getTaskConfig().getDataSources()) {
        for (Lookup lookup : dataSource.getLookup()) {
            String oldVal = lookup.getValue();
            String newVal = replaceTaskValue(oldVal, EXPR_PATTERN, task);
            lookup.setValue(newVal);
        }
    }
}

Example 9 with DataSource

use of org.motechproject.tasks.domain.mds.task.DataSource in project motech by motech.

the class TaskConfigTest method shouldRemoveOnlyFilterSets.

@Test
public void shouldRemoveOnlyFilterSets() throws Exception {
    TaskConfig config = new TaskConfig().add(new FilterSet(), new FilterSet(), new FilterSet()).add(new DataSource(), new DataSource(), new DataSource());
    assertNotNull(config.getSteps());
    assertFalse(config.getSteps().isEmpty());
    config.removeFilterSets();
    assertNotNull(config.getSteps());
    assertFalse(config.getSteps().isEmpty());
    assertTrue(config.getFilters().isEmpty());
    assertFalse(config.getDataSources().isEmpty());
}

Example 10 with DataSource

use of org.motechproject.tasks.domain.mds.task.DataSource in project motech by motech.

the class TaskConfigTest method shouldAddStepAndSetNewOrderNumber.

@Test
public void shouldAddStepAndSetNewOrderNumber() throws Exception {
    List<TaskConfigStep> steps = new ArrayList<>();
    steps.add(new FilterSet());
    steps.add(new DataSource());
    steps.add(new DataSource());
    steps.add(new FilterSet());
    Random random = new Random();
    for (TaskConfigStep step : steps) {
        step.setOrder(random.nextInt());
    }
    TaskConfig config = new TaskConfig();
    for (TaskConfigStep step : steps) {
        config.add(step);
    }
    List<Integer> orders = new ArrayList<>(steps.size());
    for (TaskConfigStep step : config.getSteps()) {
        orders.add(step.getOrder());
    }
    assertThat(orders, hasItems(0, 1, 2, 3));
}