StatisticsController.java

package org.dicadeveloper.weplantaforest.statistics;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;
import java.util.Locale;

import org.dicadeveloper.weplantaforest.support.Uris;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import lombok.NonNull;
import lombok.RequiredArgsConstructor;

@RestController
@RequiredArgsConstructor(onConstructor = @__(@Autowired) )
// queries from StatisticsRepository run only on mysql, not on h2, so the
// belonging controller is also excluded from this profile
// @Profile({ "!h2", "!test" })
@Profile({ "production", "mysql", "staging" })
public class StatisticsController {

    private @NonNull StatisticsRepository _statisticsRepository;

    @RequestMapping(value = Uris.TREE_STATISTIC_PER_MONTH, method = RequestMethod.GET)
    public ResponseEntity<?> getTreeStatisticForYear(@RequestParam String year) {
        List<TreeAmountStatisticData> treeStatistic = _statisticsRepository.getTreesPerMonthForYear(year);
        List<TreeAmountStatisticData> orderedList = new ArrayList<>();
        for (Integer i = 1; i <= 12; i++) {
            for (TreeAmountStatisticData entry : treeStatistic) {
                if (entry.getLabel()
                         .equals(i.toString())) {
                    orderedList.add(entry);
                }
            }
        }
        return new ResponseEntity<>(orderedList, HttpStatus.OK);
    }

    @RequestMapping(value = Uris.TREE_STATISTIC_PER_YEAR, method = RequestMethod.GET)
    public ResponseEntity<?> getOverallTreeStatistic() {
        List<TreeAmountStatisticData> treeStatistic = _statisticsRepository.getTreesPerYear();
        List<TreeAmountStatisticData> orderedList = new ArrayList<>();

        Calendar year2007 = new GregorianCalendar(2007, 1, 1);
        int years = getDiffYears(year2007, new Date(System.currentTimeMillis()));

        for (Integer i = 2007; i <= 2007 + years; i++) {
            for (TreeAmountStatisticData entry : treeStatistic) {
                if (entry.getLabel()
                         .equals(i.toString())) {
                    orderedList.add(entry);
                }
            }
        }
        return new ResponseEntity<>(orderedList, HttpStatus.OK);
    }

    @RequestMapping(value = Uris.TREE_STATISTIC_PER_ORGTYPE, method = RequestMethod.GET)
    public ResponseEntity<?> getTreesPerOrgType() {
        List<TreeOrgTypeStatisticData> treeStatistic = _statisticsRepository.getTreesPerOrgType();
        return new ResponseEntity<>(treeStatistic, HttpStatus.OK);
    }

    @RequestMapping(value = Uris.CO2_STATISTIC, method = RequestMethod.GET)
    public ResponseEntity<?> getCo2Statistic() {
        List<Co2StatisticData> co2Statistic = _statisticsRepository.getCo2PerYear(System.currentTimeMillis());
        List<TreeAmountStatisticData> treeStatistic = _statisticsRepository.getTreesPerYear();
        List<Co2StatisticData> orderedList = new ArrayList<>();
        Calendar year2007 = new GregorianCalendar(2007, 1, 1);
        int years = getDiffYears(year2007, new Date(System.currentTimeMillis()));

        for (Integer i = 2007; i <= 2007 + years; i++) {
            double amountOfCo2TillThisYear = 0;
            for (Co2StatisticData entry : co2Statistic) {
                if (entry.getLabel()
                         .equals(i.toString())) {
                    for (TreeAmountStatisticData treeData : treeStatistic) {
                        int year = new Integer(treeData.getLabel());
                        if (i > year) {
                            long amountOfTrees = treeData.getAmount();
                            int diffOfYears = i - year;
                            amountOfCo2TillThisYear += diffOfYears * amountOfTrees * 0.015;
                        }
                    }
                    double valueThisYear = entry.getCo2();
                    valueThisYear += amountOfCo2TillThisYear;
                    entry.setCo2(valueThisYear);
                    orderedList.add(entry);
                    amountOfCo2TillThisYear = 0;
                }
            }
        }
        return new ResponseEntity<>(orderedList, HttpStatus.OK);
    }

    private int getDiffYears(Calendar first, Date last) {
        Calendar b = getCalendar(last);
        int diff = b.get(Calendar.YEAR) - first.get(Calendar.YEAR);
        if (first.get(Calendar.MONTH) > b.get(Calendar.MONTH) || (first.get(Calendar.MONTH) == b.get(Calendar.MONTH) && first.get(Calendar.DATE) > b.get(Calendar.DATE))) {
            diff--;
        }
        return diff;
    }

    private Calendar getCalendar(Date date) {
        Calendar cal = Calendar.getInstance(Locale.GERMANY);
        cal.setTime(date);
        return cal;
    }

}