com.capitalone.dashboard.service.BinaryArtifactServiceImpl 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.model.BinaryArtifact;
import com.capitalone.dashboard.model.Build;
import com.capitalone.dashboard.model.DataResponse;
import com.capitalone.dashboard.model.JobCollectorItem;
import com.capitalone.dashboard.repository.BinaryArtifactRepository;
import com.capitalone.dashboard.repository.BuildRepository;
import com.capitalone.dashboard.repository.JobRepository;
import com.capitalone.dashboard.request.BinaryArtifactCreateRequest;
import com.capitalone.dashboard.request.BinaryArtifactSearchRequest;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang3.ObjectUtils;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class BinaryArtifactServiceImpl implements BinaryArtifactService {

    private final BinaryArtifactRepository artifactRepository;
    private final BuildRepository buildRepository;
    private final JobRepository jobRepository;

    @Autowired
    public BinaryArtifactServiceImpl(BinaryArtifactRepository artifactRepository, BuildRepository buildRepository, JobRepository jobRepository) {
        this.artifactRepository = artifactRepository;
        this.buildRepository = buildRepository;
        this.jobRepository = jobRepository;
    }

    @Override
    public DataResponse> search(BinaryArtifactSearchRequest request) {
    	
    	Map attributes = new HashMap<>();
    	
    	if (request.getArtifactGroup() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_GROUP_ID, request.getArtifactGroup());
    	}
    	
    	if (request.getArtifactModule() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_MODULE, request.getArtifactModule());
    	}
    	
    	if (request.getArtifactVersion() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_VERSION, request.getArtifactVersion());
    	}
    	
    	if (request.getArtifactName() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_NAME, request.getArtifactName());
    	}
    	
    	if (request.getArtifactClassifier() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_CLASSIFIER, request.getArtifactClassifier());
    	}
    	
    	if (request.getArtifactExtension() != null) {
    		attributes.put(BinaryArtifactRepository.ARTIFACT_EXTENSION, request.getArtifactExtension());
    	}
    	
    	if (request.getBuildId() != null) {
    		attributes.put(BinaryArtifactRepository.BUILD_INFO_ID, request.getBuildId());
    	}
		
		Iterable rt = artifactRepository.findByAttributes(attributes);
        
        return new DataResponse<>(rt, System.currentTimeMillis());

    }
    
    private Build getBuildById(ObjectId buildId){
    	return buildRepository.findOne(buildId);
    }

    @Override
    public String create(BinaryArtifactCreateRequest request) {
		BinaryArtifact ba = new BinaryArtifact();
		ba.setArtifactName(request.getArtifactName());
		ba.setCanonicalName(request.getCanonicalName());
		ba.setArtifactGroupId(request.getArtifactGroup());
		ba.setArtifactVersion(request.getArtifactVersion());
		ba.setArtifactModule(request.getArtifactModule() != null? request.getArtifactModule() : request.getArtifactName());
		ba.setArtifactClassifier(request.getArtifactClassifier());
		ba.setArtifactExtension(request.getArtifactExtension());
		ba.setTimestamp(request.getTimestamp());

		Map metadata = request.getMetadata();
		
		if (metadata != null) {
			for (Map.Entry e : metadata.entrySet()) {
				ba.getMetadata().put(e.getKey(), String.valueOf(e.getValue()));
			}
		}
		
		// Set the build information if we have it
		ObjectId buildId = (request.getBuildId() != null && !request.getBuildId().isEmpty())?
				new ObjectId(request.getBuildId()) : null;
		if (buildId != null) {
			setBuildInformation(ba, buildId);
		}
		
		BinaryArtifact existing = existing(ba, buildId);
		if (existing == null) {
			BinaryArtifact savedArt = artifactRepository.save(ba);
			if (savedArt == null)
				return "";
			return savedArt.getId().toString();
		}
		return existing.getId().toString();
	}
    
    private void setBuildInformation(BinaryArtifact ba, ObjectId buildId) {
		Build build = getBuildById(buildId);
		ba.setBuildInfos(Arrays.asList(build));
		
		// Attempt to deduce metadata information
		if (build != null) { 
			if (ba.getBuildUrl() == null) {
				ba.setBuildUrl(build.getBuildUrl());
			}
			if (ba.getBuildNumber() == null) {
				ba.setBuildNumber(build.getNumber());
			}
			
			JobCollectorItem ci = jobRepository.findOne(build.getCollectorItemId());
			if (ci != null) {
				if (ba.getInstanceUrl() == null) {
					ba.setInstanceUrl(ci.getInstanceUrl());
				}
				if (ba.getJobName() == null) {
					ba.setJobName(ci.getJobName());
				}
				if (ba.getJobUrl() == null) {
					ba.setJobUrl(ci.getJobUrl());
				}
			}
		}
    }


    private BinaryArtifact existing(BinaryArtifact artifact, ObjectId buildId) {
        Iterable bas = artifactRepository.findByAttributes(artifact.getArtifactGroupId(), artifact.getArtifactModule(), 
        		artifact.getArtifactVersion(), artifact.getArtifactName(), artifact.getArtifactClassifier(), artifact.getArtifactExtension());

        for (BinaryArtifact ba : bas) {
        	
        	// could be null due to old documents
        	if (ba.getMetadata() != null 
        			&& artifact.getBuildUrl() != null
        			&& ObjectUtils.equals(artifact.getBuildUrl(), ba.getBuildUrl())) {
                return ba;
            } else if (buildId != null 
            		&& ba.getBuildInfos() != null
            		&& findBuildId(ba.getBuildInfos(),buildId)) {
            	return ba;
            }
        }
        return null;
    }

    private boolean findBuildId(List builds,ObjectId buildId){
    	return builds.stream().filter(build -> buildId.equals(build.getId())).findAny().isPresent();
	}
}