org.camunda.bpm.engine.rest.sub.metrics.MetricsResourceImpl Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of camunda-engine-rest-core Show documentation
There is a newer version: 7.22.0-alpha3
Show newest version
/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH
 * under one or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information regarding copyright
 * ownership. Camunda licenses this file to you under the Apache License,
 * Version 2.0; you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     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.camunda.bpm.engine.rest.sub.metrics;

import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.Date;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.UriInfo;

import org.camunda.bpm.engine.ProcessEngine;
import org.camunda.bpm.engine.management.Metrics;
import org.camunda.bpm.engine.management.MetricsQuery;
import org.camunda.bpm.engine.rest.dto.converter.DateConverter;
import org.camunda.bpm.engine.rest.dto.metrics.MetricsResultDto;


/**
 * @author Daniel Meyer
 *
 */
public class MetricsResourceImpl implements MetricsResource {

  protected String metricsName;
  protected ProcessEngine processEngine;
  protected ObjectMapper objectMapper;

  public MetricsResourceImpl(String metricsName, ProcessEngine processEngine, ObjectMapper objectMapper) {
    this.metricsName = metricsName;
    this.processEngine = processEngine;
    this.objectMapper = objectMapper;
  }

  @Override
  public MetricsResultDto sum(UriInfo uriInfo) {
    MultivaluedMap queryParameters = uriInfo.getQueryParameters();

    DateConverter dateConverter = new DateConverter();
    dateConverter.setObjectMapper(objectMapper);

    Number result = null;

    if (Metrics.UNIQUE_TASK_WORKERS.equals(metricsName)) {
      result = processEngine.getManagementService().getUniqueTaskWorkerCount(
          extractStartDate(queryParameters, dateConverter),
          extractEndDate(queryParameters, dateConverter));
    } else {
      MetricsQuery query = processEngine.getManagementService()
        .createMetricsQuery()
        .name(metricsName);

      applyQueryParams(queryParameters, dateConverter, query);
      result = query.sum();
    }
    return new MetricsResultDto(result);
  }

  protected void applyQueryParams(MultivaluedMap queryParameters, DateConverter dateConverter, MetricsQuery query) {
    Date startDate = extractStartDate(queryParameters, dateConverter);
    Date endDate = extractEndDate(queryParameters, dateConverter);
    if (startDate != null) {
      query.startDate(startDate);
    }
    if (endDate != null) {
      query.endDate(endDate);
    }
  }

  protected Date extractEndDate(MultivaluedMap queryParameters, DateConverter dateConverter) {
    if(queryParameters.getFirst("endDate") != null) {
      return dateConverter.convertQueryParameterToType(queryParameters.getFirst("endDate"));
    }
    return null;
  }

  protected Date extractStartDate(MultivaluedMap queryParameters, DateConverter dateConverter) {
    if(queryParameters.getFirst("startDate") != null) {
      return dateConverter.convertQueryParameterToType(queryParameters.getFirst("startDate"));
    }
    return null;
  }

}