mirror of
https://github.com/Suwayomi/Suwayomi-Server.git
synced 2026-07-04 03:14:40 -05:00
Support Token Expiry properly (#878)
* Support token expiry properly * Small fix * Lint * Use newer fixes for expiry * Lint
This commit is contained in:
@@ -67,6 +67,19 @@ class TrackerScoresDataLoader : KotlinDataLoader<Int, List<String>> {
|
||||
}
|
||||
}
|
||||
|
||||
class TrackerTokenExpiredDataLoader : KotlinDataLoader<Int, Boolean> {
|
||||
override val dataLoaderName = "TrackerTokenExpiredDataLoader"
|
||||
|
||||
override fun getDataLoader(): DataLoader<Int, Boolean> =
|
||||
DataLoaderFactory.newDataLoader { ids ->
|
||||
future {
|
||||
ids.map { id ->
|
||||
TrackerManager.getTracker(id)?.getIfAuthExpired()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class TrackRecordsForMangaIdDataLoader : KotlinDataLoader<Int, TrackRecordNodeList> {
|
||||
override val dataLoaderName = "TrackRecordsForMangaIdDataLoader"
|
||||
|
||||
|
||||
@@ -38,6 +38,7 @@ import suwayomi.tachidesk.graphql.dataLoaders.TrackRecordsForTrackerIdDataLoader
|
||||
import suwayomi.tachidesk.graphql.dataLoaders.TrackerDataLoader
|
||||
import suwayomi.tachidesk.graphql.dataLoaders.TrackerScoresDataLoader
|
||||
import suwayomi.tachidesk.graphql.dataLoaders.TrackerStatusesDataLoader
|
||||
import suwayomi.tachidesk.graphql.dataLoaders.TrackerTokenExpiredDataLoader
|
||||
import suwayomi.tachidesk.graphql.dataLoaders.UnreadChapterCountForMangaDataLoader
|
||||
|
||||
class TachideskDataLoaderRegistryFactory {
|
||||
@@ -71,6 +72,7 @@ class TachideskDataLoaderRegistryFactory {
|
||||
TrackerDataLoader(),
|
||||
TrackerStatusesDataLoader(),
|
||||
TrackerScoresDataLoader(),
|
||||
TrackerTokenExpiredDataLoader(),
|
||||
TrackRecordsForMangaIdDataLoader(),
|
||||
DisplayScoreForTrackRecordDataLoader(),
|
||||
TrackRecordsForTrackerIdDataLoader(),
|
||||
|
||||
@@ -49,6 +49,10 @@ class TrackerType(
|
||||
fun trackRecords(dataFetchingEnvironment: DataFetchingEnvironment): CompletableFuture<TrackRecordNodeList> {
|
||||
return dataFetchingEnvironment.getValueFromDataLoader<Int, TrackRecordNodeList>("TrackRecordsForTrackerIdDataLoader", id)
|
||||
}
|
||||
|
||||
fun isTokenExpired(dataFetchingEnvironment: DataFetchingEnvironment): CompletableFuture<Boolean> {
|
||||
return dataFetchingEnvironment.getValueFromDataLoader<Int, Boolean>("TrackerTokenExpiredDataLoader", id)
|
||||
}
|
||||
}
|
||||
|
||||
class TrackStatusType(
|
||||
|
||||
Reference in New Issue
Block a user