com.capitalone.dashboard.model.Component Maven / Gradle / Ivy

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

import org.springframework.data.mongodb.core.mapping.Document;

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

/**
 * A self-contained, independently deployable piece of the larger application. Each component of an application
 * has a different source repo, build job, deploy job, etc.
 *
 */
@Document(collection="components")
public class Component extends BaseModel {
    private String name; // must be unique to the application
    private String owner;
    private Map> collectorItems = new HashMap<>();

    public Component() {
    }

    public Component(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getOwner() {
        return owner;
    }

    public void setOwner(String owner) {
        this.owner = owner;
    }

    public Map> getCollectorItems() {
        return collectorItems;
    }

    public List getCollectorItems(CollectorType type) {
        return collectorItems.get(type);
    }

    public void addCollectorItem(CollectorType collectorType, CollectorItem collectorItem) {
        // Currently only one collectorItem per collectorType is supported
        if (collectorItems.get(collectorType) == null) {
            collectorItems.put(collectorType, Arrays.asList(collectorItem));
        } else {
            List existing = new ArrayList<> (collectorItems.get(collectorType));
            if (isNewCollectorItem(existing, collectorItem)) {
                existing.add(collectorItem);
                collectorItems.put(collectorType, existing);
            }
        }
    }

    private boolean isNewCollectorItem (List existing, CollectorItem item) {
        for (CollectorItem ci : existing) {
            if (ci.getId().equals(item.getId())) return false;
        }
        return true;
    }

    public CollectorItem getFirstCollectorItemForType(CollectorType type){

        if(getCollectorItems().get(type) == null) {
            return null;
        }
        List collectorItems = new ArrayList<>();
        collectorItems.addAll(getCollectorItems().get(type));
        return collectorItems.get(0);
    }
}