dorkbox/Utilities
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Cleaned up error messages, added retry-count

Browse Source
This commit is contained in:
Robinson 2022-03-04 01:32:57 +01:00
parent 14601b43b1
commit b920e8f759
No known key found for this signature in database
GPG Key ID: 8E7DB78588BD6F5C
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/dorkbox/util/WebUtil.kt
View File

@ -542,11 +542,12 @@ object WebUtil {
/**
* Runs the 'action' function when the scheme+domain+path(s) when it was successful. Runs the 'onError' function when it fails.
*/
suspend fun fetchData(scheme: String, domain: String, vararg paths: String,
suspend fun fetchData(scheme: String, domain: String, vararg paths: String, retryCount: Int = 10,
onError: (String) ->Unit,
onSuccess: suspend (InputStream)->Unit) = withContext(Dispatchers.IO) {
val encodedPath = paths.joinToString(separator = "/") { URLEncoder.encodePathSegment(it, Charsets.UTF_8) }
var location = "$scheme://$domain/$encodedPath"
var alreadyTriedOtherScheme = false
// logger.trace{ "Getting data: $location" }
@ -557,8 +558,8 @@ object WebUtil {
while (true) {
visitedCount += 1
if (visitedCount > 10) {
onError("Stuck in a loop for '$location' --- more than $visitedCount tries to get the domain '$location'")
if (visitedCount > retryCount) {
onError("Stuck in a loop for '$location' --- more than $visitedCount attempts")
return@withContext
}
@ -596,27 +597,28 @@ object WebUtil {
return@withContext
}
HttpsURLConnection.HTTP_NOT_FOUND -> {
// if we are HTTPS, retry again as HTTP.
if (location.startsWith(scheme)) {
visitedCount = 0
location = if (scheme == "http") {
"https://$domain/$encodedPath"
} else {
"http://$domain/$encodedPath"
}
// loop again with the new location
return@with
} else {
onError("Error '$responseCode' getting domain '$location' HTTPS option exhausted.")
if (alreadyTriedOtherScheme) {
onError("Error '$responseCode' getting location '$location' HTTPS option exhausted.")
// done
return@withContext
}
// if we are HTTPS, retry again as HTTP.
alreadyTriedOtherScheme = true
visitedCount = 0
location = if (location.startsWith("https")) {
"http://$domain/$encodedPath"
} else {
"https://$domain/$encodedPath"
}
// loop again with the new location
return@with
}
else -> {
onError("Error '$responseCode' getting domain '$location'")
onError("Error '$responseCode' getting location '$location'")
// done
return@withContext