io.imunity.furms.ui.charts.service.XChartUnionArgumentsGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of furms-web-ui Show documentation
There is a newer version: 4.3.1
/*
 * Copyright (c) 2020 Bixbit s.c. All rights reserved.
 * See LICENSE file for licensing information.
 */

package io.imunity.furms.ui.charts.service;

import io.imunity.furms.domain.community_allocation.CommunityAllocationResolved;
import io.imunity.furms.domain.project_allocation.ProjectAllocationResolved;
import io.imunity.furms.domain.project_allocation_installation.ProjectAllocationChunk;
import io.imunity.furms.domain.resource_usage.ResourceUsage;
import io.imunity.furms.domain.resource_usage.UserResourceUsage;
import org.springframework.stereotype.Component;

import java.time.LocalDate;
import java.time.ZonedDateTime;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import static io.imunity.furms.utils.UTCTimeUtils.convertToUTCTime;
import static java.util.Comparator.comparing;
import static java.util.Optional.ofNullable;
import static java.util.function.Function.identity;
import static java.util.stream.Collectors.toList;

@Component
class XChartUnionArgumentsGenerator {

	List getXArgumentsForProjectAlloc(ProjectAllocationResolved projectAllocation,
	                                             Set allChunks,
	                                             Set allResourceUsageHistory, LocalDate lastChunkTime) {
		return prepareXArguments(
			projectAllocation.resourceCredit.utcStartTime.toLocalDate(),
			lastChunkTime,
			Stream.of(
					allChunks.stream().map(chunk -> chunk.validFrom.toLocalDate()).collect(Collectors.toSet()),
					allResourceUsageHistory.stream().map(usage -> usage.utcProbedAt.toLocalDate()).collect(Collectors.toSet()),
					Set.of(getToday())
				).flatMap(Collection::stream)
				.collect(Collectors.toSet())
		);
	}

	List getXArgumentsForProjectAllocWithUserUsages(ProjectAllocationResolved projectAllocation,
	                                                           Set allChunks,
	                                                           Set allUserResourceUsageHistory,
	                                                           Set allResourceUsageHistory,
	                                                           LocalDate lastChunkTime) {
		return prepareXArguments(
			projectAllocation.resourceCredit.utcStartTime.toLocalDate(),
			lastChunkTime,
			Stream.of(
					allChunks.stream().map(chunk -> chunk.validFrom.toLocalDate()).collect(Collectors.toSet()),
					allResourceUsageHistory.stream().map(usage -> usage.utcProbedAt.toLocalDate()).collect(Collectors.toSet()),
					allUserResourceUsageHistory.stream().map(usage -> usage.utcConsumedUntil.toLocalDate()).collect(Collectors.toSet()),
					Set.of(getToday())
				).flatMap(Collection::stream)
				.collect(Collectors.toSet())
		);
	}

	List getXArgumentsForCommunityAlloc(CommunityAllocationResolved communityAllocation,
	                                               Set allResourceUsageHistory) {
		Set collect =
			allResourceUsageHistory.stream()
				.map(usage -> usage.utcProbedAt.toLocalDate())
				.collect(Collectors.toSet());
		collect.add(getToday());
		return prepareXArguments(
			communityAllocation.resourceCredit.utcStartTime.toLocalDate(),
			null,
			collect
		);
	}

	List prepareXArguments(LocalDate minChartDate,
	                                  LocalDate maxChartDate,
	                                  Set datesToMark) {
		return getUnionOfAllChartDates(minChartDate, maxChartDate, datesToMark)
			.filter(Objects::nonNull)
			.distinct()
			.filter(date -> isAfterOrEqual(maxChartDate, date))
			.sorted(comparing(identity()))
			.collect(toList());
	}

	private Stream getUnionOfAllChartDates(LocalDate minChartDate, LocalDate maxChartDate,
	                                                          Set datesToMark) {
		return Stream.of(
			Stream.of(minChartDate.minusDays(1), minChartDate, maxChartDate),
			datesToMark.stream()
		).flatMap(identity());
	}

	private Boolean isAfterOrEqual(LocalDate endDate, LocalDate date) {
		return ofNullable(endDate)
			.map(localDate -> localDate.isAfter(date) || localDate.isEqual(date))
			.orElse(true);
	}

	private LocalDate getToday() {
		return convertToUTCTime(ZonedDateTime.now()).toLocalDate();
	}
}