HLTech / judge-d

import com.hltech.judged.server.domain.environment.Environment;

import com.hltech.judged.server.domain.environment.EnvironmentRepository;

import com.hltech.judged.server.domain.ServiceId;

import com.hltech.judged.server.domain.environment.Space;

import lombok.RequiredArgsConstructor;

import org.springframework.stereotype.Repository;

import javax.persistence.NoResultException;

import java.util.HashSet;

import java.util.Optional;

import java.util.Set;

import java.util.stream.Collectors;

@Repository

@RequiredArgsConstructor

public class JPAEnvironmentRepository implements EnvironmentRepository {

    private final SpringDataEnvironmentRepository springDataEnvironmentRepository;

    @Override

    public Environment persist(Environment environment) {

        EnvironmentTuple environmentTuple = toEnvironmentTuple(environment);

        return toEnvironment(springDataEnvironmentRepository.saveAndFlush(environmentTuple));

    }

    @Override

    public Set<String> getNames() {

        return springDataEnvironmentRepository.getNames();

    }

    @Override

    public Optional<Environment> find(String name) {

        return springDataEnvironmentRepository.findById(name)

            .map(this::toEnvironment);

    }

    private EnvironmentTuple toEnvironmentTuple(Environment environment) {

        return new EnvironmentTuple(environment.getName(), getSpaceServiceVersions(environment));

    }

    private Set<ServiceVersion> getSpaceServiceVersions(Environment environment) {

        return environment.getSpaceNames().stream()

            .flatMap(space -> environment.getServices(space).stream()

                .map(service -> new ServiceVersion(space, service.getName(), service.getVersion())))

            .collect(Collectors.toUnmodifiableSet());

    }

}