From ae358fd690c5f821947295bc45cbf4c97df1e8b2 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Thu, 26 Oct 2023 19:16:34 +0200
Subject: [PATCH] Added findBy methods to cover unique indices

---
 .../timetable/bahnApi/repository/BahnStationRepository.kt | 7 +++++--
 .../bahnApi/repository/BahnTimetableRepository.kt         | 7 ++++++-
 .../bahnApi/repository/ScheduledFetchTaskRepository.kt    | 8 ++++++--
 3 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnStationRepository.kt b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnStationRepository.kt
index 25bf5ea..b494efb 100644
--- a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnStationRepository.kt
+++ b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnStationRepository.kt
@@ -1,7 +1,10 @@
 package de.twomartens.timetable.bahnApi.repository
 
+import de.twomartens.timetable.bahnApi.model.Eva
 import de.twomartens.timetable.bahnApi.model.db.BahnStation
+import org.bson.types.ObjectId
 import org.springframework.data.mongodb.repository.MongoRepository
 
-// TODO: figure out what ID type should be
-interface BahnStationRepository : MongoRepository<BahnStation, String>
+interface BahnStationRepository : MongoRepository<BahnStation, ObjectId> {
+    fun findByEva(eva: Eva): BahnStation?
+}
diff --git a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnTimetableRepository.kt b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnTimetableRepository.kt
index c6f89fd..e0fa71a 100644
--- a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnTimetableRepository.kt
+++ b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/BahnTimetableRepository.kt
@@ -1,7 +1,12 @@
 package de.twomartens.timetable.bahnApi.repository
 
+import de.twomartens.timetable.bahnApi.model.Eva
 import de.twomartens.timetable.bahnApi.model.db.BahnTimetable
+import de.twomartens.timetable.types.HourAtDay
+import org.bson.types.ObjectId
 import org.springframework.data.mongodb.repository.MongoRepository
 
 // TODO: figure out what ID type should be
-interface BahnTimetableRepository : MongoRepository<BahnTimetable, String>
\ No newline at end of file
+interface BahnTimetableRepository : MongoRepository<BahnTimetable, ObjectId> {
+    fun findByEvaAndHourAtDay(eva: Eva, hourAtDay: HourAtDay): BahnTimetable?
+}
\ No newline at end of file
diff --git a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/ScheduledFetchTaskRepository.kt b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/ScheduledFetchTaskRepository.kt
index 85dee16..138c34e 100644
--- a/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/ScheduledFetchTaskRepository.kt
+++ b/module-server/src/main/kotlin/de/twomartens/timetable/bahnApi/repository/ScheduledFetchTaskRepository.kt
@@ -1,7 +1,11 @@
 package de.twomartens.timetable.bahnApi.repository
 
+import de.twomartens.timetable.bahnApi.model.Eva
 import de.twomartens.timetable.bahnApi.model.db.ScheduledFetchTask
+import de.twomartens.timetable.types.HourAtDay
+import org.bson.types.ObjectId
 import org.springframework.data.mongodb.repository.MongoRepository
 
-// TODO: what should be ID type be?
-interface ScheduledFetchTaskRepository : MongoRepository<ScheduledFetchTask, String>
+interface ScheduledFetchTaskRepository : MongoRepository<ScheduledFetchTask, ObjectId> {
+    fun findByEvaAndFetchedDateTime(eva: Eva, fetchedDateTime: HourAtDay): ScheduledFetchTask?
+}