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

Go to download
package com.capitalone.dashboard.service;

import com.capitalone.dashboard.config.collector.CloudConfig;
import com.capitalone.dashboard.misc.HygieiaException;
import com.capitalone.dashboard.model.CloudInstance;
import com.capitalone.dashboard.model.CloudInstanceHistory;
import com.capitalone.dashboard.model.CollectorItem;
import com.capitalone.dashboard.model.CollectorType;
import com.capitalone.dashboard.model.Component;
import com.capitalone.dashboard.model.NameValue;
import com.capitalone.dashboard.repository.CloudInstanceHistoryRepository;
import com.capitalone.dashboard.repository.CloudInstanceRepository;
import com.capitalone.dashboard.repository.ComponentRepository;
import com.capitalone.dashboard.request.CloudInstanceCreateRequest;
import com.capitalone.dashboard.request.CloudInstanceListRefreshRequest;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

@Service
public class CloudInstanceServiceImpl implements CloudInstanceService {

    private final CloudInstanceRepository cloudInstanceRepository;
    private final CloudInstanceHistoryRepository cloudInstanceHistoryRepository;
    private final ComponentRepository componentRepository;

    @Autowired
    public CloudInstanceServiceImpl(CloudInstanceRepository cloudInstanceRepository,
                                    CloudInstanceHistoryRepository cloudInstanceHistoryRepository,
                                    ComponentRepository cloudConfigRepository) {
        this.cloudInstanceRepository = cloudInstanceRepository;
        this.cloudInstanceHistoryRepository = cloudInstanceHistoryRepository;
        this.componentRepository = cloudConfigRepository;
    }

    public Collection getInstanceDetails(CollectorItem item) {
        Collection instances = new HashSet<>();
        if ((item != null) && (item instanceof CloudConfig)) {
            CloudConfig config = (CloudConfig) item;
            instances.addAll(getInstanceDetailsByTags(config.getTags()));
        }
        return instances;
    }

    private CollectorItem getCollectorItem(ObjectId componentId) {
        Component component = componentRepository.findOne(componentId);
        if (CollectionUtils.isEmpty(component.getCollectorItems())) return null;
        return component.getCollectorItems().get(CollectorType.Cloud).get(0);
    }

    @Override
    public Collection getInstanceDetailsByComponentId(String componentIdString) {
        return getInstanceDetails(getCollectorItem(new ObjectId(componentIdString)));
    }

    @Override
    public CloudInstance getInstanceDetailsByInstanceId(String instanceId) {
        return cloudInstanceRepository.findByInstanceId(instanceId);
    }

    @Override
    public Collection getInstanceDetailsByInstanceIds(List instanceIds) {
        return cloudInstanceRepository.findByInstanceIdIn(instanceIds);
    }

    @Override
    public Collection getInstanceDetailsByTags(List tags) {
        Set instances = new HashSet<>();
        for (NameValue nv : tags) {
            instances.addAll(cloudInstanceRepository.findByTagNameAndValue(nv.getName(), nv.getValue()));
        }
        return instances;
    }

    @Override
    public Collection getInstanceDetailsByAccount(String accountNumber) {
        return cloudInstanceRepository.findByAccountNumber(accountNumber);
    }


    @Override
    public Collection getInstanceHistoryByAccount(String account) {
        return cloudInstanceHistoryRepository.findByAccountNumber(account);
    }

    @Override
    public Collection refreshInstances(CloudInstanceListRefreshRequest request) {
        Collection existing = cloudInstanceRepository.findByAccountNumber(request.getAccountNumber());
        Set toDelete = new HashSet<>();
        Set deletedIds = new HashSet<>();
        if (CollectionUtils.isEmpty(request.getInstanceIds()) || CollectionUtils.isEmpty(existing))
            return new ArrayList<>();

        for (CloudInstance ci : existing) {
            if (!request.getInstanceIds().contains(ci.getInstanceId())) {
                toDelete.add(ci);
                deletedIds.add(ci.getInstanceId());
            }
        }
        if (CollectionUtils.isEmpty(toDelete)) {
            cloudInstanceRepository.delete(toDelete);
        }
        return deletedIds;
    }



