D-Net

package eu.dnetlib.repo.manager.service;

import eu.dnetlib.repo.manager.domain.*;

import eu.dnetlib.repo.manager.domain.broker.BrowseEntry;

import org.apache.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;

import java.util.ArrayList;

import java.util.List;

@Service("dashboardService")

public class DashboardServiceImpl implements DashboardService {

    private static final Logger logger = Logger.getLogger(DashboardServiceImpl.class);

    @Autowired

    private EmailUtils emailUtils;

    @Autowired

    private RepositoryService repositoryService;

    @Autowired

    private BrokerService brokerService;

    @Override

    public List<RepositorySummaryInfo> getRepositoriesSummaryInfo(String userEmail,

                                                                  String page,

                                                                  String size) {

        List<RepositorySummaryInfo> repositorySummaryInfoList = new ArrayList<>();

        try {

            List<RepositorySnippet> repositoriesOfUser = repositoryService.getRepositoriesSnippetsOfUser(userEmail, page, size);

            for (RepositorySnippet repository : repositoriesOfUser) {

                RepositorySummaryInfo repositorySummaryInfo = new RepositorySummaryInfo();

                repositorySummaryInfo.setId(repository.getId());

                repositorySummaryInfo.setRepositoryName(repository.getOfficialname());

                repositorySummaryInfo.setLogoURL(repository.getLogoUrl());

                //TODO getRepositoryAggregations returns only the 20 more recent items. Is it positive that we will find an indexed version there?

                long start = System.currentTimeMillis();

                List<AggregationDetails> aggregationDetailsList = repositoryService.getRepositoryAggregations(repository.getId(), 0, 20);

                for (AggregationDetails aggregationDetails : aggregationDetailsList) {

                    if (aggregationDetails.getIndexedVersion()) {

                        repositorySummaryInfo.setRecordsCollected(aggregationDetails.getNumberOfRecords());

                        repositorySummaryInfo.setLastIndexedVersion(aggregationDetails.getDate());

                        break;

                    }

                }

                long end = System.currentTimeMillis();

                System.out.println("Got repo aggregations in " + (end - start) + "ms");

                try {

                    MetricsInfo metricsInfo = repositoryService.getMetricsInfoForRepository(repository.getId());

                    repositorySummaryInfo.setTotalDownloads(metricsInfo.getMetricsNumbers().getTotalDownloads());

                    repositorySummaryInfo.setTotalViews(metricsInfo.getMetricsNumbers().getTotalViews());

                } catch (RepositoryServiceException e) {

                    logger.error("Exception getting metrics info for repository: " + repository.getId(), e);

                }

                try {

                    List<BrowseEntry> events = brokerService.getTopicsForDatasource(repository.getOfficialname());

                    Long totalEvents = 0L;

                    for (BrowseEntry browseEntry : events)

                        totalEvents += browseEntry.getSize();

                    repositorySummaryInfo.setEnrichmentEvents(totalEvents);

                } catch (BrokerException e) {

                    logger.error("Exception getting broker events for repository: " + repository.getId(), e);

                }

                repositorySummaryInfoList.add(repositorySummaryInfo);

            }

        } catch (Exception e) {

            logger.error("Something baad happened!", e);

        }

        return repositorySummaryInfoList;

    }

}