feat : Explicit connection error

IZIVIA · Jul 8, 2024 · 2dd6255 · 2dd6255
1 parent 65bf0e6
commit 2dd6255
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ocpp-wamp/src/main/kotlin/com/izivia/ocpp/wamp/server/impl/OcppWampServerApp.kt b/ocpp-wamp/src/main/kotlin/com/izivia/ocpp/wamp/server/impl/OcppWampServerApp.kt
@@ -243,6 +243,7 @@ class OcppWampServerApp(
     fun sendBlocking(ocppId: CSOcppId, message: WampMessage, startedCallAt: Instant = Clock.System.now()): WampMessage =
         getChargingStationConnection(ocppId).sendBlocking(message, startedCallAt)
 
+    // Throws NoConnectionException when no connection is found for the specified ocpp id
     private fun getChargingStationConnection(ocppId: CSOcppId): ChargingStationConnection {
         var backOffRetryMs = 10L
         var backOffRetryAttempts = 5
@@ -253,9 +254,8 @@ class OcppWampServerApp(
             backOffRetryMs *= 2
             connection = connections[ocppId]
         }
-        return connection ?: throw IllegalStateException("no connection to $ocppId")
+        return connection ?: throw NoConnectionException("no connection to $ocppId")
     }
-
     fun getChargingStationOcppVersion(ocppId: CSOcppId): OcppVersion =
         getChargingStationConnection(ocppId).ocppVersion
 
@@ -305,5 +305,7 @@ class OcppWampServerApp(
     }
 }
 
+class NoConnectionException(e: String) : IllegalStateException(e)
+
 private fun Throwable.isConnectionReset(): Boolean =
     this is SocketException && this.message.equals("Connection reset")