• Home
• com.softwaremill.sttp.client3
• fs2ce2_sjs1_2.13
• 3.9.8
• source code
• Fs2SimpleQueue.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

sttp.client3.impl.fs2.Fs2SimpleQueue.scala Maven / Gradle / Ivy

package sttp.client3.impl.fs2

import cats.effect.{Effect, IO}
import fs2.concurrent.InspectableQueue
import sttp.client3.internal.ws.SimpleQueue
import sttp.ws.WebSocketBufferFull

class Fs2SimpleQueue[F[_], A](queue: InspectableQueue[F, A], capacity: Option[Int])(implicit F: Effect[F])
    extends SimpleQueue[F, A] {
  override def offer(t: A): Unit = {
    F.toIO(queue.offer1(t))
      .flatMap {
        case true  => IO.unit
        case false => IO.raiseError(WebSocketBufferFull(capacity.getOrElse(Int.MaxValue)))
      }
      .unsafeRunSync()
  }

  override def poll: F[A] = queue.dequeue1
}