split keyword arguments (#6)

Author: bicycle1885

src/CodecZstd.jl

@@ -16,6 +16,16 @@ import TranscodingStreams:
     initialize,
     finalize
 
+# TODO: This method will be added in the next version of TranscodingStreams.jl.
+function splitkwargs(kwargs, keys)
+    hits = []
+    others = []
+    for kwarg in kwargs
+        push!(kwarg[1] ∈ keys ? hits : others, kwarg)
+    end
+    return hits, others
+end
+
 include("libzstd.jl")
 include("compression.jl")
 include("decompression.jl")

src/compression.jl

@@ -37,7 +37,8 @@ const ZstdCompressorStream{S} = TranscodingStream{ZstdCompressor,S} where S<:IO
 Create a new zstd compression stream (see `ZstdCompressor` for `kwargs`).
 """
 function ZstdCompressorStream(stream::IO; kwargs...)
-    return TranscodingStream(ZstdCompressor(;kwargs...), stream)
+    x, y = splitkwargs(kwargs, (:level,))
+    return TranscodingStream(ZstdCompressor(;x...), stream; y...)
 end
 
 

src/decompression.jl

@@ -21,12 +21,12 @@ end
 const ZstdDecompressorStream{S} = TranscodingStream{ZstdDecompressor,S} where S<:IO
 
 """
-    ZstdDecompressorStream(stream::IO)
+    ZstdDecompressorStream(stream::IO; kwargs...)
 
-Create a new zstd decompression stream.
+Create a new zstd decompression stream (`kwargs` are passed to `TranscodingStream`).
 """
-function ZstdDecompressorStream(stream::IO)
-    return TranscodingStream(ZstdDecompressor(), stream)
+function ZstdDecompressorStream(stream::IO; kwargs...)
+    return TranscodingStream(ZstdDecompressor(), stream; kwargs...)
 end