Implement Websocket for push Notification
create auth with jwt in query string
This commit is contained in:
52
src/main/kotlin/fr/dcproject/routes/Notification.kt
Normal file
52
src/main/kotlin/fr/dcproject/routes/Notification.kt
Normal file
@@ -0,0 +1,52 @@
|
||||
package fr.dcproject.routes
|
||||
|
||||
import fr.dcproject.citizen
|
||||
import io.ktor.client.HttpClient
|
||||
import io.ktor.http.cio.websocket.Frame
|
||||
import io.ktor.http.cio.websocket.readText
|
||||
import io.ktor.locations.KtorExperimentalLocationsAPI
|
||||
import io.ktor.routing.Route
|
||||
import io.ktor.websocket.webSocket
|
||||
import io.lettuce.core.Range
|
||||
import io.lettuce.core.api.async.RedisAsyncCommands
|
||||
import kotlinx.coroutines.ExperimentalCoroutinesApi
|
||||
import kotlinx.coroutines.delay
|
||||
import kotlinx.coroutines.flow.collect
|
||||
import kotlinx.coroutines.flow.consumeAsFlow
|
||||
import kotlinx.coroutines.flow.mapNotNull
|
||||
import kotlinx.coroutines.launch
|
||||
|
||||
@ExperimentalCoroutinesApi
|
||||
@KtorExperimentalLocationsAPI
|
||||
fun Route.notificationArticle(redis: RedisAsyncCommands<String, String>, client: HttpClient) {
|
||||
webSocket("/notifications") {
|
||||
val citizenId = call.citizen.id
|
||||
|
||||
launch {
|
||||
var score = 0.0
|
||||
while (true) {
|
||||
val result = redis.zrangebyscoreWithScores(
|
||||
"notification:$citizenId",
|
||||
Range.from(
|
||||
Range.Boundary.excluding(score),
|
||||
Range.Boundary.including(Double.POSITIVE_INFINITY)
|
||||
)
|
||||
)
|
||||
|
||||
result.get().forEach {
|
||||
outgoing.send(Frame.Text(it.value))
|
||||
if (it.score > score) score = it.score
|
||||
}
|
||||
delay(1000)
|
||||
// TODO terminate coroutine after connection close !
|
||||
}
|
||||
}
|
||||
|
||||
// TODO mark notification as read
|
||||
incoming.consumeAsFlow().mapNotNull { it as? Frame.Text }.collect {
|
||||
val text = it.readText()
|
||||
outgoing.send(Frame.Text(text))
|
||||
delay(100)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user