Тип данных Scala ZIO Stream

Question

Если нет возможности узнать размер ZIO Stream, у меня есть два потока, и я хотел найти внутренний продукт между ними, моя проблема в том, как я могу найти размер, чтобы перебрать их. Я решил проблему с помощью map и foldMap, но я хотел проверить производительность двумя способами? Это моя работа с использованием Fold and Map:

import zio._
import zio.console.{Console, putStrLn}
import zio.stream._
object InnerProduct extends App {

  val stream1: Stream[Nothing, Int] = Stream(1,2,3,4)
  val stream2: Stream[Nothing, Int] = Stream(4,5,6,7)


  def inner_fold [A,B,C,D,E](stream1:Stream[Nothing,A])(stream2:Stream[Nothing,B])(tms: (A, B) => C)(pls: (D, C) => D)(zero: D)= for{
    acc <- Ref.make(zero)
    acc2 <- acc.get
    i = 0
    compStream = stream1.zip(stream2)
    str = compStream.map(i => tms(i._1,i._2))
    streamFold<- str.fold(acc2)(pls)
    _ <-acc.set(streamFold)
  } yield acc

  override def run(args: List[String]) =
    for {
    res <- inner_fold[Int, Int, Int, Int,Int](stream1)(stream2)(_ * _)(_ + _ )(0)
      res2 <-res.get
      _ <- putStrLn(res2.toString)
    } yield (0)

}

Выход:60

Answer 1

Я предполагаю, что под "внутренним продуктом" вы подразумеваете два Stream[Double] которые вы хотите рассматривать как математические векторы и вычислять одно выходное значение после итерации по всем потокам.

Для этого легко использовать zip для вычисления поэлементного произведения двух потоков, а затем foldLeft для вычисления окончательного результата.

val products: Stream[Double] = stream1.zip(stream2).map(_ * _)
val innerProduct: UIO[Double] = products.run(Sink.foldLeft(_ * _))