Move a bit more pure code out of delay

evolution-gaming · Oct 9, 2023 · f776bc1 · f776bc1
1 parent c614930
commit f776bc1
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/actor/src/main/scala/com/evolutiongaming/akkaeffect/util/Serially.scala b/actor/src/main/scala/com/evolutiongaming/akkaeffect/util/Serially.scala
@@ -36,19 +36,18 @@ private[akkaeffect] object Serially {
       (a, task).tailRecM { case (a, task) =>
         for {
           a <- task(a)
-          a <- Sync[F].delay {
-            val s = ref.getAndUpdate {
+          s <- Sync[F].delay {
+            ref.getAndUpdate {
               case _: S.Active => S.Active
               case S.Active    => S.Idle(a)
               case _: S.Idle   => S.Idle(a)
             }
-            s match {
-              case s: S.Active => (a, s.task).asLeft[Unit]
-              case S.Active    => unit
-              case _: S.Idle   => unit
-            }
           }
-        } yield a
+        } yield s match {
+          case s: S.Active => (a, s.task).asLeft[Unit]
+          case S.Active    => unit
+          case _: S.Idle   => unit
+        }
       }
     }