fix handling backend errors

krbob · Nov 13, 2024 · 3f4c030 · 3f4c030
1 parent 2f14c68
commit 3f4c030
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/main/kotlin/net/bobinski/backend/AnalysisEndpoint.kt b/src/main/kotlin/net/bobinski/backend/AnalysisEndpoint.kt
@@ -21,7 +21,7 @@ object AnalysisEndpoint {
     suspend fun forStock(symbol: String): Analysis {
         val backendData = getLock(symbol).withLock {
             val info = Backend.getInfo(symbol)
-            if (info.name == null) throw IllegalArgumentException("Unknown symbol: $symbol")
+            if (info?.name == null) throw IllegalArgumentException("Unknown symbol: $symbol")
 
             var lacking = false
             var history = Backend.getHistory(symbol, Backend.Period._5y)

diff --git a/src/main/kotlin/net/bobinski/source/Backend.kt b/src/main/kotlin/net/bobinski/source/Backend.kt
@@ -7,6 +7,7 @@ import io.ktor.client.plugins.cache.HttpCache
 import io.ktor.client.plugins.cache.storage.FileStorage
 import io.ktor.client.plugins.contentnegotiation.ContentNegotiation
 import io.ktor.client.request.get
+import io.ktor.http.isSuccess
 import io.ktor.serialization.kotlinx.json.json
 import net.bobinski.Config
 import net.bobinski.data.BasicInfo
@@ -29,13 +30,13 @@ object Backend {
     suspend fun getHistory(symbol: String, period: Period): Collection<HistoricalPrice> {
         return client
             .get("${Config.backendUrl}/history/$symbol/${period.value}")
-            .body()
+            .run { if (status.isSuccess()) body() else emptySet() }
     }
 
-    suspend fun getInfo(symbol: String): BasicInfo {
+    suspend fun getInfo(symbol: String): BasicInfo? {
         return client
             .get("${Config.backendUrl}/info/$symbol")
-            .body()
+            .run { if (status.isSuccess()) body() else null }
     }
 
     enum class Period(val value: String) {