    private CloudInstance upsertCloudInstanceObject(CloudInstanceCreateRequest request, CloudInstance existing) throws HygieiaException {

        CloudInstance instance = (existing == null) ? new CloudInstance() : existing;
        if (StringUtils.isEmpty(request.getAccountNumber()) || (StringUtils.isEmpty(request.getInstanceId()))) {
            throw new HygieiaException("Missing required fields (account number, instance id). ", HygieiaException.ERROR_INSERTING_DATA);
        }

        instance.setAccountNumber(request.getAccountNumber());
        instance.setInstanceId(request.getInstanceId());

        //Anything null or resulting in parsing error will be thrown back to caller.
        if (request.getRootDeviceName() != null) instance.setRootDeviceName(request.getRootDeviceName());
        if (request.getCpuUtilization() != null)
            instance.setCpuUtilization(Double.parseDouble(request.getCpuUtilization()));
        if (request.getVirtualNetworkId() != null) instance.setVirtualNetworkId(request.getVirtualNetworkId());
        if (request.getSubnetId() != null) instance.setSubnetId(request.getSubnetId());
        if (request.getStatus() != null) instance.setStatus(request.getStatus());
        if (request.getAge() != null) instance.setAge(Integer.parseInt(request.getAge()));
        if (request.getDiskRead() != null) instance.setDiskRead(Double.parseDouble(request.getDiskRead()));
        if (request.getDiskWrite() != null) instance.setDiskWrite(Double.parseDouble(request.getDiskWrite()));
        if (request.getImageApproved() != null)
            instance.setImageApproved(Boolean.parseBoolean(request.getImageApproved()));
        if (request.getImageId() != null) instance.setImageId(request.getImageId());
        if (request.getImageExpirationDate() != null)
            instance.setImageExpirationDate(Long.parseLong(request.getImageExpirationDate()));

        if (request.getInstanceOwner() != null) instance.setInstanceOwner(request.getInstanceOwner());
        if (request.getInstanceType() != null) instance.setInstanceType(request.getInstanceType());
        if (request.getLastAction() != null) instance.setLastAction(request.getLastAction());
        if (request.getIsMonitored() != null) instance.setIsMonitored(Boolean.parseBoolean(request.getIsMonitored()));
        if (request.getNetworkIn() != null) instance.setNetworkIn(Double.parseDouble(request.getNetworkIn()));
        if (request.getNetworkOut() != null) instance.setNetworkOut(Double.parseDouble(request.getNetworkOut()));
        if (request.getLastUpdatedDate() != null)
            instance.setLastUpdatedDate(Long.parseLong(request.getLastUpdatedDate()));
        if (request.getPrivateDns() != null) instance.setPrivateDns(request.getPrivateDns());
        if (request.getPublicIp() != null) instance.setPublicIp(request.getPublicIp());
        if (request.getIsStopped() != null) instance.setIsStopped(Boolean.parseBoolean(request.getIsStopped()));
        if (request.getIsTagged() != null) instance.setIsTagged(Boolean.parseBoolean(request.getIsTagged()));
        if (request.getAutoScaleName() != null) instance.setAutoScaleName(request.getAutoScaleName());
        if (!CollectionUtils.isEmpty(request.getTags())) {
            instance.getTags().clear();
            instance.getTags().addAll(request.getTags());
        }
        if (!CollectionUtils.isEmpty(request.getSecurityGroups())) {
            instance.getSecurityGroups().clear();
            instance.getSecurityGroups().addAll(request.getSecurityGroups());
        }
        return instance;
    }


    @Override
    public List upsertInstance(List instances) throws HygieiaException {
        List objectIds = new ArrayList<>();
        if (CollectionUtils.isEmpty(instances)) return objectIds;
        for (CloudInstanceCreateRequest cir : instances) {
            CloudInstance existing = cloudInstanceRepository.findByInstanceId(cir.getInstanceId());
            CloudInstance upsertObject = cloudInstanceRepository.save(upsertCloudInstanceObject(cir, existing));
            objectIds.add(upsertObject.getId().toString());
        }
        return objectIds;
    }
}