D-Net

package eu.dnetlib.uoanotificationservice.services;

import eu.dnetlib.uoanotificationservice.dao.NotificationDAO;

import eu.dnetlib.uoanotificationservice.dao.UserDAO;

import eu.dnetlib.uoanotificationservice.entities.Notification;

import eu.dnetlib.uoanotificationservice.entities.User;

import org.apache.log4j.Logger;

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

import org.springframework.scheduling.annotation.Scheduled;

import org.springframework.stereotype.Service;

import java.util.Calendar;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

@Service

public class NotificationService {

    Logger logger = Logger.getLogger(NotificationService.class);

    @Autowired

    private NotificationDAO notificationDAO;

    @Autowired

    private UserDAO userDAO;

    public List<Notification> getAllNotifications() {

        return notificationDAO.findByOrderByDateDesc();

    }

    public List<Notification> getMyNotifications(String userId, String service, List<String> roles) {

        logger.info("Getting notifications of user " + userId);

        Set<String> services = new HashSet<>();

        services.add(service);

        services.add("all");

        Set<String> groups = new HashSet<>(roles);

        groups.add(userId);

        groups.add("all");

        List<Notification> notifications = notificationDAO.findByUserNotAndServicesInAndGroupsInOrderByDateDesc(userId, services, groups);

        User user = userDAO.findById(userId);

        if(user != null) {

            notifications.forEach(notification -> {

                notification.setRead(user.getRead().contains(notification.getId()));

            });

        }

        return notifications;

    }

    public Notification save(Notification notification) {

        return notificationDAO.save(notification);

    }

    public User readNotification(String userId, String id) {

        logger.info("Read notification " + id + " for  user " + userId);

        User user = userDAO.findById(userId);

        if(user == null) {

            user = new User();

            user.setId(userId);

            user.setRead(new HashSet<>());

        }

        user.getRead().add(id);

        return userDAO.save(user);

    }

    public void delete(String id) {

        List<User> users = userDAO.findByReadContains(id);

        users.forEach(user -> {

            user.getRead().remove(id);

            userDAO.save(user);

        });

        notificationDAO.delete(id);

    }

    // every min for testing

    //@Scheduled(cron = "0 * * * * *"

    // every day at midnight

    @Scheduled(cron = "0 0 0 1/1 * ?")

    protected void deleteNotifications() {

        logger.info("Deleting notifications that have been created more than a week ago");

        Calendar calendar = Calendar.getInstance();

        calendar.add(Calendar.DAY_OF_MONTH, -7);

        List<Notification> notifications = notificationDAO.findByDateBefore(calendar.getTime());

        notifications.forEach(notification -> {

            delete(notification.getId());

        });

    }

}