javaMain.com.arkivanov.mvikotlin.timetravel.proto.internal.io.WriterThread.kt Maven / Gradle / Ivy

Go to download
package com.arkivanov.mvikotlin.timetravel.proto.internal.io

import com.arkivanov.mvikotlin.timetravel.proto.internal.data.ProtoObject
import java.io.IOException
import java.net.Socket
import java.util.concurrent.LinkedBlockingQueue

class WriterThread(
    private val socket: Socket,
    private val onDisconnected: () -> Unit = {},
    private val onError: (IOException) -> Unit = {}
) : Thread() {

    private val queue = LinkedBlockingQueue()

    override fun run() {
        try {
            val output = socket.getOutputStream().buffered()
            val frameEncoder = ProtoFrameEncoder { data, size -> output.write(data, 0, size) }
            val protoEncoder = ProtoEncoder(frameEncoder::accept)
            while (!isInterrupted) {
                protoEncoder.encode(queue.take())
                output.flush()
            }
        } catch (e: IOException) {
            onError(e)
        } catch (e: InterruptedException) {
            interrupt()
        } finally {
            socket.closeSafe()
            onDisconnected()
        }
    }

    fun write(protoObject: ProtoObject) {
        queue += protoObject
    }
}