Add example of pull streams

effekt-lang · Nov 25, 2024 · c6e2202 · c6e2202
1 parent b69033d
commit c6e2202
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 1 deletion.
diff --git a/effekt/jvm/src/test/scala/effekt/StdlibTests.scala b/effekt/jvm/src/test/scala/effekt/StdlibTests.scala
@@ -24,7 +24,8 @@ abstract class StdlibChezTests extends StdlibTests {
     // Not implemented yet
     examplesDir / "stdlib" / "bytearray",
     examplesDir / "stdlib" / "io",
-    examplesDir / "stdlib" / "stream" / "characters.effekt"
+    examplesDir / "stdlib" / "stream" / "characters.effekt",
+    examplesDir / "stdlib" / "stream" / "fuse_newlines.effekt"
   )
 }
 class StdlibChezSchemeMonadicTests extends StdlibChezTests {
@@ -46,6 +47,7 @@ class StdlibLLVMTests extends StdlibTests {
     // Valgrind leak/failure
     examplesDir / "stdlib" / "bytearray" / "bytearray.effekt",
     examplesDir / "stdlib" / "stream" / "characters.effekt",
+    examplesDir / "stdlib" / "stream" / "fuse_newlines.effekt",
     examplesDir / "stdlib" / "io" / "filesystem" / "async_file_io.effekt",
     examplesDir / "stdlib" / "io" / "filesystem" / "files.effekt",
     examplesDir / "stdlib" / "io" / "filesystem" / "wordcount.effekt",

diff --git a/examples/stdlib/stream/fuse_newlines.check b/examples/stdlib/stream/fuse_newlines.check
@@ -0,0 +1,3 @@
+ab
+c
+de
diff --git a/examples/stdlib/stream/fuse_newlines.effekt b/examples/stdlib/stream/fuse_newlines.effekt
@@ -0,0 +1,32 @@
+import stream
+
+def fuseNewlines(): Nothing / {read[Char], emit[Char], stop} = {
+  val c = do read[Char]()
+  if (c == '\n') {
+    do emit(c)
+    skipNewlines()
+  } else {
+    do emit(c)
+    fuseNewlines()
+  }
+}
+
+def skipNewlines(): Nothing / {read[Char], emit[Char], stop} = {
+  val c = do read[Char]()
+  if (c == '\n') {
+    skipNewlines()
+  } else {
+    do emit(c)
+    fuseNewlines()
+  }
+}
+
+def main() = {
+  with feed("ab\n\nc\nde")
+  println(collectString {
+    with exhaustively
+    fuseNewlines()
+  })
+}
+
+
diff --git a/libraries/common/stream.effekt b/libraries/common/stream.effekt
@@ -93,6 +93,7 @@ def boundary { program: () => Any / stop }: Unit = {
   ()
 }
 
+/// Run `program` forever until `stop` is thrown.
 def exhaustively { program: () => Any / stop }: Unit =
   try {
     def go(): Unit = {