mirror of https://github.com/bvn13/covid19-ru.git
58 lines
2.0 KiB
Java
58 lines
2.0 KiB
Java
package com.bvn13.covid19.api.controllers;
|
|
|
|
import com.bvn13.covid19.api.model.CovidStatsInfo;
|
|
import com.bvn13.covid19.api.model.CovidStatsResponse;
|
|
import com.bvn13.covid19.api.service.CovidStatsMaker;
|
|
import com.bvn13.covid19.model.entities.CovidStat;
|
|
import lombok.RequiredArgsConstructor;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import javax.annotation.PostConstruct;
|
|
import java.time.ZoneId;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
|
|
@RequiredArgsConstructor
|
|
@RestController
|
|
@RequestMapping("/")
|
|
public class MainController {
|
|
|
|
private final CovidStatsMaker covidStatsMaker;
|
|
|
|
@Value("${app.zone-id}")
|
|
private String zoneIdStr;
|
|
private ZoneId zoneId;
|
|
|
|
@PostConstruct
|
|
public void init() {
|
|
zoneId = ZoneId.of(zoneIdStr);
|
|
}
|
|
|
|
@GetMapping
|
|
public CovidStatsResponse getStatistics() {
|
|
return covidStatsMaker.findLastUpdateInfo()
|
|
.map(updateInfo -> CovidStatsResponse.builder()
|
|
.datetime(updateInfo.getDatetime())
|
|
.updatedOn(updateInfo.getCreatedOn().atZone(zoneId))
|
|
.stats(convertStats(covidStatsMaker.findCovidStatsByUpdateInfoId(updateInfo.getId())))
|
|
.build())
|
|
.orElse(CovidStatsResponse.builder().build());
|
|
}
|
|
|
|
private List<CovidStatsInfo> convertStats(Collection<CovidStat> stats) {
|
|
return stats.stream()
|
|
.map(stat -> CovidStatsInfo.builder()
|
|
.region(stat.getRegion().getName())
|
|
.sick(stat.getSick())
|
|
.healed(stat.getHealed())
|
|
.died(stat.getDied())
|
|
.build())
|
|
.collect(Collectors.toList());
|
|
}
|
|
|
|
}
|