com.capitalone.dashboard.service.PerformanceServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Hygieia Rest API Layer
There is a newer version: 3.4.53
package com.capitalone.dashboard.service;

import com.capitalone.dashboard.misc.HygieiaException;
import com.capitalone.dashboard.model.Collector;
import com.capitalone.dashboard.model.CollectorItem;
import com.capitalone.dashboard.model.CollectorType;
import com.capitalone.dashboard.model.Component;
import com.capitalone.dashboard.model.DataResponse;
import com.capitalone.dashboard.model.Performance;
import com.capitalone.dashboard.model.PerformanceType;
import com.capitalone.dashboard.model.QPerformance;
import com.capitalone.dashboard.repository.CollectorRepository;
import com.capitalone.dashboard.repository.ComponentRepository;
import com.capitalone.dashboard.repository.PerformanceRepository;
import com.capitalone.dashboard.request.CollectorRequest;
import com.capitalone.dashboard.request.PerformanceCreateRequest;
import com.capitalone.dashboard.request.PerformanceSearchRequest;
import com.google.common.base.MoreObjects;
import com.google.common.base.Objects;
import com.google.common.collect.Iterables;
import com.querydsl.core.BooleanBuilder;
import org.apache.commons.lang3.StringUtils;
import org.bson.types.ObjectId;
import org.joda.time.LocalDate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class PerformanceServiceImpl implements PerformanceService {

    private final PerformanceRepository performanceRepository;
    private final ComponentRepository componentRepository;
    private final CollectorRepository collectorRepository;
    private final CollectorService collectorService;

    @Autowired
    public PerformanceServiceImpl(PerformanceRepository performanceRepository,
                                  ComponentRepository componentRepository,
                                  CollectorRepository collectorRepository,
                                  CollectorService collectorService) {
        this.performanceRepository = performanceRepository;
        this.componentRepository = componentRepository;
        this.collectorRepository = collectorRepository;
        this.collectorService = collectorService;
    }

    @Override
    public DataResponse> search(PerformanceSearchRequest request) {
        if (request == null) {
            return emptyResponse();
        }

        if (request.getType() == null) {
            return emptyResponse();
        }

        return searchType(request);
    }

    private DataResponse> emptyResponse() {
        return new DataResponse<>(null, System.currentTimeMillis());
    }

    private DataResponse> searchType(PerformanceSearchRequest request) {
        CollectorItem item = getCollectorItem(request);
        if (item == null) {
            return emptyResponse();
        }

        QPerformance performance = new QPerformance("performance");
        BooleanBuilder builder = new BooleanBuilder();

        builder.and(performance.collectorItemId.eq(item.getId()));

        if (request.getNumberOfDays() != null) {
            long endTimeTarget =
                    new LocalDate().minusDays(request.getNumberOfDays()).toDate().getTime();
            builder.and(performance.timestamp.goe(endTimeTarget));
        } else if (request.validDateRange()) {
            builder.and(performance.timestamp.between(request.getDateBegins(), request.getDateEnds()));
        }
        Iterable result;
        if (request.getMax() == null) {
            result = performanceRepository.findAll(builder.getValue(), performance.timestamp.desc());
        } else {
            PageRequest pageRequest =
                    new PageRequest(0, request.getMax(), Sort.Direction.DESC, "timestamp");
            result = performanceRepository.findAll(builder.getValue(), pageRequest).getContent();
        }
        Collector collector = collectorRepository.findOne(item.getCollectorId());
        long lastExecuted = (collector == null) ? 0 : collector.getLastExecuted();
        return new DataResponse<>(result, lastExecuted);
    }

    protected CollectorItem getCollectorItem(PerformanceSearchRequest request) {
        Component component = componentRepository.findOne(request.getComponentId());
        if (component == null) {
            return null;
        }

        CollectorItem item = null;
        PerformanceType qualityType = MoreObjects.firstNonNull(request.getType(),
                PerformanceType.ApplicationPerformance);
        List items = component.getCollectorItems().get(qualityType.collectorType());
        if (items != null) {
            item = Iterables.getFirst(items, null);
        }
        return item;
    }


    @Override
    public String create(PerformanceCreateRequest request) throws HygieiaException {
        /**
         * Step 1: create Collector if not there
         * Step 2: create Collector item if not there
         * Step 3: Insert Quality data if new. If existing, update it.
         */
        Collector collector = createCollector(request.getCollectorName());

        if (collector == null) {
            throw new HygieiaException("Failed creating code quality collector.", HygieiaException.COLLECTOR_CREATE_ERROR);
        }

        CollectorItem collectorItem = createCollectorItem(collector, request);

        if (collectorItem == null) {
            throw new HygieiaException("Failed creating code quality collector item.", HygieiaException.COLLECTOR_ITEM_CREATE_ERROR);
        }

        Performance quality = createPerformance(collectorItem, request);

        if (quality == null) {
            throw new HygieiaException("Failed inserting/updating Quality information.", HygieiaException.ERROR_INSERTING_DATA);
        }

        return quality.getId().toString() + "," + quality.getCollectorItemId().toString();

    }


    private Collector createCollector(String collectorName) {
        CollectorRequest collectorReq = new CollectorRequest();
        collectorReq.setName(collectorName);  //for now hardcode it.
        collectorReq.setCollectorType(CollectorType.AppPerformance);
        Collector col = collectorReq.toCollector();
        col.setEnabled(true);
        col.setOnline(true);
        col.setLastExecuted(System.currentTimeMillis());
        return collectorService.createCollector(col);
    }

    private CollectorItem createCollectorItem(Collector collector, PerformanceCreateRequest request) throws HygieiaException {
        CollectorItem tempCi = new CollectorItem();
        tempCi.setCollectorId(collector.getId());
        tempCi.setDescription(request.getProjectName());
        tempCi.setPushed(true);
        tempCi.setLastUpdated(System.currentTimeMillis());
        Map option = new HashMap<>();
        option.put("projectName", request.getProjectName());
        option.put("projectId", request.getProjectId());
        option.put("instanceUrl", request.getServerUrl());
        tempCi.getOptions().putAll(option);
        tempCi.setNiceName(request.getNiceName());

        if (StringUtils.isEmpty(tempCi.getNiceName())) {
            return collectorService.createCollectorItem(tempCi);
        }
        return collectorService.createCollectorItemByNiceNameAndProjectId(tempCi, request.getProjectId());
    }

    private Performance createPerformance(CollectorItem collectorItem, PerformanceCreateRequest request) {
        Performance performance = performanceRepository.findByCollectorItemIdAndTimestamp(
                collectorItem.getId(), request.getTimestamp());
        if (performance == null) {
            performance = new Performance();
        }
        ObjectId hygieiaId = request.getHygieiaId() != null && !request.getHygieiaId().isEmpty() ? new ObjectId(request.getHygieiaId()) : new ObjectId();
        performance.setCollectorItemId(collectorItem.getId());
        performance.setExecutionId(hygieiaId);
        performance.setType(PerformanceType.ApplicationPerformance);
        performance.setUrl(request.getProjectUrl());
        performance.setVersion(request.getProjectVersion());
        performance.setTimestamp(System.currentTimeMillis());
        performance.setTargetAppName(request.getTargetAppName());
        performance.setTargetEnvName(request.getTargetEnvName());
        performance.setMetrics(new HashMap<>(request.getMetrics()));
        return performanceRepository.save(performance); // Save = Update (if ID present) or Insert (if ID not there)
    }

}