#101 event service edited

src/main/java/ru/ulstu/conference/service/ConferenceService.java

@@ -27,6 +27,7 @@ import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Date;
 import java.util.List;
+import java.util.Collections;
 import java.util.stream.Collectors;
 
 import static org.springframework.util.ObjectUtils.isEmpty;
@@ -116,7 +117,7 @@ public class ConferenceService extends BaseService {
         Conference newConference = copyFromDto(new Conference(), conferenceDto);
         newConference = conferenceRepository.save(newConference);
         conferenceNotificationService.sendCreateNotification(newConference);
-        eventService.createFromConference(newConference);
+        eventService.createFromObject(newConference, Collections.emptyList());
         return newConference;
     }
 

src/main/java/ru/ulstu/grant/service/GrantService.java

@@ -90,7 +90,7 @@ public class GrantService extends BaseService {
     public Integer create(GrantDto grantDto) throws IOException {
         Grant newGrant = copyFromDto(new Grant(), grantDto);
         newGrant = grantRepository.save(newGrant);
-        eventService.createFromGrant(newGrant);
+        eventService.createFromObject(newGrant, Collections.emptyList());
         grantNotificationService.sendCreateNotification(newGrant);
         return newGrant.getId();
     }
@@ -178,7 +178,7 @@ public class GrantService extends BaseService {
         grant.getPapers().add(paper);
         grant = grantRepository.save(grant);
 
-        eventService.createFromGrant(grant);
+        eventService.createFromObject(grant, Collections.emptyList());
         grantNotificationService.sendCreateNotification(grant);
 
         return grant;

src/main/java/ru/ulstu/project/service/ProjectService.java

@@ -17,8 +17,8 @@ import ru.ulstu.user.service.UserService;
 
 import java.io.IOException;
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
-import java.util.Set;
 
 import static java.util.stream.Collectors.toList;
 import static org.springframework.util.ObjectUtils.isEmpty;
@@ -72,7 +72,7 @@ public class ProjectService {
     public Project create(ProjectDto projectDto) throws IOException {
         Project newProject = copyFromDto(new Project(), projectDto);
         newProject = projectRepository.save(newProject);
-        eventService.createFromProject(newProject);
+        eventService.createFromObject(newProject, Collections.emptyList());
         return newProject;
     }
 

src/main/java/ru/ulstu/students/service/TaskService.java

@@ -18,14 +18,15 @@ import ru.ulstu.tags.service.TagService;
 import ru.ulstu.timeline.service.EventService;
 
 import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.Calendar;
+import java.util.Collections;
 import java.util.Date;
 import java.util.List;
-import java.util.Map;
 import java.util.Set;
+import java.util.Map;
 import java.util.TreeMap;
+import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.stream.Collectors;
 
 import static org.springframework.util.ObjectUtils.isEmpty;
@@ -83,7 +84,7 @@ public class TaskService {
     public Integer create(TaskDto taskDto) throws IOException {
         Task newTask = copyFromDto(new Task(), taskDto);
         newTask = taskRepository.save(newTask);
-        eventService.createFromTask(newTask);
+        eventService.createFromObject(newTask, Collections.emptyList());
         return newTask.getId();
     }
 

src/main/java/ru/ulstu/timeline/service/EventService.java

@@ -1,10 +1,12 @@
 package ru.ulstu.timeline.service;
 
+import com.sun.jna.platform.win32.COM.util.Convert;
 import org.apache.commons.lang3.time.DateUtils;
 import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import ru.ulstu.conference.model.Conference;
+import ru.ulstu.core.model.BaseEntity;
 import ru.ulstu.deadline.model.Deadline;
 import ru.ulstu.grant.model.Grant;
 import ru.ulstu.paper.model.Paper;
@@ -34,6 +36,13 @@ public class EventService {
     private final TimelineService timelineService;
     private final UserService userService;
 
+    private String objectClassName = "";
+    private Paper newPaper = new Paper();
+    private Conference newConference = new Conference();
+    private Grant newGrant = new Grant();
+    private Project newProject = new Project();
+    private Task newTask = new Task();
+
     public EventService(EventRepository eventRepository,
                         @Lazy TimelineService timelineService,
                         UserService userService) {
@@ -107,35 +116,135 @@ public class EventService {
     }
 
     public void createFromPaper(Paper newPaper) {
-        createFromPaper(newPaper, Collections.emptyList());
+        createFromObject(newPaper, Collections.emptyList());
     }
 
-    public void createFromPaper(Paper newPaper, List<Event> events) {
+    public void createFromObject(Object newObject, List<Event> events) {
         List<Timeline> timelines = timelineService.findAll();
         Timeline timeline = timelines.isEmpty() ? new Timeline() : timelines.get(0);
-        timeline.getEvents().removeAll(events);
+        objectClassName = newObject.getClass().getName();
-        for (Deadline deadline : newPaper.getDeadlines()
+
-                .stream()
+        switch(objectClassName) {
-                .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
+            case "ru.ulstu.paper.model.Paper":
-                .collect(Collectors.toList())) {
+                newPaper = (Paper)newObject;
-            Event newEvent = new Event();
+                timeline.getEvents().removeAll(events);
-            newEvent.setTitle("Дедлайн статьи");
+                for (Deadline deadline : newPaper.getDeadlines()
-            newEvent.setStatus(Event.EventStatus.NEW);
+                        .stream()
-            newEvent.setExecuteDate(deadline.getDate());
+                        .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
-            newEvent.setCreateDate(new Date());
+                        .collect(Collectors.toList())) {
-            newEvent.setUpdateDate(new Date());
+                    Event newEvent = new Event();
-            newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' cтатьи '" + newPaper.getTitle() + "'");
+                    newEvent.setTitle("Дедлайн статьи");
-            newEvent.setRecipients(new ArrayList(newPaper.getAuthors()));
+                    newEvent.setStatus(Event.EventStatus.NEW);
-            newEvent.setPaper(newPaper);
+                    newEvent.setExecuteDate(deadline.getDate());
-            timeline.getEvents().add(eventRepository.save(newEvent));
+                    newEvent.setCreateDate(new Date());
+                    newEvent.setUpdateDate(new Date());
+                    newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' cтатьи '" + newPaper.getTitle() + "'");
+                    newEvent.setRecipients(new ArrayList(newPaper.getAuthors()));
+                    newEvent.setPaper(newPaper);
+                    timeline.getEvents().add(eventRepository.save(newEvent));
+                }
+                timelineService.save(timeline);
+                break;
+            case "ru.ulstu.conference.model.Conference":
+                newConference = (Conference)newObject;
+                for (Deadline deadline : newConference.getDeadlines()
+                        .stream()
+                        .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
+                        .collect(Collectors.toList())) {
+                    Event newEvent = new Event();
+                    newEvent.setTitle("Дедлайн конференции");
+                    newEvent.setStatus(Event.EventStatus.NEW);
+                    newEvent.setExecuteDate(deadline.getDate());
+                    newEvent.setCreateDate(new Date());
+                    newEvent.setUpdateDate(new Date());
+                    newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' конференции '" + newConference.getTitle() + "'");
+                    newConference.getUsers().forEach(conferenceUser -> newEvent.getRecipients().add(conferenceUser.getUser()));
+                    newEvent.setConference(newConference);
+                    save(newEvent);
+
+                    timeline.getEvents().add(newEvent);
+                    timelineService.save(timeline);
+                }
+                break;
+            case "ru.ulstu.grant.model.Grant":
+                newGrant = (Grant)newObject;
+                for (Deadline deadline : newGrant.getDeadlines()
+                        .stream()
+                        .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
+                        .collect(Collectors.toList())) {
+                    Event newEvent = new Event();
+                    newEvent.setTitle("Дедлайн гранта");
+                    newEvent.setStatus(Event.EventStatus.NEW);
+                    newEvent.setExecuteDate(deadline.getDate());
+                    newEvent.setCreateDate(new Date());
+                    newEvent.setUpdateDate(new Date());
+                    newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' гранта '" + newGrant.getTitle() + "'");
+                    if (newGrant.getAuthors() != null) {
+                        newEvent.setRecipients(new ArrayList(newGrant.getAuthors()));
+                    }
+                    newEvent.getRecipients().add(newGrant.getLeader());
+                    newEvent.setGrant(newGrant);
+                    eventRepository.save(newEvent);
+
+                    timeline.getEvents().add(newEvent);
+                    timelineService.save(timeline);
+                }
+                break;
+            case "ru.ulstu.project.model.Project":
+                newProject = (Project)newObject;
+                for (Deadline deadline : newProject.getDeadlines()
+                        .stream()
+                        .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
+                        .collect(Collectors.toList())) {
+                    Event newEvent = new Event();
+                    newEvent.setTitle("Дедлайн проекта");
+                    newEvent.setStatus(Event.EventStatus.NEW);
+                    newEvent.setExecuteDate(deadline.getDate());
+                    newEvent.setCreateDate(new Date());
+                    newEvent.setUpdateDate(new Date());
+                    newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' проекта '" + newProject.getTitle() + "'");
+                    if (newProject.getExecutors() != null) {
+                        newEvent.setRecipients(new ArrayList(newProject.getExecutors()));
+                    }
+                    newEvent.setProject(newProject);
+                    eventRepository.save(newEvent);
+
+                    timeline.getEvents().add(newEvent);
+                    timelineService.save(timeline);
+                }
+                break;
+            case "ru.ulstu.students.model.Task":
+                newTask = (Task)newObject;
+                for (Deadline deadline : newTask.getDeadlines()
+                        .stream()
+                        .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
+                        .collect(Collectors.toList())) {
+                    Event newEvent = new Event();
+                    newEvent.setTitle("Дедлайн задачи");
+                    newEvent.setStatus(Event.EventStatus.NEW);
+                    newEvent.setExecuteDate(deadline.getDate());
+                    newEvent.setCreateDate(new Date());
+                    newEvent.setUpdateDate(new Date());
+                    newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' задачи '" + newTask.getTitle() + "'");
+                    newEvent.getRecipients().add(userService.getCurrentUser());
+                    newEvent.setTask(newTask);
+                    eventRepository.save(newEvent);
+
+                    timeline.getEvents().add(newEvent);
+                    timelineService.save(timeline);
+                }
+                System.out.println("5");
+                break;
+            default:
+                System.out.println("No such object");
+                break;
         }
-        timelineService.save(timeline);
     }
 
     public void updatePaperDeadlines(Paper paper) {
         List<Event> foundEvents = eventRepository.findAllByPaper(paper);
         eventRepository.delete(foundEvents);
-        createFromPaper(paper, foundEvents);
+        createFromObject(paper, foundEvents);
     }
 
     public List<Event> findByCurrentDate() {
@@ -150,97 +259,19 @@ public class EventService {
         return convert(findAllFuture(), EventDto::new);
     }
 
-    public void createFromConference(Conference newConference) {
-        List<Timeline> timelines = timelineService.findAll();
-        Timeline timeline = timelines.isEmpty() ? new Timeline() : timelines.get(0);
-
-        for (Deadline deadline : newConference.getDeadlines()
-                .stream()
-                .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
-                .collect(Collectors.toList())) {
-            Event newEvent = new Event();
-            newEvent.setTitle("Дедлайн конференции");
-            newEvent.setStatus(Event.EventStatus.NEW);
-            newEvent.setExecuteDate(deadline.getDate());
-            newEvent.setCreateDate(new Date());
-            newEvent.setUpdateDate(new Date());
-            newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' конференции '" + newConference.getTitle() + "'");
-            newConference.getUsers().forEach(conferenceUser -> newEvent.getRecipients().add(conferenceUser.getUser()));
-            newEvent.setConference(newConference);
-            save(newEvent);
-
-            timeline.getEvents().add(newEvent);
-            timelineService.save(timeline);
-        }
-    }
-
     public void updateConferenceDeadlines(Conference conference) {
         eventRepository.delete(eventRepository.findAllByConference(conference));
-        createFromConference(conference);
+        createFromObject(conference, Collections.emptyList());
-    }
-
-    public void createFromGrant(Grant newGrant) {
-        List<Timeline> timelines = timelineService.findAll();
-        Timeline timeline = timelines.isEmpty() ? new Timeline() : timelines.get(0);
-
-        for (Deadline deadline : newGrant.getDeadlines()
-                .stream()
-                .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
-                .collect(Collectors.toList())) {
-            Event newEvent = new Event();
-            newEvent.setTitle("Дедлайн гранта");
-            newEvent.setStatus(Event.EventStatus.NEW);
-            newEvent.setExecuteDate(deadline.getDate());
-            newEvent.setCreateDate(new Date());
-            newEvent.setUpdateDate(new Date());
-            newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' гранта '" + newGrant.getTitle() + "'");
-            if (newGrant.getAuthors() != null) {
-                newEvent.setRecipients(new ArrayList(newGrant.getAuthors()));
-            }
-            newEvent.getRecipients().add(newGrant.getLeader());
-            newEvent.setGrant(newGrant);
-            eventRepository.save(newEvent);
-
-            timeline.getEvents().add(newEvent);
-            timelineService.save(timeline);
-        }
     }
 
     public void updateGrantDeadlines(Grant grant) {
         eventRepository.delete(eventRepository.findAllByGrant(grant));
-        createFromGrant(grant);
+        createFromObject(grant, Collections.emptyList());
-    }
-
-    public void createFromProject(Project newProject) {
-        List<Timeline> timelines = timelineService.findAll();
-        Timeline timeline = timelines.isEmpty() ? new Timeline() : timelines.get(0);
-
-        for (Deadline deadline : newProject.getDeadlines()
-                .stream()
-                .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
-                .collect(Collectors.toList())) {
-            Event newEvent = new Event();
-            newEvent.setTitle("Дедлайн проекта");
-            newEvent.setStatus(Event.EventStatus.NEW);
-            newEvent.setExecuteDate(deadline.getDate());
-            newEvent.setCreateDate(new Date());
-            newEvent.setUpdateDate(new Date());
-            newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' проекта '" + newProject.getTitle() + "'");
-            if (newProject.getExecutors() != null) {
-                newEvent.setRecipients(new ArrayList(newProject.getExecutors()));
-            }
-//            newEvent.getRecipients().add((User) newProject.getExecutors());
-            newEvent.setProject(newProject);
-            eventRepository.save(newEvent);
-
-            timeline.getEvents().add(newEvent);
-            timelineService.save(timeline);
-        }
     }
 
     public void updateProjectDeadlines(Project project) {
         eventRepository.delete(eventRepository.findAllByProject(project));
-        createFromProject(project);
+        createFromObject(project, Collections.emptyList());
     }
 
     public void removeConferencesEvent(Conference conference) {
@@ -248,32 +279,8 @@ public class EventService {
         eventList.forEach(event -> eventRepository.delete(event.getId()));
     }
 
-    public void createFromTask(Task newTask) {
-        List<Timeline> timelines = timelineService.findAll();
-        Timeline timeline = timelines.isEmpty() ? new Timeline() : timelines.get(0);
-
-        for (Deadline deadline : newTask.getDeadlines()
-                .stream()
-                .filter(d -> d.getDate().after(new Date()) || DateUtils.isSameDay(d.getDate(), new Date()))
-                .collect(Collectors.toList())) {
-            Event newEvent = new Event();
-            newEvent.setTitle("Дедлайн задачи");
-            newEvent.setStatus(Event.EventStatus.NEW);
-            newEvent.setExecuteDate(deadline.getDate());
-            newEvent.setCreateDate(new Date());
-            newEvent.setUpdateDate(new Date());
-            newEvent.setDescription("Дедлайн '" + deadline.getDescription() + "' задачи '" + newTask.getTitle() + "'");
-            newEvent.getRecipients().add(userService.getCurrentUser());
-            newEvent.setTask(newTask);
-            eventRepository.save(newEvent);
-
-            timeline.getEvents().add(newEvent);
-            timelineService.save(timeline);
-        }
-    }
-
     public void updateTaskDeadlines(Task task) {
         eventRepository.delete(eventRepository.findAllByTask(task));
-        createFromTask(task);
+        createFromObject(task, Collections.emptyList());
     }
 }

src/main/resources/db/changelog-master.xml

@@ -49,6 +49,6 @@
     <include file="db/changelog-20190523_000000-schema.xml"/>
     <include file="db/changelog-20190528_000000-schema.xml"/>
     <include file="db/changelog-20190528_000002-schema.xml"/>
-    <include file="db/changelog-20190529_000000-schema.xml"/>
+    <include file="db/changelog-20190529_000001-schema.xml"/>
     <include file="db/changelog-20190601_000001-schema.xml"/>
 </databaseChangeLog>