raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

=docs use stream compression in recipe (#23823)

Browse source 
* =docs use stream compression in recipe

* Also use Compression.gzip in scaladsl example
This commit is contained in:
Johannes Rudolph 2017-12-05 16:46:54 +01:00 • committed by Arnout Engelen
parent 52f30a8043
commit f996731c42
2 changed files with 22 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

36
akka-docs/src/test/java/jdocs/stream/javadsl/cookbook/RecipeDecompress.java
View file

@ -8,19 +8,15 @@ import akka.actor.ActorSystem;
import akka.stream.ActorMaterializer;
import akka.stream.Materializer;
import akka.stream.javadsl.Compression;
import akka.stream.javadsl.Sink;
import akka.stream.javadsl.Source;
import akka.testkit.javadsl.TestKit;
import akka.util.ByteString;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.TimeUnit;
import java.util.zip.GZIPOutputStream;
public class RecipeDecompress extends RecipeTest {
@ -40,28 +36,26 @@ public class RecipeDecompress extends RecipeTest {
mat = null;
}
private ByteString gzip(final String s) throws IOException {
final ByteArrayOutputStream buf = new ByteArrayOutputStream();
final GZIPOutputStream out = new GZIPOutputStream(buf);
try {
out.write(s.getBytes(StandardCharsets.UTF_8));
} finally {
out.close();
}
return ByteString.fromArray(buf.toByteArray());
}
@Test
public void parseLines() throws Exception {
final Source<ByteString, NotUsed> compressed = Source.single(gzip("Hello World"));
final Source<ByteString, NotUsed> dataStream =
Source.single(ByteString.fromString("Hello World"));
final Source<ByteString, NotUsed> compressedStream =
dataStream.via(Compression.gzip());
//#decompress-gzip
final Source<String, NotUsed> uncompressed = compressed
.via(Compression.gunzip(100))
.map(b -> b.utf8String());
final Source<ByteString, NotUsed> decompressedStream =
compressedStream.via(Compression.gunzip(100));
//#decompress-gzip
uncompressed.runWith(Sink.head(), mat).toCompletableFuture().get(1, TimeUnit.SECONDS);
ByteString decompressedData =
decompressedStream
.runFold(ByteString.empty(), ByteString::concat, mat)
.toCompletableFuture()
.get(1, TimeUnit.SECONDS);
String decompressedString = decompressedData.utf8String();
Assert.assertEquals("Hello World", decompressedString);
}
}