From 8d6c24752b483d7670e6d88e265a754259d79ec0 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Thu, 20 Mar 2025 18:14:20 +0100
Subject: [PATCH] fix: Trigger timetable fetch without restarting

---
 .../bahnApi/events/FetchTasksCreatedEvent.kt       |  5 +++++
 .../bahnApi/service/ScheduledTaskService.kt        | 14 +++++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/events/FetchTasksCreatedEvent.kt

diff --git a/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/events/FetchTasksCreatedEvent.kt b/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/events/FetchTasksCreatedEvent.kt
new file mode 100644
index 0000000..3429dbd
--- /dev/null
+++ b/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/events/FetchTasksCreatedEvent.kt
@@ -0,0 +1,5 @@
+package de.twomartens.timetable.bahnApi.events
+
+import org.springframework.context.ApplicationEvent
+
+class FetchTasksCreatedEvent(source: Any) : ApplicationEvent(source)
\ No newline at end of file
diff --git a/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/service/ScheduledTaskService.kt b/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/service/ScheduledTaskService.kt
index 1023541..9ee251c 100644
--- a/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/service/ScheduledTaskService.kt
+++ b/module-bahnApi/src/main/kotlin/de/twomartens/timetable/bahnApi/service/ScheduledTaskService.kt
@@ -1,5 +1,6 @@
 package de.twomartens.timetable.bahnApi.service
 
+import de.twomartens.timetable.bahnApi.events.FetchTasksCreatedEvent
 import de.twomartens.timetable.bahnApi.model.Eva
 import de.twomartens.timetable.bahnApi.model.FetchDates
 import de.twomartens.timetable.bahnApi.model.TaskFactory
@@ -10,6 +11,7 @@ import de.twomartens.timetable.types.Hour
 import de.twomartens.timetable.types.HourAtDay
 import mu.KotlinLogging
 import org.springframework.boot.context.event.ApplicationReadyEvent
+import org.springframework.context.ApplicationEventPublisher
 import org.springframework.context.event.EventListener
 import org.springframework.data.mongodb.core.BulkOperations
 import org.springframework.data.mongodb.core.MongoTemplate
@@ -22,7 +24,8 @@ class ScheduledTaskService(
         private val scheduledFetchTaskRepository: ScheduledFetchTaskRepository,
         private val taskFactory: TaskFactory,
         private val fetchTaskScheduler: FetchTaskScheduler,
-        private val mongoTemplate: MongoTemplate
+        private val mongoTemplate: MongoTemplate,
+        private val eventPublisher: ApplicationEventPublisher
 ) {
     private var createdTime: Instant = Instant.EPOCH
     private var lastUpdate: Instant = Instant.EPOCH
@@ -34,6 +37,13 @@ class ScheduledTaskService(
         updateTaskCounterAndScheduleTasks(updateTime)
     }
 
+    @EventListener(ApplicationReadyEvent::class)
+    fun onFetchTasksCreated(event: FetchTasksCreatedEvent) {
+        log.info { "Scheduled tasks created" }
+        val updateTime = Instant.ofEpochMilli(event.timestamp)
+        updateTaskCounterAndScheduleTasks(updateTime)
+    }
+
     private fun updateTaskCounterAndScheduleTasks(updateTime: Instant) {
         log.info { "Update tasks from database and schedule" }
         val createdTasks = findTasksCreatedSince(lastUpdate)
@@ -50,6 +60,8 @@ class ScheduledTaskService(
         val newTasks = buildScheduledTasks(tswRoute, fetchDates)
 
         storeTasksInDatabaseAndStoreCreationTime(newTasks)
+        val event = FetchTasksCreatedEvent(this)
+        eventPublisher.publishEvent(event)
     }
 
     private fun calculateDatesToFetch(fetchedDate: LocalDate): FetchDates {