kyo.server.internal.KyoUtil.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kyo-tapir_3 Show documentation

kyo-tapir

There is a newer version: 0.15.1

package kyo.server.internal

import io.netty.channel.Channel
import io.netty.channel.ChannelFuture
import kyo.*
import scala.concurrent.CancellationException

object KyoUtil:
    def nettyChannelFutureToScala(nettyFuture: ChannelFuture): Channel < Fibers =
        Fibers.initPromise[Channel].map { p =>
            p.onComplete(_ => IOs(nettyFuture.cancel(true).unit)).andThen {
                nettyFuture.addListener((future: ChannelFuture) =>
                    discard {
                        IOs.run {
                            if future.isSuccess then p.complete(future.channel())
                            else if future.isCancelled then
                                p.complete(IOs.fail(new CancellationException))
                            else p.complete(IOs.fail(future.cause()))
                        }
                    }
                )
                p.get
            }
        }

    def nettyFutureToScala[T](f: io.netty.util.concurrent.Future[T]): T < Fibers =
        Fibers.initPromise[T].map { p =>
            p.onComplete(_ => IOs(f.cancel(true)).unit).andThen {
                f.addListener((future: io.netty.util.concurrent.Future[T]) =>
                    discard {
                        IOs.run {
                            if future.isSuccess then p.complete(future.getNow)
                            else if future.isCancelled then
                                p.complete(IOs.fail(new CancellationException))
                            else p.complete(IOs.fail(future.cause()))
                        }
                    }
                )
                p.get
            }
        }
end KyoUtil