Search in sources :

Example 1 with DataSetLookupException

use of org.dashbuilder.dataset.exception.DataSetLookupException in project jbpm by kiegroup.

the class QueryServiceImpl method replaceQuery.

@Override
public void replaceQuery(QueryDefinition queryDefinition) {
    logger.debug("About to register {} query...", queryDefinition);
    if (queryDefinition instanceof SqlQueryDefinition) {
        SqlQueryDefinition sqlQueryDefinition = (SqlQueryDefinition) queryDefinition;
        SQLDataSetDefBuilder<?> builder = DataSetDefFactory.newSQLDataSetDef().uuid(sqlQueryDefinition.getName()).name(sqlQueryDefinition.getName() + "::" + sqlQueryDefinition.getTarget().toString()).dataSource(sqlQueryDefinition.getSource()).dbSQL(sqlQueryDefinition.getExpression(), true);
        DataSetDef sqlDef = builder.buildDef();
        try {
            dataSetDefRegistry.registerDataSetDef(sqlDef);
            DataSetMetadata metadata = dataSetManager.getDataSetMetadata(sqlDef.getUUID());
            if (queryDefinition.getTarget().equals(Target.BA_TASK)) {
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new BusinessAdminTasksPreprocessor(identityProvider, metadata));
            } else if (queryDefinition.getTarget().equals(Target.PO_TASK)) {
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new PotOwnerTasksPreprocessor(identityProvider, metadata));
            } else if (queryDefinition.getTarget().equals(Target.FILTERED_PROCESS)) {
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new DeploymentIdsPreprocessor(deploymentRolesManager, identityProvider, COLUMN_EXTERNALID));
            } else if (queryDefinition.getTarget().equals(Target.FILTERED_BA_TASK)) {
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new BusinessAdminTasksPreprocessor(identityProvider, metadata));
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new DeploymentIdsPreprocessor(deploymentRolesManager, identityProvider, COLUMN_DEPLOYMENTID));
            } else if (queryDefinition.getTarget().equals(Target.FILTERED_PO_TASK)) {
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new PotOwnerTasksPreprocessor(identityProvider, metadata));
                dataSetDefRegistry.registerPreprocessor(sqlDef.getUUID(), new DeploymentIdsPreprocessor(deploymentRolesManager, identityProvider, COLUMN_DEPLOYMENTID));
            }
            for (String columnId : metadata.getColumnIds()) {
                logger.debug("Column {} is of type {}", columnId, metadata.getColumnType(columnId));
                sqlDef.addColumn(columnId, metadata.getColumnType(columnId));
            }
            logger.info("Registered {} query successfully", queryDefinition.getName());
        } catch (DataSetLookupException e) {
            unregisterQuery(queryDefinition.getName());
            throw e;
        }
    }
}
Also used : DataSetDef(org.dashbuilder.dataset.def.DataSetDef) BusinessAdminTasksPreprocessor(org.jbpm.kie.services.impl.query.preprocessor.BusinessAdminTasksPreprocessor) DataSetLookupException(org.dashbuilder.dataset.exception.DataSetLookupException) PotOwnerTasksPreprocessor(org.jbpm.kie.services.impl.query.preprocessor.PotOwnerTasksPreprocessor) DeploymentIdsPreprocessor(org.jbpm.kie.services.impl.query.preprocessor.DeploymentIdsPreprocessor) DataSetMetadata(org.dashbuilder.dataset.DataSetMetadata)

Aggregations

DataSetMetadata (org.dashbuilder.dataset.DataSetMetadata)1 DataSetDef (org.dashbuilder.dataset.def.DataSetDef)1 DataSetLookupException (org.dashbuilder.dataset.exception.DataSetLookupException)1 BusinessAdminTasksPreprocessor (org.jbpm.kie.services.impl.query.preprocessor.BusinessAdminTasksPreprocessor)1 DeploymentIdsPreprocessor (org.jbpm.kie.services.impl.query.preprocessor.DeploymentIdsPreprocessor)1 PotOwnerTasksPreprocessor (org.jbpm.kie.services.impl.query.preprocessor.PotOwnerTasksPreprocessor)1