/*******************************************************************************
* Copyright (c) 2012-2017 Codenvy, S.A.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Codenvy, S.A. - initial API and implementation
*******************************************************************************/
package org.eclipse.che.api.agent.server.impl;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import org.eclipse.che.api.agent.server.AgentRegistry;
import org.eclipse.che.api.agent.server.exception.AgentException;
import org.eclipse.che.api.agent.server.exception.AgentNotFoundException;
import org.eclipse.che.api.agent.shared.model.Agent;
import org.eclipse.che.api.agent.shared.model.AgentKey;
import org.eclipse.che.api.agent.shared.model.impl.AgentKeyImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;
import static java.lang.String.format;
import static java.util.Collections.unmodifiableCollection;
/**
* Local implementation of the {@link AgentRegistry}.
* The name of the agent might represent a url.
*
* @author Anatoliy Bazko
*/
@Singleton
public class AgentRegistryImpl implements AgentRegistry {
protected static final Logger LOG = LoggerFactory.getLogger(AgentRegistryImpl.class);
private final Map<AgentKey, Agent> agents;
/**
* AgentRegistryImpl constructor.
*
* @param agents
* list of agents to register
* @throws IllegalArgumentException
* if there are several agents with same id and version
*/
@Inject
public AgentRegistryImpl(Set<Agent> agents) throws IllegalArgumentException {
this.agents = new HashMap<>(agents.size());
for (Agent agent : agents) {
AgentKeyImpl key = new AgentKeyImpl(agent);
Agent registeredAgent = this.agents.put(key, agent);
if (registeredAgent != null) {
throw new IllegalArgumentException(format("Agent with key %s has been registered already.", key));
}
}
}
@Override
public Agent getAgent(AgentKey agentKey) throws AgentException {
return doGetAgent(agentKey);
}
@Override
public List<String> getVersions(String id) throws AgentException {
return agents.entrySet().stream()
.filter(e -> e.getKey().getId().equals(id))
.map(e -> e.getKey().getVersion())
.collect(Collectors.toList());
}
@Override
public Collection<Agent> getAgents() throws AgentException {
return unmodifiableCollection(agents.values());
}
private Agent doGetAgent(AgentKey key) throws AgentException {
Optional<Agent> agent = Optional.ofNullable(agents.get(key));
return agent.orElseThrow(() -> new AgentNotFoundException(format("Agent %s not found", key.getId())));
}
}