kopia lustrzana https://github.com/vitorpamplona/amethyst
55 wiersze
1.9 KiB
Kotlin
55 wiersze
1.9 KiB
Kotlin
/**
|
|
* Copyright (c) 2024 Vitor Pamplona
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
* this software and associated documentation files (the "Software"), to deal in
|
|
* the Software without restriction, including without limitation the rights to use,
|
|
* copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
|
|
* Software, and to permit persons to whom the Software is furnished to do so,
|
|
* subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
|
|
* AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
package com.vitorpamplona.amethyst.service.playback
|
|
|
|
import okhttp3.WebSocket
|
|
import okhttp3.WebSocketListener
|
|
import okio.ByteString
|
|
import java.util.concurrent.ConcurrentSkipListSet
|
|
|
|
class WssDataStreamCollector : WebSocketListener() {
|
|
private val wssData = ConcurrentSkipListSet<ByteString>()
|
|
|
|
override fun onMessage(
|
|
webSocket: WebSocket,
|
|
bytes: ByteString,
|
|
) {
|
|
wssData.add(bytes)
|
|
}
|
|
|
|
override fun onClosing(
|
|
webSocket: WebSocket,
|
|
code: Int,
|
|
reason: String,
|
|
) {
|
|
super.onClosing(webSocket, code, reason)
|
|
wssData.removeAll(wssData)
|
|
}
|
|
|
|
fun canStream(): Boolean {
|
|
return wssData.size > 0
|
|
}
|
|
|
|
fun getNextStream(): ByteString {
|
|
return wssData.pollFirst()
|
|
}
|
|
}
|