67 lines
2.7 KiB
Java
67 lines
2.7 KiB
Java
package ru.soune.nocopy.scheduler;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
|
import lombok.RequiredArgsConstructor;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.scheduling.annotation.Scheduled;
|
|
import org.springframework.stereotype.Component;
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
import ru.soune.nocopy.entity.monitoring.FileMonitoringEntity;
|
|
import ru.soune.nocopy.entity.notification.NotificationType;
|
|
import ru.soune.nocopy.repository.FileMonitoringRepository;
|
|
import ru.soune.nocopy.service.monitoring.MonitoringSearchService;
|
|
import ru.soune.nocopy.service.notification.NotificationService;
|
|
|
|
import java.time.LocalDateTime;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Slf4j
|
|
@Component
|
|
@RequiredArgsConstructor
|
|
public class MonitoringScheduler {
|
|
|
|
private final FileMonitoringRepository monitoringRepository;
|
|
|
|
private final MonitoringSearchService monitoringSearchService;
|
|
|
|
private final NotificationService notificationService;
|
|
|
|
@Scheduled(cron = "0 1 2 * * *", zone = "Europe/Moscow")
|
|
@Transactional
|
|
public void processScheduledMonitoring() {
|
|
log.info("Starting scheduled monitoring check at 02:01 MSK");
|
|
|
|
LocalDateTime now = LocalDateTime.now();
|
|
List<FileMonitoringEntity> readyForRun = monitoringRepository.findReadyForRun(now);
|
|
Map<Long, List<FileMonitoringEntity>> groupingFilesMonitoring = readyForRun.stream()
|
|
.collect(Collectors.groupingBy(FileMonitoringEntity::getUserId));
|
|
|
|
log.info("Found {} files ready for monitoring", readyForRun.size());
|
|
|
|
//TODO check tokens for apply
|
|
for (FileMonitoringEntity monitoring : readyForRun) {
|
|
try {
|
|
monitoringSearchService.processFileSearch(monitoring);
|
|
log.info("Successfully queued search for file: {}", monitoring.getFile().getId());
|
|
} catch (Exception e) {
|
|
log.error("Failed to process monitoring for file: {}", monitoring.getFile().getId(), e);
|
|
monitoring.setLastRunStatus("ERROR: " + e.getMessage());
|
|
monitoringRepository.save(monitoring);
|
|
}
|
|
}
|
|
|
|
Set<Long> usersForNotifications = groupingFilesMonitoring.keySet();
|
|
for (Long userId : usersForNotifications) {
|
|
ObjectMapper mapper = new ObjectMapper();
|
|
ObjectNode context = mapper.createObjectNode();
|
|
context.put("message", "Monitoring files on monitoring");
|
|
|
|
notificationService.addNotification(NotificationType.MONITORING_RESULT, userId, context.toString());
|
|
}
|
|
}
|
|
} |