com.lewisd.maven.lint.CachingModelFactory Maven / Gradle / Ivy

Go to download
package com.lewisd.maven.lint;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import org.apache.maven.project.MavenProject;

public class CachingModelFactory implements ModelFactory {
	
	private final Map projectModelsByProjectId = new HashMap();
	private final Map modelBuilders = new HashMap();

	public CachingModelFactory() {
		
	}
	
	public Map getModels(final MavenProject mavenProject, final Set requiredModels) {
		final ProjectModels projectModels = getProjectModels(mavenProject);
		
		return projectModels.getModels(requiredModels);
	}
	
	private ProjectModels getProjectModels(MavenProject mavenProject) {
		final String id = mavenProject.getId();
		ProjectModels projectModels = projectModelsByProjectId.get(id);
		if (projectModels == null) {
			projectModels = new ProjectModels(mavenProject, Collections.unmodifiableMap(modelBuilders));
			projectModelsByProjectId.put(id, projectModels);
		}
		
		return projectModels;
	}

	public void addModelBuilder(final ModelBuilder modelBuilder) {
		modelBuilders.put(modelBuilder.getModelId(), modelBuilder);
	}

}