org.kie.server.services.jbpm.QueryDataServiceBase Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of kie-server-services-jbpm Show documentation
Show all versions of kie-server-services-jbpm Show documentation
KIE jBPM Execution Server Extension
/*
* Copyright 2016 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.kie.server.services.jbpm;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.jbpm.kie.services.impl.query.SqlQueryDefinition;
import org.jbpm.services.api.model.ProcessInstanceDesc;
import org.jbpm.services.api.model.ProcessInstanceWithVarsDesc;
import org.jbpm.services.api.model.UserTaskInstanceDesc;
import org.jbpm.services.api.model.UserTaskInstanceWithVarsDesc;
import org.jbpm.services.api.query.QueryAlreadyRegisteredException;
import org.jbpm.services.api.query.QueryMapperRegistry;
import org.jbpm.services.api.query.QueryNotFoundException;
import org.jbpm.services.api.query.QueryParamBuilderFactory;
import org.jbpm.services.api.query.QueryResultMapper;
import org.jbpm.services.api.query.QueryService;
import org.jbpm.services.api.query.model.QueryParam;
import org.kie.api.runtime.query.QueryContext;
import org.kie.api.task.model.TaskSummary;
import org.kie.server.api.model.definition.QueryDefinition;
import org.kie.server.api.model.definition.QueryDefinitionList;
import org.kie.server.api.model.definition.QueryFilterSpec;
import org.kie.server.services.api.KieServerRegistry;
import org.kie.server.services.impl.marshal.MarshallerHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.kie.server.services.jbpm.ConvertUtils.*;
public class QueryDataServiceBase {
public static final Logger logger = LoggerFactory.getLogger(QueryDataServiceBase.class);
private QueryService queryService;
private MarshallerHelper marshallerHelper;
public QueryDataServiceBase(QueryService queryService, KieServerRegistry context) {
this.queryService = queryService;
this.marshallerHelper = new MarshallerHelper(context);
}
public void registerQuery(String queryName, String payload, String marshallingType) throws QueryAlreadyRegisteredException {
logger.debug("About to unmarshal queryDefinition from payload: '{}'", payload);
QueryDefinition queryDefinition = marshallerHelper.unmarshal(payload, marshallingType, QueryDefinition.class);
queryDefinition.setName(queryName);
SqlQueryDefinition actualDefinition = build(queryDefinition);
logger.debug("Built sql query definition for {} with content {}", queryName, actualDefinition);
queryService.registerQuery(actualDefinition);
}
public void replaceQuery(String queryName, String payload, String marshallingType) {
logger.debug("About to unmarshal queryDefinition from payload: '{}'", payload);
QueryDefinition queryDefinition = marshallerHelper.unmarshal(payload, marshallingType, QueryDefinition.class);
queryDefinition.setName(queryName);
SqlQueryDefinition actualDefinition = build(queryDefinition);
logger.debug("Built sql query definition for {} with content {}", queryName, actualDefinition);
queryService.replaceQuery(actualDefinition);
}
public void unregisterQuery(String uniqueQueryName) throws QueryNotFoundException {
queryService.unregisterQuery(uniqueQueryName);
}
public QueryDefinition getQuery(String uniqueQueryName) throws QueryNotFoundException {
org.jbpm.services.api.query.model.QueryDefinition query = queryService.getQuery(uniqueQueryName);
return convertQueryDefinition(query);
}
public QueryDefinitionList getQueries(Integer page, Integer pageSize) throws QueryNotFoundException {
List queries = queryService.getQueries(buildQueryContext(page, pageSize));
return convertToQueryDefinitionList(queries);
}
public Object query(String queryName, String mapper, String orderBy, Integer page, Integer pageSize) {
QueryResultMapper resultMapper = QueryMapperRegistry.get().mapperFor(mapper, null);
QueryContext queryContext = buildQueryContext(page, pageSize);
if (orderBy != null && !orderBy.isEmpty()) {
queryContext.setOrderBy(orderBy);
queryContext.setAscending(true);
}
logger.debug("About to perform query '{}' with sort {} and page {} and page size {}", queryName, orderBy, page, pageSize);
Object result = queryService.query(queryName, resultMapper, queryContext);
logger.debug("Result returned from the query {} mapped with {}", result, resultMapper);
return transform(result, resultMapper);
}
public Object queryFiltered(String queryName, String mapper, Integer page, Integer pageSize, String payload, String marshallingType) {
QueryParam[] params = new QueryParam[0];
Map columnMapping = null;
QueryContext queryContext = buildQueryContext(page, pageSize);
if (payload != null && !payload.isEmpty()) {
logger.debug("About to unmarshal queryDefinition from payload: '{}'", payload);
QueryFilterSpec filterSpec = marshallerHelper.unmarshal(payload, marshallingType, QueryFilterSpec.class);
queryContext.setOrderBy(filterSpec.getOrderBy());
queryContext.setAscending(filterSpec.isAscending());
// build parameters for filtering the query
if (filterSpec.getParameters() != null) {
params = new QueryParam[filterSpec.getParameters().length];
int index = 0;
for (org.kie.server.api.model.definition.QueryParam param : filterSpec.getParameters()) {
params[index] = new QueryParam(param.getColumn(), param.getOperator(), param.getValue());
index++;
}
}
columnMapping = filterSpec.getColumnMapping();
}
QueryResultMapper resultMapper = QueryMapperRegistry.get().mapperFor(mapper, columnMapping);
logger.debug("About to perform query '{}' with page {} and page size {}", queryName, page, pageSize);
Object result = queryService.query(queryName, resultMapper, queryContext, params);
logger.debug("Result returned from the query {} mapped with {}", result, resultMapper);
return transform(result, resultMapper);
}
public Object queryFilteredWithBuilder(String queryName, String mapper, String builder, Integer page, Integer pageSize, String payload, String marshallingType) {
QueryResultMapper resultMapper = QueryMapperRegistry.get().mapperFor(mapper, null);
QueryParamBuilderFactory paramBuilderFactory = QueryParamBuilderManager.get().find(builder);
if (paramBuilderFactory == null) {
new RuntimeException("No query param builder found for " + builder);
}
QueryContext queryContext = buildQueryContext(page, pageSize);
Map queryParameters = new HashMap();
if (payload != null && !payload.isEmpty()) {
logger.debug("About to unmarshal query params from payload: '{}'", payload);
queryParameters = marshallerHelper.unmarshal(payload, marshallingType, Map.class);
}
logger.debug("About to perform query '{}' with page {} and page size {}", queryName, page, pageSize);
Object result = queryService.query(queryName, resultMapper, queryContext, paramBuilderFactory.newInstance(queryParameters));
logger.debug("Result returned from the query {} mapped with {}", result, resultMapper);
return transform(result, resultMapper);
}
/*
* helper methods
*/
protected Object transform(Object result, QueryResultMapper resultMapper) {
Object actualResult = null;
if (result instanceof Collection) {
if (ProcessInstanceWithVarsDesc.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of ProcessInstanceWithVarsDesc to ProcessInstanceList");
actualResult = convertToProcessInstanceWithVarsList((Collection) result);
} else if (ProcessInstanceDesc.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of ProcessInstanceDesc to ProcessInstanceList");
actualResult = convertToProcessInstanceList((Collection) result);
} else if (UserTaskInstanceWithVarsDesc.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of UserTaskInstanceWithVarsDesc to TaskInstanceList");
actualResult = convertToTaskInstanceWithVarsList((Collection) result);
} else if (UserTaskInstanceDesc.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of UserTaskInstanceDesc to TaskInstanceList");
actualResult = convertToTaskInstanceList((Collection) result);
} else if (TaskSummary.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of TaskSummary to TaskSummaryList");
actualResult = convertToTaskSummaryList((Collection) result);
}
else if (List.class.isAssignableFrom(resultMapper.getType())) {
logger.debug("Converting collection of List to ArrayList");
actualResult = new ArrayList((Collection)result);
}
else {
logger.debug("Convert not supported for custom type {}", resultMapper.getType());
actualResult = result;
}
logger.debug("Actual result after converting is {}", actualResult);
} else {
logger.debug("Result is not a collection - {}, skipping any conversion", result);
actualResult = result;
}
return actualResult;
}
protected SqlQueryDefinition build(QueryDefinition queryDefinition) {
SqlQueryDefinition actualDefinition = new SqlQueryDefinition(queryDefinition.getName(), queryDefinition.getSource());
actualDefinition.setExpression(queryDefinition.getExpression());
actualDefinition.setTarget(org.jbpm.services.api.query.model.QueryDefinition.Target.valueOf(queryDefinition.getTarget()));
return actualDefinition;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy