recover service after crash

greenart7c3 · Jan 3, 2025 · da65496 · da65496
1 parent 6049b79
commit da65496
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/app/src/main/java/com/greenart7c3/citrine/server/CustomWebSocketServer.kt b/app/src/main/java/com/greenart7c3/citrine/server/CustomWebSocketServer.kt
@@ -9,6 +9,7 @@ import com.greenart7c3.citrine.BuildConfig
 import com.greenart7c3.citrine.Citrine
 import com.greenart7c3.citrine.database.AppDatabase
 import com.greenart7c3.citrine.database.toEventWithTags
+import com.greenart7c3.citrine.service.CustomWebSocketService
 import com.greenart7c3.citrine.service.LocalPreferences
 import com.greenart7c3.citrine.utils.isParameterizedReplaceable
 import com.greenart7c3.citrine.utils.shouldDelete
@@ -17,6 +18,8 @@ import com.vitorpamplona.quartz.events.Event
 import io.ktor.http.ContentType
 import io.ktor.http.HttpMethod
 import io.ktor.http.HttpStatusCode
+import io.ktor.server.application.ApplicationStarted
+import io.ktor.server.application.ApplicationStopped
 import io.ktor.server.application.install
 import io.ktor.server.cio.CIO
 import io.ktor.server.cio.CIOApplicationEngine
@@ -276,6 +279,15 @@ class CustomWebSocketServer(
             port = port,
             host = host,
         ) {
+            monitor.subscribe(ApplicationStarted) {
+                Log.d(Citrine.TAG, "Server started on $host:$port")
+                CustomWebSocketService.hasStarted = true
+            }
+
+            monitor.subscribe(ApplicationStopped) {
+                Log.d(Citrine.TAG, "Server stopped")
+                CustomWebSocketService.hasStarted = false
+            }
             install(WebSockets) {
                 pingPeriodMillis = 1000L
                 timeoutMillis = 300000L

diff --git a/app/src/main/java/com/greenart7c3/citrine/service/CustomWebSocketService.kt b/app/src/main/java/com/greenart7c3/citrine/service/CustomWebSocketService.kt
@@ -4,4 +4,5 @@ import com.greenart7c3.citrine.server.CustomWebSocketServer
 
 object CustomWebSocketService {
     var server: CustomWebSocketServer? = null
+    var hasStarted = false
 }
diff --git a/app/src/main/java/com/greenart7c3/citrine/service/WebSocketServerService.kt b/app/src/main/java/com/greenart7c3/citrine/service/WebSocketServerService.kt
@@ -42,6 +42,11 @@ class WebSocketServerService : Service() {
         fun getService(): WebSocketServerService = this@WebSocketServerService
     }
 
+    override fun onStartCommand(intent: Intent?, flags: Int, startId: Int): Int {
+        Log.d(Citrine.TAG, "onStartCommand ${CustomWebSocketService.hasStarted}")
+        return START_STICKY
+    }
+
     @SuppressLint("UnspecifiedRegisterReceiverFlag")
     override fun onCreate() {
         super.onCreate()