From bd66ed93afc0b6b1bb7293f7930ab6c2d2c55a49 Mon Sep 17 00:00:00 2001 From: Aleksandar Pokopec Date: Sun, 25 Apr 2010 19:58:01 +0000 Subject: Fixes #3350. review by extempore --- src/library/scala/collection/immutable/Stream.scala | 10 ++++++++++ test/files/run/stream_length.scala | 15 +++++++++++++++ 2 files changed, 25 insertions(+) create mode 100644 test/files/run/stream_length.scala diff --git a/src/library/scala/collection/immutable/Stream.scala b/src/library/scala/collection/immutable/Stream.scala index 50fa07c2cf..fbda0b918b 100644 --- a/src/library/scala/collection/immutable/Stream.scala +++ b/src/library/scala/collection/immutable/Stream.scala @@ -104,6 +104,16 @@ self => loop(this, "") } + override def length: Int = { + var len = 0 + var left = this + while (!left.isEmpty) { + len += 1 + left = left.tail + } + len + } + /** It's an imperfect world, but at least we can bottle up the * imperfection in a capsule. */ diff --git a/test/files/run/stream_length.scala b/test/files/run/stream_length.scala new file mode 100644 index 0000000000..68e9cad5ac --- /dev/null +++ b/test/files/run/stream_length.scala @@ -0,0 +1,15 @@ + + +object Test { + def walk(depth: Int, bias: String): Stream[String] = { + if (depth == 0) + Stream(bias) + else { + Stream.concat(Stream.range(1, 100).map((x: Int) => walk(depth-1, bias + x))) + } + } + + def main(args: Array[String]) { + println("Length: " + walk(3, "---").length) + } +} -- cgit v1.2.3