Merge pull request #1 from guardian/rk-flatmap

Add Box.flatMap
guardian · Sep 7, 2017 · 6e10fe8 · 6e10fe8
2 parents 09b20e2 + 7d5aa6c
commit 6e10fe8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/scala/com/gu/Box.scala b/src/main/scala/com/gu/Box.scala
@@ -15,6 +15,7 @@ abstract class Box[T] {
   def alter(f: T => T): Future[T]
 
   def map[A](f: T => A): Box[A]
+  def flatMap[A](f: T => Box[A]): Box[A]
 }
 
 object Box {
@@ -34,5 +35,6 @@ private class AtomicRefBox[T](t: T) extends Box[T] {
   def alter(f: T => T): Future[T] = Future.successful(ref.updateAndGet(t => f(t)))
 
   def map[A](f: T => A): Box[A] = new AtomicRefBox[A](f(get()))
+  def flatMap[A](f: T => Box[A]): Box[A] = f(get())
 }