51 lines
1.5 KiB
Swift
51 lines
1.5 KiB
Swift
import CoreData
|
|
import Foundation
|
|
import Models
|
|
import Utils
|
|
|
|
public extension DataService {
|
|
func prefetchPages(itemIDs: [String], username: String) async {
|
|
await withTaskGroup(of: Void.self) { group in
|
|
for itemID in itemIDs {
|
|
group.addTask {
|
|
await self.prefetchPage(pendingLink: PendingLink(itemID: itemID, retryCount: 1), username: username)
|
|
}
|
|
}
|
|
await group.waitForAll()
|
|
}
|
|
}
|
|
|
|
func loadArticleContentWithRetries(
|
|
itemID: String,
|
|
username: String,
|
|
requestCount: Int = 1
|
|
) async throws -> ArticleContent {
|
|
guard requestCount < 7 else {
|
|
throw ContentFetchError.badData
|
|
}
|
|
|
|
let fetchedContent = try await loadArticleContent(username: username, itemID: itemID, useCache: true)
|
|
|
|
switch fetchedContent.contentStatus {
|
|
case .failed:
|
|
throw ContentFetchError.badData
|
|
case .processing:
|
|
let retryDelayInNanoSeconds = UInt64(requestCount * 2 * 1_000_000_000)
|
|
try await Task.sleep(nanoseconds: retryDelayInNanoSeconds)
|
|
logger.debug("fetching content for \(itemID). request count: \(requestCount)")
|
|
|
|
// Check for an updated itemID
|
|
// May have changed in the loadArticleContent call
|
|
let updatedItemID = await linkedItemID(from: itemID)
|
|
|
|
return try await loadArticleContentWithRetries(
|
|
itemID: updatedItemID ?? itemID,
|
|
username: username,
|
|
requestCount: requestCount + 1
|
|
)
|
|
case .succeeded, .unknown:
|
|
return fetchedContent
|
|
}
|
|
}
|
|
}
|